apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add float16 type size

Browse Source
This commit is contained in:
zhoushunjie
2022-10-02 03:38:12 +00:00
parent 6f73cc109d
commit 84ee24ff67
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/core/fd_type.cc
View File

@@ -14,12 +14,12 @@
#include "fastdeploy/core/fd_type.h"
#include "fastdeploy/core/float16.h"
#include "fastdeploy/utils/utils.h"
namespace fastdeploy {
int FDDataTypeSize(const FDDataType& data_type) {
FDASSERT(data_type != FDDataType::FP16, "Float16 is not supported.");
if (data_type == FDDataType::BOOL) {
return sizeof(bool);
} else if (data_type == FDDataType::INT16) {
@@ -36,6 +36,8 @@ int FDDataTypeSize(const FDDataType& data_type) {
return sizeof(uint8_t);
} else if (data_type == FDDataType::INT8) {
return sizeof(int8_t);
} else if (data_type == FDDataType::FP16) {
return sizeof(float16);
} else {
FDASSERT(false, "Unexpected data type: %s", Str(data_type).c_str());
}