From 6f7d5f7cd4bb4350af44cf0241ef6825c1ce2cc9 Mon Sep 17 00:00:00 2001 From: lixian <179220644@qq.com> Date: Mon, 17 Aug 2020 20:32:54 +0800 Subject: [PATCH] fix 1x1 conv of fp16 --- .../lite/src/runtime/kernel/arm/fp16/convolution_1x1_fp16.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/src/runtime/kernel/arm/fp16/convolution_1x1_fp16.cc b/mindspore/lite/src/runtime/kernel/arm/fp16/convolution_1x1_fp16.cc index 764a22eb51..b1bdaa4f2a 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp16/convolution_1x1_fp16.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp16/convolution_1x1_fp16.cc @@ -81,7 +81,8 @@ int Convolution1x1FP16CPUKernel::InitWeightBias() { return RET_ERROR; } memset(bias_data_, 0, matmul_param_->col_8_ * sizeof(float16_t)); - memcpy(bias_data_, in_tensors_[2]->Data(), conv_param_->output_channel_ * sizeof(float16_t)); + Float32ToFloat16(reinterpret_cast(in_tensors_[2]->Data()), reinterpret_cast(bias_data_), + conv_param_->output_channel_); } else { bias_data_ = nullptr; }