Skip to content

Commit d8d00b4

Browse files
committedDec 23, 2024
xe: conv: apply skip-w optimization for large strides
1 parent b8303a5 commit d8d00b4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed
 

‎src/gpu/intel/jit/conv/config.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -1005,7 +1005,7 @@ bwd_d_optimize_kind_t bwd_d_optimize_kind_hint(const conv_problem_t &prb) {
10051005
if (prb.is_stride1()) return bwd_d_optimize_kind_t::none;
10061006

10071007
auto hint = bwd_d_optimize_kind_t::skip_strided_dhw;
1008-
if (prb.iw % prb.sw != 0 || prb.mb < 16)
1008+
if (prb.iw % prb.sw != 0 || (prb.mb < 16 && prb.sw <= 8))
10091009
hint = bwd_d_optimize_kind_t::skip_strided_dh;
10101010
return hint;
10111011
}

0 commit comments

Comments
 (0)