Properly adapt to upstream

EnzymeAD · Jan 15, 2025 · 12dc0bf · 12dc0bf
1 parent 0d2bbce
commit 12dc0bf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/third_party/xla/workspace.bzl b/third_party/xla/workspace.bzl
@@ -1,10 +1,8 @@
 """Loads XLA."""
 
 load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
-# load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256")
+load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256")
 load("//:workspace.bzl", "XLA_PATCHES")
-XLA_COMMIT = "1bb4fc18e73faa1c001d96bfe3a22f733987b018"
-XLA_SHA256 = ""
 
 def repo():
     http_archive(

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