jax
62d1b791 - Add pytest_rocm.yml workflow for ROCm testing

Commit
66 days ago
Add pytest_rocm.yml workflow for ROCm testing Create reusable workflow for ROCm pytest execution, following pytest_cuda.yml pattern. Runs in ROCm team's GHCR containers with GPU device access, supports configurable JAX versions (default 0.8.0), and uses custom jaxlib from GitHub releases or PyPI. Installs GitHub CLI for release downloads and executes run_pytest_rocm.sh for test execution.
Author
Committer
Parents
Loading