提交 5fe789ab 编写于 作者: M Megvii Engine Team

fix(mge/distributed): skip check when machine_ranks is None

GitOrigin-RevId: a18ee052e9dc23a689a3396b2646a0a5365bbd3a
上级 298c4443
......@@ -94,7 +94,7 @@ class Group:
return self.is_single_machine_cache
assert _sd is not None, "please call init_process_group first"
for rank in self.proc_ranks:
if rank not in _sd.machine_ranks:
if _sd.machine_ranks is None or rank not in _sd.machine_ranks:
self.is_single_machine_cache = False
return False
self.is_single_machine_cache = True
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册