提交 75a5bb17 编写于 作者: B baocheny 提交者: cuicheng01

add 2 more custom devices intel_gpu and apple mps

上级 3d0c0eb5
......@@ -101,7 +101,7 @@ class Engine(object):
# set device
assert self.config["Global"][
"device"] in ["cpu", "gpu", "xpu", "npu", "mlu", "ascend"]
"device"] in ["cpu", "gpu", "xpu", "npu", "mlu", "ascend", "intel_gpu", "mps"]
self.device = paddle.set_device(self.config["Global"]["device"])
logger.info('train with paddle {} and device {}'.format(
paddle.__version__, self.device))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册