Skip to content

Commit

Permalink
x64: brgdeconv: dispatch to VNNI for zp with big buffer on AMX
Browse files Browse the repository at this point in the history
tczeszun authored and tprimak committed Mar 15, 2024
1 parent 15c7916 commit 7e50e15
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp
Original file line number Diff line number Diff line change
@@ -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<dim_t>(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<dim_t>(jcp.ngroups) * jcp.nb_ic

0 comments on commit 7e50e15

Please sign in to comment.