jax
42553c7e - fixup! Add ROCm pytest workflow to nightly/release CI

Commit
28 days ago
fixup! Add ROCm pytest workflow to nightly/release CI Rename parameter to be generic for both CUDA and ROCm workflows. Change skip-download-jaxlib-and-cuda-plugins-from-gcs to skip-download-jaxlib-and-plugins-from-gcs to better reflect that this parameter controls skipping for all accelerator types (CUDA, ROCm, TPU), not just CUDA. This makes the parameter name more accurate and easier to understand.
Author
Committer
Parents
Loading