[CI] test xpu device

ModelCloud · Jan 8, 2025 · f03ebee · f03ebee
1 parent 5e4e790
commit f03ebee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
@@ -166,7 +166,7 @@ jobs:
 
       - name: test gpu
         if: always() && !cancelled()
-        run: python tests/gpu.py
+        run: python tests/xpu.py
 
   m4:
     runs-on:  [self-hosted, m4]

diff --git a/tests/gpu.py b/tests/gpu.py
@@ -1,4 +1,5 @@
 from device_smi import Device
+import torch
 
 dev = Device("gpu")
 print(dev)