gaotingquan hai 9 meses
pai
achega
64712bdb93
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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):