Browse Source

fix(mge/distributed): fix single_machine_cache error

GitOrigin-RevId: be57111408
release-1.7
Megvii Engine Team 3 years ago
parent
commit
05ee603843
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      imperative/python/megengine/distributed/group.py

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

@@ -56,6 +56,7 @@ class Group:
def reset(self, proc_ranks):
self.check(proc_ranks)
self.proc_ranks = proc_ranks
self.is_single_machine_cache = None
self.stream = _sh.get_next()

def check(self, proc_ranks):


Loading…
Cancel
Save