From 771dcf4b918a46be307adec9f15fbb8446192d16 Mon Sep 17 00:00:00 2001 From: ChaoII <849453582@qq.com> Date: Wed, 16 Aug 2023 14:12:58 +0800 Subject: [PATCH] [Backend]Update sophgo_backend.cc for FP16 data type (#2161) --- fastdeploy/runtime/backends/sophgo/sophgo_backend.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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;