From bc64926943adf7a26ae08bf8698cf529905f57cd Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Tue, 3 Oct 2023 15:04:49 -0400 Subject: [PATCH] Advance binary build matrix for release 2.1 (#4611) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit ### 🤖 Generated by Copilot at 0f3f385 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. --- tools/scripts/generate_binary_build_matrix.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index cc6419a2d0..7545d17c2d 100644 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -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"] @@ -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 @@ -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")