diff --git a/.github/workflows/multi_arch_release_linux_jax_wheels.yml b/.github/workflows/multi_arch_release_linux_jax_wheels.yml index 31653fcb..52edefad 100644 --- a/.github/workflows/multi_arch_release_linux_jax_wheels.yml +++ b/.github/workflows/multi_arch_release_linux_jax_wheels.yml @@ -15,15 +15,19 @@ 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_find_links_url: + description: "URL for pip --find-links to install ROCm packages for the JAX manylinux build." + 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: - description: Branch, tag, or SHA to checkout. + description: "Branch, tag, or SHA to checkout." type: string default: "" repository: @@ -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 || '' }}