fix apple build

EnzymeAD · Jan 15, 2025 · 0d2bbce · 0d2bbce
1 parent 362f33f
commit 0d2bbce
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/enzyme_ad/jax/BUILD b/src/enzyme_ad/jax/BUILD
@@ -313,6 +313,7 @@ cc_library(
         "@llvm-project//llvm:Passes",
         "@llvm-project//llvm:Scalar",
         "@llvm-project//llvm:InstCombine",
+        "@llvm-project//llvm:IPO",
         ":mhlo-derivatives",
         ":stablehlo-derivatives",
         ":chlo-derivatives",

diff --git a/workspace.bzl b/workspace.bzl
@@ -1,4 +1,4 @@
-JAX_COMMIT = "aeac6b03833f9f16fffa4b034defc03b6d9a98c6"
+JAX_COMMIT = "afcb21ddf1075c908a040f23be07f083eb58f473"
 JAX_SHA256 = ""
 
 ENZYME_COMMIT = "565163635fe98c7e0da004b4ac6dd2cd45bc880c"