From d0c14c204782945b3732bd83b7329c314c3339c1 Mon Sep 17 00:00:00 2001 From: "Chereshnev, Eugene" Date: Mon, 21 Nov 2022 18:00:43 -0800 Subject: [PATCH] gpu: jit: conv: always allow src reorder from NCHW --- src/gpu/jit/conv/config.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/src/gpu/jit/conv/config.cpp b/src/gpu/jit/conv/config.cpp index 12d741f15e0..042d762120f 100644 --- 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; }