diff --git a/src/gpu/jit/conv/config.cpp b/src/gpu/jit/conv/config.cpp index 7a0a2aac36a..6ef86f0338f 100644 --- a/src/gpu/jit/conv/config.cpp +++ b/src/gpu/jit/conv/config.cpp @@ -177,7 +177,8 @@ status_t conv_config_t::init_fwd(convolution_pd_t *conv_pd) { if (is_dw) bh->set_pref_tg_block(osp_name); bh->allow_split({osp_name, "mb"}); bh->reorder({osp_name, "mb"}); - if (!fuse_spatial && mb < 16 && iw % 8 != 0 && !is_dw) { + if (!is_int8_dst() && !fuse_spatial && mb < 16 && iw % 8 != 0 + && !is_dw) { bh->set_max_m_tg_dim(1); } } else {