apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 16:48:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix bug of get input/output information from PaddleBackend (#339)

Browse Source 
* Fix bug of get input/output information from PaddleBackend

* Support Paddle Inference with TensorRT (#340)

* Fix bug
This commit is contained in:
Jason
2022-10-12 11:37:26 +08:00
committed by GitHub
parent 3faaeaea36
commit 945e197bd1
8 changed files with 152 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
fastdeploy/backends/paddle/util.cc
View File

@@ -89,4 +89,26 @@ FDDataType PaddleDataTypeToFD(const paddle_infer::DataType& dtype) {
return fd_dtype;
}
FDDataType ReaderDataTypeToFD(int32_t dtype) {
auto fd_dtype = FDDataType::FP32;
if (dtype == 0) {
fd_dtype = FDDataType::FP32;
} else if (dtype == 1) {
fd_dtype = FDDataType::FP64;
} else if (dtype == 2) {
fd_dtype = FDDataType::UINT8;
} else if (dtype == 3) {
fd_dtype = FDDataType::INT8;
} else if (dtype == 4) {
fd_dtype = FDDataType::INT32;
} else if (dtype == 5) {
fd_dtype = FDDataType::INT64;
} else if (dtype == 6) {
fd_dtype = FDDataType::FP16;
} else {
FDASSERT(false, "Unexpected data type: %d while call ReaderDataTypeToFD in PaddleBackend.", dtype);
}
return fd_dtype;
}
} // namespace fastdeploy