diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index dd0e0108a2..ebd1519c9b 100755 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -445,9 +445,6 @@ def generate_wheels_matrix( upload_to_base_bucket = "no" if os in (LINUX, WINDOWS): arches += CUDA_ARCHES - # todo: remove once windows cuda 13.0 binaries are available - if channel == NIGHTLY and os != LINUX: - arches.remove("13.0") elif os == LINUX_AARCH64: arches += CUDA_AARCH64_ARCHES diff --git a/tools/tests/assets/build_matrix_windows_wheel_cuda.json b/tools/tests/assets/build_matrix_windows_wheel_cuda.json index 9813dfb151..14c15ca145 100644 --- a/tools/tests/assets/build_matrix_windows_wheel_cuda.json +++ b/tools/tests/assets/build_matrix_windows_wheel_cuda.json @@ -45,6 +45,21 @@ "stable_version": "2.8.0", "use_split_build": false }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "package_type": "wheel", + "build_name": "wheel-py3_10-cuda13_0", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.8.0", + "use_split_build": false + }, { "python_version": "3.10", "gpu_arch_type": "xpu", @@ -105,6 +120,21 @@ "stable_version": "2.8.0", "use_split_build": false }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "package_type": "wheel", + "build_name": "wheel-py3_11-cuda13_0", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.8.0", + "use_split_build": false + }, { "python_version": "3.11", "gpu_arch_type": "xpu", @@ -165,6 +195,21 @@ "stable_version": "2.8.0", "use_split_build": false }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "package_type": "wheel", + "build_name": "wheel-py3_12-cuda13_0", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.8.0", + "use_split_build": false + }, { "python_version": "3.12", "gpu_arch_type": "xpu", @@ -225,6 +270,21 @@ "stable_version": "2.8.0", "use_split_build": false }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "package_type": "wheel", + "build_name": "wheel-py3_13-cuda13_0", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.8.0", + "use_split_build": false + }, { "python_version": "3.13", "gpu_arch_type": "xpu", @@ -285,6 +345,21 @@ "stable_version": "2.8.0", "use_split_build": false }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "package_type": "wheel", + "build_name": "wheel-py3_13t-cuda13_0", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.8.0", + "use_split_build": false + }, { "python_version": "3.13t", "gpu_arch_type": "xpu",