|
28 | 28 | CUDA_ARCHES_DICT = {
|
29 | 29 | "nightly": ["11.8", "12.1"],
|
30 | 30 | "test": ["11.8", "12.1"],
|
31 |
| - "release": ["11.7", "11.8"], |
| 31 | + "release": ["11.8", "12.1"], |
32 | 32 | }
|
33 | 33 | ROCM_ARCHES_DICT = {
|
34 | 34 | "nightly": ["5.6", "5.7"],
|
35 | 35 | "test": ["5.5", "5.6"],
|
36 |
| - "release": ["5.3", "5.4.2"], |
| 36 | + "release": ["5.5", "5.6"], |
37 | 37 | }
|
38 | 38 |
|
39 | 39 | CPU_AARCH64_ARCH = ["cpu-aarch64"]
|
|
46 | 46 | TEST = "test"
|
47 | 47 |
|
48 | 48 | CURRENT_CANDIDATE_VERSION = "2.1.0"
|
49 |
| -CURRENT_STABLE_VERSION = "2.0.1" |
| 49 | +CURRENT_STABLE_VERSION = "2.1.0" |
50 | 50 | mod.CURRENT_VERSION = CURRENT_STABLE_VERSION
|
51 | 51 |
|
52 | 52 | # By default use Nightly for CUDA arches
|
@@ -259,7 +259,7 @@ def get_wheel_install_command(
|
259 | 259 |
|
260 | 260 | index_url_option = "--index-url" if os != "linux-aarch64" else "--extra-index-url"
|
261 | 261 | if channel == RELEASE and (
|
262 |
| - (gpu_arch_version == "11.7" and os == "linux") |
| 262 | + (gpu_arch_version == "12.1" and os == "linux") |
263 | 263 | or (
|
264 | 264 | gpu_arch_type == "cpu"
|
265 | 265 | and (os == "windows" or os == "macos" or os == "macos-arm64")
|
|
0 commit comments