Update generate_build_matrix.py (#1309)

k2-fsa · Nov 22, 2024 · f5a59be · f5a59be
1 parent e7390b2
commit f5a59be
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/scripts/github_actions/generate_build_matrix.py b/scripts/github_actions/generate_build_matrix.py
@@ -281,13 +281,25 @@ def generate_build_matrix(
                 else ["11.8.0", "12.1.0", "12.4.0"]
             ),
         },
+        "2.5.1": {
+            # Only Linux supports python 3.13
+            "python-version": ["3.9", "3.10", "3.11", "3.12", "3.13"],
+            "cuda": (
+                ["11.8", "12.1", "12.4"]  # default 12.4
+                if not for_windows
+                else ["11.8.0", "12.1.0", "12.4.0"]
+            ),
+        },
         # https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
     }
     if test_only_latest_torch:
-        latest = "2.5.0"
+        latest = "2.5.1"
         matrix = {latest: matrix[latest]}
 
     if for_windows or for_macos:
+        if "2.5.1" in matrix:
+            matrix["2.5.1"]["python-version"].remove("3.13")
+
         if "2.5.0" in matrix:
             matrix["2.5.0"]["python-version"].remove("3.13")