Skip to content

Commit 548a4bc

Browse files
atalmanatalman
authored
Add python 3.14 builds for windows (#7108)
Linux and MacOS builds are enabled by #7022
1 parent 1f6355d commit 548a4bc

File tree

4 files changed

+182
-7
lines changed

4 files changed

+182
-7
lines changed

‎.github/actions/setup-binary-builds/action.yml‎

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,6 @@ runs:
128128
export CONDA_EXTRA_PARAM=""
129129
130130
if [[ "${PYTHON_VERSION:-}" == *t || "${PYTHON_VERSION:-}" == 3.14* ]]; then
131-
132131
# downgrade conda version for python 3.13t, 3.14t, 3.14 install.
133132
# TODO: remove this once python 3.13t, 3.14t, 3.14 is fully supported on conda
134133
# Please see : https://github.com/conda/conda/issues/14554
@@ -147,11 +146,11 @@ runs:
147146
# shellcheck disable=SC2153
148147
case $PYTHON_VERSION in
149148
3.14t)
150-
export PYTHON_VERSION=3.14.0rc1
149+
export PYTHON_VERSION=3.14.0rc2
151150
export CONDA_EXTRA_PARAM=" python-freethreading -c conda-forge/label/python_rc -c conda-forge"
152151
;;
153152
3.14)
154-
export PYTHON_VERSION=3.14.0rc1
153+
export PYTHON_VERSION=3.14.0rc2
155154
export CONDA_EXTRA_PARAM=" -c conda-forge/label/python_rc -c conda-forge"
156155
;;
157156
3.13t)

‎tools/scripts/generate_binary_build_matrix.py‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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)