mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!19755 [MS][LITE][CPU] x86 conv shape优化 

Merge pull request !19755 from liuzhongkai/thread2
This commit is contained in:
i-robot 2021-07-11 10:05:34 +00:00 committed by Gitee
parent 72cf841432 74caf214b4
commit ac06df9640
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/src/runtime/kernel/arm/fp32/convolution_delegate_fp32.cc
View File

@ -168,7 +168,7 @@ kernel::InnerKernel *ConvolutionDelegateCPUKernel::CpuConvFp32KernelSelect() {
} else {
#ifdef ENABLE_AVX
if (conv_param->input_channel_ / op_parameter_->thread_num_ > 64 ||
conv_param->input_h_ < conv_param->thread_num_) {
conv_param->input_h_ < conv_param->thread_num_ || conv_param->kernel_h_ >= 7 || conv_param->kernel_w_ >= 7) {
kernel = new (std::nothrow) kernel::ConvolutionCPUKernel(
op_parameter_, in_tensors_, out_tensors_, static_cast<const lite::InnerContext *>(this->context_),
origin_weight_, origin_bias_);