|
@@ -26,7 +26,7 @@ from .custom_device_whitelist import (
|
|
|
GCU_WHITELIST,
|
|
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):
|
|
def _constr_device(device_type, device_ids):
|