jax
15ed381b - Upgrade upstream ROCm CI from 7.2.0 to 7.2.2

Commit
24 days ago
Upgrade upstream ROCm CI from 7.2.0 to 7.2.2 Point the three upstream ROCm workflows at the new 7.2.2 base images published by ROCm/rocm-jax: - build_rocm_artifacts.yml: switch the manylinux_2_28 builder image from jax-manylinux_2_28-rocm-7.2.0:latest to ...-rocm-7.2.2:latest. - bazel_rocm.yml / pytest_rocm.yml: bump the default rocm-tag from rocm720 to rocm722 (and the default rocm-version from 7.2.0 to 7.2.2 for pytest_rocm.yml), so the jobs pull the jax-dev-ubu24.rocm722 / jax-base-ubu24.rocm722 containers. - wheel_tests_continuous.yml / wheel_tests_nightly_release.yml: update the rocm matrix entry to {version: "7.2.2", tag: "rocm722"}. - download-jax-rocm-wheels composite action: bump the default rocm-version to 7.2.2 to match.
Author
Committer
Parents
Loading