From 5fe789abb9e16212c5ae2619d2c0cded81781d8d Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 3 Aug 2021 14:52:31 +0800 Subject: [PATCH] fix(mge/distributed): skip check when machine_ranks is None GitOrigin-RevId: a18ee052e9dc23a689a3396b2646a0a5365bbd3a --- imperative/python/megengine/distributed/group.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imperative/python/megengine/distributed/group.py b/imperative/python/megengine/distributed/group.py index 8fc0634d1..630175079 100644 --- a/imperative/python/megengine/distributed/group.py +++ b/imperative/python/megengine/distributed/group.py @@ -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 -- GitLab