Skip to content

Commit 58842fd

Browse files
huydhnhuydhn
committed
Bring back CUDA 12.9 builds for domains (pytorch#7352)
This is basically a partial revert of pytorch#7074 and pytorch#7345 to resume building nightly, test, and release binaries for domains. This is to address pytorch/pytorch#165165 --------- Signed-off-by: Huy Do <[email protected]>
1 parent f576b33 commit 58842fd

6 files changed

+1476
-2
lines changed

‎tools/scripts/generate_binary_build_matrix.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def validation_runner(arch_type: str, os: str) -> str:
148148
return LINUX_CPU_RUNNER
149149

150150

151-
def initialize_globals(channel: str, build_python_only: bool) -> None:
151+
def initialize_globals(channel: str, os: str, build_python_only: bool) -> None:
152152
global CURRENT_VERSION, CUDA_ARCHES, ROCM_ARCHES, PYTHON_ARCHES
153153
global WHEEL_CONTAINER_IMAGES, LIBTORCH_CONTAINER_IMAGES
154154
if channel == TEST:
@@ -157,6 +157,10 @@ def initialize_globals(channel: str, build_python_only: bool) -> None:
157157
CURRENT_VERSION = CURRENT_STABLE_VERSION
158158

159159
CUDA_ARCHES = CUDA_ARCHES_DICT[channel]
160+
if channel != "release" and os == LINUX:
161+
# TODO (huydhn): Only build CUDA 12.9 for Linux. This logic is to be cleaned up
162+
# in 2.10
163+
CUDA_ARCHES.append("12.9")
160164
ROCM_ARCHES = ROCM_ARCHES_DICT[channel]
161165
if build_python_only:
162166
# Only select the oldest version of python if building a python only package
@@ -549,7 +553,7 @@ def generate_build_matrix(
549553

550554
for channel in channels:
551555
for package in package_types:
552-
initialize_globals(channel, build_python_only == ENABLE)
556+
initialize_globals(channel, operating_system, build_python_only == ENABLE)
553557
includes.extend(
554558
GENERATING_FUNCTIONS_BY_PACKAGE_TYPE[package](
555559
operating_system,

‎tools/tests/assets/build_matrix_linux_wheel_cuda.json‎

Lines changed: 105 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,21 @@
6060
"stable_version": "2.8.0",
6161
"use_split_build": false
6262
},
63+
{
64+
"python_version": "3.10",
65+
"gpu_arch_type": "cuda",
66+
"gpu_arch_version": "12.9",
67+
"desired_cuda": "cu129",
68+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
69+
"package_type": "manywheel",
70+
"build_name": "manywheel-py3_10-cuda12_9",
71+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
72+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
73+
"channel": "nightly",
74+
"upload_to_base_bucket": "no",
75+
"stable_version": "2.8.0",
76+
"use_split_build": false
77+
},
6378
{
6479
"python_version": "3.10",
6580
"gpu_arch_type": "rocm",
@@ -150,6 +165,21 @@
150165
"stable_version": "2.8.0",
151166
"use_split_build": false
152167
},
168+
{
169+
"python_version": "3.11",
170+
"gpu_arch_type": "cuda",
171+
"gpu_arch_version": "12.9",
172+
"desired_cuda": "cu129",
173+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
174+
"package_type": "manywheel",
175+
"build_name": "manywheel-py3_11-cuda12_9",
176+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
177+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
178+
"channel": "nightly",
179+
"upload_to_base_bucket": "no",
180+
"stable_version": "2.8.0",
181+
"use_split_build": false
182+
},
153183
{
154184
"python_version": "3.11",
155185
"gpu_arch_type": "rocm",
@@ -240,6 +270,21 @@
240270
"stable_version": "2.8.0",
241271
"use_split_build": false
242272
},
273+
{
274+
"python_version": "3.12",
275+
"gpu_arch_type": "cuda",
276+
"gpu_arch_version": "12.9",
277+
"desired_cuda": "cu129",
278+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
279+
"package_type": "manywheel",
280+
"build_name": "manywheel-py3_12-cuda12_9",
281+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
282+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
283+
"channel": "nightly",
284+
"upload_to_base_bucket": "no",
285+
"stable_version": "2.8.0",
286+
"use_split_build": false
287+
},
243288
{
244289
"python_version": "3.12",
245290
"gpu_arch_type": "rocm",
@@ -330,6 +375,21 @@
330375
"stable_version": "2.8.0",
331376
"use_split_build": false
332377
},
378+
{
379+
"python_version": "3.13",
380+
"gpu_arch_type": "cuda",
381+
"gpu_arch_version": "12.9",
382+
"desired_cuda": "cu129",
383+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
384+
"package_type": "manywheel",
385+
"build_name": "manywheel-py3_13-cuda12_9",
386+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
387+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
388+
"channel": "nightly",
389+
"upload_to_base_bucket": "no",
390+
"stable_version": "2.8.0",
391+
"use_split_build": false
392+
},
333393
{
334394
"python_version": "3.13",
335395
"gpu_arch_type": "rocm",
@@ -420,6 +480,21 @@
420480
"stable_version": "2.8.0",
421481
"use_split_build": false
422482
},
483+
{
484+
"python_version": "3.13t",
485+
"gpu_arch_type": "cuda",
486+
"gpu_arch_version": "12.9",
487+
"desired_cuda": "cu129",
488+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
489+
"package_type": "manywheel",
490+
"build_name": "manywheel-py3_13t-cuda12_9",
491+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
492+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
493+
"channel": "nightly",
494+
"upload_to_base_bucket": "no",
495+
"stable_version": "2.8.0",
496+
"use_split_build": false
497+
},
423498
{
424499
"python_version": "3.13t",
425500
"gpu_arch_type": "rocm",
@@ -510,6 +585,21 @@
510585
"stable_version": "2.8.0",
511586
"use_split_build": false
512587
},
588+
{
589+
"python_version": "3.14",
590+
"gpu_arch_type": "cuda",
591+
"gpu_arch_version": "12.9",
592+
"desired_cuda": "cu129",
593+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
594+
"package_type": "manywheel",
595+
"build_name": "manywheel-py3_14-cuda12_9",
596+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
597+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
598+
"channel": "nightly",
599+
"upload_to_base_bucket": "no",
600+
"stable_version": "2.8.0",
601+
"use_split_build": false
602+
},
513603
{
514604
"python_version": "3.14",
515605
"gpu_arch_type": "rocm",
@@ -600,6 +690,21 @@
600690
"stable_version": "2.8.0",
601691
"use_split_build": false
602692
},
693+
{
694+
"python_version": "3.14t",
695+
"gpu_arch_type": "cuda",
696+
"gpu_arch_version": "12.9",
697+
"desired_cuda": "cu129",
698+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
699+
"package_type": "manywheel",
700+
"build_name": "manywheel-py3_14t-cuda12_9",
701+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
702+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
703+
"channel": "nightly",
704+
"upload_to_base_bucket": "no",
705+
"stable_version": "2.8.0",
706+
"use_split_build": false
707+
},
603708
{
604709
"python_version": "3.14t",
605710
"gpu_arch_type": "rocm",

0 commit comments

Comments
 (0)