From 8bc641f386305ca017c3346c0e2671fc5e0a3a1a Mon Sep 17 00:00:00 2001 From: ethanwee1 Date: Mon, 8 Jun 2026 17:15:39 +0000 Subject: [PATCH 1/2] Allow ROCm packages from a PEP 503 index (e.g. repo.amd.com) Add an optional rocm_package_index_url input and forward it to the TheRock reusable workflow, and make rocm_package_find_links_url optional. This lets the multi-arch PyTorch wheel release build install ROCm from a PEP 503 index such as the released wheels at repo.amd.com/rocm/whl-multi-arch instead of only a flat find-links page. --- .../multi_arch_release_linux_pytorch_wheels.yml | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/.github/workflows/multi_arch_release_linux_pytorch_wheels.yml b/.github/workflows/multi_arch_release_linux_pytorch_wheels.yml index da95e6c5..e75c87bb 100644 --- a/.github/workflows/multi_arch_release_linux_pytorch_wheels.yml +++ b/.github/workflows/multi_arch_release_linux_pytorch_wheels.yml @@ -14,9 +14,13 @@ on: type: string required: true rocm_package_find_links_url: - description: URL for pip --find-links to install ROCm packages + description: URL for pip --find-links to install ROCm packages (flat page; use this OR rocm_package_index_url) type: string - required: true + default: "" + rocm_package_index_url: + description: URL for pip --index-url to install ROCm packages (PEP 503 index, e.g. https://repo.amd.com/rocm/whl-multi-arch/; use this OR rocm_package_find_links_url) + type: string + default: "" release_type: description: 'Release type: "dev", "nightly", or "prerelease". Selects the S3 bucket.' type: choice @@ -51,6 +55,7 @@ jobs: amdgpu_families: ${{ inputs.amdgpu_families }} rocm_version: ${{ inputs.rocm_version }} rocm_package_find_links_url: ${{ inputs.rocm_package_find_links_url }} + rocm_package_index_url: ${{ inputs.rocm_package_index_url }} release_type: ${{ inputs.release_type }} cache_type: ${{ inputs.cache_type }} repository: "ROCm/TheRock" From 10eb38a3acd3c6811fd530b8f78fb66fb74f7a1f Mon Sep 17 00:00:00 2001 From: ethanwee1 Date: Mon, 8 Jun 2026 18:48:38 +0000 Subject: [PATCH 2/2] Temporarily use TheRock index-url workflow branch Route the rockrel wrapper to the TheRock PR branch so the pre-merge ROCm 7.13 released-index build can be dispatched before the reusable workflow change lands on main. --- .github/workflows/multi_arch_release_linux_pytorch_wheels.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/multi_arch_release_linux_pytorch_wheels.yml b/.github/workflows/multi_arch_release_linux_pytorch_wheels.yml index e75c87bb..d2fd354d 100644 --- a/.github/workflows/multi_arch_release_linux_pytorch_wheels.yml +++ b/.github/workflows/multi_arch_release_linux_pytorch_wheels.yml @@ -50,7 +50,7 @@ run-name: Multi-Arch Release Linux PyTorch Wheels (${{ inputs.amdgpu_families }} jobs: release: - uses: ROCm/TheRock/.github/workflows/multi_arch_release_linux_pytorch_wheels.yml@main + uses: ROCm/TheRock/.github/workflows/multi_arch_release_linux_pytorch_wheels.yml@ethanwee/pytorch-wheels-rocm-index-url with: amdgpu_families: ${{ inputs.amdgpu_families }} rocm_version: ${{ inputs.rocm_version }}