From d79e8f93654f341802ed00c985851bd178af565b Mon Sep 17 00:00:00 2001 From: Ting Lu Date: Thu, 20 Feb 2025 11:12:06 -0800 Subject: [PATCH] Add CU128 Windows Builds --- tools/scripts/generate_binary_build_matrix.py | 3 - .../build_matrix_windows_wheel_cuda.json | 90 +++++++++++++++++++ 2 files changed, 90 insertions(+), 3 deletions(-) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index 4863b27cc8..2919647e8b 100755 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -462,9 +462,6 @@ def generate_wheels_matrix( upload_to_base_bucket = "no" if os in (LINUX, WINDOWS): arches += CUDA_ARCHES - # todo: remove once windows cuda 12.8 binaries are available - if channel == NIGHTLY and os == WINDOWS: - arches.remove("12.8") if with_rocm == ENABLE and os == LINUX: arches += ROCM_ARCHES diff --git a/tools/tests/assets/build_matrix_windows_wheel_cuda.json b/tools/tests/assets/build_matrix_windows_wheel_cuda.json index 987a09b9d3..ff7e8db5b1 100644 --- a/tools/tests/assets/build_matrix_windows_wheel_cuda.json +++ b/tools/tests/assets/build_matrix_windows_wheel_cuda.json @@ -60,6 +60,21 @@ "stable_version": "2.6.0", "use_split_build": false }, + { + "python_version": "3.9", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.8", + "desired_cuda": "cu128", + "container_image": "pytorch/manylinux2_28-builder:cuda12.8", + "package_type": "wheel", + "build_name": "wheel-py3_9-cuda12_8", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.6.0", + "use_split_build": false + }, { "python_version": "3.9", "gpu_arch_type": "xpu", @@ -135,6 +150,21 @@ "stable_version": "2.6.0", "use_split_build": false }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.8", + "desired_cuda": "cu128", + "container_image": "pytorch/manylinux2_28-builder:cuda12.8", + "package_type": "wheel", + "build_name": "wheel-py3_10-cuda12_8", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.6.0", + "use_split_build": false + }, { "python_version": "3.10", "gpu_arch_type": "xpu", @@ -210,6 +240,21 @@ "stable_version": "2.6.0", "use_split_build": false }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.8", + "desired_cuda": "cu128", + "container_image": "pytorch/manylinux2_28-builder:cuda12.8", + "package_type": "wheel", + "build_name": "wheel-py3_11-cuda12_8", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.6.0", + "use_split_build": false + }, { "python_version": "3.11", "gpu_arch_type": "xpu", @@ -285,6 +330,21 @@ "stable_version": "2.6.0", "use_split_build": false }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.8", + "desired_cuda": "cu128", + "container_image": "pytorch/manylinux2_28-builder:cuda12.8", + "package_type": "wheel", + "build_name": "wheel-py3_12-cuda12_8", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.6.0", + "use_split_build": false + }, { "python_version": "3.12", "gpu_arch_type": "xpu", @@ -360,6 +420,21 @@ "stable_version": "2.6.0", "use_split_build": false }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.8", + "desired_cuda": "cu128", + "container_image": "pytorch/manylinux2_28-builder:cuda12.8", + "package_type": "wheel", + "build_name": "wheel-py3_13-cuda12_8", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.6.0", + "use_split_build": false + }, { "python_version": "3.13", "gpu_arch_type": "xpu", @@ -434,6 +509,21 @@ "upload_to_base_bucket": "no", "stable_version": "2.6.0", "use_split_build": false + }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.8", + "desired_cuda": "cu128", + "container_image": "pytorch/manylinux2_28-builder:cuda12.8", + "package_type": "wheel", + "build_name": "wheel-py3_13t-cuda12_8", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.6.0", + "use_split_build": false } ] } \ No newline at end of file