From 404c4ea35b4f279ebcfd8787b1fdc8055833a6f0 Mon Sep 17 00:00:00 2001 From: clee2000 <44682903+clee2000@users.noreply.github.com> Date: Wed, 27 Nov 2024 12:39:29 -0800 Subject: [PATCH] Update test_generate_binary_build_matrix.py reference files after #5986 (#5993) Also adds a way to update the files by just running the test script through `python -m tools.tests.test_generate_binary_build_matrix --update-reference-files` --- .../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 | 65 ++- .../build_matrix_windows_conda_cuda.json | 4 +- .../build_matrix_windows_wheel_cuda.json | 245 +++++++++- .../build_matrix_windows_wheel_xpu.json | 125 +++++- .../test_generate_binary_build_matrix.py | 25 +- 11 files changed, 1844 insertions(+), 13 deletions(-) diff --git a/tools/tests/assets/build_matrix_linux_conda_cuda.json b/tools/tests/assets/build_matrix_linux_conda_cuda.json index efafe1abef..d3f37ec39c 100644 --- a/tools/tests/assets/build_matrix_linux_conda_cuda.json +++ b/tools/tests/assets/build_matrix_linux_conda_cuda.json @@ -1 +1,3 @@ -{"include": []} +{ + "include": [] +} \ No newline at end of file diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda.json b/tools/tests/assets/build_matrix_linux_wheel_cuda.json index cf53a9ffd6..7c57b7468b 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_cuda.json +++ b/tools/tests/assets/build_matrix_linux_wheel_cuda.json @@ -1 +1,424 @@ -{"include": [{"python_version": "3.9", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux-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", "desired_cuda": "rocm6.2", "container_image": "pytorch/manylinux-builder:rocm6.2", "package_type": "manywheel", "build_name": "manywheel-py3_9-rocm6_2", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", "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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux-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", "desired_cuda": "rocm6.2", "container_image": "pytorch/manylinux-builder:rocm6.2", "package_type": "manywheel", "build_name": "manywheel-py3_10-rocm6_2", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", "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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux-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", "desired_cuda": "rocm6.2", "container_image": "pytorch/manylinux-builder:rocm6.2", "package_type": "manywheel", "build_name": "manywheel-py3_11-rocm6_2", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", "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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux-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", "desired_cuda": "rocm6.2", "container_image": "pytorch/manylinux-builder:rocm6.2", "package_type": "manywheel", "build_name": "manywheel-py3_12-rocm6_2", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", "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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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}]} +{ + "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/manylinux-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", + "desired_cuda": "rocm6.2", + "container_image": "pytorch/manylinux-builder:rocm6.2", + "package_type": "manywheel", + "build_name": "manywheel-py3_9-rocm6_2", + "validation_runner": "linux.2xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", + "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/manylinux-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", + "desired_cuda": "rocm6.2", + "container_image": "pytorch/manylinux-builder:rocm6.2", + "package_type": "manywheel", + "build_name": "manywheel-py3_10-rocm6_2", + "validation_runner": "linux.2xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", + "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/manylinux-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", + "desired_cuda": "rocm6.2", + "container_image": "pytorch/manylinux-builder:rocm6.2", + "package_type": "manywheel", + "build_name": "manywheel-py3_11-rocm6_2", + "validation_runner": "linux.2xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", + "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/manylinux-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", + "desired_cuda": "rocm6.2", + "container_image": "pytorch/manylinux-builder:rocm6.2", + "package_type": "manywheel", + "build_name": "manywheel-py3_12-rocm6_2", + "validation_runner": "linux.2xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", + "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 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 68803b8338..1af8daf1a8 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json +++ b/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json @@ -1 +1,304 @@ -{"include": [{"python_version": "3.9", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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}]} +{ + "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 diff --git a/tools/tests/assets/build_matrix_linux_wheel_nocpu.json b/tools/tests/assets/build_matrix_linux_wheel_nocpu.json index 778c9dc049..3d139504fb 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_nocpu.json +++ b/tools/tests/assets/build_matrix_linux_wheel_nocpu.json @@ -1 +1,349 @@ -{"include": [{"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux2_28-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/manylinux2_28-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/manylinux-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", "desired_cuda": "rocm6.2", "container_image": "pytorch/manylinux-builder:rocm6.2", "package_type": "manywheel", "build_name": "manywheel-py3_9-rocm6_2", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", "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/manylinux2_28-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/manylinux2_28-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/manylinux-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", "desired_cuda": "rocm6.2", "container_image": "pytorch/manylinux-builder:rocm6.2", "package_type": "manywheel", "build_name": "manywheel-py3_10-rocm6_2", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", "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/manylinux2_28-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/manylinux2_28-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/manylinux-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", "desired_cuda": "rocm6.2", "container_image": "pytorch/manylinux-builder:rocm6.2", "package_type": "manywheel", "build_name": "manywheel-py3_11-rocm6_2", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", "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/manylinux2_28-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/manylinux2_28-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/manylinux-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", "desired_cuda": "rocm6.2", "container_image": "pytorch/manylinux-builder:rocm6.2", "package_type": "manywheel", "build_name": "manywheel-py3_12-rocm6_2", "validation_runner": "linux.2xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", "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/manylinux2_28-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/manylinux2_28-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}]} +{ + "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/manylinux-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", + "desired_cuda": "rocm6.2", + "container_image": "pytorch/manylinux-builder:rocm6.2", + "package_type": "manywheel", + "build_name": "manywheel-py3_9-rocm6_2", + "validation_runner": "linux.2xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", + "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/manylinux-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", + "desired_cuda": "rocm6.2", + "container_image": "pytorch/manylinux-builder:rocm6.2", + "package_type": "manywheel", + "build_name": "manywheel-py3_10-rocm6_2", + "validation_runner": "linux.2xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", + "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/manylinux-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", + "desired_cuda": "rocm6.2", + "container_image": "pytorch/manylinux-builder:rocm6.2", + "package_type": "manywheel", + "build_name": "manywheel-py3_11-rocm6_2", + "validation_runner": "linux.2xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", + "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/manylinux-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", + "desired_cuda": "rocm6.2", + "container_image": "pytorch/manylinux-builder:rocm6.2", + "package_type": "manywheel", + "build_name": "manywheel-py3_12-rocm6_2", + "validation_runner": "linux.2xlarge", + "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2", + "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 diff --git a/tools/tests/assets/build_matrix_linux_wheel_xpu.json b/tools/tests/assets/build_matrix_linux_wheel_xpu.json index 99d565629c..cdb2afc7da 100644 --- a/tools/tests/assets/build_matrix_linux_wheel_xpu.json +++ b/tools/tests/assets/build_matrix_linux_wheel_xpu.json @@ -1 +1,304 @@ -{"include": [{"python_version": "3.9", "gpu_arch_type": "cuda", "gpu_arch_version": "11.8", "desired_cuda": "cu118", "container_image": "pytorch/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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}]} +{ + "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 diff --git a/tools/tests/assets/build_matrix_macos_conda.json b/tools/tests/assets/build_matrix_macos_conda.json index efafe1abef..d3f37ec39c 100644 --- a/tools/tests/assets/build_matrix_macos_conda.json +++ b/tools/tests/assets/build_matrix_macos_conda.json @@ -1 +1,3 @@ -{"include": []} +{ + "include": [] +} \ No newline at end of file diff --git a/tools/tests/assets/build_matrix_macos_wheel.json b/tools/tests/assets/build_matrix_macos_wheel.json index bbf028a59e..627a2e8d50 100644 --- a/tools/tests/assets/build_matrix_macos_wheel.json +++ b/tools/tests/assets/build_matrix_macos_wheel.json @@ -1 +1,64 @@ -{"include": [{"python_version": "3.9", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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}]} +{ + "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 + } + ] +} \ No newline at end of file diff --git a/tools/tests/assets/build_matrix_windows_conda_cuda.json b/tools/tests/assets/build_matrix_windows_conda_cuda.json index efafe1abef..d3f37ec39c 100644 --- a/tools/tests/assets/build_matrix_windows_conda_cuda.json +++ b/tools/tests/assets/build_matrix_windows_conda_cuda.json @@ -1 +1,3 @@ -{"include": []} +{ + "include": [] +} \ No newline at end of file diff --git a/tools/tests/assets/build_matrix_windows_wheel_cuda.json b/tools/tests/assets/build_matrix_windows_wheel_cuda.json index 298fd01c47..c5fdf12b83 100644 --- a/tools/tests/assets/build_matrix_windows_wheel_cuda.json +++ b/tools/tests/assets/build_matrix_windows_wheel_cuda.json @@ -1 +1,244 @@ -{"include": [{"python_version": "3.9", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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": "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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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": "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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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": "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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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": "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}]} +{ + "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": "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": "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": "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": "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 + } + ] +} \ No newline at end of file diff --git a/tools/tests/assets/build_matrix_windows_wheel_xpu.json b/tools/tests/assets/build_matrix_windows_wheel_xpu.json index 4ba8b0af98..8cb810207f 100644 --- a/tools/tests/assets/build_matrix_windows_wheel_xpu.json +++ b/tools/tests/assets/build_matrix_windows_wheel_xpu.json @@ -1 +1,124 @@ -{"include": [{"python_version": "3.9", "gpu_arch_type": "cpu", "gpu_arch_version": "", "desired_cuda": "cpu", "container_image": "pytorch/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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/manylinux2_28-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}]} +{ + "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 + } + ] +} \ No newline at end of file diff --git a/tools/tests/test_generate_binary_build_matrix.py b/tools/tests/test_generate_binary_build_matrix.py index 61b52656cc..d1cae0adbc 100644 --- a/tools/tests/test_generate_binary_build_matrix.py +++ b/tools/tests/test_generate_binary_build_matrix.py @@ -1,6 +1,8 @@ import json import os +import argparse +import sys from unittest import main, TestCase from tools.scripts.generate_binary_build_matrix import generate_build_matrix @@ -9,6 +11,8 @@ class GenerateBuildMatrixTest(TestCase): + update_reference_files = False + def matrix_compare_helper( self, package_type: str, @@ -31,11 +35,14 @@ def matrix_compare_helper( "false", "false", "enable" if build_python_only else "disable", - ) expected_json_filename = os.path.join(ASSETS_DIR, reference_output_file) + if self.update_reference_files: + with open(expected_json_filename, "w") as f: + json.dump(out, f, indent=2) + with open(expected_json_filename) as f: expected = json.load(f) @@ -140,7 +147,7 @@ def test_linux_wheel_cuda_rocm_nocpu(self): xpu=False, reference_output_file="build_matrix_linux_wheel_nocpu.json", ) - + def test_linux_wheel_cuda_xpu_nocpu(self): self.matrix_compare_helper( package_type="wheel", @@ -153,5 +160,17 @@ def test_linux_wheel_cuda_xpu_nocpu(self): ) +def parse_args(): + parser = argparse.ArgumentParser(description="Test generate build matrix") + parser.add_argument( + "--update-reference-files", + action="store_true", + help="Update reference files with the generated output", + ) + return parser.parse_known_args() + + if __name__ == "__main__": - main() + args, unittest_args = parse_args() + GenerateBuildMatrixTest.update_reference_files = args.update_reference_files + main(argv=[sys.argv[0]] + unittest_args)