diff --git a/.gitignore b/.gitignore index babbc03dca..f714acdefa 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 a735dee884..e2456bfdf4 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;