Skip to content

Commit

Permalink
Update XLA/JAX/libtpu pins to 2024/05/02 (#7020)
Browse files Browse the repository at this point in the history
  • Loading branch information
will-cromar authored May 2, 2024
1 parent 93ce054 commit 213b72b
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions WORKSPACE
Original file line number Diff line number Diff line change
Expand Up @@ -50,9 +50,9 @@ http_archive(
"//openxla_patches:gpu_race_condition.diff",
"//openxla_patches:f16_abi_clang.diff",
],
strip_prefix = "xla-fe08041b23d8baa0d00967913a1d6e8a0c348df3",
strip_prefix = "xla-80462ef5b22360df177fe24fc13c81b235d3f3a2",
urls = [
"https://github.com/openxla/xla/archive/fe08041b23d8baa0d00967913a1d6e8a0c348df3.tar.gz",
"https://github.com/openxla/xla/archive/80462ef5b22360df177fe24fc13c81b235d3f3a2.tar.gz",
],
)

Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@

base_dir = os.path.dirname(os.path.abspath(__file__))

_date = '20240425'
_date = '20240502'
_libtpu_version = f'0.1.dev{_date}'
_libtpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}-py3-none-any.whl'
_jax_version = f'0.4.27.dev{_date}'
Expand Down

0 comments on commit 213b72b

Please sign in to comment.