From f03ebeec55a7485297ba664bdb3469f216aa8673 Mon Sep 17 00:00:00 2001
From: CSY <csy@modelcloud.ai>
Date: Wed, 8 Jan 2025 08:13:56 +0000
Subject: [PATCH] [CI] test xpu device

---
 .github/workflows/unit_tests.yml | 2 +-
 tests/gpu.py                     | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
index a5cab2a..19e3ac5 100644
--- 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
index cf7884b..8a72e6c 100644
--- a/tests/gpu.py
+++ b/tests/gpu.py
@@ -1,4 +1,5 @@
 from device_smi import Device
+import torch
 
 dev = Device("gpu")
 print(dev)