jax
bd4cbc8c - Exit the CUDA test job early if not all wheels can be downloaded.

Commit

185 days ago

Exit the CUDA test job early if not all wheels can be downloaded. Should fix errors such as https://github.com/jax-ml/jax/actions/runs/16368343158/job/46250704316 where the test job didn't correctly exit right the "Download wheels from GCS" step. PiperOrigin-RevId: 784669391

References

#30316 - Exit the CUDA test job early if not all wheels can be downloaded.

#31381 - Remove old ROCm build code

#31720 - Fix ann_test.py numerical bug in target reshape

#31768 - [ROCm] Support lowering through PJRT_Triton_Extension

#32115 - Relax version requirements for ROCm Jax Plugin wheels

#33157 - Resolve undefined behavior in bitshift unit test

#579 - Create rocm-test-requirements.txt

#581 - Fix/pallas tests shared memory

#584 - Use plain bazel to test jax, use hermetic rocm dependency

#585 - update a test for checking zero ROCm GPU event

#34135 - [ROCm] update to test if there are GPU events when doing profiling on…