gpu: jit: conv: limit mad_g_small_oc case to small g

oneapi-src · Apr 18, 2023 · 8710839 · 8710839
1 parent 36aa622
commit 8710839
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/gpu/jit/conv/config.cpp b/src/gpu/jit/conv/config.cpp
@@ -125,8 +125,8 @@ bool is_small_oc(const conv_problem_t &prb) {
 }
 
 bool is_mad_g_small_oc(const conv_config_t &cfg) {
-    return (cfg.prb().g > 1 && cfg.fma_kind() == fma_kind_t::mad
-            && is_small_oc(cfg.prb()));
+    return (cfg.prb().g > 1 && cfg.prb().g < 16
+            && cfg.fma_kind() == fma_kind_t::mad && is_small_oc(cfg.prb()));
 }
 
 bool is_dw_large_mb(const conv_problem_t &prb) {