Skip to content

Commit c3b1dcd

Browse files
committed
x64: fix inp_buffer_size for strided convolutions
1 parent 5e76d8b commit c3b1dcd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/cpu/x64/jit_brgemm_conv_utils.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2068,9 +2068,9 @@ status_t init_conf(jit_brgemm_conv_conf_t &jcp, cpu_isa_t isa,
20682068
if (jcp.is_os_blocking)
20692069
hs = div_up(rnd_up(hs * jcp.iwp, jcp.brgM), jcp.iwp);
20702070

2071-
jcp.inp_buffer_size = rnd_up(ds * hs * jcp.iwp * jcp.ngroups * jcp.nb_ic
2072-
* jcp.ic_block * jcp.kh_sets * jcp.kw_sets,
2073-
P4K);
2071+
jcp.inp_buffer_size = rnd_up(
2072+
ds * hs * jcp.iwp * jcp.ngroups * jcp.nb_ic * jcp.LDA, P4K);
2073+
20742074
jcp.inp_buffer_mask_size = rnd_up(static_cast<dim_t>(jcp.nb_od)
20752075
* jcp.nb_oh * jcp.nb_ow * jcp.ngroups * jcp.nb_ic,
20762076
P4K);

0 commit comments

Comments
 (0)