diff --git a/src/gpu/intel/jit/conv/config.cpp b/src/gpu/intel/jit/conv/config.cpp index d326f6bd172..4d5ed3fd588 100644 --- a/src/gpu/intel/jit/conv/config.cpp +++ b/src/gpu/intel/jit/conv/config.cpp @@ -1023,7 +1023,7 @@ bwd_d_optimize_kind_t bwd_d_optimize_kind_hint(const conv_problem_t &prb) { if (prb.is_stride1()) return bwd_d_optimize_kind_t::none; auto hint = bwd_d_optimize_kind_t::skip_strided_dhw; - if (prb.iw % prb.sw != 0 || prb.mb < 16) + if (prb.iw % prb.sw != 0 || (prb.mb < 16 && prb.sw <= 8)) hint = bwd_d_optimize_kind_t::skip_strided_dh; return hint; }