[6.4.0] Add blake3 NEON instructions on linux arm64 (#19804)

otherwise getting libunix_jni.so: undefined symbol: blake3_hash_many_neon see d0de5e0 Closes #19384. Commit 31812fe PiperOrigin-RevId: 562479599 Change-Id: I8f0ca6b68486f5ea8db698f3f99526eb1c0de8a8 Co-authored-by: Mauricio G <[email protected]>
bazelbuild · Oct 12, 2023 · ceb4f2d · ceb4f2d
1 parent 924bee9
commit ceb4f2d
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/third_party/blake3/blake3.BUILD b/third_party/blake3/blake3.BUILD
@@ -36,12 +36,18 @@ cc_library(
             "c/blake3_sse2_x86-64_unix.S",
             "c/blake3_sse41_x86-64_unix.S",
         ],
+        "@bazel_tools//src/conditions:linux_aarch64": [
+            "c/blake3_neon.c",
+        ],
         "@bazel_tools//src/conditions:windows_x64": [
             "c/blake3_avx2_x86-64_windows_msvc.asm",
             "c/blake3_avx512_x86-64_windows_msvc.asm",
             "c/blake3_sse2_x86-64_windows_msvc.asm",
             "c/blake3_sse41_x86-64_windows_msvc.asm",
         ],
+        "@bazel_tools//src/conditions:windows_arm64": [
+            "c/blake3_neon.c",
+        ],
         "@bazel_tools//src/conditions:darwin_arm64": [
             "c/blake3_neon.c",
         ],
@@ -57,9 +63,12 @@ cc_library(
             # lacking the headers to compile AVX512.
 	    "-DBLAKE3_NO_AVX512",
 	],
+        "@bazel_tools//src/conditions:linux_aarch64": [
+            "-DBLAKE3_USE_NEON=1",
+        ],
         "@bazel_tools//src/conditions:windows_x64": [],
         "@bazel_tools//src/conditions:windows_arm64": [
-            "-DBLAKE3_USE_NEON=0",
+            "-DBLAKE3_USE_NEON=1",
         ],
         "@bazel_tools//src/conditions:darwin_arm64": [
             "-DBLAKE3_USE_NEON=1",