cpu: x64: fix kd/kh blocking for brg deconv

oneapi-src · Oct 17, 2023 · 01204ed · 01204ed
1 parent 1f9a3db
commit 01204ed
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp b/src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp
@@ -1080,16 +1080,17 @@ status_t brg_blocking_t::calc_blocks() {
     // results then we need the out buffer
     const auto maybe_use_buffer = (dst_dt != acc_dt || with_sum);
 
+    // kd/kh block should be either kd/kh or a multiple of sd/sh
     std::vector<int> kd_blocks(1), kh_blocks(1);
     kd_blocks[0] = kd;
     kh_blocks[0] = kh;
     if (kd != 1) {
         kd_blocks.resize(2);
-        kd_blocks[1] = 1;
+        kd_blocks[1] = stride_d;
     }
     if (kh != 1) {
         kh_blocks.resize(2);
-        kh_blocks[1] = 1;
+        kh_blocks[1] = stride_h;
     }
 
     const auto thr_eff_threshold = 0.9f;