xe: conv: apply skip-w optimization for large strides

oneapi-src · Dec 23, 2024 · d8d00b4 · d8d00b4
1 parent b8303a5
commit d8d00b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gpu/intel/jit/conv/config.cpp b/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) {
     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;
 }