x64: brgconv: fixup K_tail

oneapi-src · Mar 25, 2024 · bec487e · bec487e
1 parent 1eab005
commit bec487e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp b/src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp
@@ -581,7 +581,7 @@ status_t brg_blocking_t::estimate_brgemm_ur() {
     N = ic >= ic_block ? ic_block : 0;
     N_tail = ic % ic_block;
     K = oc >= oc_block ? oc_block : 0;
-    K_tail = oc_block;
+    K_tail = oc % oc_block;
 
     const auto vK = K > 0 ? K : K_tail;
     const auto vM = M > 0 ? M : M_tail;