Update XLA Pin (#7797)

Co-authored-by: Siyuan Liu <[email protected]>
pytorch · Aug 2, 2024 · dd3b00c · dd3b00c
1 parent dfd7b00
commit dd3b00c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -50,7 +50,7 @@ new_local_repository(
 #    curl -L https://github.com/openxla/xla/archive/<git hash>.tar.gz | sha256sum
 #    and update the sha256 with the result.
 
-xla_hash = '6c5d5686515f4ea2e505b9f7dd85cbde78f51b4f'
+xla_hash = '336c9a99eeebb04a4ad9a8e6ce3b8d96e5ecd5e7'
 
 http_archive(
     name = "xla",

diff --git a/setup.py b/setup.py
@@ -64,10 +64,10 @@
 
 base_dir = os.path.dirname(os.path.abspath(__file__))
 
-_date = '20240726'
+_date = '20240801'
 _libtpu_version = f'0.1.dev{_date}'
-_libtpu_storage_path = f'https://storage.googleapis.com/libtpu-nightly-releases/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}+default-py3-none-any.whl'
-_jax_version = f'0.4.31.dev{_date}'
+_libtpu_storage_path = f'https://storage.googleapis.com/libtpu-nightly-releases/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}+nightly-py3-none-any.whl'
+_jax_version = f'0.4.32.dev{_date}'
 
 
 def _get_build_mode():