From 8a778a8350b301ef9cc4768261c0bf2e3146433c Mon Sep 17 00:00:00 2001 From: erman-gurses Date: Wed, 24 Jun 2026 06:23:39 +0000 Subject: [PATCH 1/9] Pass ROCm package index URL through JAX release workflow --- .github/workflows/multi_arch_release_linux_jax_wheels.yml | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/.github/workflows/multi_arch_release_linux_jax_wheels.yml b/.github/workflows/multi_arch_release_linux_jax_wheels.yml index 31653fcb..aafa28ea 100644 --- a/.github/workflows/multi_arch_release_linux_jax_wheels.yml +++ b/.github/workflows/multi_arch_release_linux_jax_wheels.yml @@ -18,6 +18,10 @@ on: description: ROCm package version to build against. type: string required: true + rocm_package_find_links_url: + description: "ROCm package index / find-links URL for the JAX manylinux build." + type: string + default: "https://rocm.devreleases.amd.com/whl-multi-arch/" tar_url: description: URL to the shared TheRock tarball used for the build. type: string @@ -45,6 +49,7 @@ jobs: test_amdgpu_family: ${{ inputs.test_amdgpu_family }} release_type: ${{ inputs.release_type }} rocm_version: ${{ inputs.rocm_version }} + rocm_package_find_links_url: ${{ inputs.rocm_package_find_links_url }} tar_url: ${{ inputs.tar_url }} repository: ${{ inputs.repository || 'ROCm/TheRock' }} ref: ${{ inputs.ref || '' }} From f956fc1396f6b865eed3a923c5c15609186083f5 Mon Sep 17 00:00:00 2001 From: erman-gurses Date: Wed, 24 Jun 2026 20:47:38 +0000 Subject: [PATCH 2/9] Add rocm_package_index_url and remove rocm_package_find_links_url --- .../workflows/multi_arch_release_linux_jax_wheels.yml | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/.github/workflows/multi_arch_release_linux_jax_wheels.yml b/.github/workflows/multi_arch_release_linux_jax_wheels.yml index aafa28ea..0f1b88fa 100644 --- a/.github/workflows/multi_arch_release_linux_jax_wheels.yml +++ b/.github/workflows/multi_arch_release_linux_jax_wheels.yml @@ -18,16 +18,16 @@ on: description: ROCm package version to build against. type: string required: true - rocm_package_find_links_url: - description: "ROCm package index / find-links URL for the JAX manylinux build." + rocm_package_index_url: + description: "URL for pip --index-url to install ROCm packages for the JAX manylinux build." type: string - default: "https://rocm.devreleases.amd.com/whl-multi-arch/" + default: "" tar_url: - description: URL to the shared TheRock tarball used for the build. + description: "URL to the shared TheRock tarball used for the build." type: string required: true ref: - description: Branch, tag, or SHA to checkout. + description: "Branch, tag, or SHA to checkout." type: string default: "" repository: @@ -49,7 +49,6 @@ jobs: test_amdgpu_family: ${{ inputs.test_amdgpu_family }} release_type: ${{ inputs.release_type }} rocm_version: ${{ inputs.rocm_version }} - rocm_package_find_links_url: ${{ inputs.rocm_package_find_links_url }} tar_url: ${{ inputs.tar_url }} repository: ${{ inputs.repository || 'ROCm/TheRock' }} ref: ${{ inputs.ref || '' }} From 5fd785cf2cb023036320e479f7d25fbae6556425 Mon Sep 17 00:00:00 2001 From: erman-gurses Date: Wed, 24 Jun 2026 20:49:46 +0000 Subject: [PATCH 3/9] Add quotes for descriptions --- .github/workflows/multi_arch_release_linux_jax_wheels.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/multi_arch_release_linux_jax_wheels.yml b/.github/workflows/multi_arch_release_linux_jax_wheels.yml index 0f1b88fa..7f24c641 100644 --- a/.github/workflows/multi_arch_release_linux_jax_wheels.yml +++ b/.github/workflows/multi_arch_release_linux_jax_wheels.yml @@ -15,7 +15,7 @@ on: type: string required: true rocm_version: - description: ROCm package version to build against. + description: "ROCm package version to build against." type: string required: true rocm_package_index_url: From 8f1e97d643934a7fc8faa1feb9346c111d1c74e0 Mon Sep 17 00:00:00 2001 From: erman-gurses Date: Wed, 24 Jun 2026 20:51:13 +0000 Subject: [PATCH 4/9] Add branch for test --- .github/workflows/multi_arch_release_linux_jax_wheels.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/multi_arch_release_linux_jax_wheels.yml b/.github/workflows/multi_arch_release_linux_jax_wheels.yml index 7f24c641..d49d86e6 100644 --- a/.github/workflows/multi_arch_release_linux_jax_wheels.yml +++ b/.github/workflows/multi_arch_release_linux_jax_wheels.yml @@ -43,7 +43,7 @@ run-name: Multi-Arch Release Linux JAX Wheels (${{ inputs.release_type }}, ${{ i jobs: build_jax_wheels: - uses: ROCm/TheRock/.github/workflows/multi_arch_release_linux_jax_wheels.yml@main + uses: ROCm/TheRock/.github/workflows/multi_arch_release_linux_jax_wheels.yml@users/erman-gurses/add-jax10x-support secrets: inherit with: test_amdgpu_family: ${{ inputs.test_amdgpu_family }} From 0a1ea936552ac7f6a4f121db5c920368f5090d37 Mon Sep 17 00:00:00 2001 From: erman-gurses Date: Wed, 24 Jun 2026 21:17:56 +0000 Subject: [PATCH 5/9] Pass rocm_package_index_url --- .github/workflows/multi_arch_release_linux_jax_wheels.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/multi_arch_release_linux_jax_wheels.yml b/.github/workflows/multi_arch_release_linux_jax_wheels.yml index d49d86e6..05808aea 100644 --- a/.github/workflows/multi_arch_release_linux_jax_wheels.yml +++ b/.github/workflows/multi_arch_release_linux_jax_wheels.yml @@ -49,6 +49,7 @@ jobs: test_amdgpu_family: ${{ inputs.test_amdgpu_family }} release_type: ${{ inputs.release_type }} rocm_version: ${{ inputs.rocm_version }} + rocm_package_index_url: ${{ inputs.rocm_package_index_url }} tar_url: ${{ inputs.tar_url }} repository: ${{ inputs.repository || 'ROCm/TheRock' }} ref: ${{ inputs.ref || '' }} From cafadf9fc91075ee2eec4b04cb05094c3f034485 Mon Sep 17 00:00:00 2001 From: erman-gurses Date: Thu, 25 Jun 2026 02:23:28 +0000 Subject: [PATCH 6/9] Test --- .github/workflows/multi_arch_release_linux_jax_wheels.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/multi_arch_release_linux_jax_wheels.yml b/.github/workflows/multi_arch_release_linux_jax_wheels.yml index 05808aea..5b69730d 100644 --- a/.github/workflows/multi_arch_release_linux_jax_wheels.yml +++ b/.github/workflows/multi_arch_release_linux_jax_wheels.yml @@ -23,7 +23,7 @@ on: type: string default: "" tar_url: - description: "URL to the shared TheRock tarball used for the build." + description: "URL to TheRock tarball used for the build." type: string required: true ref: From aa90a579deb3fc688906d0ee3c45c1cae13d9ce6 Mon Sep 17 00:00:00 2001 From: erman-gurses Date: Thu, 25 Jun 2026 19:51:04 +0000 Subject: [PATCH 7/9] Test --- .github/workflows/multi_arch_release.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/multi_arch_release.yml b/.github/workflows/multi_arch_release.yml index a8449fa6..cfe481c7 100644 --- a/.github/workflows/multi_arch_release.yml +++ b/.github/workflows/multi_arch_release.yml @@ -51,7 +51,7 @@ run-name: >- jobs: release: - uses: ROCm/TheRock/.github/workflows/multi_arch_release.yml@main + uses: ROCm/TheRock/.github/workflows/multi_arch_release.yml@users/erman-gurses/add-jax10x-support secrets: inherit with: release_type: ${{ inputs.release_type || 'nightly' }} From dc9f3941eeca93adf292810005c29c74ccda538f Mon Sep 17 00:00:00 2001 From: erman-gurses Date: Fri, 26 Jun 2026 00:42:00 +0000 Subject: [PATCH 8/9] Replace rocm_package_index_url with rocm_package_find_links_url --- .github/workflows/multi_arch_release_linux_jax_wheels.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/multi_arch_release_linux_jax_wheels.yml b/.github/workflows/multi_arch_release_linux_jax_wheels.yml index 5b69730d..e8869bdf 100644 --- a/.github/workflows/multi_arch_release_linux_jax_wheels.yml +++ b/.github/workflows/multi_arch_release_linux_jax_wheels.yml @@ -18,8 +18,8 @@ on: description: "ROCm package version to build against." type: string required: true - rocm_package_index_url: - description: "URL for pip --index-url to install ROCm packages for the JAX manylinux build." + rocm_package_find_links_url: + description: "URL for pip --find-links to install ROCm packages for the JAX manylinux build." type: string default: "" tar_url: @@ -49,7 +49,7 @@ jobs: test_amdgpu_family: ${{ inputs.test_amdgpu_family }} release_type: ${{ inputs.release_type }} rocm_version: ${{ inputs.rocm_version }} - rocm_package_index_url: ${{ inputs.rocm_package_index_url }} + rocm_package_find_links_url: ${{ inputs.rocm_package_find_links_url }} tar_url: ${{ inputs.tar_url }} repository: ${{ inputs.repository || 'ROCm/TheRock' }} ref: ${{ inputs.ref || '' }} From 5e77fc1f4c9f1d495b394c3e9e9ba6978b686dc6 Mon Sep 17 00:00:00 2001 From: erman-gurses Date: Fri, 26 Jun 2026 13:45:37 +0000 Subject: [PATCH 9/9] Revert pin to main branch --- .github/workflows/multi_arch_release.yml | 2 +- .github/workflows/multi_arch_release_linux_jax_wheels.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/multi_arch_release.yml b/.github/workflows/multi_arch_release.yml index cfe481c7..a8449fa6 100644 --- a/.github/workflows/multi_arch_release.yml +++ b/.github/workflows/multi_arch_release.yml @@ -51,7 +51,7 @@ run-name: >- jobs: release: - uses: ROCm/TheRock/.github/workflows/multi_arch_release.yml@users/erman-gurses/add-jax10x-support + uses: ROCm/TheRock/.github/workflows/multi_arch_release.yml@main secrets: inherit with: release_type: ${{ inputs.release_type || 'nightly' }} diff --git a/.github/workflows/multi_arch_release_linux_jax_wheels.yml b/.github/workflows/multi_arch_release_linux_jax_wheels.yml index e8869bdf..52edefad 100644 --- a/.github/workflows/multi_arch_release_linux_jax_wheels.yml +++ b/.github/workflows/multi_arch_release_linux_jax_wheels.yml @@ -43,7 +43,7 @@ run-name: Multi-Arch Release Linux JAX Wheels (${{ inputs.release_type }}, ${{ i jobs: build_jax_wheels: - uses: ROCm/TheRock/.github/workflows/multi_arch_release_linux_jax_wheels.yml@users/erman-gurses/add-jax10x-support + uses: ROCm/TheRock/.github/workflows/multi_arch_release_linux_jax_wheels.yml@main secrets: inherit with: test_amdgpu_family: ${{ inputs.test_amdgpu_family }}