diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index abb95462a8..8cdaa81450 100644 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -33,7 +33,7 @@ ROCM_ARCHES_DICT = { "nightly": ["5.7", "6.0"], "test": ["5.7", "6.0"], - "release": ["5.7", "6.0"], + "release": ["5.6", "5.7"], } PACKAGE_TYPES = ["wheel", "conda", "libtorch"]