Skip to content

Commit 017a2e3

Browse files
committed
Update heuristic
1 parent 1789b1e commit 017a2e3

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/cpu/acl/acl_convolution_utils.cpp

+6-6
Original file line numberDiff line numberDiff line change
@@ -355,12 +355,12 @@ status_t init_conf_wino(acl_conv_conf_t &acp, memory_desc_t &src_md,
355355
// Under these conditions, fallback to faster GEMM-based convolution
356356
// unless the user explicitly specifies Winograd algorithm
357357
// clang-format off
358-
if (one_of(true, src_md.dims[2] > 112, // ih
359-
src_md.dims[3] > 112, // iw
360-
src_md.dims[1] < 64, // ic
361-
dst_md.dims[1] < 64, // oc
362-
dnnl_get_max_threads() > 28)
363-
&& cd.alg_kind == alg_kind::convolution_auto) {
358+
if (one_of(true, one_of(true, src_md.dims[2] < 32, // ih
359+
src_md.dims[3] < 32) // iw
360+
&& dnnl_get_max_threads() > 32,
361+
one_of(true, src_md.dims[1] < 64, // ic
362+
dst_md.dims[1] < 64)) // oc
363+
&& cd.alg_kind == alg_kind::convolution_auto) {
364364
return status::unimplemented;
365365
}
366366
// clang-format on

0 commit comments

Comments
 (0)