提交 8fb6cc53 编写于 作者: G gaotingquan 提交者: Wei Shengyu

fix: debug

上级 f3aaa735
......@@ -93,8 +93,8 @@ def main(args):
use_npu = global_config.get("use_npu", False)
use_mlu = global_config.get("use_mlu", False)
assert (
use_gpu and use_xpu and use_npu and use_mlu
) is not True, "gpu, xpu, npu and mlu can not be true in the same time in static mode!"
use_gpu + use_xpu + use_npu + use_mlu + use_ascend <= 1
), "gpu, xpu, npu, mlu and ascend can not be true in the same time in static mode!"
if use_gpu:
device = paddle.set_device('gpu')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册