add test gpu back

ModelCloud · Jan 8, 2025 · 0794cf3 · 0794cf3
1 parent ae516f4
commit 0794cf3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
@@ -116,10 +116,13 @@ jobs:
           gpu_id=$(curl -s "http://10.0.14.248/gpu/get?id=${{ github.run_id }}&timestamp=$(date +%s%3N)&force=1")
           echo "CUDA_VISIBLE_DEVICES=$gpu_id" >> $GITHUB_ENV
 
-      - name: test gpu
-        if: always()
+      - name: test cpu
         run: python tests/cpu.py
 
+      - name: test gpu
+        if: always() && !cancelled()
+        run: python tests/gpu.py
+
   m4:
     runs-on:  [self-hosted, m4]
     steps: