advance rocm to 5.7 and 6.0 in nightly (#4939)

pytorch · Feb 9, 2024 · d90dee0 · d90dee0
1 parent 41c1892
commit d90dee0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -33,7 +33,7 @@
     "release": ["11.8", "12.1"],
 }
 ROCM_ARCHES_DICT = {
-    "nightly": ["5.6", "5.7"],
+    "nightly": ["5.7", "6.0"],
     "test": ["5.6", "5.7"],
     "release": ["5.6", "5.7"],
 }