Skip to content

Commit c00a90b

Browse files
authored
1 parent 1a26b61 commit c00a90b

6 files changed

Lines changed: 3 additions & 493 deletions

tools/scripts/generate_binary_build_matrix.py

Lines changed: 3 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", "13.2"],
39-
"test": ["12.6", "12.8", "13.0", "13.2"],
38+
"nightly": ["12.6", "13.0", "13.2"],
39+
"test": ["12.6", "13.0", "13.2"],
4040
"release": ["12.6", "12.8", "13.0"],
4141
}
4242

@@ -59,7 +59,7 @@
5959
"release": "13.0",
6060
}
6161

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

6464
PACKAGE_TYPES = ["wheel", "libtorch"]
6565
CXX11_ABI = "cxx11-abi"

tools/tests/assets/build_matrix_linux_wheel_cuda.json

Lines changed: 0 additions & 98 deletions
Original file line numberDiff line numberDiff line change
@@ -28,20 +28,6 @@
2828
"upload_to_base_bucket": "no",
2929
"stable_version": "2.11.0"
3030
},
31-
{
32-
"python_version": "3.10",
33-
"gpu_arch_type": "cuda",
34-
"gpu_arch_version": "12.8",
35-
"desired_cuda": "cu128",
36-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
37-
"package_type": "manywheel",
38-
"build_name": "manywheel-py3_10-cuda12_8",
39-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
40-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
41-
"channel": "nightly",
42-
"upload_to_base_bucket": "no",
43-
"stable_version": "2.11.0"
44-
},
4531
{
4632
"python_version": "3.10",
4733
"gpu_arch_type": "cuda",
@@ -126,20 +112,6 @@
126112
"upload_to_base_bucket": "no",
127113
"stable_version": "2.11.0"
128114
},
129-
{
130-
"python_version": "3.11",
131-
"gpu_arch_type": "cuda",
132-
"gpu_arch_version": "12.8",
133-
"desired_cuda": "cu128",
134-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
135-
"package_type": "manywheel",
136-
"build_name": "manywheel-py3_11-cuda12_8",
137-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
138-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
139-
"channel": "nightly",
140-
"upload_to_base_bucket": "no",
141-
"stable_version": "2.11.0"
142-
},
143115
{
144116
"python_version": "3.11",
145117
"gpu_arch_type": "cuda",
@@ -224,20 +196,6 @@
224196
"upload_to_base_bucket": "no",
225197
"stable_version": "2.11.0"
226198
},
227-
{
228-
"python_version": "3.12",
229-
"gpu_arch_type": "cuda",
230-
"gpu_arch_version": "12.8",
231-
"desired_cuda": "cu128",
232-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
233-
"package_type": "manywheel",
234-
"build_name": "manywheel-py3_12-cuda12_8",
235-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
236-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
237-
"channel": "nightly",
238-
"upload_to_base_bucket": "no",
239-
"stable_version": "2.11.0"
240-
},
241199
{
242200
"python_version": "3.12",
243201
"gpu_arch_type": "cuda",
@@ -322,20 +280,6 @@
322280
"upload_to_base_bucket": "no",
323281
"stable_version": "2.11.0"
324282
},
325-
{
326-
"python_version": "3.13",
327-
"gpu_arch_type": "cuda",
328-
"gpu_arch_version": "12.8",
329-
"desired_cuda": "cu128",
330-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
331-
"package_type": "manywheel",
332-
"build_name": "manywheel-py3_13-cuda12_8",
333-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
334-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
335-
"channel": "nightly",
336-
"upload_to_base_bucket": "no",
337-
"stable_version": "2.11.0"
338-
},
339283
{
340284
"python_version": "3.13",
341285
"gpu_arch_type": "cuda",
@@ -420,20 +364,6 @@
420364
"upload_to_base_bucket": "no",
421365
"stable_version": "2.11.0"
422366
},
423-
{
424-
"python_version": "3.13t",
425-
"gpu_arch_type": "cuda",
426-
"gpu_arch_version": "12.8",
427-
"desired_cuda": "cu128",
428-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
429-
"package_type": "manywheel",
430-
"build_name": "manywheel-py3_13t-cuda12_8",
431-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
432-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
433-
"channel": "nightly",
434-
"upload_to_base_bucket": "no",
435-
"stable_version": "2.11.0"
436-
},
437367
{
438368
"python_version": "3.13t",
439369
"gpu_arch_type": "cuda",
@@ -518,20 +448,6 @@
518448
"upload_to_base_bucket": "no",
519449
"stable_version": "2.11.0"
520450
},
521-
{
522-
"python_version": "3.14",
523-
"gpu_arch_type": "cuda",
524-
"gpu_arch_version": "12.8",
525-
"desired_cuda": "cu128",
526-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
527-
"package_type": "manywheel",
528-
"build_name": "manywheel-py3_14-cuda12_8",
529-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
530-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
531-
"channel": "nightly",
532-
"upload_to_base_bucket": "no",
533-
"stable_version": "2.11.0"
534-
},
535451
{
536452
"python_version": "3.14",
537453
"gpu_arch_type": "cuda",
@@ -616,20 +532,6 @@
616532
"upload_to_base_bucket": "no",
617533
"stable_version": "2.11.0"
618534
},
619-
{
620-
"python_version": "3.14t",
621-
"gpu_arch_type": "cuda",
622-
"gpu_arch_version": "12.8",
623-
"desired_cuda": "cu128",
624-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
625-
"package_type": "manywheel",
626-
"build_name": "manywheel-py3_14t-cuda12_8",
627-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
628-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
629-
"channel": "nightly",
630-
"upload_to_base_bucket": "no",
631-
"stable_version": "2.11.0"
632-
},
633535
{
634536
"python_version": "3.14t",
635537
"gpu_arch_type": "cuda",

tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json

Lines changed: 0 additions & 98 deletions
Original file line numberDiff line numberDiff line change
@@ -28,20 +28,6 @@
2828
"upload_to_base_bucket": "no",
2929
"stable_version": "2.11.0"
3030
},
31-
{
32-
"python_version": "3.10",
33-
"gpu_arch_type": "cuda",
34-
"gpu_arch_version": "12.8",
35-
"desired_cuda": "cu128",
36-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
37-
"package_type": "manywheel",
38-
"build_name": "manywheel-py3_10-cuda12_8",
39-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
40-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
41-
"channel": "nightly",
42-
"upload_to_base_bucket": "no",
43-
"stable_version": "2.11.0"
44-
},
4531
{
4632
"python_version": "3.10",
4733
"gpu_arch_type": "cuda",
@@ -98,20 +84,6 @@
9884
"upload_to_base_bucket": "no",
9985
"stable_version": "2.11.0"
10086
},
101-
{
102-
"python_version": "3.11",
103-
"gpu_arch_type": "cuda",
104-
"gpu_arch_version": "12.8",
105-
"desired_cuda": "cu128",
106-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
107-
"package_type": "manywheel",
108-
"build_name": "manywheel-py3_11-cuda12_8",
109-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
110-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
111-
"channel": "nightly",
112-
"upload_to_base_bucket": "no",
113-
"stable_version": "2.11.0"
114-
},
11587
{
11688
"python_version": "3.11",
11789
"gpu_arch_type": "cuda",
@@ -168,20 +140,6 @@
168140
"upload_to_base_bucket": "no",
169141
"stable_version": "2.11.0"
170142
},
171-
{
172-
"python_version": "3.12",
173-
"gpu_arch_type": "cuda",
174-
"gpu_arch_version": "12.8",
175-
"desired_cuda": "cu128",
176-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
177-
"package_type": "manywheel",
178-
"build_name": "manywheel-py3_12-cuda12_8",
179-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
180-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
181-
"channel": "nightly",
182-
"upload_to_base_bucket": "no",
183-
"stable_version": "2.11.0"
184-
},
185143
{
186144
"python_version": "3.12",
187145
"gpu_arch_type": "cuda",
@@ -238,20 +196,6 @@
238196
"upload_to_base_bucket": "no",
239197
"stable_version": "2.11.0"
240198
},
241-
{
242-
"python_version": "3.13",
243-
"gpu_arch_type": "cuda",
244-
"gpu_arch_version": "12.8",
245-
"desired_cuda": "cu128",
246-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
247-
"package_type": "manywheel",
248-
"build_name": "manywheel-py3_13-cuda12_8",
249-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
250-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
251-
"channel": "nightly",
252-
"upload_to_base_bucket": "no",
253-
"stable_version": "2.11.0"
254-
},
255199
{
256200
"python_version": "3.13",
257201
"gpu_arch_type": "cuda",
@@ -308,20 +252,6 @@
308252
"upload_to_base_bucket": "no",
309253
"stable_version": "2.11.0"
310254
},
311-
{
312-
"python_version": "3.13t",
313-
"gpu_arch_type": "cuda",
314-
"gpu_arch_version": "12.8",
315-
"desired_cuda": "cu128",
316-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
317-
"package_type": "manywheel",
318-
"build_name": "manywheel-py3_13t-cuda12_8",
319-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
320-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
321-
"channel": "nightly",
322-
"upload_to_base_bucket": "no",
323-
"stable_version": "2.11.0"
324-
},
325255
{
326256
"python_version": "3.13t",
327257
"gpu_arch_type": "cuda",
@@ -378,20 +308,6 @@
378308
"upload_to_base_bucket": "no",
379309
"stable_version": "2.11.0"
380310
},
381-
{
382-
"python_version": "3.14",
383-
"gpu_arch_type": "cuda",
384-
"gpu_arch_version": "12.8",
385-
"desired_cuda": "cu128",
386-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
387-
"package_type": "manywheel",
388-
"build_name": "manywheel-py3_14-cuda12_8",
389-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
390-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
391-
"channel": "nightly",
392-
"upload_to_base_bucket": "no",
393-
"stable_version": "2.11.0"
394-
},
395311
{
396312
"python_version": "3.14",
397313
"gpu_arch_type": "cuda",
@@ -448,20 +364,6 @@
448364
"upload_to_base_bucket": "no",
449365
"stable_version": "2.11.0"
450366
},
451-
{
452-
"python_version": "3.14t",
453-
"gpu_arch_type": "cuda",
454-
"gpu_arch_version": "12.8",
455-
"desired_cuda": "cu128",
456-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
457-
"package_type": "manywheel",
458-
"build_name": "manywheel-py3_14t-cuda12_8",
459-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
460-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
461-
"channel": "nightly",
462-
"upload_to_base_bucket": "no",
463-
"stable_version": "2.11.0"
464-
},
465367
{
466368
"python_version": "3.14t",
467369
"gpu_arch_type": "cuda",

0 commit comments

Comments
 (0)