Upgrade nightly wheels to ROCm6.1 (#5154)

pytorch · May 2, 2024 · fa9ae96 · fa9ae96
1 parent 93379c5
commit fa9ae96
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -31,7 +31,7 @@
     "release": ["11.8", "12.1"],
 }
 ROCM_ARCHES_DICT = {
-    "nightly": ["5.7", "6.0"],
+    "nightly": ["6.0", "6.1"],
     "test": ["5.7", "6.0"],
     "release": ["5.7", "6.0"],
 }