We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d307dfd commit 8186817Copy full SHA for 8186817
src/cpu/x64/jit_avx2_1x1_convolution.cpp
@@ -681,7 +681,8 @@ void jit_avx2_1x1_convolution_bwd_weights_t::execute_backward_weights(
681
sp_step = nstl::min(sp_dim - sp, mb_sp_end - mb_sp);
682
683
const bool first_image = img == img_start;
684
- if (first_image && rw->balancer().nthr_per_group_ > 1) {
+ if (is_ddst_layout_nxc && first_image
685
+ && rw->balancer().nthr_per_group_ > 1) {
686
// Zero-pad the scratchpad when nthr > 1 (since most threads
687
// write to scratchpad) so that zero-padding is maintained
688
// for the final output after reduction
0 commit comments