add cuda12.1 build check ci (#782)

* update cuda12.1 build check ci * use matrix
InternLM · Dec 4, 2023 · 7990d25 · 7990d25
1 parent 7f943a2
commit 7990d25
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/.github/workflows/linux-x64-gpu.yml b/.github/workflows/linux-x64-gpu.yml
@@ -25,7 +25,11 @@ permissions:
   contents: read
 
 jobs:
-  cuda-118:
+  build:
+    strategy:
+      matrix:
+        cudaver: [11.8, 12.1]
+    name: cuda-${{ matrix.cudaver }}
     runs-on: ubuntu-latest
     steps:
       - name: Free disk space
@@ -45,8 +49,8 @@ jobs:
       - name: Build
         uses: addnab/docker-run-action@v3
         with:
-          image: openmmlab/lmdeploy-builder:cuda11.8
-          options: -v ${{ github.workspace }}:/work --cpus=1.8
+          image: openmmlab/lmdeploy-builder:cuda${{ matrix.cudaver }}
+          options: -v ${{ github.workspace }}:/work
           run: |
             cd /work
             source /opt/conda/bin/activate

diff --git a/.github/workflows/windows-x64-gpu.yml b/.github/workflows/windows-x64-gpu.yml
@@ -25,7 +25,11 @@ permissions:
   contents: read
 
 jobs:
-  cuda-118:
+  build:
+    strategy:
+      matrix:
+        cudaver: [11.8.0, 12.1.0]
+    name: cuda-${{ matrix.cudaver }}
     runs-on: windows-latest
     steps:
       - name: Checkout repository
@@ -40,7 +44,7 @@ jobs:
       - uses: Jimver/[email protected]
         id: cuda-toolkit
         with:
-          cuda: '11.8.0'
+          cuda: ${{ matrix.cudaver }}
           use-github-cache: false
       - name: Build wheel
         run: |