[CI] update device type

ModelCloud · Jan 8, 2025 · d8533e0 · d8533e0
1 parent 52f92b1
commit d8533e0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/device_smi/amd.py b/device_smi/amd.py
@@ -10,7 +10,7 @@ class AMDGPUMetrics(BaseMetrics):
 
 class AMDDevice(GPUDevice):
     def __init__(self, cls, index):
-        super().__init__(cls, index)
+        super().__init__(cls, index, "xpu")
         self.gpu_id = self._get_gpu_id()
 
         try:

diff --git a/device_smi/base.py b/device_smi/base.py
@@ -20,8 +20,8 @@ def to_dict(self, text, split: str = ":"):
 
 
 class GPUDevice(BaseDevice):
-    def __init__(self, cls, index):
-        super().__init__(cls, "gpu")
+    def __init__(self, cls, index, type= "gpu"):
+        super().__init__(cls, type)
         self.index = index
 
     @abstractmethod