Skip to content

Commit 7e50e15

Browse files
tczeszuntprimak
authored andcommitted
x64: brgdeconv: dispatch to VNNI for zp with big buffer on AMX
1 parent 15c7916 commit 7e50e15

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2024,6 +2024,11 @@ status_t init_conf(jit_brgemm_conv_conf_t &jcp, cpu_isa_t isa,
20242024
"heuristic to skip amx implementation for given data dimensions");
20252025

20262026
if (jcp.req_cal_comp_pad) {
2027+
VDISPATCH_CONV_IC(!(is_amx(jcp.isa)
2028+
&& static_cast<dim_t>(jcp.ngroups) * jcp.nb_ic
2029+
* jcp.ic_block * jcp.iw
2030+
> 4096),
2031+
"heuristic to skip amx implementation because of buffer size");
20272032
const auto comp_buffer_iw = jcp.exec_type == exec_trans ? jcp.iw : 1;
20282033
jcp.ker_ranges_size = precalculate_comp_pad_kernels(jcp);
20292034
jcp.comp_a_buffer_size = static_cast<dim_t>(jcp.ngroups) * jcp.nb_ic

0 commit comments

Comments
 (0)