commit 479f87e6fbdc2477e5cfff8c7e89fce5cdf040a8
author Marat Dukhan <maratek@google.com> Wed Nov 27 15:17:06 2019 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Wed Nov 27 15:17:40 2019 -0800
tree 044628b8c7720f8ec90fd50590bf2b2a97f13acc
parent 91f8d8610481ba637a5e43c904221ff828845b22

AVX512F implementation of DWCONV micro-kernels

PiperOrigin-RevId: 282842351

src/init.c

diff --git a/src/init.c b/src/init.c
index fd2aa12..4b9e0ce 100644
--- a/src/init.c
+++ b/src/init.c
@@ -666,7 +666,23 @@
       .nr = 2,
       .log2_kr = 2,
     };
-    if (!XNN_PLATFORM_MOBILE && cpuinfo_has_x86_fma3()) {
+    if (!XNN_PLATFORM_MOBILE && cpuinfo_has_x86_avx512f()) {
+      xnn_params.f32.dwconv[0] = (struct dwconv_parameters) {
+        .up = (xnn_dwconv_up_ukernel_function) xnn_f32_dwconv_ukernel_up16x4__avx512f,
+        .cr = 16,
+        .mr = 4,
+      };
+      xnn_params.f32.dwconv[1] = (struct dwconv_parameters) {
+        .up = (xnn_dwconv_up_ukernel_function) xnn_f32_dwconv_ukernel_up16x9__avx512f,
+        .cr = 16,
+        .mr = 9,
+      };
+      xnn_params.f32.dwconv[2] = (struct dwconv_parameters) {
+        .up = (xnn_dwconv_up_ukernel_function) xnn_f32_dwconv_ukernel_up16x25__avx512f,
+        .cr = 16,
+        .mr = 25,
+      };
+    } else if (!XNN_PLATFORM_MOBILE && cpuinfo_has_x86_fma3()) {
       xnn_params.f32.dwconv[0] = (struct dwconv_parameters) {
         .up = (xnn_dwconv_up_ukernel_function) xnn_f32_dwconv_ukernel_up16x4__fma3,
         .cr = 16,