From 23f2397d41f2ffedbe77253362cec3a15f632d12 Mon Sep 17 00:00:00 2001 From: zhujingxuan Date: Fri, 9 Apr 2021 15:20:03 +0800 Subject: [PATCH] fix lenet --- mindspore/lite/micro/cmake/file_list.cmake | 1 + .../opcoders/nnacl/fp32/convolution_winograd_fp32_coder.cc | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/mindspore/lite/micro/cmake/file_list.cmake b/mindspore/lite/micro/cmake/file_list.cmake index 8777af0662..d9118e653d 100644 --- a/mindspore/lite/micro/cmake/file_list.cmake +++ b/mindspore/lite/micro/cmake/file_list.cmake @@ -173,6 +173,7 @@ set(LITE_SRC ${LITE_DIR}/src/ops/populate/strided_slice_populate.cc ${LITE_DIR}/src/ops/populate/lstm_populate.cc ${LITE_DIR}/src/ops/populate/squeeze_populate.cc + ${LITE_DIR}/src/ops/populate/transpose_populate.cc ### tools ${LITE_DIR}/tools/common/flag_parser.cc ) diff --git a/mindspore/lite/micro/coder/opcoders/nnacl/fp32/convolution_winograd_fp32_coder.cc b/mindspore/lite/micro/coder/opcoders/nnacl/fp32/convolution_winograd_fp32_coder.cc index 685f1c7b73..c16331a3bd 100644 --- a/mindspore/lite/micro/coder/opcoders/nnacl/fp32/convolution_winograd_fp32_coder.cc +++ b/mindspore/lite/micro/coder/opcoders/nnacl/fp32/convolution_winograd_fp32_coder.cc @@ -162,9 +162,9 @@ int ConvolutionWinogradFP32Coder::InitWeightBias() { if (input_tensors_.size() == kInputSize2) { auto ori_bias_addr = reinterpret_cast(bias_tensor_->data_c()); MS_CHECK_RET_CODE(memcpy_s(new_bias_, new_bias_ele_size, ori_bias_addr, out_channel_size * sizeof(float)), - "memset_s failed!"); + "memcpy_s failed!"); } else { - return RET_ERROR; + MS_CHECK_RET_CODE(memset_s(new_bias_, new_bias_ele_size, 0, new_bias_ele_size), "memset_s failed!"); } return RET_OK; }