commit 0f349c496f82fa49aa09adbb257959eea4e74fd0
author Marat Dukhan <maratek@google.com> Wed Nov 27 11:58:54 2019 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Wed Nov 27 11:59:28 2019 -0800
tree a444fc9fb93b7a866c6728e2e9daf0ff0ebafd40
parent c72fa1e65f314612a9746a6ce783c0b218695f1b

AVX512F implementation of GEMM & IGEMM micro-kernels

PiperOrigin-RevId: 282810752

src/init.c

diff --git a/src/init.c b/src/init.c
index 2be2415..fd2aa12 100644
--- a/src/init.c
+++ b/src/init.c
@@ -622,7 +622,16 @@
 
   /**************************** F32 micro-kernels ****************************/
   #ifndef XNN_NO_F32_OPERATORS
-    if (!XNN_PLATFORM_MOBILE && cpuinfo_has_x86_fma3()) {
+    if (!XNN_PLATFORM_MOBILE && cpuinfo_has_x86_avx512f()) {
+      xnn_params.f32.gemm = (struct gemm_parameters) {
+        .gemm = (xnn_gemm_ukernel_function) xnn_f32_gemm_ukernel_7x16__avx512f_broadcast,
+        .igemm = (xnn_igemm_ukernel_function) xnn_f32_igemm_ukernel_7x16__avx512f_broadcast,
+        .gemm1 = (xnn_gemm_ukernel_function) xnn_f32_gemm_ukernel_1x16__avx512f_broadcast,
+        .igemm1 = (xnn_igemm_ukernel_function) xnn_f32_igemm_ukernel_1x16__avx512f_broadcast,
+        .mr = 7,
+        .nr = 16,
+      };
+    } else if (!XNN_PLATFORM_MOBILE && cpuinfo_has_x86_fma3()) {
       xnn_params.f32.gemm = (struct gemm_parameters) {
         .gemm = (xnn_gemm_ukernel_function) xnn_f32_gemm_ukernel_7x8__fma3_broadcast,
         .igemm = (xnn_igemm_ukernel_function) xnn_f32_igemm_ukernel_7x8__fma3_broadcast,