Update generate_binary_build_matrix.py for 2.5.0

pytorch · Oct 16, 2024 · 9504bf3 · 9504bf3
1 parent e76c422
commit 9504bf3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -22,7 +22,7 @@
 PYTHON_ARCHES_DICT = {
     "nightly": ["3.9", "3.10", "3.11", "3.12"],
     "test": ["3.9", "3.10", "3.11", "3.12"],
-    "release": ["3.8", "3.9", "3.10", "3.11", "3.12"],
+    "release": ["3.9", "3.10", "3.11", "3.12"],
 }
 CUDA_ARCHES_DICT = {
     "nightly": ["11.8", "12.1", "12.4"],
@@ -32,7 +32,7 @@
 ROCM_ARCHES_DICT = {
     "nightly": ["6.1", "6.2"],
     "test": ["6.1", "6.2"],
-    "release": ["6.0", "6.1"],
+    "release": ["6.1", "6.2"],
 }
 
 CUDA_CUDDN_VERSIONS = {
@@ -67,7 +67,7 @@
 
 CURRENT_NIGHTLY_VERSION = "2.6.0"
 CURRENT_CANDIDATE_VERSION = "2.5.0"
-CURRENT_STABLE_VERSION = "2.4.1"
+CURRENT_STABLE_VERSION = "2.5.0"
 CURRENT_VERSION = CURRENT_STABLE_VERSION
 
 # By default use Nightly for CUDA arches