Revert "Go back to old XLA"

This reverts commit 2fc98df.
ROCm · Nov 13, 2024 · 3ee34af · 3ee34af
1 parent 2fc98df
commit 3ee34af
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/build/rocm/tools/build_wheels.py b/build/rocm/tools/build_wheels.py
@@ -303,7 +303,7 @@ def main():
             LOG.info("Copying %s into %s" % (whl, wheelhouse_dir))
             shutil.copy(whl, wheelhouse_dir)
 
-    # delete the 'dist' directory since it causes permissions issues
+# delete the 'dist' directory since it causes permissions issues
     logging.info('Deleting dist, egg-info and cache directory')
     shutil.rmtree(os.path.join(args.jax_path, "dist"))
     shutil.rmtree(os.path.join(args.jax_path, "jax.egg-info"))

diff --git a/third_party/xla/workspace.bzl b/third_party/xla/workspace.bzl
@@ -21,8 +21,8 @@ load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
 #    curl -L https://github.com/openxla/xla/archive/<git hash>.tar.gz | sha256sum
 #    and update XLA_SHA256 with the result.
 
-XLA_COMMIT = "95e3eea8d2aebd55160ed4185a38345ae98ab500"
-XLA_SHA256 = "7e23f70fe0b2459ac8a27e6b7b53e3643a053346244bd7e83c6b944827d0b9e1"
+XLA_COMMIT = "e93a258e4494231626c7d3b6a6447e746ea72f9c"
+XLA_SHA256 = "99f3a6b06230becf013f00009afeee4c89f52818e7a4a1ea4851157dc853830e"
 
 def repo():
     tf_http_archive(