cpu: x64: brgemm: fix treat_A_as_plain for batch > 1

Author: yair-obodovsky

src/cpu/x64/matmul/brgemm_matmul_utils.cpp

@@ -1341,7 +1341,7 @@ status_t init_brgemm_matmul_conf(cpu_isa_t isa, brgemm_matmul_conf_t &bgmmc,
     const bool transposed_A = bm_conf_utils.check_is_transposed(bgmmc.src_tag);
     // When M == 1 MatMul always considers A to be non-transposed even if A md
     // was created using "ba" tag.
-    bgmmc.treat_A_as_plain = bgmmc.M == 1;
+    bgmmc.treat_A_as_plain = bgmmc.M == 1 && bgmmc.batch == 1;
     bgmmc.transposed_A = ((transposed_A && !bgmmc.treat_A_as_plain)
             || bgmmc.src_tag == adbc);
     // For batched problems with plain A and C and fully broadcasted across B