new heuristic

openvinotoolkit · Jan 15, 2025 · 2089a05 · 2089a05
1 parent c7ecd8f
commit 2089a05
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/cpu/acl/acl_convolution_utils.cpp b/src/cpu/acl/acl_convolution_utils.cpp
@@ -355,12 +355,12 @@ status_t init_conf_wino(acl_conv_conf_t &acp, memory_desc_t &src_md,
     // Under these conditions, fallback to faster GEMM-based convolution
     // unless the user explicitly specifies Winograd algorithm
     // clang-format off
-    if (one_of(true, src_md.dims[2] > 112, // ih
-                src_md.dims[3] > 112, // iw
-                src_md.dims[1] < 64, // ic
-                dst_md.dims[1] < 64, // oc
-                dnnl_get_max_threads() > 28)
-            && cd.alg_kind == alg_kind::convolution_auto) {
+    if (one_of(true, one_of(true, src_md.dims[2] < 32,  // ih
+                                  src_md.dims[3] < 32)  // iw
+                                  && dnnl_get_max_threads() > 64,
+                     one_of(true, src_md.dims[1] < 64,  // ic
+                                  dst_md.dims[1] < 64)) // oc
+        && cd.alg_kind == alg_kind::convolution_auto) {
         return status::unimplemented;
     }
     // clang-format on