src: cpu: bwd_d: fix input-channel w/stride_d shapes

oneapi-src · Oct 30, 2019 · 002e3ab · 002e3ab
1 parent 846eba1
commit 002e3ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cpu/jit_avx512_common_conv_kernel.cpp b/src/cpu/jit_avx512_common_conv_kernel.cpp
@@ -2429,7 +2429,7 @@ status_t jit_avx512_common_conv_bwd_data_kernel_f32::init_conf(
             if (!(jcp.kw > 3
                         || (jcp.kw == 3 && ker_total_size < L1_cache_size
                                 && jcp.ow > 8))
-                    && jcp.stride_h == 1)
+                    && jcp.stride_h == 1 && jcp.stride_d == 1)
                 if (jcp.nb_ic % try_nb_ic_blocking == 0) {
                     jcp.nb_ic_blocking = try_nb_ic_blocking;
                     jcp.ur_w = 31 / (jcp.nb_ic_blocking + 1);