From 4544cb8c1a91c0fa6126f0b2b298b0b0c31fd08f Mon Sep 17 00:00:00 2001 From: lixian Date: Fri, 9 Apr 2021 15:26:29 +0800 Subject: [PATCH] fix dw-conv 3x3 check --- mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c b/mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c index 9aa7e02718..b57952cb87 100644 --- a/mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c +++ b/mindspore/lite/nnacl/fp32/conv_depthwise_fp32.c @@ -339,7 +339,7 @@ bool CheckConvDw1DWinograd(const ConvParameter *conv_param, int thread_num) { return conv_param->kernel_h_ == 3 && conv_param->kernel_w_ == 3 && conv_param->stride_w_ == 1 && conv_param->stride_h_ == 1 && conv_param->dilation_h_ == 1 && conv_param->dilation_w_ == 1 && conv_param->pad_u_ == 1 && conv_param->pad_d_ == 1 && conv_param->pad_l_ == 1 && conv_param->pad_r_ == 1 && - conv_param->input_channel_ == conv_param->output_channel_ && + conv_param->input_channel_ == conv_param->output_channel_ && conv_param->output_w_ >= 4 && conv_param->output_h_ >= thread_num * 4; // better had more than 4 rows for each thread }