x64: matmul: fixup: parallel_k_reduction is not supported for batched…

… matmul
oneapi-src · Aug 10, 2023 · 1a9b80d · 1a9b80d
1 parent 8c20f62
commit 1a9b80d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/cpu/x64/matmul/brgemm_matmul_utils.cpp b/src/cpu/x64/matmul/brgemm_matmul_utils.cpp
@@ -666,7 +666,8 @@ float compute_blocking_heuristic_avx512(brgemm_matmul_conf_t &bgmmc,
         }
 
         // Parallelize across K for shapes with big 'K' dimension
-        bool bwd_w_par_k_blk = bm_conf_utils.check_is_transposed(bgmmc.src_tag)
+        bool bwd_w_par_k_blk = bgmmc.batch == 1
+                && bm_conf_utils.check_is_transposed(bgmmc.src_tag)
                 && IMPLICATION(bm_conf_utils.is_bf16(), math::is_pow2(matmul.K))
                 && matmul.K >= 2048;
         if (bwd_w_par_k_blk) {