jax
b3ebfd48 - Add pytest_rocm.yml workflow for ROCm testing

Commit
52 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