more darwin targets

EnzymeAD · Apr 25, 2024 · 5198f99 · 5198f99
1 parent f4dfb97
commit 5198f99
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/deps/ReactantExtra/BUILD b/deps/ReactantExtra/BUILD
@@ -128,6 +128,22 @@ platform(
     ],
 )
 
+
+platform(
+    name = "darwin_x86_64_2",
+    constraint_values = [
+        "@platforms//os:macos",
+        "@platforms//cpu:darwin_x86_64",
+    ],
+)
+platform(
+    name = "darwin_arm64",
+    constraint_values = [
+        "@platforms//os:macos",
+        "@platforms//cpu:darwin_arm64",
+    ],
+)
+
 cc_library(
     name = "ReactantExtraLib",
     srcs = glob(

diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE
@@ -23,6 +23,9 @@ load("@enzyme_ad//:workspace.bzl", "JAX_COMMIT", "JAX_SHA256", "ENZYME_COMMIT",
 
 XLA_PATCHES = XLA_PATCHES + [
 """
+sed -i.bak0 "s/@tsl\\/\\/tsl:linux_x86_64/@bazel_tools\\/\\/src\\/conditions:linux/g" xla/tsl/mkl/build_defs.bzl
+""",
+"""
 sed -i.bak0 "s/\\\"@onednn\\/\\/:mkl_dnn\\\"//g" xla/tsl/mkl/build_defs.bzl
 """,
 """
-Original file line number
+Diff line change
@@ Expand Up @@
     XLA_PATCHES = XLA_PATCHES + [
     """
+    sed -i.bak0 "s/@tsl\\/\\/tsl:linux_x86_64/@bazel_tools\\/\\/src\\/conditions:linux/g" xla/tsl/mkl/build_defs.bzl
+    """,
+    """
     sed -i.bak0 "s/\\\"@onednn\\/\\/:mkl_dnn\\\"//g" xla/tsl/mkl/build_defs.bzl
     """,
     """
@@ Expand Down @@