jax
8a61f1fb - [ROCm] Version-tier S3 layout, standardized CI

Commit
5 days ago
[ROCm] Version-tier S3 layout, standardized CI job naming, and nightly build workflow Version-tier S3 artifact layout: - Add rocm-release-version tier to S3 artifact paths to prevent wheel collisions across ROCm versions. - Add per-repo/branch/version LATEST pointer for wheel resolution. - Parameterize manylinux container image in build_rocm_artifacts.yml. - Plumb s3_download_uri through bazel_rocm.yml. Standardize CI job naming and matrix schema: - Unify ROCm matrix entries across all caller workflows to use a canonical schema with label, wheel-version, release-version, and tag fields. - Update job names to use matrix.rocm.label. - Add rocm-release-version input to pytest_rocm, bazel_rocm, and download-jax-rocm-wheels to decouple wheel filename matching from S3 path construction. - Add manylinux-image input to build_rocm_artifacts so callers can override the container image. Nightly workflow: - Add a dedicated build-rocm-artifacts job to the nightly workflow that builds wheels with pinned XLA (clone_main_xla: 0) and uploads to a nightly-specific S3 path. - Update nightly pytest and bazel ROCm jobs to depend on the new build.
Author
Committer
Parents
Loading