提交 3d0c0eb5 编写于 作者: B baocheny 提交者: cuicheng01

add 2 more custom devices intel_gpu and apple mps

上级 62c56ce2
...@@ -90,8 +90,9 @@ def main(args): ...@@ -90,8 +90,9 @@ def main(args):
fleet.init(is_collective=True) fleet.init(is_collective=True)
# assign the device # assign the device
assert global_config[ assert global_config["device"] in [
"device"] in ["cpu", "gpu", "xpu", "npu", "mlu", "ascend"] "cpu", "gpu", "xpu", "npu", "mlu", "ascend", "intel_gpu", "mps"
]
device = paddle.set_device(global_config["device"]) device = paddle.set_device(global_config["device"])
# amp related config # amp related config
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册