diff --git a/fastdeploy/runtime/backends/sophgo/sophgo_backend.cc b/fastdeploy/runtime/backends/sophgo/sophgo_backend.cc index 5770fae64..858fa7a07 100644 --- a/fastdeploy/runtime/backends/sophgo/sophgo_backend.cc +++ b/fastdeploy/runtime/backends/sophgo/sophgo_backend.cc @@ -248,7 +248,7 @@ bool SophgoBackend::Infer(std::vector& inputs, ***************************************************************/ FDDataType SophgoBackend::SophgoTensorTypeToFDDataType(bm_data_type_t type) { if (type == BM_FLOAT16) { - return FDDataType::FP32; + return FDDataType::FP16; } if (type == BM_FLOAT32) { return FDDataType::FP32;