Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
openxla pin update to 20250123
Browse files Browse the repository at this point in the history
zpcore committed Jan 24, 2025
1 parent d382fee commit 79b33c2
Showing 2 changed files with 5 additions and 4 deletions.
3 changes: 2 additions & 1 deletion WORKSPACE
Original file line number Diff line number Diff line change
@@ -50,7 +50,8 @@ 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 = '8d06f3680ad046ea44f8e7159f52c728bb66c069'
# (based on openxla commit 6e91ff19dad528ab7d2025a9bb46150618a3bc7d)
xla_hash = 'a820fc0c16ec40728748874dcde92bc7aa5d18daa01e72e9ca2ed4e706e19f3d'

http_archive(
name = "xla",
6 changes: 3 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
@@ -65,10 +65,10 @@
base_dir = os.path.dirname(os.path.abspath(__file__))

USE_NIGHTLY = True # whether to use nightly or stable libtpu and jax
_date = '20250113'
_date = '20250123'
_libtpu_version = f'0.0.8'
_jax_version = f'0.4.39'
_jaxlib_version = f'0.4.39'
_jax_version = f'0.5.1'
_jaxlib_version = f'0.5.1'
_libtpu_wheel_name = f'libtpu-{_libtpu_version}'
_libtpu_storage_directory = 'libtpu-lts-releases'

0 comments on commit 79b33c2

Please sign in to comment.