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

nivas-x86 · vpirogov · commit 36f355e0773f · 2023-06-30T16:40:00.000-07:00
diff --git a/src/cpu/x64/matmul/brgemm_matmul_utils.cpp b/src/cpu/x64/matmul/brgemm_matmul_utils.cpp
@@ -705,7 +705,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) {

Original file line number	Diff line number	Diff line change
`@@ -705,7 +705,8 @@ float compute_blocking_heuristic_avx512(brgemm_matmul_conf_t &bgmmc,`
`705`	`705`	`}`
`706`	`706`
`707`	`707`	`// Parallelize across K for shapes with big 'K' dimension`
`708`		`- bool bwd_w_par_k_blk = bm_conf_utils.check_is_transposed(bgmmc.src_tag)`
	`708`	`+ bool bwd_w_par_k_blk = bgmmc.batch == 1`
	`709`	`+ && bm_conf_utils.check_is_transposed(bgmmc.src_tag)`
`709`	`710`	`&& IMPLICATION(bm_conf_utils.is_bf16(), math::is_pow2(matmul.K))`
`710`	`711`	`&& matmul.K >= 2048;`
`711`	`712`	`if (bwd_w_par_k_blk) {`