fix: add cuda suffix for win build, add cuda 12 support (#1019)

* fix: add cuda suffix for win build, add cuda 12 support * resolve comment * fix ci
TabbyML · Dec 14, 2023 · cf89342 · cf89342
1 parent a90db70
commit cf89342
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -26,7 +26,8 @@ jobs:
     container: ${{ matrix.container }}
     strategy:
       matrix:
-        binary: [aarch64-apple-darwin, x86_64-manylinux2014, x86_64-manylinux2014-cuda117, x86_64-pc-windows-msvc]
+        binary: [aarch64-apple-darwin, x86_64-manylinux2014, x86_64-manylinux2014-cuda117,
+                 x86_64-windows-msvc-cuda117, x86_64-windows-msvc-cuda122]
         include:
           - os: macos-latest
             target: aarch64-apple-darwin
@@ -42,9 +43,16 @@ jobs:
             build_args: --features cuda
           - os: windows-latest
             target: x86_64-pc-windows-msvc
-            binary: x86_64-pc-windows-msvc
+            binary: x86_64-windows-msvc-cuda117
             ext: .exe
             build_args: --features cuda
+            windows_cuda: '11.7.1'
+          - os: windows-latest
+            target: x86_64-pc-windows-msvc
+            binary: x86_64-windows-msvc-cuda122
+            ext: .exe
+            build_args: --features cuda
+            windows_cuda: '12.2.0'
 
     env:
       SCCACHE_GHA_ENABLED: true
@@ -93,7 +101,7 @@ jobs:
       - name: Install CUDA toolkit for Windows
         uses: Jimver/cuda-toolkit@v0.2.11
         with:
-          cuda: '11.7.1'
+          cuda: ${{ matrix.windows_cuda }}
           method: 'network'
           sub-packages: '["nvcc", "cudart", "cublas", "cublas_dev", "thrust", "visual_studio_integration"]'
         if: runner.os == 'Windows'