jax
0a12efc0 - fixup! Add ROCm pytest workflow to nightly/release CI

Commit
28 days ago
fixup! Add ROCm pytest workflow to nightly/release CI Update ROCm job to use GCS downloads and match CUDA workflow pattern. Changes: - Remove parameters: jax-version, github-release-tag, download-jax-from-pypi - Change jaxlib-version from github_release to head - Add gcs_download_uri parameter from workflow inputs - Rename skip-download-jaxlib-and-rocm-plugins to skip-download-jaxlib-and-rocm-plugins-from-gcs - ROCm plugins now downloaded from ROCm/jax-plugin latest release
Author
Committer
Parents
Loading