Skip to content

Allow ROCm packages from a PEP 503 index (e.g. repo.amd.com) #53

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
ethanwee1 wants to merge 2 commits into
base: main
Choose a base branch
from
Draft

Allow ROCm packages from a PEP 503 index (e.g. repo.amd.com) #53

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
8bc641f
Allow ROCm packages from a PEP 503 index (e.g. repo.amd.com)
ethanwee1 Jun 8, 2026
10eb38a
Temporarily use TheRock index-url workflow branch
ethanwee1 Jun 8, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 8 additions & 3 deletions .github/workflows/multi_arch_release_linux_pytorch_wheels.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -46,11 +50,12 @@ 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 }}
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"
Expand Down
Loading