Split gpu ci workflow into single-gpu and multi-gpu

Author: edknv

.github/workflows/gpu-ci.yml

@@ -12,8 +12,24 @@ on:
 
 jobs:
   build:
-    runs-on: 2GPU
+    runs-on: 1GPU
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+      - name: Run tests
+        run: |
+          ref_type=${{ github.ref_type }}
+          branch=main
+          if [[ $ref_type == "tag"* ]]
+          then
+            raw=$(git branch -r --contains ${{ github.ref_name }})
+            branch=${raw/origin\/}
+          fi
+          cd ${{ github.workspace }}; tox -r -e test-gpu -- $branch
 
+  multigpu:
+    runs-on: 2GPU
     steps:
       - uses: actions/checkout@v3
         with:
@@ -27,4 +43,4 @@ jobs:
             raw=$(git branch -r --contains ${{ github.ref_name }})
             branch=${raw/origin\/}
           fi
-          cd ${{ github.workspace }}; tox -e test-gpu -- $branch
+          cd ${{ github.workspace }}; tox -r -e test-gpu-multigpu -- $branch