Skip to content

Commit

Permalink
Update XLA pin to 11/22 (pytorch#8411)
Browse files Browse the repository at this point in the history
  • Loading branch information
JackCaoG authored Nov 25, 2024
1 parent 51575db commit 39e67b5
Show file tree
Hide file tree
Showing 4 changed files with 6 additions and 6 deletions.
2 changes: 1 addition & 1 deletion BUILD
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
load(
"@tsl//tsl/platform/default:cuda_build_defs.bzl",
"@xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)

Expand Down
2 changes: 1 addition & 1 deletion WORKSPACE
Original file line number Diff line number Diff line change
Expand Up @@ -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 = 'ed936d026b1aba1265f0030c2aa1ca6d5ad9d67b'
xla_hash = '30422ba2564eb108e12fb8a01629b2d0cf1f4393'

http_archive(
name = "xla",
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 = '20241118'
_date = '20241122'
_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}+nightly-py3-none-any.whl'
_jax_version = f'0.4.36.dev{_date}'
Expand Down
6 changes: 3 additions & 3 deletions torch_xla/csrc/runtime/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ load(
"ptxla_cc_test",
)
load(
"@tsl//tsl/platform/default:cuda_build_defs.bzl",
"@xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)

Expand Down Expand Up @@ -83,7 +83,7 @@ cc_library(
":tf_logging",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/types:span",
"@tsl//tsl/platform/cloud:gcs_file_system",
"@xla//xla/tsl/platform/cloud:gcs_file_system",
"@tsl//tsl/profiler/lib:traceme",
"@xla//xla:literal",
"@xla//xla:shape_util",
Expand Down Expand Up @@ -122,7 +122,7 @@ cc_library(
"@com_google_absl//absl/synchronization",
"@com_google_absl//absl/types:span",
"@tsl//tsl/platform:env",
"@tsl//tsl/platform/cloud:gcs_file_system",
"@xla//xla/tsl/platform/cloud:gcs_file_system",
"@tsl//tsl/profiler/lib:traceme",
"@xla//xla:literal",
"@xla//xla:shape_util",
Expand Down

0 comments on commit 39e67b5

Please sign in to comment.