support torch 2.2.1

csukuangfj · Feb 23, 2024 · 18b2377 · 18b2377
1 parent 2a8a993
commit 18b2377
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/scripts/github_actions/generate_build_matrix.py b/scripts/github_actions/generate_build_matrix.py
@@ -183,10 +183,16 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_
             if not for_windows
             else ["11.8.0", "12.1.0"],
         },
+        "2.2.1": {
+            "python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
+            "cuda": ["11.8", "12.1"]  # default 12.1
+            if not for_windows
+            else ["11.8.0", "12.1.0"],
+        },
         # https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
     }
     if test_only_latest_torch:
-        latest = "2.2.0"
+        latest = "2.2.1"
         matrix = {latest: matrix[latest]}
 
     if for_windows or for_macos: