src: gpu: jit: conv: allow non-trivial m dim tg for int8

oneapi-src · Oct 7, 2022 · 18764fb · 18764fb
1 parent c3781c6
commit 18764fb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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 {