From 0f3f385d85336827019f6a6428867fa63d6e4161 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 3 Oct 2023 11:16:51 -0700 Subject: [PATCH] Advance binary build matrix for release 2.1 --- tools/scripts/generate_binary_build_matrix.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index cc6419a2d0..7545d17c2d 100644 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -28,12 +28,12 @@ CUDA_ARCHES_DICT = { "nightly": ["11.8", "12.1"], "test": ["11.8", "12.1"], - "release": ["11.7", "11.8"], + "release": ["11.8", "12.1"], } ROCM_ARCHES_DICT = { "nightly": ["5.6", "5.7"], "test": ["5.5", "5.6"], - "release": ["5.3", "5.4.2"], + "release": ["5.5", "5.6"], } CPU_AARCH64_ARCH = ["cpu-aarch64"] @@ -46,7 +46,7 @@ TEST = "test" CURRENT_CANDIDATE_VERSION = "2.1.0" -CURRENT_STABLE_VERSION = "2.0.1" +CURRENT_STABLE_VERSION = "2.1.0" mod.CURRENT_VERSION = CURRENT_STABLE_VERSION # By default use Nightly for CUDA arches @@ -259,7 +259,7 @@ def get_wheel_install_command( index_url_option = "--index-url" if os != "linux-aarch64" else "--extra-index-url" if channel == RELEASE and ( - (gpu_arch_version == "11.7" and os == "linux") + (gpu_arch_version == "12.1" and os == "linux") or ( gpu_arch_type == "cpu" and (os == "windows" or os == "macos" or os == "macos-arm64")