gaotingquan 9 months ago
parent
commit
64712bdb93
1 changed files with 1 additions and 1 deletions
  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):