Skip to content

Comments

Add script for running bazel tests on ROCm#632

Draft
charleshofer wants to merge 1 commit intorocm-jaxlib-v0.8.2from
add-run-bazel-test-rocm-rbe-0.8.2
Draft

Add script for running bazel tests on ROCm#632
charleshofer wants to merge 1 commit intorocm-jaxlib-v0.8.2from
add-run-bazel-test-rocm-rbe-0.8.2

Conversation

@charleshofer
Copy link
Collaborator

@charleshofer charleshofer commented Jan 21, 2026

Add a script for running Bazel tests on ROCm using a similar script to https://github.com/jax-ml/jax/blob/main/ci/run_bazel_test_cuda_rbe.sh

TODO:

  • Don't comment out CUDA-related targets to make this script work
  • Set up workflows to run this script (try and re-use CUDA as much as possible)
  • Either wait for the 0.8.2. release for use in tests, or find a way to run by building jaxlib in-place

@alekstheod
Copy link

Check my changes here. They somehow include this part as well!
jax-ml#34462
Also a patch from my rbe PR to rocm-jax

@charleshofer charleshofer force-pushed the add-run-bazel-test-rocm-rbe-0.8.2 branch 3 times, most recently from bfa2cfe to fa37273 Compare January 26, 2026 20:12
@charleshofer charleshofer force-pushed the add-run-bazel-test-rocm-rbe-0.8.2 branch from fa37273 to 7638166 Compare January 26, 2026 20:22
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants