Skip to content

Commit 8867a5a

Browse files
tinglvvtinglvv
authored
Add CUDA 13.2 domain builds (#7865)
pytorch/pytorch#177067 Add 13.2 builds for torchvision and torchaudio
1 parent c40b2b3 commit 8867a5a

8 files changed

Lines changed: 503 additions & 4 deletions

‎release/pypi/promote_wheel_variants.sh‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,7 @@ declare -a TORCH_VARIANTS=(
105105
"cu126|manylinux_2_28_x86_64|%2Bcu126"
106106
"cu128|manylinux_2_28_x86_64|%2Bcu128"
107107
"cu130|manylinux_2_28_x86_64|%2Bcu130"
108+
"cu132|manylinux_2_28_x86_64|%2Bcu132"
108109
"xpu|linux_x86_64|%2Bxpu"
109110
"rocm7.0|manylinux_2_28_x86_64|%2Brocm7.0"
110111
"rocm7.1|manylinux_2_28_x86_64|%2Brocm7.1"
@@ -113,11 +114,13 @@ declare -a TORCH_VARIANTS=(
113114
"cu126|manylinux_2_28_aarch64|%2Bcu126"
114115
"cu128|manylinux_2_28_aarch64|%2Bcu128"
115116
"cu130|manylinux_2_28_aarch64|%2Bcu130"
117+
"cu132|manylinux_2_28_aarch64|%2Bcu132"
116118
# Windows
117119
"cpu|win_amd64|%2Bcpu"
118120
"cu126|win_amd64|%2Bcu126"
119121
"cu128|win_amd64|%2Bcu128"
120122
"cu130|win_amd64|%2Bcu130"
123+
"cu132|win_amd64|%2Bcu132"
121124
"xpu|win_amd64|%2Bxpu"
122125
# macOS (no suffix for CPU-only)
123126
"cpu|macosx_.*_arm64|"
@@ -129,6 +132,7 @@ declare -a TORCHVISION_VARIANTS=(
129132
"cu126|manylinux_2_28_x86_64|%2Bcu126"
130133
"cu128|manylinux_2_28_x86_64|%2Bcu128"
131134
"cu130|manylinux_2_28_x86_64|%2Bcu130"
135+
"cu132|manylinux_2_28_x86_64|%2Bcu132"
132136
"xpu|manylinux_2_28_x86_64|%2Bxpu"
133137
"rocm7.0|manylinux_2_28_x86_64|%2Brocm7.0"
134138
"rocm7.1|manylinux_2_28_x86_64|%2Brocm7.1"
@@ -137,11 +141,13 @@ declare -a TORCHVISION_VARIANTS=(
137141
"cu126|manylinux_2_28_aarch64|"
138142
"cu128|manylinux_2_28_aarch64|"
139143
"cu130|manylinux_2_28_aarch64|"
144+
"cu132|manylinux_2_28_aarch64|"
140145
# Windows
141146
"cpu|win_amd64|%2Bcpu"
142147
"cu126|win_amd64|%2Bcu126"
143148
"cu128|win_amd64|%2Bcu128"
144149
"cu130|win_amd64|%2Bcu130"
150+
"cu132|win_amd64|%2Bcu132"
145151
"xpu|win_amd64|%2Bxpu"
146152
# macOS
147153
"cpu|macosx_.*_arm64|"
@@ -153,13 +159,15 @@ declare -a TORCHAUDIO_VARIANTS=(
153159
"cu126|manylinux_2_28_x86_64|%2Bcu126"
154160
"cu128|manylinux_2_28_x86_64|%2Bcu128"
155161
"cu130|manylinux_2_28_x86_64|%2Bcu130"
162+
"cu132|manylinux_2_28_x86_64|%2Bcu132"
156163
# Linux aarch64
157164
"cpu|manylinux_2_28_aarch64|%2Bcpu"
158165
# Windows
159166
"cpu|win_amd64|%2Bcpu"
160167
"cu126|win_amd64|%2Bcu126"
161168
"cu128|win_amd64|%2Bcu128"
162169
"cu130|win_amd64|%2Bcu130"
170+
"cu132|win_amd64|%2Bcu132"
163171
# macOS
164172
"cpu|macosx_.*_arm64|"
165173
)

‎tools/pkg-helpers/pytorch_pkg_helpers/cuda.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def get_cuda_arch_list(
1616
# however we would like to keep sm_70 architecture see: https://github.com/pytorch/pytorch/issues/157517
1717
if sanitized_version == "12.8":
1818
return "7.0;7.5;8.0;8.6;9.0;10.0;12.0"
19-
elif sanitized_version == "13.0":
19+
elif sanitized_version == "13.0" or sanitized_version == "13.2":
2020
arch_list = "7.5;8.0;8.6;9.0;10.0;12.0+PTX"
2121
# Add sm_110 for aarch64
2222
if "aarch64" in platform:

‎tools/scripts/generate_binary_build_matrix.py‎

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,8 @@
3535
"3.14": "3.14.3",
3636
}
3737
CUDA_ARCHES_DICT = {
38-
"nightly": ["12.6", "12.8", "13.0"],
39-
"test": ["12.6", "12.8", "13.0"],
38+
"nightly": ["12.6", "12.8", "13.0", "13.2"],
39+
"test": ["12.6", "12.8", "13.0", "13.2"],
4040
"release": ["12.6", "12.8", "13.0"],
4141
}
4242

@@ -51,6 +51,7 @@
5151
"12.8": {"cuda": "12.8.0", "cudnn": "9"},
5252
"12.9": {"cuda": "12.9.1", "cudnn": "9"},
5353
"13.0": {"cuda": "13.0.0", "cudnn": "9"},
54+
"13.2": {"cuda": "13.2.0", "cudnn": "9"},
5455
}
5556

5657
STABLE_CUDA_VERSIONS = {
@@ -59,7 +60,7 @@
5960
"release": "13.0",
6061
}
6162

62-
CUDA_AARCH64_ARCHES = ["12.6-aarch64", "12.8-aarch64", "13.0-aarch64"]
63+
CUDA_AARCH64_ARCHES = ["12.6-aarch64", "12.8-aarch64", "13.0-aarch64", "13.2-aarch64"]
6364

6465
PACKAGE_TYPES = ["wheel", "libtorch"]
6566
CXX11_ABI = "cxx11-abi"

‎tools/tests/assets/build_matrix_linux_wheel_cuda.json‎

Lines changed: 98 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,20 @@
5656
"upload_to_base_bucket": "no",
5757
"stable_version": "2.11.0"
5858
},
59+
{
60+
"python_version": "3.10",
61+
"gpu_arch_type": "cuda",
62+
"gpu_arch_version": "13.2",
63+
"desired_cuda": "cu132",
64+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
65+
"package_type": "manywheel",
66+
"build_name": "manywheel-py3_10-cuda13_2",
67+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
68+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
69+
"channel": "nightly",
70+
"upload_to_base_bucket": "no",
71+
"stable_version": "2.11.0"
72+
},
5973
{
6074
"python_version": "3.10",
6175
"gpu_arch_type": "cuda",
@@ -154,6 +168,20 @@
154168
"upload_to_base_bucket": "no",
155169
"stable_version": "2.11.0"
156170
},
171+
{
172+
"python_version": "3.11",
173+
"gpu_arch_type": "cuda",
174+
"gpu_arch_version": "13.2",
175+
"desired_cuda": "cu132",
176+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
177+
"package_type": "manywheel",
178+
"build_name": "manywheel-py3_11-cuda13_2",
179+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
180+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
181+
"channel": "nightly",
182+
"upload_to_base_bucket": "no",
183+
"stable_version": "2.11.0"
184+
},
157185
{
158186
"python_version": "3.11",
159187
"gpu_arch_type": "cuda",
@@ -252,6 +280,20 @@
252280
"upload_to_base_bucket": "no",
253281
"stable_version": "2.11.0"
254282
},
283+
{
284+
"python_version": "3.12",
285+
"gpu_arch_type": "cuda",
286+
"gpu_arch_version": "13.2",
287+
"desired_cuda": "cu132",
288+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
289+
"package_type": "manywheel",
290+
"build_name": "manywheel-py3_12-cuda13_2",
291+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
292+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
293+
"channel": "nightly",
294+
"upload_to_base_bucket": "no",
295+
"stable_version": "2.11.0"
296+
},
255297
{
256298
"python_version": "3.12",
257299
"gpu_arch_type": "cuda",
@@ -350,6 +392,20 @@
350392
"upload_to_base_bucket": "no",
351393
"stable_version": "2.11.0"
352394
},
395+
{
396+
"python_version": "3.13",
397+
"gpu_arch_type": "cuda",
398+
"gpu_arch_version": "13.2",
399+
"desired_cuda": "cu132",
400+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
401+
"package_type": "manywheel",
402+
"build_name": "manywheel-py3_13-cuda13_2",
403+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
404+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
405+
"channel": "nightly",
406+
"upload_to_base_bucket": "no",
407+
"stable_version": "2.11.0"
408+
},
353409
{
354410
"python_version": "3.13",
355411
"gpu_arch_type": "cuda",
@@ -448,6 +504,20 @@
448504
"upload_to_base_bucket": "no",
449505
"stable_version": "2.11.0"
450506
},
507+
{
508+
"python_version": "3.13t",
509+
"gpu_arch_type": "cuda",
510+
"gpu_arch_version": "13.2",
511+
"desired_cuda": "cu132",
512+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
513+
"package_type": "manywheel",
514+
"build_name": "manywheel-py3_13t-cuda13_2",
515+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
516+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
517+
"channel": "nightly",
518+
"upload_to_base_bucket": "no",
519+
"stable_version": "2.11.0"
520+
},
451521
{
452522
"python_version": "3.13t",
453523
"gpu_arch_type": "cuda",
@@ -546,6 +616,20 @@
546616
"upload_to_base_bucket": "no",
547617
"stable_version": "2.11.0"
548618
},
619+
{
620+
"python_version": "3.14",
621+
"gpu_arch_type": "cuda",
622+
"gpu_arch_version": "13.2",
623+
"desired_cuda": "cu132",
624+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
625+
"package_type": "manywheel",
626+
"build_name": "manywheel-py3_14-cuda13_2",
627+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
628+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
629+
"channel": "nightly",
630+
"upload_to_base_bucket": "no",
631+
"stable_version": "2.11.0"
632+
},
549633
{
550634
"python_version": "3.14",
551635
"gpu_arch_type": "cuda",
@@ -644,6 +728,20 @@
644728
"upload_to_base_bucket": "no",
645729
"stable_version": "2.11.0"
646730
},
731+
{
732+
"python_version": "3.14t",
733+
"gpu_arch_type": "cuda",
734+
"gpu_arch_version": "13.2",
735+
"desired_cuda": "cu132",
736+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
737+
"package_type": "manywheel",
738+
"build_name": "manywheel-py3_14t-cuda13_2",
739+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
740+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
741+
"channel": "nightly",
742+
"upload_to_base_bucket": "no",
743+
"stable_version": "2.11.0"
744+
},
647745
{
648746
"python_version": "3.14t",
649747
"gpu_arch_type": "cuda",

‎tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json‎

Lines changed: 98 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,20 @@
5656
"upload_to_base_bucket": "no",
5757
"stable_version": "2.11.0"
5858
},
59+
{
60+
"python_version": "3.10",
61+
"gpu_arch_type": "cuda",
62+
"gpu_arch_version": "13.2",
63+
"desired_cuda": "cu132",
64+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
65+
"package_type": "manywheel",
66+
"build_name": "manywheel-py3_10-cuda13_2",
67+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
68+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
69+
"channel": "nightly",
70+
"upload_to_base_bucket": "no",
71+
"stable_version": "2.11.0"
72+
},
5973
{
6074
"python_version": "3.10",
6175
"gpu_arch_type": "cuda",
@@ -140,6 +154,20 @@
140154
"upload_to_base_bucket": "no",
141155
"stable_version": "2.11.0"
142156
},
157+
{
158+
"python_version": "3.11",
159+
"gpu_arch_type": "cuda",
160+
"gpu_arch_version": "13.2",
161+
"desired_cuda": "cu132",
162+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
163+
"package_type": "manywheel",
164+
"build_name": "manywheel-py3_11-cuda13_2",
165+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
166+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
167+
"channel": "nightly",
168+
"upload_to_base_bucket": "no",
169+
"stable_version": "2.11.0"
170+
},
143171
{
144172
"python_version": "3.11",
145173
"gpu_arch_type": "cuda",
@@ -224,6 +252,20 @@
224252
"upload_to_base_bucket": "no",
225253
"stable_version": "2.11.0"
226254
},
255+
{
256+
"python_version": "3.12",
257+
"gpu_arch_type": "cuda",
258+
"gpu_arch_version": "13.2",
259+
"desired_cuda": "cu132",
260+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
261+
"package_type": "manywheel",
262+
"build_name": "manywheel-py3_12-cuda13_2",
263+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
264+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
265+
"channel": "nightly",
266+
"upload_to_base_bucket": "no",
267+
"stable_version": "2.11.0"
268+
},
227269
{
228270
"python_version": "3.12",
229271
"gpu_arch_type": "cuda",
@@ -308,6 +350,20 @@
308350
"upload_to_base_bucket": "no",
309351
"stable_version": "2.11.0"
310352
},
353+
{
354+
"python_version": "3.13",
355+
"gpu_arch_type": "cuda",
356+
"gpu_arch_version": "13.2",
357+
"desired_cuda": "cu132",
358+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
359+
"package_type": "manywheel",
360+
"build_name": "manywheel-py3_13-cuda13_2",
361+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
362+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
363+
"channel": "nightly",
364+
"upload_to_base_bucket": "no",
365+
"stable_version": "2.11.0"
366+
},
311367
{
312368
"python_version": "3.13",
313369
"gpu_arch_type": "cuda",
@@ -392,6 +448,20 @@
392448
"upload_to_base_bucket": "no",
393449
"stable_version": "2.11.0"
394450
},
451+
{
452+
"python_version": "3.13t",
453+
"gpu_arch_type": "cuda",
454+
"gpu_arch_version": "13.2",
455+
"desired_cuda": "cu132",
456+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
457+
"package_type": "manywheel",
458+
"build_name": "manywheel-py3_13t-cuda13_2",
459+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
460+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
461+
"channel": "nightly",
462+
"upload_to_base_bucket": "no",
463+
"stable_version": "2.11.0"
464+
},
395465
{
396466
"python_version": "3.13t",
397467
"gpu_arch_type": "cuda",
@@ -476,6 +546,20 @@
476546
"upload_to_base_bucket": "no",
477547
"stable_version": "2.11.0"
478548
},
549+
{
550+
"python_version": "3.14",
551+
"gpu_arch_type": "cuda",
552+
"gpu_arch_version": "13.2",
553+
"desired_cuda": "cu132",
554+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
555+
"package_type": "manywheel",
556+
"build_name": "manywheel-py3_14-cuda13_2",
557+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
558+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
559+
"channel": "nightly",
560+
"upload_to_base_bucket": "no",
561+
"stable_version": "2.11.0"
562+
},
479563
{
480564
"python_version": "3.14",
481565
"gpu_arch_type": "cuda",
@@ -560,6 +644,20 @@
560644
"upload_to_base_bucket": "no",
561645
"stable_version": "2.11.0"
562646
},
647+
{
648+
"python_version": "3.14t",
649+
"gpu_arch_type": "cuda",
650+
"gpu_arch_version": "13.2",
651+
"desired_cuda": "cu132",
652+
"container_image": "pytorch/manylinux2_28-builder:cuda13.2",
653+
"package_type": "manywheel",
654+
"build_name": "manywheel-py3_14t-cuda13_2",
655+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
656+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu132",
657+
"channel": "nightly",
658+
"upload_to_base_bucket": "no",
659+
"stable_version": "2.11.0"
660+
},
563661
{
564662
"python_version": "3.14t",
565663
"gpu_arch_type": "cuda",

0 commit comments

Comments
 (0)