From c3b1dcd2605cae5609d7175fcf5223da16e03fb9 Mon Sep 17 00:00:00 2001 From: Andrey Kalinin Date: Fri, 24 Mar 2023 11:30:12 -0700 Subject: [PATCH] x64: fix inp_buffer_size for strided convolutions --- src/cpu/x64/jit_brgemm_conv_utils.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/cpu/x64/jit_brgemm_conv_utils.cpp b/src/cpu/x64/jit_brgemm_conv_utils.cpp index ae209175697..ee306201f89 100644 --- a/src/cpu/x64/jit_brgemm_conv_utils.cpp +++ b/src/cpu/x64/jit_brgemm_conv_utils.cpp @@ -2068,9 +2068,9 @@ status_t init_conf(jit_brgemm_conv_conf_t &jcp, cpu_isa_t isa, if (jcp.is_os_blocking) hs = div_up(rnd_up(hs * jcp.iwp, jcp.brgM), jcp.iwp); - jcp.inp_buffer_size = rnd_up(ds * hs * jcp.iwp * jcp.ngroups * jcp.nb_ic - * jcp.ic_block * jcp.kh_sets * jcp.kw_sets, - P4K); + jcp.inp_buffer_size = rnd_up( + ds * hs * jcp.iwp * jcp.ngroups * jcp.nb_ic * jcp.LDA, P4K); + jcp.inp_buffer_mask_size = rnd_up(static_cast(jcp.nb_od) * jcp.nb_oh * jcp.nb_ow * jcp.ngroups * jcp.nb_ic, P4K);