diff --git a/.gitignore b/.gitignore index babbc03dca3cf85475f0b4ff2b4f77bf16d72db4..f714acdefad8fee158f49bf044e59be9c3cf8c87 100644 --- a/.gitignore +++ b/.gitignore @@ -86,3 +86,6 @@ test_temp_summary_event_file/ mindspore/version.py mindspore/default_config.py mindspore/.commit_id + +# lite test file +mindspore/lite/test/do_test/ diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/convolution_1x1.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/convolution_1x1.cc index a735dee884ad867e688aa232c98124bff8870771..e2456bfdf45662df5228fcada3decdc623dbd12a 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp32/convolution_1x1.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp32/convolution_1x1.cc @@ -80,12 +80,12 @@ int Convolution1x1CPUKernel::InitConv1x1BiasWeight() { bias_data_ = nullptr; } - weight_ptr_ = reinterpret_cast(malloc(matmul_param_->row_8_ * matmul_param_->col_8_ * sizeof(float))); + weight_ptr_ = reinterpret_cast(malloc(matmul_param_->deep_ * matmul_param_->col_8_ * sizeof(float))); if (weight_ptr_ == nullptr) { MS_LOG(ERROR) << "Conv1x1 Malloc weight_ptr_ error!"; return RET_ERROR; } - memset(weight_ptr_, 0, matmul_param_->row_8_ * matmul_param_->col_8_ * sizeof(float)); + memset(weight_ptr_, 0, matmul_param_->deep_ * matmul_param_->col_8_ * sizeof(float)); RowMajor2Col8Major(reinterpret_cast(inputs_[1]->Data()), weight_ptr_, matmul_param_->col_, matmul_param_->deep_); return RET_OK;