From cf7d0f8da9b577ecef134b4522c06c4d7cfc165c Mon Sep 17 00:00:00 2001 From: Ting Lu Date: Mon, 16 Dec 2024 10:52:31 -0800 Subject: [PATCH] add cu126 torchvision and torchaudio aarch64 build --- tools/scripts/generate_binary_build_matrix.py | 2 +- .../assets/build_matrix_linux_conda_cuda.json | 4 +- .../assets/build_matrix_linux_wheel_cuda.json | 425 +----------------- .../build_matrix_linux_wheel_cuda_norocm.json | 305 +------------ .../build_matrix_linux_wheel_nocpu.json | 350 +-------------- .../assets/build_matrix_linux_wheel_xpu.json | 305 +------------ .../assets/build_matrix_macos_conda.json | 4 +- .../assets/build_matrix_macos_wheel.json | 80 +--- .../build_matrix_windows_conda_cuda.json | 4 +- .../build_matrix_windows_wheel_cuda.json | 380 +--------------- .../build_matrix_windows_wheel_xpu.json | 155 +------ 11 files changed, 11 insertions(+), 2003 deletions(-) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index 6449eb44fc..50638046d9 100755 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -191,7 +191,7 @@ def translate_desired_cuda(gpu_arch_type: str, gpu_arch_version: str) -> str: return { CPU: "cpu", CPU_AARCH64: CPU, - CUDA_AARCH64: "cu124", + CUDA_AARCH64: "cu126", CUDA: f"cu{gpu_arch_version.replace('.', '')}", ROCM: f"rocm{gpu_arch_version}", XPU: "xpu", diff --git a/tools/tests/assets/build_matrix_linux_conda_cuda.json b/tools/tests/assets/build_matrix_linux_conda_cuda.json index d3f37ec39c..efafe1abef 100644 --- a/tools/tests/assets/build_matrix_linux_conda_cuda.json +++ b/tools/tests/assets/build_matrix_linux_conda_cuda.json @@ -1,3 +1 @@ -{ - "include": [] -} \ No newline at end of file +{"include": []} diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda.json b/tools/tests/assets/build_matrix_linux_wheel_cuda.json index 62c34c3eaf..9e62756878 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_cuda.json +++ b/tools/tests/assets/build_matrix_linux_wheel_cuda.json @@ -1,424 +1 @@ -{ - "include": [ - { - "python_version": "3.9", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.1", - "desired_cuda": "rocm6.1", - "container_image": "pytorch/manylinux2_28-builder:rocm6.1", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-rocm6_1", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.2.4", - "desired_cuda": "rocm6.2.4", - "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-rocm6_2_4", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.1", - "desired_cuda": "rocm6.1", - "container_image": "pytorch/manylinux2_28-builder:rocm6.1", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-rocm6_1", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.2.4", - "desired_cuda": "rocm6.2.4", - "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-rocm6_2_4", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.1", - "desired_cuda": "rocm6.1", - "container_image": "pytorch/manylinux2_28-builder:rocm6.1", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-rocm6_1", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.2.4", - "desired_cuda": "rocm6.2.4", - "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-rocm6_2_4", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.1", - "desired_cuda": "rocm6.1", - "container_image": "pytorch/manylinux2_28-builder:rocm6.1", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-rocm6_1", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.2.4", - "desired_cuda": "rocm6.2.4", - "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-rocm6_2_4", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - } - ] -} \ No newline at end of file +{"include": [{"python_version": "3.9", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "manywheel", "build_name": "manywheel-py3_9-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "rocm", "gpu_arch_version": "6.1", "desired_cuda": "rocm6.1", "container_image": "pytorch/manylinux2_28-builder:rocm6.1", "package_type": "manywheel", "build_name": "manywheel-py3_9-rocm6_1", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "rocm", "gpu_arch_version": "6.2.4", "desired_cuda": "rocm6.2.4", "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", "package_type": "manywheel", "build_name": "manywheel-py3_9-rocm6_2_4", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "manywheel", "build_name": "manywheel-py3_10-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "rocm", "gpu_arch_version": "6.1", "desired_cuda": "rocm6.1", "container_image": "pytorch/manylinux2_28-builder:rocm6.1", "package_type": "manywheel", "build_name": "manywheel-py3_10-rocm6_1", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "rocm", "gpu_arch_version": "6.2.4", "desired_cuda": "rocm6.2.4", "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", "package_type": "manywheel", "build_name": "manywheel-py3_10-rocm6_2_4", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "manywheel", "build_name": "manywheel-py3_11-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "rocm", "gpu_arch_version": "6.1", "desired_cuda": "rocm6.1", "container_image": "pytorch/manylinux2_28-builder:rocm6.1", "package_type": "manywheel", "build_name": "manywheel-py3_11-rocm6_1", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "rocm", "gpu_arch_version": "6.2.4", "desired_cuda": "rocm6.2.4", "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", "package_type": "manywheel", "build_name": "manywheel-py3_11-rocm6_2_4", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "manywheel", "build_name": "manywheel-py3_12-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "rocm", "gpu_arch_version": "6.1", "desired_cuda": "rocm6.1", "container_image": "pytorch/manylinux2_28-builder:rocm6.1", "package_type": "manywheel", "build_name": "manywheel-py3_12-rocm6_1", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "rocm", "gpu_arch_version": "6.2.4", "desired_cuda": "rocm6.2.4", "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", "package_type": "manywheel", "build_name": "manywheel-py3_12-rocm6_2_4", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "manywheel", "build_name": "manywheel-py3_13-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}]} 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 1af8daf1a8..e6c298c09e 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json +++ b/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json @@ -1,304 +1 @@ -{ - "include": [ - { - "python_version": "3.9", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - } - ] -} \ No newline at end of file +{"include": [{"python_version": "3.9", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "manywheel", "build_name": "manywheel-py3_9-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "manywheel", "build_name": "manywheel-py3_10-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "manywheel", "build_name": "manywheel-py3_11-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "manywheel", "build_name": "manywheel-py3_12-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "manywheel", "build_name": "manywheel-py3_13-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}]} diff --git a/tools/tests/assets/build_matrix_linux_wheel_nocpu.json b/tools/tests/assets/build_matrix_linux_wheel_nocpu.json index dbf87f230d..109ef5073c 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_nocpu.json +++ b/tools/tests/assets/build_matrix_linux_wheel_nocpu.json @@ -1,349 +1 @@ -{ - "include": [ - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.1", - "desired_cuda": "rocm6.1", - "container_image": "pytorch/manylinux2_28-builder:rocm6.1", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-rocm6_1", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.2.4", - "desired_cuda": "rocm6.2.4", - "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-rocm6_2_4", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.1", - "desired_cuda": "rocm6.1", - "container_image": "pytorch/manylinux2_28-builder:rocm6.1", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-rocm6_1", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.2.4", - "desired_cuda": "rocm6.2.4", - "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-rocm6_2_4", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.1", - "desired_cuda": "rocm6.1", - "container_image": "pytorch/manylinux2_28-builder:rocm6.1", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-rocm6_1", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.2.4", - "desired_cuda": "rocm6.2.4", - "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-rocm6_2_4", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.1", - "desired_cuda": "rocm6.1", - "container_image": "pytorch/manylinux2_28-builder:rocm6.1", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-rocm6_1", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "rocm", - "gpu_arch_version": "6.2.4", - "desired_cuda": "rocm6.2.4", - "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-rocm6_2_4", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - } - ] -} \ No newline at end of file +{"include": [{"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "rocm", "gpu_arch_version": "6.1", "desired_cuda": "rocm6.1", "container_image": "pytorch/manylinux2_28-builder:rocm6.1", "package_type": "manywheel", "build_name": "manywheel-py3_9-rocm6_1", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "rocm", "gpu_arch_version": "6.2.4", "desired_cuda": "rocm6.2.4", "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", "package_type": "manywheel", "build_name": "manywheel-py3_9-rocm6_2_4", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "rocm", "gpu_arch_version": "6.1", "desired_cuda": "rocm6.1", "container_image": "pytorch/manylinux2_28-builder:rocm6.1", "package_type": "manywheel", "build_name": "manywheel-py3_10-rocm6_1", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "rocm", "gpu_arch_version": "6.2.4", "desired_cuda": "rocm6.2.4", "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", "package_type": "manywheel", "build_name": "manywheel-py3_10-rocm6_2_4", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "rocm", "gpu_arch_version": "6.1", "desired_cuda": "rocm6.1", "container_image": "pytorch/manylinux2_28-builder:rocm6.1", "package_type": "manywheel", "build_name": "manywheel-py3_11-rocm6_1", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "rocm", "gpu_arch_version": "6.2.4", "desired_cuda": "rocm6.2.4", "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", "package_type": "manywheel", "build_name": "manywheel-py3_11-rocm6_2_4", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "rocm", "gpu_arch_version": "6.1", "desired_cuda": "rocm6.1", "container_image": "pytorch/manylinux2_28-builder:rocm6.1", "package_type": "manywheel", "build_name": "manywheel-py3_12-rocm6_1", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "rocm", "gpu_arch_version": "6.2.4", "desired_cuda": "rocm6.2.4", "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4", "package_type": "manywheel", "build_name": "manywheel-py3_12-rocm6_2_4", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}]} diff --git a/tools/tests/assets/build_matrix_linux_wheel_xpu.json b/tools/tests/assets/build_matrix_linux_wheel_xpu.json index cdb2afc7da..bb2258c19e 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_xpu.json +++ b/tools/tests/assets/build_matrix_linux_wheel_xpu.json @@ -1,304 +1 @@ -{ - "include": [ - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_9-xpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_10-xpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_11-xpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_12-xpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda11_8", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda12_4", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-cuda12_6", - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "manywheel", - "build_name": "manywheel-py3_13-xpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - } - ] -} \ No newline at end of file +{"include": [{"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_9-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "manywheel", "build_name": "manywheel-py3_9-xpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_10-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "manywheel", "build_name": "manywheel-py3_10-xpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_11-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "manywheel", "build_name": "manywheel-py3_11-xpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_12-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "manywheel", "build_name": "manywheel-py3_12-xpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda11_8", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda12_4", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "manywheel", "build_name": "manywheel-py3_13-cuda12_6", "validation_runner": "linux.g5.4xlarge.nvidia.gpu", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "manywheel", "build_name": "manywheel-py3_13-xpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}]} diff --git a/tools/tests/assets/build_matrix_macos_conda.json b/tools/tests/assets/build_matrix_macos_conda.json index d3f37ec39c..efafe1abef 100644 --- a/tools/tests/assets/build_matrix_macos_conda.json +++ b/tools/tests/assets/build_matrix_macos_conda.json @@ -1,3 +1 @@ -{ - "include": [] -} \ No newline at end of file +{"include": []} diff --git a/tools/tests/assets/build_matrix_macos_wheel.json b/tools/tests/assets/build_matrix_macos_wheel.json index 4c492f6df1..4ae2d82a05 100644 --- a/tools/tests/assets/build_matrix_macos_wheel.json +++ b/tools/tests/assets/build_matrix_macos_wheel.json @@ -1,79 +1 @@ -{ - "include": [ - { - "python_version": "3.9", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_9-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_10-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_11-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_12-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_13-cpu", - "validation_runner": "linux.2xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - } - ] -} \ No newline at end of file +{"include": [{"python_version": "3.9", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_9-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_10-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_11-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_12-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_13-cpu", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}]} diff --git a/tools/tests/assets/build_matrix_windows_conda_cuda.json b/tools/tests/assets/build_matrix_windows_conda_cuda.json index d3f37ec39c..efafe1abef 100644 --- a/tools/tests/assets/build_matrix_windows_conda_cuda.json +++ b/tools/tests/assets/build_matrix_windows_conda_cuda.json @@ -1,3 +1 @@ -{ - "include": [] -} \ No newline at end of file +{"include": []} diff --git a/tools/tests/assets/build_matrix_windows_wheel_cuda.json b/tools/tests/assets/build_matrix_windows_wheel_cuda.json index 966ba6e0bb..168b5bf052 100644 --- a/tools/tests/assets/build_matrix_windows_wheel_cuda.json +++ b/tools/tests/assets/build_matrix_windows_wheel_cuda.json @@ -1,379 +1 @@ -{ - "include": [ - { - "python_version": "3.9", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_9-cpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "wheel", - "build_name": "wheel-py3_9-cuda11_8", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "wheel", - "build_name": "wheel-py3_9-cuda12_4", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "wheel", - "build_name": "wheel-py3_9-cuda12_6", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "wheel", - "build_name": "wheel-py3_9-xpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_10-cpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "wheel", - "build_name": "wheel-py3_10-cuda11_8", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "wheel", - "build_name": "wheel-py3_10-cuda12_4", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "wheel", - "build_name": "wheel-py3_10-cuda12_6", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "wheel", - "build_name": "wheel-py3_10-xpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_11-cpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "wheel", - "build_name": "wheel-py3_11-cuda11_8", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "wheel", - "build_name": "wheel-py3_11-cuda12_4", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "wheel", - "build_name": "wheel-py3_11-cuda12_6", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "wheel", - "build_name": "wheel-py3_11-xpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_12-cpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "wheel", - "build_name": "wheel-py3_12-cuda11_8", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "wheel", - "build_name": "wheel-py3_12-cuda12_4", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "wheel", - "build_name": "wheel-py3_12-cuda12_6", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "wheel", - "build_name": "wheel-py3_12-xpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_13-cpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "11.8", - "desired_cuda": "cu118", - "container_image": "pytorch/manylinux-builder:cuda11.8", - "package_type": "wheel", - "build_name": "wheel-py3_13-cuda11_8", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.4", - "desired_cuda": "cu124", - "container_image": "pytorch/manylinux-builder:cuda12.4", - "package_type": "wheel", - "build_name": "wheel-py3_13-cuda12_4", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cuda", - "gpu_arch_version": "12.6", - "desired_cuda": "cu126", - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", - "package_type": "wheel", - "build_name": "wheel-py3_13-cuda12_6", - "validation_runner": "windows.g4dn.xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "wheel", - "build_name": "wheel-py3_13-xpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "no", - "stable_version": "2.5.1", - "use_split_build": false - } - ] -} \ No newline at end of file +{"include": [{"python_version": "3.9", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_9-cpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "wheel", "build_name": "wheel-py3_9-cuda11_8", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "wheel", "build_name": "wheel-py3_9-cuda12_4", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "wheel", "build_name": "wheel-py3_9-cuda12_6", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "wheel", "build_name": "wheel-py3_9-xpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_10-cpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "wheel", "build_name": "wheel-py3_10-cuda11_8", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "wheel", "build_name": "wheel-py3_10-cuda12_4", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "wheel", "build_name": "wheel-py3_10-cuda12_6", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "wheel", "build_name": "wheel-py3_10-xpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_11-cpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "wheel", "build_name": "wheel-py3_11-cuda11_8", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "wheel", "build_name": "wheel-py3_11-cuda12_4", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "wheel", "build_name": "wheel-py3_11-cuda12_6", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "wheel", "build_name": "wheel-py3_11-xpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_12-cpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "wheel", "build_name": "wheel-py3_12-cuda11_8", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "wheel", "build_name": "wheel-py3_12-cuda12_4", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "wheel", "build_name": "wheel-py3_12-cuda12_6", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "wheel", "build_name": "wheel-py3_12-xpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_13-cpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux-builder:cuda11.8", "package_type": "wheel", "build_name": "wheel-py3_13-cuda11_8", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "12.4", "desired_cuda": "cu124", "container_image": "pytorch/manylinux-builder:cuda12.4", "package_type": "wheel", "build_name": "wheel-py3_13-cuda12_4", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cuda", "gpu_arch_version": "12.6", "desired_cuda": "cu126", "container_image": "pytorch/manylinux2_28-builder:cuda12.6", "package_type": "wheel", "build_name": "wheel-py3_13-cuda12_6", "validation_runner": "windows.g4dn.xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "wheel", "build_name": "wheel-py3_13-xpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.5.1", "use_split_build": false}]} diff --git a/tools/tests/assets/build_matrix_windows_wheel_xpu.json b/tools/tests/assets/build_matrix_windows_wheel_xpu.json index 2b49fc9284..1483e3d024 100644 --- a/tools/tests/assets/build_matrix_windows_wheel_xpu.json +++ b/tools/tests/assets/build_matrix_windows_wheel_xpu.json @@ -1,154 +1 @@ -{ - "include": [ - { - "python_version": "3.9", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_9-cpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.9", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "wheel", - "build_name": "wheel-py3_9-xpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_10-cpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.10", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "wheel", - "build_name": "wheel-py3_10-xpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_11-cpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.11", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "wheel", - "build_name": "wheel-py3_11-xpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_12-cpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.12", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "wheel", - "build_name": "wheel-py3_12-xpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "cpu", - "gpu_arch_version": "", - "desired_cuda": "cpu", - "container_image": "pytorch/manylinux-builder:cpu", - "package_type": "wheel", - "build_name": "wheel-py3_13-cpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - }, - { - "python_version": "3.13", - "gpu_arch_type": "xpu", - "gpu_arch_version": "", - "desired_cuda": "xpu", - "container_image": "pytorch/manylinux2_28-builder:xpu", - "package_type": "wheel", - "build_name": "wheel-py3_13-xpu", - "validation_runner": "windows.4xlarge", - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", - "channel": "nightly", - "upload_to_base_bucket": "yes", - "stable_version": "2.5.1", - "use_split_build": false - } - ] -} \ No newline at end of file +{"include": [{"python_version": "3.9", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_9-cpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.9", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "wheel", "build_name": "wheel-py3_9-xpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_10-cpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.10", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "wheel", "build_name": "wheel-py3_10-xpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_11-cpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.11", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "wheel", "build_name": "wheel-py3_11-xpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_12-cpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.12", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "wheel", "build_name": "wheel-py3_12-xpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux-builder:cpu", "package_type": "wheel", "build_name": "wheel-py3_13-cpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}, {"python_version": "3.13", "gpu_arch_type": "xpu", "gpu_arch_version": "", "desired_cuda": "xpu", "container_image": "pytorch/manylinux2_28-builder:xpu", "package_type": "wheel", "build_name": "wheel-py3_13-xpu", "validation_runner": "windows.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu", "channel": "nightly", "upload_to_base_bucket": "yes", "stable_version": "2.5.1", "use_split_build": false}]}