Skip to content

Commit e943605

Browse files
atalmanatalman
committed
[release/2.14] Remove CUDA 13.4 from the binary build matrix
CUDA 13.4 is not shipping in the 2.14 release, so the release branch should not be building or testing it. The `release` channel already excluded 13.4; this drops it from the `nightly` and `test` channels too, along with the supporting arch/cudnn entries, so nothing on release/2.14 generates a 13.4 job. Changes in `tools/scripts/generate_binary_build_matrix.py`: - `CUDA_ARCHES_DICT`: drop `13.4` from `nightly` and `test` - `CUDA_CUDNN_VERSIONS`: drop the `13.4` entry - `CUDA_AARCH64_ARCHES`: drop `13.4-aarch64` - `CUDA_ARCHES_NO_WINDOWS`: now empty (13.4 was its only member) Test assets regenerated for the four affected matrices; each loses exactly the 8 CUDA 13.4 entries (one per Python version) with no other change: - `build_matrix_linux_wheel_cuda.json` 56 -> 48 - `build_matrix_linux_wheel_cuda_norocm.json` 40 -> 32 - `build_matrix_linux_wheel_nocpu.json` 48 -> 40 - `build_matrix_linux_wheel_xpu.json` 40 -> 32 The macOS and Windows assets are unchanged, as expected: macOS has no CUDA builds and 13.4 was already excluded from Windows. `tools/pkg-helpers/tests/test_cuda.py` still parametrizes over 13.4, which is left alone deliberately -- it exercises version-string sanitization on an arbitrary version and does not depend on the build matrix.
1 parent 2252fef commit e943605

5 files changed

Lines changed: 4 additions & 453 deletions

‎tools/scripts/generate_binary_build_matrix.py‎

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,8 @@
4444
"3.14": "3.14.3",
4545
}
4646
CUDA_ARCHES_DICT = {
47-
"nightly": ["12.6", "13.0", "13.2", "13.4"],
48-
"test": ["12.6", "13.0", "13.2", "13.4"],
47+
"nightly": ["12.6", "13.0", "13.2"],
48+
"test": ["12.6", "13.0", "13.2"],
4949
"release": ["12.6", "13.0", "13.2"],
5050
}
5151

@@ -60,7 +60,6 @@
6060
"12.8": {"cuda": "12.8.0", "cudnn": "9"},
6161
"13.0": {"cuda": "13.0.0", "cudnn": "9"},
6262
"13.2": {"cuda": "13.2.0", "cudnn": "9"},
63-
"13.4": {"cuda": "13.4.0", "cudnn": "9"},
6463
}
6564

6665
STABLE_CUDA_VERSIONS = {
@@ -69,11 +68,11 @@
6968
"release": "13.0",
7069
}
7170

72-
CUDA_AARCH64_ARCHES = ["12.6-aarch64", "13.0-aarch64", "13.2-aarch64", "13.4-aarch64"]
71+
CUDA_AARCH64_ARCHES = ["12.6-aarch64", "13.0-aarch64", "13.2-aarch64"]
7372

7473
# CUDA versions with no Windows torch build to depend on; see
7574
# CUDA_ARCHES_NO_WINDOWS in pytorch/pytorch.
76-
CUDA_ARCHES_NO_WINDOWS = ["13.4"]
75+
CUDA_ARCHES_NO_WINDOWS: list[str] = []
7776

7877
PACKAGE_TYPES = ["wheel", "libtorch"]
7978
CXX11_ABI = "cxx11-abi"

‎tools/tests/assets/build_matrix_linux_wheel_cuda.json‎

Lines changed: 0 additions & 112 deletions
Original file line numberDiff line numberDiff line change
@@ -56,20 +56,6 @@
5656
"upload_to_base_bucket": "no",
5757
"stable_version": "2.13.0"
5858
},
59-
{
60-
"python_version": "3.10",
61-
"gpu_arch_type": "cuda",
62-
"gpu_arch_version": "13.4",
63-
"desired_cuda": "cu134",
64-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
65-
"package_type": "manywheel",
66-
"build_name": "manywheel-py3_10-cuda13_4",
67-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
68-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
69-
"channel": "nightly",
70-
"upload_to_base_bucket": "no",
71-
"stable_version": "2.13.0"
72-
},
7359
{
7460
"python_version": "3.10",
7561
"gpu_arch_type": "rocm",
@@ -154,20 +140,6 @@
154140
"upload_to_base_bucket": "no",
155141
"stable_version": "2.13.0"
156142
},
157-
{
158-
"python_version": "3.11",
159-
"gpu_arch_type": "cuda",
160-
"gpu_arch_version": "13.4",
161-
"desired_cuda": "cu134",
162-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
163-
"package_type": "manywheel",
164-
"build_name": "manywheel-py3_11-cuda13_4",
165-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
166-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
167-
"channel": "nightly",
168-
"upload_to_base_bucket": "no",
169-
"stable_version": "2.13.0"
170-
},
171143
{
172144
"python_version": "3.11",
173145
"gpu_arch_type": "rocm",
@@ -252,20 +224,6 @@
252224
"upload_to_base_bucket": "no",
253225
"stable_version": "2.13.0"
254226
},
255-
{
256-
"python_version": "3.12",
257-
"gpu_arch_type": "cuda",
258-
"gpu_arch_version": "13.4",
259-
"desired_cuda": "cu134",
260-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
261-
"package_type": "manywheel",
262-
"build_name": "manywheel-py3_12-cuda13_4",
263-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
264-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
265-
"channel": "nightly",
266-
"upload_to_base_bucket": "no",
267-
"stable_version": "2.13.0"
268-
},
269227
{
270228
"python_version": "3.12",
271229
"gpu_arch_type": "rocm",
@@ -350,20 +308,6 @@
350308
"upload_to_base_bucket": "no",
351309
"stable_version": "2.13.0"
352310
},
353-
{
354-
"python_version": "3.13",
355-
"gpu_arch_type": "cuda",
356-
"gpu_arch_version": "13.4",
357-
"desired_cuda": "cu134",
358-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
359-
"package_type": "manywheel",
360-
"build_name": "manywheel-py3_13-cuda13_4",
361-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
362-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
363-
"channel": "nightly",
364-
"upload_to_base_bucket": "no",
365-
"stable_version": "2.13.0"
366-
},
367311
{
368312
"python_version": "3.13",
369313
"gpu_arch_type": "rocm",
@@ -448,20 +392,6 @@
448392
"upload_to_base_bucket": "no",
449393
"stable_version": "2.13.0"
450394
},
451-
{
452-
"python_version": "3.14",
453-
"gpu_arch_type": "cuda",
454-
"gpu_arch_version": "13.4",
455-
"desired_cuda": "cu134",
456-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
457-
"package_type": "manywheel",
458-
"build_name": "manywheel-py3_14-cuda13_4",
459-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
460-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
461-
"channel": "nightly",
462-
"upload_to_base_bucket": "no",
463-
"stable_version": "2.13.0"
464-
},
465395
{
466396
"python_version": "3.14",
467397
"gpu_arch_type": "rocm",
@@ -546,20 +476,6 @@
546476
"upload_to_base_bucket": "no",
547477
"stable_version": "2.13.0"
548478
},
549-
{
550-
"python_version": "3.14t",
551-
"gpu_arch_type": "cuda",
552-
"gpu_arch_version": "13.4",
553-
"desired_cuda": "cu134",
554-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
555-
"package_type": "manywheel",
556-
"build_name": "manywheel-py3_14t-cuda13_4",
557-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
558-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
559-
"channel": "nightly",
560-
"upload_to_base_bucket": "no",
561-
"stable_version": "2.13.0"
562-
},
563479
{
564480
"python_version": "3.14t",
565481
"gpu_arch_type": "rocm",
@@ -644,20 +560,6 @@
644560
"upload_to_base_bucket": "no",
645561
"stable_version": "2.13.0"
646562
},
647-
{
648-
"python_version": "3.15",
649-
"gpu_arch_type": "cuda",
650-
"gpu_arch_version": "13.4",
651-
"desired_cuda": "cu134",
652-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
653-
"package_type": "manywheel",
654-
"build_name": "manywheel-py3_15-cuda13_4",
655-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
656-
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu134",
657-
"channel": "nightly",
658-
"upload_to_base_bucket": "no",
659-
"stable_version": "2.13.0"
660-
},
661563
{
662564
"python_version": "3.15",
663565
"gpu_arch_type": "rocm",
@@ -742,20 +644,6 @@
742644
"upload_to_base_bucket": "no",
743645
"stable_version": "2.13.0"
744646
},
745-
{
746-
"python_version": "3.15t",
747-
"gpu_arch_type": "cuda",
748-
"gpu_arch_version": "13.4",
749-
"desired_cuda": "cu134",
750-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
751-
"package_type": "manywheel",
752-
"build_name": "manywheel-py3_15t-cuda13_4",
753-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
754-
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu134",
755-
"channel": "nightly",
756-
"upload_to_base_bucket": "no",
757-
"stable_version": "2.13.0"
758-
},
759647
{
760648
"python_version": "3.15t",
761649
"gpu_arch_type": "rocm",

‎tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json‎

Lines changed: 0 additions & 112 deletions
Original file line numberDiff line numberDiff line change
@@ -56,20 +56,6 @@
5656
"upload_to_base_bucket": "no",
5757
"stable_version": "2.13.0"
5858
},
59-
{
60-
"python_version": "3.10",
61-
"gpu_arch_type": "cuda",
62-
"gpu_arch_version": "13.4",
63-
"desired_cuda": "cu134",
64-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
65-
"package_type": "manywheel",
66-
"build_name": "manywheel-py3_10-cuda13_4",
67-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
68-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
69-
"channel": "nightly",
70-
"upload_to_base_bucket": "no",
71-
"stable_version": "2.13.0"
72-
},
7359
{
7460
"python_version": "3.11",
7561
"gpu_arch_type": "cpu",
@@ -126,20 +112,6 @@
126112
"upload_to_base_bucket": "no",
127113
"stable_version": "2.13.0"
128114
},
129-
{
130-
"python_version": "3.11",
131-
"gpu_arch_type": "cuda",
132-
"gpu_arch_version": "13.4",
133-
"desired_cuda": "cu134",
134-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
135-
"package_type": "manywheel",
136-
"build_name": "manywheel-py3_11-cuda13_4",
137-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
138-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
139-
"channel": "nightly",
140-
"upload_to_base_bucket": "no",
141-
"stable_version": "2.13.0"
142-
},
143115
{
144116
"python_version": "3.12",
145117
"gpu_arch_type": "cpu",
@@ -196,20 +168,6 @@
196168
"upload_to_base_bucket": "no",
197169
"stable_version": "2.13.0"
198170
},
199-
{
200-
"python_version": "3.12",
201-
"gpu_arch_type": "cuda",
202-
"gpu_arch_version": "13.4",
203-
"desired_cuda": "cu134",
204-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
205-
"package_type": "manywheel",
206-
"build_name": "manywheel-py3_12-cuda13_4",
207-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
208-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
209-
"channel": "nightly",
210-
"upload_to_base_bucket": "no",
211-
"stable_version": "2.13.0"
212-
},
213171
{
214172
"python_version": "3.13",
215173
"gpu_arch_type": "cpu",
@@ -266,20 +224,6 @@
266224
"upload_to_base_bucket": "no",
267225
"stable_version": "2.13.0"
268226
},
269-
{
270-
"python_version": "3.13",
271-
"gpu_arch_type": "cuda",
272-
"gpu_arch_version": "13.4",
273-
"desired_cuda": "cu134",
274-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
275-
"package_type": "manywheel",
276-
"build_name": "manywheel-py3_13-cuda13_4",
277-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
278-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
279-
"channel": "nightly",
280-
"upload_to_base_bucket": "no",
281-
"stable_version": "2.13.0"
282-
},
283227
{
284228
"python_version": "3.14",
285229
"gpu_arch_type": "cpu",
@@ -336,20 +280,6 @@
336280
"upload_to_base_bucket": "no",
337281
"stable_version": "2.13.0"
338282
},
339-
{
340-
"python_version": "3.14",
341-
"gpu_arch_type": "cuda",
342-
"gpu_arch_version": "13.4",
343-
"desired_cuda": "cu134",
344-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
345-
"package_type": "manywheel",
346-
"build_name": "manywheel-py3_14-cuda13_4",
347-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
348-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
349-
"channel": "nightly",
350-
"upload_to_base_bucket": "no",
351-
"stable_version": "2.13.0"
352-
},
353283
{
354284
"python_version": "3.14t",
355285
"gpu_arch_type": "cpu",
@@ -406,20 +336,6 @@
406336
"upload_to_base_bucket": "no",
407337
"stable_version": "2.13.0"
408338
},
409-
{
410-
"python_version": "3.14t",
411-
"gpu_arch_type": "cuda",
412-
"gpu_arch_version": "13.4",
413-
"desired_cuda": "cu134",
414-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
415-
"package_type": "manywheel",
416-
"build_name": "manywheel-py3_14t-cuda13_4",
417-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
418-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu134",
419-
"channel": "nightly",
420-
"upload_to_base_bucket": "no",
421-
"stable_version": "2.13.0"
422-
},
423339
{
424340
"python_version": "3.15",
425341
"gpu_arch_type": "cpu",
@@ -476,20 +392,6 @@
476392
"upload_to_base_bucket": "no",
477393
"stable_version": "2.13.0"
478394
},
479-
{
480-
"python_version": "3.15",
481-
"gpu_arch_type": "cuda",
482-
"gpu_arch_version": "13.4",
483-
"desired_cuda": "cu134",
484-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
485-
"package_type": "manywheel",
486-
"build_name": "manywheel-py3_15-cuda13_4",
487-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
488-
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu134",
489-
"channel": "nightly",
490-
"upload_to_base_bucket": "no",
491-
"stable_version": "2.13.0"
492-
},
493395
{
494396
"python_version": "3.15t",
495397
"gpu_arch_type": "cpu",
@@ -545,20 +447,6 @@
545447
"channel": "nightly",
546448
"upload_to_base_bucket": "no",
547449
"stable_version": "2.13.0"
548-
},
549-
{
550-
"python_version": "3.15t",
551-
"gpu_arch_type": "cuda",
552-
"gpu_arch_version": "13.4",
553-
"desired_cuda": "cu134",
554-
"container_image": "pytorch/manylinux2_28-builder:cuda13.4",
555-
"package_type": "manywheel",
556-
"build_name": "manywheel-py3_15t-cuda13_4",
557-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
558-
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu134",
559-
"channel": "nightly",
560-
"upload_to_base_bucket": "no",
561-
"stable_version": "2.13.0"
562450
}
563451
]
564452
}

0 commit comments

Comments
 (0)