Enable -mavx512fp16 needed for avx512fp16 microkernels

PiperOrigin-RevId: 633580841
google · May 15, 2024 · f1ba939 · f1ba939
1 parent 1ea7244
commit f1ba939
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -224,7 +224,7 @@ jobs:
           add-to-path: false
       - name: Build and run tests
         run: |
-          bazel test test:all
+          bazel test --define=xnn_enable_avx512fp16=false test:all
         working-directory: ${{ github.workspace }}
         env:
           ANDROID_NDK_HOME: ${{ steps.setup-ndk.outputs.ndk-path }}
diff --git a/BUILD.bazel b/BUILD.bazel
@@ -2715,7 +2715,7 @@ xnnpack_cc_library(
         "-mavx512vl",
         "-mavx512vnni",
         "-mgfni",
-        "-march=sapphirerapids",
+        "-mavx512fp16",
     ],
     mingw_copts = ["-fno-asynchronous-unwind-tables"],
     msvc_copts = xnnpack_msvc_std_copts(),
@@ -2747,7 +2747,7 @@ xnnpack_cc_library(
         "-mavx512vl",
         "-mavx512vnni",
         "-mgfni",
-        "-march=sapphirerapids",
+        "-mavx512fp16",
     ],
     mingw_copts = ["-fno-asynchronous-unwind-tables"],
     msvc_copts = xnnpack_msvc_std_copts(),
@@ -2783,7 +2783,7 @@ xnnpack_cc_library(
         "-mavx512vl",
         "-mavx512vnni",
         "-mgfni",
-        "-march=sapphirerapids",
+        "-mavx512fp16",
     ],
     mingw_copts = ["-fno-asynchronous-unwind-tables"],
     msvc_copts = xnnpack_msvc_std_copts(),