From d8d00b47ba5cb26df3ebe7bfbaf004c94b6e9bc0 Mon Sep 17 00:00:00 2001 From: "Chereshnev, Eugene" Date: Fri, 20 Dec 2024 12:22:59 -0800 Subject: [PATCH] xe: conv: apply skip-w optimization for large strides --- src/gpu/intel/jit/conv/config.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/gpu/intel/jit/conv/config.cpp b/src/gpu/intel/jit/conv/config.cpp index a9ae3d1c9af..7fe8341173f 100644 --- 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; }