gaotingquan 9 月之前
父節點
當前提交
64712bdb93
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/utils/device.py

+ 1 - 1
paddlex/utils/device.py

@@ -26,7 +26,7 @@ from .custom_device_whitelist import (
     GCU_WHITELIST,
 )
 
-SUPPORTED_DEVICE_TYPE = ["cpu", "gpu", "xpu", "npu", "mlu", "gcu"]
+SUPPORTED_DEVICE_TYPE = ["cpu", "gpu", "xpu", "npu", "mlu", "gcu", "dcu"]
 
 
 def _constr_device(device_type, device_ids):