From 1ce2d722922efb80da52a6efe2152a9aecdddebf Mon Sep 17 00:00:00 2001
From: "Xuxin, Zeng" <xuxin.zeng@intel.com>
Date: Tue, 27 Aug 2024 19:02:47 -0700
Subject: [PATCH] cpu: x64: correct the condition for dimensions transformation
 in brgemm conv

---
 src/cpu/x64/jit_brgemm_conv_utils.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/cpu/x64/jit_brgemm_conv_utils.cpp b/src/cpu/x64/jit_brgemm_conv_utils.cpp
index 98eea342b7e..3898adccfa1 100644
--- a/src/cpu/x64/jit_brgemm_conv_utils.cpp
+++ b/src/cpu/x64/jit_brgemm_conv_utils.cpp
@@ -1706,7 +1706,8 @@ status_t init_jcp(jit_brgemm_conv_conf_t &jcp, cpu_isa_t isa,
             if (IMPLICATION(!pure_1d, jcp.iw % i == 0)
                     && IMPLICATION(jcp.ic * i > jcp.simd_w,
                             (jcp.ic * i) % jcp.simd_w == 0)
-                    && jcp.kw % i == 0 && jcp.stride_w % i == 0)
+                    && jcp.iw % i == 0 && jcp.kw % i == 0
+                    && jcp.stride_w % i == 0)
                 jcp.trans_dim_koef = i;
         }
         if (jcp.trans_dim_koef > 1) {