From 2089a05b5ea2405a2e1231930f56a0ca3f0828d3 Mon Sep 17 00:00:00 2001 From: Alexander Nesterov Date: Wed, 15 Jan 2025 15:26:12 +0100 Subject: [PATCH] new heuristic --- src/cpu/acl/acl_convolution_utils.cpp | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/cpu/acl/acl_convolution_utils.cpp b/src/cpu/acl/acl_convolution_utils.cpp index 0ff95018d66..3916a65a617 100644 --- 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