Skip to content

Commit

Permalink
Advance binary build matrix for release 2.1
Browse files Browse the repository at this point in the history
  • Loading branch information
atalman committed Oct 3, 2023
1 parent 9f460f7 commit 0f3f385
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tools/scripts/generate_binary_build_matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"]
Expand All @@ -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
Expand Down Expand Up @@ -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")
Expand Down

0 comments on commit 0f3f385

Please sign in to comment.