commit 0b18cb384701feed540cdcad20e2a711936145d1
author Frank Barchard <fbarchard@google.com> Mon Nov 23 10:50:44 2020 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Mon Nov 23 10:51:34 2020 -0800
tree 17e279a1810d57e9c9aa9ba44cc27106290ceb3d
parent 3b80045e32e8cbb2e81fbcc7028269a4c2eee8ad

Enable dwconv2d_chw_3x3p1__ssse3_2x4_acc2 microkernel

PiperOrigin-RevId: 343882814

src/init.c

diff --git a/src/init.c b/src/init.c
index 6ac32ed..761d85f 100644
--- a/src/init.c
+++ b/src/init.c
@@ -1939,12 +1939,21 @@
         .output_height_tile = 2,
         .output_width_tile = 2,
       };
-      xnn_params.f32.dwconv2d_chw_3x3 = (struct dwconv2d_chw_parameters) {
-        .ukernel = (xnn_dwconv2d_chw_ukernel_function) xnn_f32_dwconv2d_chw_ukernel_3x3p1__sse_2x4_acc2,
-        .input_width_tile = 4,
-        .output_width_tile = 4,
-        .output_height_tile = 2,
-      };
+      if (!XNN_PLATFORM_MOBILE && cpuinfo_has_x86_ssse3()) {
+        xnn_params.f32.dwconv2d_chw_3x3 = (struct dwconv2d_chw_parameters) {
+          .ukernel = (xnn_dwconv2d_chw_ukernel_function) xnn_f32_dwconv2d_chw_ukernel_3x3p1__ssse3_2x4_acc2,
+          .input_width_tile = 4,
+          .output_width_tile = 4,
+          .output_height_tile = 2,
+        };
+      } else {
+        xnn_params.f32.dwconv2d_chw_3x3 = (struct dwconv2d_chw_parameters) {
+          .ukernel = (xnn_dwconv2d_chw_ukernel_function) xnn_f32_dwconv2d_chw_ukernel_3x3p1__sse_2x4_acc2,
+          .input_width_tile = 4,
+          .output_width_tile = 4,
+          .output_height_tile = 2,
+        };
+      }
       xnn_params.f32.dwconv2d_chw_3x3s2 = (struct dwconv2d_chw_parameters) {
         .ukernel = (xnn_dwconv2d_chw_ukernel_function) xnn_f32_dwconv2d_chw_ukernel_3x3s2p1__sse_1x4_acc3,
         .input_width_tile = 4,