Skip to content

Commit 1ce2d72

Browse files
xuxinzenluweizhou2016
authored andcommitted
cpu: x64: correct the condition for dimensions transformation in brgemm conv
1 parent aa986d8 commit 1ce2d72

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/cpu/x64/jit_brgemm_conv_utils.cpp

+2-1
Original file line numberDiff line numberDiff line change
@@ -1706,7 +1706,8 @@ status_t init_jcp(jit_brgemm_conv_conf_t &jcp, cpu_isa_t isa,
17061706
if (IMPLICATION(!pure_1d, jcp.iw % i == 0)
17071707
&& IMPLICATION(jcp.ic * i > jcp.simd_w,
17081708
(jcp.ic * i) % jcp.simd_w == 0)
1709-
&& jcp.kw % i == 0 && jcp.stride_w % i == 0)
1709+
&& jcp.iw % i == 0 && jcp.kw % i == 0
1710+
&& jcp.stride_w % i == 0)
17101711
jcp.trans_dim_koef = i;
17111712
}
17121713
if (jcp.trans_dim_koef > 1) {

0 commit comments

Comments
 (0)