Browse Source

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

GitOrigin-RevId: a18ee052e9
tags/v1.6.0-rc1
Megvii Engine Team 3 years ago
parent
commit
5fe789abb9
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      imperative/python/megengine/distributed/group.py

+ 1
- 1
imperative/python/megengine/distributed/group.py View File

@@ -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


Loading…
Cancel
Save