We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b8303a5 commit d8d00b4Copy full SHA for d8d00b4
src/gpu/intel/jit/conv/config.cpp
@@ -1005,7 +1005,7 @@ bwd_d_optimize_kind_t bwd_d_optimize_kind_hint(const conv_problem_t &prb) {
1005
if (prb.is_stride1()) return bwd_d_optimize_kind_t::none;
1006
1007
auto hint = bwd_d_optimize_kind_t::skip_strided_dhw;
1008
- if (prb.iw % prb.sw != 0 || prb.mb < 16)
+ if (prb.iw % prb.sw != 0 || (prb.mb < 16 && prb.sw <= 8))
1009
hint = bwd_d_optimize_kind_t::skip_strided_dh;
1010
return hint;
1011
}
0 commit comments