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)