From d81a96f14feec1148b39be6d42043db198072fc5 Mon Sep 17 00:00:00 2001
From: Andrey Talman <atalman@fb.com>
Date: Tue, 17 Dec 2024 15:21:56 +0000
Subject: [PATCH] Fix test after enabling rocm 3.13 builds (#6079)

Fix tests after https://github.com/pytorch/test-infra/pull/6070
---
 .../assets/build_matrix_linux_wheel_cuda.json | 30 +++++++++++++++++++
 .../build_matrix_linux_wheel_nocpu.json       | 30 +++++++++++++++++++
 2 files changed, 60 insertions(+)

diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda.json b/tools/tests/assets/build_matrix_linux_wheel_cuda.json
index 62c34c3eaf..719bdbbfb7 100644
--- 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
     }
   ]
 }
\ 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 dbf87f230d..536c79eff5 100644
--- 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
     }
   ]
 }
\ No newline at end of file