x64: deconv: dispatch low performant shapes to jit

oneapi-src · May 6, 2024 · f46fffb · f46fffb
1 parent d39e1b7
commit f46fffb
Showing 1 changed file with 25 additions and 1 deletion.
diff --git a/src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp b/src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp
@@ -1535,7 +1535,7 @@ status_t init_jcp(jit_brgemm_conv_conf_t &jcp, cpu_isa_t isa,
     const bool is_f32
             = utils::everyone_is(f32, jcp.src_dt, jcp.wei_dt, jcp.dst_dt);
 
-    // Disable 4 shapes that cause performance regression
+    // Disable shapes that cause performance regression
     const auto is_regression_shape = jcp.id == 1 && jcp.od == 1
             && ((jcp.ic == 128 && jcp.oc == 256 && jcp.ih == 101 && jcp.oh == 49
                         && jcp.iw == 85 && jcp.ow == 41)
@@ -1550,6 +1550,30 @@ status_t init_jcp(jit_brgemm_conv_conf_t &jcp, cpu_isa_t isa,
                             && everyone_is(4, jcp.ih, jcp.iw)
                             && everyone_is(2, jcp.oh, jcp.ow)
                             && everyone_is(4, jcp.kh, jcp.kw)
+                            && everyone_is(2, jcp.stride_h, jcp.stride_w))
+                    || (jcp.ic == 64 && jcp.oc == 128
+                            && everyone_is(513, jcp.ih, jcp.iw)
+                            && everyone_is(256, jcp.oh, jcp.ow)
+                            && everyone_is(3, jcp.kh, jcp.kw)
+                            && everyone_is(2, jcp.stride_h, jcp.stride_w))
+                    || (jcp.ic == 32 && jcp.oc == 64
+                            && everyone_is(1025, jcp.ih, jcp.iw)
+                            && everyone_is(512, jcp.oh, jcp.ow)
+                            && everyone_is(3, jcp.kh, jcp.kw)
+                            && everyone_is(2, jcp.stride_h, jcp.stride_w))
+                    || (jcp.ic == 128 && jcp.oc == 256
+                            && everyone_is(257, jcp.ih, jcp.iw)
+                            && everyone_is(128, jcp.oh, jcp.ow)
+                            && everyone_is(3, jcp.kh, jcp.kw)
+                            && everyone_is(2, jcp.stride_h, jcp.stride_w))
+                    || (jcp.ic == 256 && jcp.oc == 512
+                            && everyone_is(129, jcp.ih, jcp.iw)
+                            && everyone_is(64, jcp.oh, jcp.ow)
+                            && everyone_is(3, jcp.kh, jcp.kw)
+                            && everyone_is(2, jcp.stride_h, jcp.stride_w))
+                    || (jcp.ic == 256 && jcp.oc == 512 && jcp.ih == 49
+                            && jcp.iw == 41 && jcp.oh == 23 && jcp.ow == 19
+                            && everyone_is(5, jcp.kh, jcp.kw)
                             && everyone_is(2, jcp.stride_h, jcp.stride_w)));
     VDISPATCH_CONV_IC(!(is_f32 && is_regression_shape),
             "implementation skipped due to low performance");