Remove ROCm 5.6 and add ROCm 6.0

pytorch · Mar 15, 2024 · e3d7561 · e3d7561
1 parent f61febd
commit e3d7561
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -32,8 +32,8 @@
 }
 ROCM_ARCHES_DICT = {
     "nightly": ["5.7", "6.0"],
-    "test": ["5.6", "5.7"],
-    "release": ["5.6", "5.7"],
+    "test": ["5.7", "6.0"],
+    "release": ["5.7", "6.0"],
 }
 
 PACKAGE_TYPES = ["wheel", "conda", "libtorch"]