Skip to content

Commit 4cdfe8f

Browse files
committed
src: cpu: int8: dilated conv: fixup case when ih<=dh
1 parent 57063eb commit 4cdfe8f

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

src/cpu/jit_avx512_core_x8s8s32x_conv_kernel.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -505,7 +505,7 @@ void _jit_avx512_core_x8s8s32x_fwd_kernel<Vmm>::kh_loop(
505505
L(no_t_overflow_label);
506506
}
507507
mov(reg_kj, ptr[param1 + GET_OFF(kh_padding)]);
508-
if ((jcp.signed_input)
508+
if ((jcp.signed_input) || (jcp.dilate_h >= jcp.ih)
509509
|| (!jcp.signed_input
510510
&& (jcp.kh - 1) * (jcp.dilate_h + 1)
511511
< nstl::max(jcp.t_pad, jcp.b_pad))) {

tests/benchdnn/inputs/conv/test_conv_regression_general

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,3 +130,6 @@ mb1_g16ic16oc16_ih5oh2kh2sh3ph0
130130

131131
# AVX JIT incorrectly calculates output when 'nb_oc_block' results in a prime number
132132
--reset --dir=FWD_I mb1ic3ih320oc51oh160kh7sh2ph3n"regression_oc-channel"
133+
134+
# Dilated convolution with ih<=dh when there is _no_ compute work.
135+
--reset --conv --cfg=u8s8u8 mb1ic16ih1iw1oc16oh2ow1kh3kw1dh1ph1

0 commit comments

Comments
 (0)