From f5bb128cdbc31d2e583b95ff5b774376e41dc786 Mon Sep 17 00:00:00 2001 From: Huy Do Date: Tue, 14 Oct 2025 21:00:55 -0700 Subject: [PATCH 1/3] Bring back CUDA 12.9 builds for domains Signed-off-by: Huy Do --- tools/scripts/generate_binary_build_matrix.py | 6 +- .../assets/build_matrix_linux_wheel_cuda.json | 98 +++++++++++++++++++ .../build_matrix_linux_wheel_cuda_norocm.json | 98 +++++++++++++++++++ .../build_matrix_linux_wheel_nocpu.json | 98 +++++++++++++++++++ .../assets/build_matrix_linux_wheel_xpu.json | 98 +++++++++++++++++++ .../build_matrix_windows_wheel_cuda.json | 98 +++++++++++++++++++ 6 files changed, 493 insertions(+), 3 deletions(-) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index 47c60b338a..9b514c1efd 100755 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -27,9 +27,9 @@ "release": ["3.10", "3.11", "3.12", "3.13", "3.14", "3.14t"], } CUDA_ARCHES_DICT = { - "nightly": ["12.6", "12.8", "13.0"], - "test": ["12.6", "12.8", "13.0"], - "release": ["12.6", "12.8", "13.0"], + "nightly": ["12.6", "12.8", "12.9", "13.0"], + "test": ["12.6", "12.8", "12.9", "13.0"], + "release": ["12.6", "12.8", "12.9", "13.0"], } ROCM_ARCHES_DICT = { diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda.json b/tools/tests/assets/build_matrix_linux_wheel_cuda.json index 495960fedb..02cc5ae0b5 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_cuda.json +++ b/tools/tests/assets/build_matrix_linux_wheel_cuda.json @@ -42,6 +42,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_10-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.10", "gpu_arch_type": "cuda", @@ -126,6 +140,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_11-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.11", "gpu_arch_type": "cuda", @@ -210,6 +238,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_12-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.12", "gpu_arch_type": "cuda", @@ -294,6 +336,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.13", "gpu_arch_type": "cuda", @@ -378,6 +434,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.13t", "gpu_arch_type": "cuda", @@ -462,6 +532,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.14", "gpu_arch_type": "cuda", @@ -546,6 +630,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.14t", "gpu_arch_type": "cuda", diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json b/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json index 44a1771c29..4071f55d1b 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json +++ b/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json @@ -42,6 +42,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_10-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.10", "gpu_arch_type": "cuda", @@ -98,6 +112,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_11-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.11", "gpu_arch_type": "cuda", @@ -154,6 +182,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_12-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.12", "gpu_arch_type": "cuda", @@ -210,6 +252,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.13", "gpu_arch_type": "cuda", @@ -266,6 +322,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.13t", "gpu_arch_type": "cuda", @@ -322,6 +392,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.14", "gpu_arch_type": "cuda", @@ -378,6 +462,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.14t", "gpu_arch_type": "cuda", diff --git a/tools/tests/assets/build_matrix_linux_wheel_nocpu.json b/tools/tests/assets/build_matrix_linux_wheel_nocpu.json index e944e09e92..a7518b4117 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_nocpu.json +++ b/tools/tests/assets/build_matrix_linux_wheel_nocpu.json @@ -28,6 +28,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_10-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.10", "gpu_arch_type": "cuda", @@ -98,6 +112,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_11-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.11", "gpu_arch_type": "cuda", @@ -168,6 +196,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_12-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.12", "gpu_arch_type": "cuda", @@ -238,6 +280,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.13", "gpu_arch_type": "cuda", @@ -308,6 +364,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.13t", "gpu_arch_type": "cuda", @@ -378,6 +448,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.14", "gpu_arch_type": "cuda", @@ -448,6 +532,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.14t", "gpu_arch_type": "cuda", diff --git a/tools/tests/assets/build_matrix_linux_wheel_xpu.json b/tools/tests/assets/build_matrix_linux_wheel_xpu.json index fa464d2fc8..a41a15617a 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_xpu.json +++ b/tools/tests/assets/build_matrix_linux_wheel_xpu.json @@ -28,6 +28,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_10-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.10", "gpu_arch_type": "cuda", @@ -84,6 +98,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_11-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.11", "gpu_arch_type": "cuda", @@ -140,6 +168,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_12-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.12", "gpu_arch_type": "cuda", @@ -196,6 +238,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.13", "gpu_arch_type": "cuda", @@ -252,6 +308,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.13t", "gpu_arch_type": "cuda", @@ -308,6 +378,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.14", "gpu_arch_type": "cuda", @@ -364,6 +448,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.14t", "gpu_arch_type": "cuda", diff --git a/tools/tests/assets/build_matrix_windows_wheel_cuda.json b/tools/tests/assets/build_matrix_windows_wheel_cuda.json index 699d0a5df4..43ed3c2a3d 100644 --- a/tools/tests/assets/build_matrix_windows_wheel_cuda.json +++ b/tools/tests/assets/build_matrix_windows_wheel_cuda.json @@ -42,6 +42,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_10-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.10", "gpu_arch_type": "cuda", @@ -112,6 +126,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_11-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.11", "gpu_arch_type": "cuda", @@ -182,6 +210,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_12-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.12", "gpu_arch_type": "cuda", @@ -252,6 +294,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_13-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.13", "gpu_arch_type": "cuda", @@ -322,6 +378,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_13t-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.13t", "gpu_arch_type": "cuda", @@ -392,6 +462,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_14-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.14", "gpu_arch_type": "cuda", @@ -462,6 +546,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_14t-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, { "python_version": "3.14t", "gpu_arch_type": "cuda", From 1b6992a9ff65e5eaab9d95f9a3809ee1c0ab6491 Mon Sep 17 00:00:00 2001 From: Huy Do Date: Wed, 15 Oct 2025 17:17:31 -0700 Subject: [PATCH 2/3] Address review comments Signed-off-by: Huy Do --- tools/scripts/generate_binary_build_matrix.py | 13 +- .../assets/build_matrix_linux_wheel_cuda.json | 140 +++---- .../build_matrix_linux_wheel_cuda_norocm.json | 168 ++++++-- .../build_matrix_linux_wheel_nocpu.json | 266 +++++++++++-- .../assets/build_matrix_linux_wheel_xpu.json | 364 ++++++++++++++++-- .../build_matrix_windows_wheel_cuda.json | 364 ++++++++++++++++-- 6 files changed, 1100 insertions(+), 215 deletions(-) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index 9b514c1efd..93a21364a7 100755 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -27,9 +27,9 @@ "release": ["3.10", "3.11", "3.12", "3.13", "3.14", "3.14t"], } CUDA_ARCHES_DICT = { - "nightly": ["12.6", "12.8", "12.9", "13.0"], - "test": ["12.6", "12.8", "12.9", "13.0"], - "release": ["12.6", "12.8", "12.9", "13.0"], + "nightly": ["12.6", "12.8", "13.0"], + "test": ["12.6", "12.8", "13.0"], + "release": ["12.6", "12.8", "13.0"], } ROCM_ARCHES_DICT = { @@ -148,7 +148,7 @@ def validation_runner(arch_type: str, os: str) -> str: return LINUX_CPU_RUNNER -def initialize_globals(channel: str, build_python_only: bool) -> None: +def initialize_globals(channel: str, os: str, build_python_only: bool) -> None: global CURRENT_VERSION, CUDA_ARCHES, ROCM_ARCHES, PYTHON_ARCHES global WHEEL_CONTAINER_IMAGES, LIBTORCH_CONTAINER_IMAGES if channel == TEST: @@ -157,6 +157,9 @@ def initialize_globals(channel: str, build_python_only: bool) -> None: CURRENT_VERSION = CURRENT_STABLE_VERSION CUDA_ARCHES = CUDA_ARCHES_DICT[channel] + if channel != "release" and os == LINUX: + # Only build CUDA 12.9 for Linux + CUDA_ARCHES.append("12.9") ROCM_ARCHES = ROCM_ARCHES_DICT[channel] if build_python_only: # Only select the oldest version of python if building a python only package @@ -523,7 +526,7 @@ def generate_build_matrix( for channel in channels: for package in package_types: - initialize_globals(channel, build_python_only == ENABLE) + initialize_globals(channel, operating_system, build_python_only == ENABLE) includes.extend( GENERATING_FUNCTIONS_BY_PACKAGE_TYPE[package]( operating_system, diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda.json b/tools/tests/assets/build_matrix_linux_wheel_cuda.json index 02cc5ae0b5..63a82b8334 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_cuda.json +++ b/tools/tests/assets/build_matrix_linux_wheel_cuda.json @@ -45,13 +45,13 @@ { "python_version": "3.10", "gpu_arch_type": "cuda", - "gpu_arch_version": "12.9", - "desired_cuda": "cu129", - "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_9", + "build_name": "manywheel-py3_10-cuda13_0", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "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.9.0" @@ -59,13 +59,13 @@ { "python_version": "3.10", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda13_0", + "build_name": "manywheel-py3_10-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -143,13 +143,13 @@ { "python_version": "3.11", "gpu_arch_type": "cuda", - "gpu_arch_version": "12.9", - "desired_cuda": "cu129", - "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda12_9", + "build_name": "manywheel-py3_11-cuda13_0", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "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.9.0" @@ -157,13 +157,13 @@ { "python_version": "3.11", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda13_0", + "build_name": "manywheel-py3_11-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -241,13 +241,13 @@ { "python_version": "3.12", "gpu_arch_type": "cuda", - "gpu_arch_version": "12.9", - "desired_cuda": "cu129", - "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda12_9", + "build_name": "manywheel-py3_12-cuda13_0", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "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.9.0" @@ -255,13 +255,13 @@ { "python_version": "3.12", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda13_0", + "build_name": "manywheel-py3_12-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -339,13 +339,13 @@ { "python_version": "3.13", "gpu_arch_type": "cuda", - "gpu_arch_version": "12.9", - "desired_cuda": "cu129", - "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda12_9", + "build_name": "manywheel-py3_13-cuda13_0", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "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.9.0" @@ -353,13 +353,13 @@ { "python_version": "3.13", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda13_0", + "build_name": "manywheel-py3_13-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -437,13 +437,13 @@ { "python_version": "3.13t", "gpu_arch_type": "cuda", - "gpu_arch_version": "12.9", - "desired_cuda": "cu129", - "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", "package_type": "manywheel", - "build_name": "manywheel-py3_13t-cuda12_9", + "build_name": "manywheel-py3_13t-cuda13_0", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "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.9.0" @@ -451,13 +451,13 @@ { "python_version": "3.13t", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_13t-cuda13_0", + "build_name": "manywheel-py3_13t-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -535,13 +535,13 @@ { "python_version": "3.14", "gpu_arch_type": "cuda", - "gpu_arch_version": "12.9", - "desired_cuda": "cu129", - "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", "package_type": "manywheel", - "build_name": "manywheel-py3_14-cuda12_9", + "build_name": "manywheel-py3_14-cuda13_0", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "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.9.0" @@ -549,13 +549,13 @@ { "python_version": "3.14", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_14-cuda13_0", + "build_name": "manywheel-py3_14-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -633,13 +633,13 @@ { "python_version": "3.14t", "gpu_arch_type": "cuda", - "gpu_arch_version": "12.9", - "desired_cuda": "cu129", - "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", "package_type": "manywheel", - "build_name": "manywheel-py3_14t-cuda12_9", + "build_name": "manywheel-py3_14t-cuda13_0", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "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.9.0" @@ -647,13 +647,13 @@ { "python_version": "3.14t", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_14t-cuda13_0", + "build_name": "manywheel-py3_14t-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json b/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json index 4071f55d1b..7e31a53f32 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json +++ b/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json @@ -42,6 +42,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_10-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.10", "gpu_arch_type": "cuda", @@ -59,13 +73,13 @@ { "python_version": "3.10", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda13_0", + "build_name": "manywheel-py3_10-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -112,6 +126,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_11-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.11", "gpu_arch_type": "cuda", @@ -129,13 +157,13 @@ { "python_version": "3.11", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda13_0", + "build_name": "manywheel-py3_11-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -182,6 +210,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_12-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.12", "gpu_arch_type": "cuda", @@ -199,13 +241,13 @@ { "python_version": "3.12", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda13_0", + "build_name": "manywheel-py3_12-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -252,6 +294,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_13-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.13", "gpu_arch_type": "cuda", @@ -269,13 +325,13 @@ { "python_version": "3.13", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda13_0", + "build_name": "manywheel-py3_13-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -322,6 +378,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_13t-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.13t", "gpu_arch_type": "cuda", @@ -339,13 +409,13 @@ { "python_version": "3.13t", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_13t-cuda13_0", + "build_name": "manywheel-py3_13t-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -392,6 +462,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "package_type": "manywheel", + "build_name": "manywheel-py3_14-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.14", "gpu_arch_type": "cuda", @@ -409,13 +493,13 @@ { "python_version": "3.14", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_14-cuda13_0", + "build_name": "manywheel-py3_14-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -462,6 +546,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "package_type": "manywheel", + "build_name": "manywheel-py3_14t-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.14t", "gpu_arch_type": "cuda", @@ -479,13 +577,13 @@ { "python_version": "3.14t", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_14t-cuda13_0", + "build_name": "manywheel-py3_14t-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" diff --git a/tools/tests/assets/build_matrix_linux_wheel_nocpu.json b/tools/tests/assets/build_matrix_linux_wheel_nocpu.json index a7518b4117..5155be9f5c 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_nocpu.json +++ b/tools/tests/assets/build_matrix_linux_wheel_nocpu.json @@ -28,6 +28,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_10-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.10", "gpu_arch_type": "cuda", @@ -45,13 +59,27 @@ { "python_version": "3.10", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda13_0", + "build_name": "manywheel-py3_10-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_10-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -112,6 +140,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_11-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.11", "gpu_arch_type": "cuda", @@ -129,13 +171,27 @@ { "python_version": "3.11", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda13_0", + "build_name": "manywheel-py3_11-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_11-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -196,6 +252,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_12-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.12", "gpu_arch_type": "cuda", @@ -213,13 +283,27 @@ { "python_version": "3.12", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda13_0", + "build_name": "manywheel-py3_12-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_12-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -280,6 +364,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_13-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.13", "gpu_arch_type": "cuda", @@ -297,13 +395,27 @@ { "python_version": "3.13", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda13_0", + "build_name": "manywheel-py3_13-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -364,6 +476,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_13t-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.13t", "gpu_arch_type": "cuda", @@ -381,13 +507,27 @@ { "python_version": "3.13t", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_13t-cuda13_0", + "build_name": "manywheel-py3_13t-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -448,6 +588,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "package_type": "manywheel", + "build_name": "manywheel-py3_14-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.14", "gpu_arch_type": "cuda", @@ -465,13 +619,27 @@ { "python_version": "3.14", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_14-cuda13_0", + "build_name": "manywheel-py3_14-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -532,6 +700,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "package_type": "manywheel", + "build_name": "manywheel-py3_14t-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.14t", "gpu_arch_type": "cuda", @@ -549,13 +731,27 @@ { "python_version": "3.14t", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_14t-cuda13_0", + "build_name": "manywheel-py3_14t-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" diff --git a/tools/tests/assets/build_matrix_linux_wheel_xpu.json b/tools/tests/assets/build_matrix_linux_wheel_xpu.json index a41a15617a..14569f2f24 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_xpu.json +++ b/tools/tests/assets/build_matrix_linux_wheel_xpu.json @@ -28,6 +28,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_10-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.10", "gpu_arch_type": "cuda", @@ -45,13 +59,41 @@ { "python_version": "3.10", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda13_0", + "build_name": "manywheel-py3_10-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_10-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_10-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -98,6 +140,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_11-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.11", "gpu_arch_type": "cuda", @@ -115,13 +171,41 @@ { "python_version": "3.11", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda13_0", + "build_name": "manywheel-py3_11-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_11-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_11-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -168,6 +252,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_12-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.12", "gpu_arch_type": "cuda", @@ -185,13 +283,41 @@ { "python_version": "3.12", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda13_0", + "build_name": "manywheel-py3_12-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_12-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_12-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -238,6 +364,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_13-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.13", "gpu_arch_type": "cuda", @@ -255,13 +395,41 @@ { "python_version": "3.13", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda13_0", + "build_name": "manywheel-py3_13-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -308,6 +476,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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": "manywheel", + "build_name": "manywheel-py3_13t-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.13t", "gpu_arch_type": "cuda", @@ -325,13 +507,41 @@ { "python_version": "3.13t", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_13t-cuda13_0", + "build_name": "manywheel-py3_13t-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_13t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -378,6 +588,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "package_type": "manywheel", + "build_name": "manywheel-py3_14-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.14", "gpu_arch_type": "cuda", @@ -395,13 +619,41 @@ { "python_version": "3.14", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_14-cuda13_0", + "build_name": "manywheel-py3_14-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -448,6 +700,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "13.0", + "desired_cuda": "cu130", + "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "package_type": "manywheel", + "build_name": "manywheel-py3_14t-cuda13_0", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "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.9.0" + }, { "python_version": "3.14t", "gpu_arch_type": "cuda", @@ -465,13 +731,41 @@ { "python_version": "3.14t", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "manywheel", - "build_name": "manywheel-py3_14t-cuda13_0", + "build_name": "manywheel-py3_14t-cuda12_9", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "manywheel", + "build_name": "manywheel-py3_14t-cuda12_9", + "validation_runner": "linux.g5.4xlarge.nvidia.gpu", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" diff --git a/tools/tests/assets/build_matrix_windows_wheel_cuda.json b/tools/tests/assets/build_matrix_windows_wheel_cuda.json index 43ed3c2a3d..cbb1b7b9e3 100644 --- a/tools/tests/assets/build_matrix_windows_wheel_cuda.json +++ b/tools/tests/assets/build_matrix_windows_wheel_cuda.json @@ -42,6 +42,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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.9.0" + }, { "python_version": "3.10", "gpu_arch_type": "cuda", @@ -59,13 +73,41 @@ { "python_version": "3.10", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "wheel", - "build_name": "wheel-py3_10-cuda13_0", + "build_name": "wheel-py3_10-cuda12_9", "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_10-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.10", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_10-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -126,6 +168,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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.9.0" + }, { "python_version": "3.11", "gpu_arch_type": "cuda", @@ -143,13 +199,41 @@ { "python_version": "3.11", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "wheel", - "build_name": "wheel-py3_11-cuda13_0", + "build_name": "wheel-py3_11-cuda12_9", "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_11-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.11", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_11-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -210,6 +294,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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.9.0" + }, { "python_version": "3.12", "gpu_arch_type": "cuda", @@ -227,13 +325,41 @@ { "python_version": "3.12", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "wheel", - "build_name": "wheel-py3_12-cuda13_0", + "build_name": "wheel-py3_12-cuda12_9", "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_12-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.12", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_12-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -294,6 +420,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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.9.0" + }, { "python_version": "3.13", "gpu_arch_type": "cuda", @@ -311,13 +451,41 @@ { "python_version": "3.13", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "wheel", - "build_name": "wheel-py3_13-cuda13_0", + "build_name": "wheel-py3_13-cuda12_9", "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_13-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.13", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_13-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -378,6 +546,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "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.9.0" + }, { "python_version": "3.13t", "gpu_arch_type": "cuda", @@ -395,13 +577,41 @@ { "python_version": "3.13t", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "wheel", - "build_name": "wheel-py3_13t-cuda13_0", + "build_name": "wheel-py3_13t-cuda12_9", "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_13t-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.13t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_13t-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -462,6 +672,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14", + "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_14-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.9.0" + }, { "python_version": "3.14", "gpu_arch_type": "cuda", @@ -479,13 +703,41 @@ { "python_version": "3.14", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "wheel", - "build_name": "wheel-py3_14-cuda13_0", + "build_name": "wheel-py3_14-cuda12_9", "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_14-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.14", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_14-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" @@ -546,6 +798,20 @@ "upload_to_base_bucket": "no", "stable_version": "2.9.0" }, + { + "python_version": "3.14t", + "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_14t-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.9.0" + }, { "python_version": "3.14t", "gpu_arch_type": "cuda", @@ -563,13 +829,41 @@ { "python_version": "3.14t", "gpu_arch_type": "cuda", - "gpu_arch_version": "13.0", - "desired_cuda": "cu130", - "container_image": "pytorch/manylinux2_28-builder:cuda13.0", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", "package_type": "wheel", - "build_name": "wheel-py3_14t-cuda13_0", + "build_name": "wheel-py3_14t-cuda12_9", "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_14t-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", + "channel": "nightly", + "upload_to_base_bucket": "no", + "stable_version": "2.9.0" + }, + { + "python_version": "3.14t", + "gpu_arch_type": "cuda", + "gpu_arch_version": "12.9", + "desired_cuda": "cu129", + "container_image": "pytorch/manylinux2_28-builder:cuda12.9", + "package_type": "wheel", + "build_name": "wheel-py3_14t-cuda12_9", + "validation_runner": "windows.g4dn.xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.9.0" From d4978015170f780ac0b12e99c154a69a8675383f Mon Sep 17 00:00:00 2001 From: Huy Do Date: Wed, 15 Oct 2025 17:19:06 -0700 Subject: [PATCH 3/3] Add more comments Signed-off-by: Huy Do --- tools/scripts/generate_binary_build_matrix.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index 93a21364a7..7246e40368 100755 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -158,7 +158,8 @@ def initialize_globals(channel: str, os: str, build_python_only: bool) -> None: CUDA_ARCHES = CUDA_ARCHES_DICT[channel] if channel != "release" and os == LINUX: - # Only build CUDA 12.9 for Linux + # TODO (huydhn): Only build CUDA 12.9 for Linux. This logic is to be cleaned up + # in 2.10 CUDA_ARCHES.append("12.9") ROCM_ARCHES = ROCM_ARCHES_DICT[channel] if build_python_only: