commit bb00b1dda26b55c80f7a737ad374dfc312bd92b0
author Marat Dukhan <maratek@google.com> Mon Aug 10 11:37:23 2020 -0700
committer XNNPACK Team <xnnpack-github-robot@google.com> Mon Aug 10 11:38:00 2020 -0700
tree 260af38d3e6c650055de9d4b11d2e95779c26653
parent facecc526bf35d09ac3319f44072d640101b2b70

AVX512 variants of QS8 GEMM and IGEMM microkernels

PiperOrigin-RevId: 325850791

BUILD.bazel

diff --git a/BUILD.bazel b/BUILD.bazel
index 2cc6155..f5f5a93 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -2416,6 +2416,17 @@
     "src/math/sqrt-avx512f-nr1fma1adj.c",
 ]
 
+AVX512SKX_UKERNELS = [
+    "src/qs8-gemm/gen/1x16c8-minmax-avx512skx.c",
+    "src/qs8-gemm/gen/2x16c8-minmax-avx512skx.c",
+    "src/qs8-gemm/gen/3x16c8-minmax-avx512skx.c",
+    "src/qs8-gemm/gen/4x16c8-minmax-avx512skx.c",
+    "src/qs8-igemm/gen/1x16c8-minmax-avx512skx.c",
+    "src/qs8-igemm/gen/2x16c8-minmax-avx512skx.c",
+    "src/qs8-igemm/gen/3x16c8-minmax-avx512skx.c",
+    "src/qs8-igemm/gen/4x16c8-minmax-avx512skx.c",
+]
+
 AARCH32_ASM_UKERNELS = [
     "src/f32-gemm/4x4-aarch32-vfp-ld64.S",
     "src/f32-gemm/4x4-minmax-aarch32-vfp-ld64.S",
@@ -3257,6 +3268,58 @@
 )
 
 xnnpack_cc_library(
+    name = "avx512skx_ukernels",
+    hdrs = INTERNAL_HDRS,
+    gcc_copts = xnnpack_gcc_std_copts(),
+    gcc_x86_copts = [
+        "-mavx512f",
+        "-mavx512cd",
+        "-mavx512bw",
+        "-mavx512dq",
+        "-mavx512vl",
+    ],
+    mingw_copts = ["-fno-asynchronous-unwind-tables"],
+    msvc_copts = xnnpack_msvc_std_copts(),
+    msvc_x86_32_copts = ["/arch:AVX512"],
+    msvc_x86_64_copts = ["/arch:AVX512"],
+    msys_copts = ["-fno-asynchronous-unwind-tables"],
+    x86_srcs = AVX512SKX_UKERNELS,
+    deps = [
+        ":tables",
+        "@FP16",
+        "@pthreadpool",
+    ],
+)
+
+xnnpack_cc_library(
+    name = "avx512skx_ukernels_test_mode",
+    hdrs = INTERNAL_HDRS,
+    copts = [
+        "-UNDEBUG",
+        "-DXNN_TEST_MODE=1",
+    ],
+    gcc_copts = xnnpack_gcc_std_copts(),
+    gcc_x86_copts = [
+        "-mavx512f",
+        "-mavx512cd",
+        "-mavx512bw",
+        "-mavx512dq",
+        "-mavx512vl",
+    ],
+    mingw_copts = ["-fno-asynchronous-unwind-tables"],
+    msvc_copts = xnnpack_msvc_std_copts(),
+    msvc_x86_32_copts = ["/arch:AVX512"],
+    msvc_x86_64_copts = ["/arch:AVX512"],
+    msys_copts = ["-fno-asynchronous-unwind-tables"],
+    x86_srcs = AVX512SKX_UKERNELS,
+    deps = [
+        ":tables",
+        "@FP16",
+        "@pthreadpool",
+    ],
+)
+
+xnnpack_cc_library(
     name = "asm_ukernels",
     hdrs = ["src/xnnpack/assembly.h"],
     aarch32_srcs = AARCH32_ASM_UKERNELS,
@@ -3322,6 +3385,7 @@
         ":fma3_ukernels",
         ":avx2_ukernels",
         ":avx512f_ukernels",
+        ":avx512skx_ukernels",
     ],
 )
 
@@ -3362,6 +3426,7 @@
         ":fma3_ukernels_test_mode",
         ":avx2_ukernels_test_mode",
         ":avx512f_ukernels_test_mode",
+        ":avx512skx_ukernels_test_mode",
     ],
 )
 
@@ -3404,6 +3469,7 @@
         ":fma3_ukernels",
         ":avx2_ukernels",
         ":avx512f_ukernels",
+        ":avx512skx_ukernels",
     ],
 )