Skip to content

Commit

Permalink
Advance binary build matrix for release 2.1 (#4611)
Browse files Browse the repository at this point in the history
<!--
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.
  • Loading branch information
atalman authored Oct 3, 2023
1 parent e6db930 commit bc64926
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 bc64926

Please sign in to comment.