Upstream ci download jax rocm wheels #592
mminutoli
wants to merge
19 commits into
upstream_rocm_ci
from
upstream_ci_download_jax_rocm_wheels
Add download-jax-rocm-wheels action for PyPI downloads
47965b7f
fixup: Remove GCS references from ROCm download action
350cacb7
adding run_pytest_rocm.sh
4b9488d9
Add GitHub release support for custom ROCm jaxlib download
c2b3554b
Add ROCm wheel patterns to install_wheels_locally.sh
f21ee5fb
Complete run_pytest_rocm.sh for CI integration
4b78046b
Add pytest_rocm.yml workflow for ROCm testing
faa72d59
Add workflow_dispatch to pytest_rocm.yml for manual triggering
71d47e86
Update pytest_rocm.yml to use AMD GFX942 runner
ee6a766f
Use ROCm team's GHCR containers for pytest_rocm workflow
198b3893
Fix container image tag generation in pytest_rocm workflow
01a1aa8b
Move env block before container block in pytest_rocm workflow
1e3e56be
Fix container image expression to use inputs directly
4b0a4872
Remove enable-x64 input from pytest_rocm workflow
69b2daea
hardwiring rocm version for test
566f6683
Fix image name
0c7a4dcc
Simplify ROCm configuration and fix wheel downloads
0e55c090
Add exec flag to the run pytest_rocm.sh script
07c3f265
Replace curl/wget with gh release download for ROCm wheels
b6cb4c58
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub