cpu: x64: correct the condition for dimensions transformation in brge…

…mm conv
openvinotoolkit · Sep 10, 2024 · 1ce2d72 · 1ce2d72
1 parent aa986d8
commit 1ce2d72
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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) {