OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

fix(mge/distributed): fix single_machine_cache error 

GitOrigin-RevId: be57111408
release-1.7
Megvii Engine Team 3 years ago
parent
0a56a5b110
commit
05ee603843
1 changed files with 1 additions and 0 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  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): def reset(self, proc_ranks):
self.check(proc_ranks) self.check(proc_ranks)
self.proc_ranks = proc_ranks self.proc_ranks = proc_ranks
self.is_single_machine_cache = None
self.stream = _sh.get_next() self.stream = _sh.get_next()


def check(self, proc_ranks): def check(self, proc_ranks):


Write Preview
Loading…
Cancel
Save