Update test_generate_binary_build_matrix.py reference files after #5986…

… (#5993) Also adds a way to update the files by just running the test script through `python -m tools.tests.test_generate_binary_build_matrix --update-reference-files`
pytorch · Nov 27, 2024 · 404c4ea · 404c4ea
1 parent c1862b5
commit 404c4ea
Show file tree

Hide file tree

Showing 11 changed files with 1,844 additions and 13 deletions.
diff --git 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": []
+}
diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda.json b/tools/tests/assets/build_matrix_linux_wheel_cuda.json
diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json b/tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json
diff --git a/tools/tests/assets/build_matrix_linux_wheel_nocpu.json b/tools/tests/assets/build_matrix_linux_wheel_nocpu.json
diff --git a/tools/tests/assets/build_matrix_linux_wheel_xpu.json b/tools/tests/assets/build_matrix_linux_wheel_xpu.json
diff --git a/tools/tests/assets/build_matrix_macos_conda.json b/tools/tests/assets/build_matrix_macos_conda.json
@@ -1 +1,3 @@
-{"include": []}
+{
+  "include": []
+}
diff --git 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
+    }
+  ]
+}
diff --git 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": []
+}