Skip to content

Commit 7d6f614

Browse files
atalmanatalman
committed
Add python 3.14 builds for domains
1 parent 12626a2 commit 7d6f614

3 files changed

Lines changed: 181 additions & 5 deletions

File tree

‎tools/scripts/generate_binary_build_matrix.py‎

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -458,7 +458,7 @@ def generate_wheels_matrix(
458458
arches += [XPU]
459459

460460
if limit_pr_builds:
461-
python_versions = [python_versions[0]]
461+
python_versions = ["3.14", "3.14t"]
462462

463463
global WHEEL_CONTAINER_IMAGES
464464

@@ -474,10 +474,6 @@ def generate_wheels_matrix(
474474
if (gpu_arch_type == "cpu-s390x") and python_version == "3.13t":
475475
continue
476476

477-
# Python 3.14 and 3.14t not yet supported on Windows
478-
if os == WINDOWS and python_version.startswith("3.14"):
479-
continue
480-
481477
desired_cuda = translate_desired_cuda(gpu_arch_type, gpu_arch_version)
482478
entry = {
483479
"python_version": python_version,

‎tools/tests/assets/build_matrix_windows_wheel_cuda.json‎

Lines changed: 120 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -299,6 +299,126 @@
299299
"upload_to_base_bucket": "no",
300300
"stable_version": "2.8.0",
301301
"use_split_build": false
302+
},
303+
{
304+
"python_version": "3.14",
305+
"gpu_arch_type": "cpu",
306+
"gpu_arch_version": "",
307+
"desired_cuda": "cpu",
308+
"container_image": "pytorch/manylinux2_28-builder:cpu",
309+
"package_type": "wheel",
310+
"build_name": "wheel-py3_14-cpu",
311+
"validation_runner": "windows.4xlarge",
312+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu",
313+
"channel": "nightly",
314+
"upload_to_base_bucket": "no",
315+
"stable_version": "2.8.0",
316+
"use_split_build": false
317+
},
318+
{
319+
"python_version": "3.14",
320+
"gpu_arch_type": "cuda",
321+
"gpu_arch_version": "12.6",
322+
"desired_cuda": "cu126",
323+
"container_image": "pytorch/manylinux2_28-builder:cuda12.6",
324+
"package_type": "wheel",
325+
"build_name": "wheel-py3_14-cuda12_6",
326+
"validation_runner": "windows.g4dn.xlarge",
327+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126",
328+
"channel": "nightly",
329+
"upload_to_base_bucket": "no",
330+
"stable_version": "2.8.0",
331+
"use_split_build": false
332+
},
333+
{
334+
"python_version": "3.14",
335+
"gpu_arch_type": "cuda",
336+
"gpu_arch_version": "12.8",
337+
"desired_cuda": "cu128",
338+
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
339+
"package_type": "wheel",
340+
"build_name": "wheel-py3_14-cuda12_8",
341+
"validation_runner": "windows.g4dn.xlarge",
342+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
343+
"channel": "nightly",
344+
"upload_to_base_bucket": "no",
345+
"stable_version": "2.8.0",
346+
"use_split_build": false
347+
},
348+
{
349+
"python_version": "3.14",
350+
"gpu_arch_type": "xpu",
351+
"gpu_arch_version": "",
352+
"desired_cuda": "xpu",
353+
"container_image": "pytorch/manylinux2_28-builder:xpu",
354+
"package_type": "wheel",
355+
"build_name": "wheel-py3_14-xpu",
356+
"validation_runner": "windows.4xlarge",
357+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu",
358+
"channel": "nightly",
359+
"upload_to_base_bucket": "no",
360+
"stable_version": "2.8.0",
361+
"use_split_build": false
362+
},
363+
{
364+
"python_version": "3.14t",
365+
"gpu_arch_type": "cpu",
366+
"gpu_arch_version": "",
367+
"desired_cuda": "cpu",
368+
"container_image": "pytorch/manylinux2_28-builder:cpu",
369+
"package_type": "wheel",
370+
"build_name": "wheel-py3_14t-cpu",
371+
"validation_runner": "windows.4xlarge",
372+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu",
373+
"channel": "nightly",
374+
"upload_to_base_bucket": "no",
375+
"stable_version": "2.8.0",
376+
"use_split_build": false
377+
},
378+
{
379+
"python_version": "3.14t",
380+
"gpu_arch_type": "cuda",
381+
"gpu_arch_version": "12.6",
382+
"desired_cuda": "cu126",
383+
"container_image": "pytorch/manylinux2_28-builder:cuda12.6",
384+
"package_type": "wheel",
385+
"build_name": "wheel-py3_14t-cuda12_6",
386+
"validation_runner": "windows.g4dn.xlarge",
387+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126",
388+
"channel": "nightly",
389+
"upload_to_base_bucket": "no",
390+
"stable_version": "2.8.0",
391+
"use_split_build": false
392+
},
393+
{
394+
"python_version": "3.14t",
395+
"gpu_arch_type": "cuda",
396+
"gpu_arch_version": "12.8",
397+
"desired_cuda": "cu128",
398+
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
399+
"package_type": "wheel",
400+
"build_name": "wheel-py3_14t-cuda12_8",
401+
"validation_runner": "windows.g4dn.xlarge",
402+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
403+
"channel": "nightly",
404+
"upload_to_base_bucket": "no",
405+
"stable_version": "2.8.0",
406+
"use_split_build": false
407+
},
408+
{
409+
"python_version": "3.14t",
410+
"gpu_arch_type": "xpu",
411+
"gpu_arch_version": "",
412+
"desired_cuda": "xpu",
413+
"container_image": "pytorch/manylinux2_28-builder:xpu",
414+
"package_type": "wheel",
415+
"build_name": "wheel-py3_14t-xpu",
416+
"validation_runner": "windows.4xlarge",
417+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu",
418+
"channel": "nightly",
419+
"upload_to_base_bucket": "no",
420+
"stable_version": "2.8.0",
421+
"use_split_build": false
302422
}
303423
]
304424
}

‎tools/tests/assets/build_matrix_windows_wheel_xpu.json‎

Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,66 @@
149149
"upload_to_base_bucket": "yes",
150150
"stable_version": "2.8.0",
151151
"use_split_build": false
152+
},
153+
{
154+
"python_version": "3.14",
155+
"gpu_arch_type": "cpu",
156+
"gpu_arch_version": "",
157+
"desired_cuda": "cpu",
158+
"container_image": "pytorch/manylinux2_28-builder:cpu",
159+
"package_type": "wheel",
160+
"build_name": "wheel-py3_14-cpu",
161+
"validation_runner": "windows.4xlarge",
162+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu",
163+
"channel": "nightly",
164+
"upload_to_base_bucket": "yes",
165+
"stable_version": "2.8.0",
166+
"use_split_build": false
167+
},
168+
{
169+
"python_version": "3.14",
170+
"gpu_arch_type": "xpu",
171+
"gpu_arch_version": "",
172+
"desired_cuda": "xpu",
173+
"container_image": "pytorch/manylinux2_28-builder:xpu",
174+
"package_type": "wheel",
175+
"build_name": "wheel-py3_14-xpu",
176+
"validation_runner": "windows.4xlarge",
177+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu",
178+
"channel": "nightly",
179+
"upload_to_base_bucket": "yes",
180+
"stable_version": "2.8.0",
181+
"use_split_build": false
182+
},
183+
{
184+
"python_version": "3.14t",
185+
"gpu_arch_type": "cpu",
186+
"gpu_arch_version": "",
187+
"desired_cuda": "cpu",
188+
"container_image": "pytorch/manylinux2_28-builder:cpu",
189+
"package_type": "wheel",
190+
"build_name": "wheel-py3_14t-cpu",
191+
"validation_runner": "windows.4xlarge",
192+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu",
193+
"channel": "nightly",
194+
"upload_to_base_bucket": "yes",
195+
"stable_version": "2.8.0",
196+
"use_split_build": false
197+
},
198+
{
199+
"python_version": "3.14t",
200+
"gpu_arch_type": "xpu",
201+
"gpu_arch_version": "",
202+
"desired_cuda": "xpu",
203+
"container_image": "pytorch/manylinux2_28-builder:xpu",
204+
"package_type": "wheel",
205+
"build_name": "wheel-py3_14t-xpu",
206+
"validation_runner": "windows.4xlarge",
207+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu",
208+
"channel": "nightly",
209+
"upload_to_base_bucket": "yes",
210+
"stable_version": "2.8.0",
211+
"use_split_build": false
152212
}
153213
]
154214
}

0 commit comments

Comments
 (0)