Skip to content

Commit bec487e

Browse files
tczeszuntprimak
authored andcommitted
x64: brgconv: fixup K_tail
1 parent 1eab005 commit bec487e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -581,7 +581,7 @@ status_t brg_blocking_t::estimate_brgemm_ur() {
581581
N = ic >= ic_block ? ic_block : 0;
582582
N_tail = ic % ic_block;
583583
K = oc >= oc_block ? oc_block : 0;
584-
K_tail = oc_block;
584+
K_tail = oc % oc_block;
585585

586586
const auto vK = K > 0 ? K : K_tail;
587587
const auto vM = M > 0 ? M : M_tail;

0 commit comments

Comments
 (0)