From 74caf214b4e0670196547c8977b8f13085d9f5d5 Mon Sep 17 00:00:00 2001 From: lzk Date: Thu, 8 Jul 2021 06:45:05 -0700 Subject: [PATCH] conv common --- .../src/runtime/kernel/arm/fp32/convolution_delegate_fp32.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/convolution_delegate_fp32.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/convolution_delegate_fp32.cc index 413614e9cc8..a93be032023 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp32/convolution_delegate_fp32.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp32/convolution_delegate_fp32.cc @@ -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(this->context_), origin_weight_, origin_bias_);