From 1ca62cf399d0693e5997deda107f32368b770eb7 Mon Sep 17 00:00:00 2001 From: Benson Ma Date: Wed, 12 Jun 2024 11:57:05 -0700 Subject: [PATCH 1/2] [cuda] Add support for CUDA 12.4 in `test` channel - Add support for CUDA 12.4 in `test` channel --- tools/scripts/generate_binary_build_matrix.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index d2d1bd599b..31e17f400e 100644 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -25,8 +25,8 @@ } CUDA_ARCHES_DICT = { "nightly": ["11.8", "12.1", "12.4"], - "test": ["11.8", "12.1"], - "release": ["11.8", "12.1"], + "test": ["11.8", "12.1", "12.4"], + "release": ["11.8", "12.1", "12.4"], } ROCM_ARCHES_DICT = { "nightly": ["6.0", "6.1"], From 1401743a9858f3f6e24a819c3b52ce19f61ad9cc Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Wed, 12 Jun 2024 16:53:52 -0400 Subject: [PATCH 2/2] Update tools/scripts/generate_binary_build_matrix.py --- tools/scripts/generate_binary_build_matrix.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index 31e17f400e..46349829e5 100644 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -26,7 +26,7 @@ CUDA_ARCHES_DICT = { "nightly": ["11.8", "12.1", "12.4"], "test": ["11.8", "12.1", "12.4"], - "release": ["11.8", "12.1", "12.4"], + "release": ["11.8", "12.1"], } ROCM_ARCHES_DICT = { "nightly": ["6.0", "6.1"],