Update repositories.bzl

anakinxc · Dec 1, 2023 · ef086fe · ef086fe
1 parent 75fe4d1
commit ef086fe
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/bazel/repositories.bzl b/bazel/repositories.bzl
@@ -52,6 +52,19 @@ def spu_deps():
         remote = "{}/yacl.git".format(SECRETFLOW_GIT),
     )
 
+    # Add homebrew openmp for macOS, somehow..homebrew installs to different location on Apple Silcon/Intel macs.. so we need two rules here
+    native.new_local_repository(
+        name = "local_homebrew_x64",
+        build_file = "@spulib//bazel:local_openmp_macos.BUILD",
+        path = "/usr/local/opt/libomp",
+    )
+
+    native.new_local_repository(
+        name = "local_homebrew_arm64",
+        build_file = "@spulib//bazel:local_openmp_macos.BUILD",
+        path = "/opt/homebrew/opt/libomp/",
+    )
+
 def _rules_proto_grpc():
     http_archive(
         name = "rules_proto_grpc",