Skip to content

Bump xla #309

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
alekstheod wants to merge 3 commits into
base: master
Choose a base branch
from
Open

Bump xla #309

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
10f0c25
Update runner labels in Llama perf workflow
psanal35 Feb 11, 2026
b091fc2
Merge branch 'master' of github.com:ROCm/rocm-jax
alekstheod Feb 12, 2026
7b140cb
Bump xla
alekstheod Feb 12, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion jax_rocm_plugin/build/rocm/targets_to_ignore.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@ TARGETS_TO_IGNORE=(
-@jax//tests:gpu_memory_flags_test_gpu
-@jax//tests:lax_numpy_test_gpu
-@jax//tests:lax_test_gpu
-@jax//tests:linalg_test_gpu
Copy link
Collaborator

@charleshofer charleshofer Feb 12, 2026 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I assume that this test case was failing in CI and that's why you're removing it? If that's the case, no complaints from me.

Copy link
Contributor Author

@alekstheod alekstheod Feb 12, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes the test was failing. I think in your pipeline you have several --test_filter cases. In bazel the last one overrides. So in fact not all the tests were executed but only the one in the last filter. Usually we do --test_filter="Test1|Test2|Test3" Not sure however if that is a correct syntax for asltest but bazelwise --test_filter several times is not correct.

)

echo "${TARGETS_TO_IGNORE[*]}"
4 changes: 2 additions & 2 deletions jax_rocm_plugin/third_party/xla/workspace.bzl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,8 @@ load("//third_party:repo.bzl", "amd_http_archive")
# curl -L https://github.com/openxla/xla/archive/<git hash>.tar.gz | sha256sum
# and update XLA_SHA256 with the result.

XLA_COMMIT = "24c5f10ae8fc24aefd20b43c501ade7f66fd0cfd"
XLA_SHA256 = "f00db8761e86bcb51b52e64bc983717181050c8752c040e33ecb1429d861c30b"
XLA_COMMIT = "a15e41b80f82e4160ba78851d80eaa90cb7e8435"
XLA_SHA256 = "777f137337ac80ed72a77f82d1e610008a217f0daed384243fac676806919ffd"

def repo():
amd_http_archive(
Expand Down
Loading