Skip to content

Commit 0f3f385

Browse files
committed
Advance binary build matrix for release 2.1
1 parent 9f460f7 commit 0f3f385

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tools/scripts/generate_binary_build_matrix.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -28,12 +28,12 @@
2828
CUDA_ARCHES_DICT = {
2929
"nightly": ["11.8", "12.1"],
3030
"test": ["11.8", "12.1"],
31-
"release": ["11.7", "11.8"],
31+
"release": ["11.8", "12.1"],
3232
}
3333
ROCM_ARCHES_DICT = {
3434
"nightly": ["5.6", "5.7"],
3535
"test": ["5.5", "5.6"],
36-
"release": ["5.3", "5.4.2"],
36+
"release": ["5.5", "5.6"],
3737
}
3838

3939
CPU_AARCH64_ARCH = ["cpu-aarch64"]
@@ -46,7 +46,7 @@
4646
TEST = "test"
4747

4848
CURRENT_CANDIDATE_VERSION = "2.1.0"
49-
CURRENT_STABLE_VERSION = "2.0.1"
49+
CURRENT_STABLE_VERSION = "2.1.0"
5050
mod.CURRENT_VERSION = CURRENT_STABLE_VERSION
5151

5252
# By default use Nightly for CUDA arches
@@ -259,7 +259,7 @@ def get_wheel_install_command(
259259

260260
index_url_option = "--index-url" if os != "linux-aarch64" else "--extra-index-url"
261261
if channel == RELEASE and (
262-
(gpu_arch_version == "11.7" and os == "linux")
262+
(gpu_arch_version == "12.1" and os == "linux")
263263
or (
264264
gpu_arch_type == "cpu"
265265
and (os == "windows" or os == "macos" or os == "macos-arm64")

0 commit comments

Comments
 (0)