Skip to content

[ROCm] Build prototype wheels against PyTorch 2.12 #1060

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

Merged
mgehre-amd merged 1 commit into from
Jul 17, 2026
Merged

[ROCm] Build prototype wheels against PyTorch 2.12 #1060

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
19 changes: 13 additions & 6 deletions .github/workflows/build-rocm-wheels.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -59,11 +59,16 @@ jobs:
# Resolve torch on the ROCm nightly index, then derive matching
# torchvision/torchaudio pins. The nightly publishes multiple
# base versions side-by-side (2.9, 2.10, 2.11, 2.12); pin to
# 2.11 to match requirements/build/rocm.txt. The three packages
# must share the same +rocm build-date suffix or they
# 2.12 to match requirements/build/rocm.txt. torch and
# torchvision must share the same +rocm build-date suffix or they
# ABI-mismatch at import ("operator torchvision::nms does not
# exist").
echo 'torch==2.11.*' > constraints.in
# exist"). torchaudio is a special case: it was retired from the
# PyTorch release process at 2.11 (I/O moved to TorchCodec, see
# pytorch/audio#3902), so there is no 2.12 build and never will
# be. It is pinned at its terminal 2.11.0 release (same +rocm
# suffix), which declares no torch version dependency and floats
# across torch minors.
echo 'torch==2.12.*' > constraints.in
# --no-deps: torch pins exact rocm[libraries]/triton versions
# in its metadata (e.g. rocm[libraries]==7.14.0a..., triton==
# 3.7.0+rocm...); we don't want those locked in constraints.
Expand All @@ -76,8 +81,10 @@ jobs:
# +rocm local version uv just picked for torch.
TORCH_LOCAL=$(grep '^torch==' constraints.txt | sed 's/.*+//')
TORCH_PATCH_SUFFIX=$(grep '^torch==' constraints.txt | sed 's/^torch==[0-9]\+\.[0-9]\+\.\([0-9]\+[^+]*\)+.*/\1/')
echo "torchvision==0.26.${TORCH_PATCH_SUFFIX}+${TORCH_LOCAL}" >> constraints.txt
echo "torchaudio==2.11.${TORCH_PATCH_SUFFIX}+${TORCH_LOCAL}" >> constraints.txt
echo "torchvision==0.27.${TORCH_PATCH_SUFFIX}+${TORCH_LOCAL}" >> constraints.txt
# torchaudio is frozen at 2.11.0 (no 2.12+ exists); hardcode the
# base version and only inherit the +rocm build-date suffix.
echo "torchaudio==2.11.0+${TORCH_LOCAL}" >> constraints.txt
cat constraints.txt
# unsafe-best-match: uv defaults to stopping at the first
# index that has a given package name, so PyPI (extra-index)
Expand Down
6 changes: 4 additions & 2 deletions requirements/build/rocm.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,10 @@


--extra-index-url https://download.pytorch.org/whl/rocm7.1
torch==2.11.0
torchvision==0.26.0
torch==2.12.0
torchvision==0.27.0
# torchaudio was retired at 2.11 (no 2.12+; see pytorch/audio#3902). Its
# terminal release declares no torch pin and floats across torch minors.
torchaudio==2.11.0
triton
cmake>=3.26.1,<4
Expand Down
Loading