提交 c032293a 编写于 作者: U USTCKAY 提交者: Wei Shengyu

change judgment logic for multi device

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