Fix test after enabling rocm 3.13 builds (#6079)

Fix tests after #6070
pytorch · Dec 17, 2024 · d81a96f · d81a96f
1 parent 863dd13
commit d81a96f
Show file tree

Hide file tree

Showing 2 changed files with 60 additions and 0 deletions.
diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda.json b/tools/tests/assets/build_matrix_linux_wheel_cuda.json
@@ -419,6 +419,36 @@
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
       "use_split_build": false
+    },
+    {
+      "python_version": "3.13",
+      "gpu_arch_type": "rocm",
+      "gpu_arch_version": "6.1",
+      "desired_cuda": "rocm6.1",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "package_type": "manywheel",
+      "build_name": "manywheel-py3_13-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.13",
+      "gpu_arch_type": "rocm",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "package_type": "manywheel",
+      "build_name": "manywheel-py3_13-rocm6_2_4",
+      "validation_runner": "linux.2xlarge",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "channel": "nightly",
+      "upload_to_base_bucket": "no",
+      "stable_version": "2.5.1",
+      "use_split_build": false
     }
   ]
 }
diff --git a/tools/tests/assets/build_matrix_linux_wheel_nocpu.json b/tools/tests/assets/build_matrix_linux_wheel_nocpu.json
@@ -344,6 +344,36 @@
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
       "use_split_build": false
+    },
+    {
+      "python_version": "3.13",
+      "gpu_arch_type": "rocm",
+      "gpu_arch_version": "6.1",
+      "desired_cuda": "rocm6.1",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "package_type": "manywheel",
+      "build_name": "manywheel-py3_13-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.13",
+      "gpu_arch_type": "rocm",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "package_type": "manywheel",
+      "build_name": "manywheel-py3_13-rocm6_2_4",
+      "validation_runner": "linux.2xlarge",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "channel": "nightly",
+      "upload_to_base_bucket": "no",
+      "stable_version": "2.5.1",
+      "use_split_build": false
     }
   ]
 }