diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index d2d1bd599b..46349829e5 100644 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -25,7 +25,7 @@ } CUDA_ARCHES_DICT = { "nightly": ["11.8", "12.1", "12.4"], - "test": ["11.8", "12.1"], + "test": ["11.8", "12.1", "12.4"], "release": ["11.8", "12.1"], } ROCM_ARCHES_DICT = {