apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-22 08:09:28 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: build error without ENABLE_PADDLE2ONNX

Browse Source
This commit is contained in:
MistEO
2024-11-19 16:32:26 +08:00
parent 3bb05ac574
commit eca2ae8b94
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
fastdeploy/runtime/backends/ort/ort_backend.cc
View File

@@ -260,8 +260,13 @@ bool OrtBackend::InitFromOnnx(const std::string& model_file,
}
char* model_content_ptr;
int model_content_size = 0;
#ifdef ENABLE_PADDLE2ONNX
paddle2onnx::ConvertFP32ToFP16(model_file.c_str(), model_file.size(),
&model_content_ptr, &model_content_size);
#else
FDERROR << "Didn't compile with PaddlePaddle Frontend, FP16 is not supported" << std::endl;
return false;
#endif
std::string onnx_model_proto(model_content_ptr,
model_content_ptr + model_content_size);
delete[] model_content_ptr;