From 55d2cecd698f865efac2e1dbf2f701b4b8095df1 Mon Sep 17 00:00:00 2001 From: "Xuxin, Zeng" Date: Fri, 6 Oct 2023 14:59:11 -0700 Subject: [PATCH] cpu: x64: fixup comp_pad_idx when od_block > 1 --- src/cpu/x64/jit_brgemm_conv.cpp | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/src/cpu/x64/jit_brgemm_conv.cpp b/src/cpu/x64/jit_brgemm_conv.cpp index 2ace73d11e9..43ca32f6fd9 100644 --- a/src/cpu/x64/jit_brgemm_conv.cpp +++ b/src/cpu/x64/jit_brgemm_conv.cpp @@ -1174,14 +1174,11 @@ status_t brgemm_convolution_fwd_t::init(engine_t *engine) { assert(k <= static_cast(jcp.ker_ranges_size)); }; - for_(int odb = 0; odb < jcp.nb_od; odb++) + for_(int od = 0; od < jcp.od; od++) for_(int ohb = 0; ohb < jcp.nb_oh; ohb++) for (int owb = 0; owb < jcp.nb_ow; owb++) { - auto od_begin = odb * jcp.od_block; - auto od_end = nstl::min(OD, od_begin + jcp.od_block); auto oh_begin = ohb * jcp.oh_block; auto oh_end = nstl::min(OH, oh_begin + jcp.oh_block); - for_(int od = od_begin; od < od_end; od++) for (int oh = oh_begin; oh < oh_end; oh++) { int kw_s {0}, kw_full_s {0}, kw_f {0}, kw_full_f {0}; const int ow = owb * jcp.ow_block;