gpu: jit: conv: always allow src reorder from NCHW

oneapi-src · Nov 30, 2022 · d0c14c2 · d0c14c2
1 parent df6f2e3
commit d0c14c2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/gpu/jit/conv/config.cpp b/src/gpu/jit/conv/config.cpp
@@ -849,6 +849,7 @@ status_t conv_config_t::init_data_layouts(convolution_pd_t *conv_pd) {
     bool allow_dst_reorder = false;
     bool src_abx = matches_tag(src_md, "abx");
     bool src_axb = matches_tag(src_md, "axb");
+    if (src_abx) allow_src_reorder = true;
     if ((src_abx || src_axb) && (is_fwd || is_bwd_w) && is_small_ic_g1) {
         allow_src_reorder = true;
     }