Update generate_binary_build_matrix.py with latest versions

pytorch · Jun 13, 2024 · 52ceb8b · 52ceb8b
1 parent 00e3423
commit 52ceb8b
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
@@ -25,12 +25,12 @@
 }
 CUDA_ARCHES_DICT = {
     "nightly": ["11.8", "12.1", "12.4"],
-    "test": ["11.8", "12.1"],
+    "test": ["11.8", "12.1", "12.4"],
     "release": ["11.8", "12.1"],
 }
 ROCM_ARCHES_DICT = {
     "nightly": ["6.0", "6.1"],
-    "test": ["5.7", "6.0"],
+    "test": ["6.0", "6.1"],
     "release": ["5.7", "6.0"],
 }
 
@@ -63,7 +63,7 @@
 
 
 CURRENT_NIGHTLY_VERSION = "2.4.0"
-CURRENT_CANDIDATE_VERSION = "2.3.1"
+CURRENT_CANDIDATE_VERSION = "2.4.0"
 CURRENT_STABLE_VERSION = "2.3.1"
 CURRENT_VERSION = CURRENT_STABLE_VERSION