diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
index 8d8e346db5..abb95462a8 100644
--- 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"]