From 7e50e152f21a79978b8910260e042b43941b601c Mon Sep 17 00:00:00 2001 From: Tomasz Czeszun Date: Fri, 15 Mar 2024 10:09:17 -0700 Subject: [PATCH] x64: brgdeconv: dispatch to VNNI for zp with big buffer on AMX --- src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp b/src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp index 98333ff1a66..df87dd7bd24 100644 --- a/src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp +++ b/src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp @@ -2024,6 +2024,11 @@ status_t init_conf(jit_brgemm_conv_conf_t &jcp, cpu_isa_t isa, "heuristic to skip amx implementation for given data dimensions"); if (jcp.req_cal_comp_pad) { + VDISPATCH_CONV_IC(!(is_amx(jcp.isa) + && static_cast(jcp.ngroups) * jcp.nb_ic + * jcp.ic_block * jcp.iw + > 4096), + "heuristic to skip amx implementation because of buffer size"); const auto comp_buffer_iw = jcp.exec_type == exec_trans ? jcp.iw : 1; jcp.ker_ranges_size = precalculate_comp_pad_kernels(jcp); jcp.comp_a_buffer_size = static_cast(jcp.ngroups) * jcp.nb_ic