Skip to content

Commit bc64926

Browse files
authored
Advance binary build matrix for release 2.1 (#4611)
<!-- copilot:summary --> ### <samp>🤖 Generated by Copilot at 0f3f385</samp> Update the release channel of the binary build matrix to use PyTorch 2.1.0 and newer GPU architectures. This allows testing and releasing binary builds for the latest PyTorch features and hardware support.
1 parent e6db930 commit bc64926

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tools/scripts/generate_binary_build_matrix.py

Lines changed: 4 additions & 4 deletions
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)