MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / reset

Method reset

imperative/python/megengine/distributed/group.py:49–53  ·  view source on GitHub ↗
(self, proc_ranks)

Source from the content-addressed store, hash-verified

47 self.reset(proc_ranks)
48
49 def reset(self, proc_ranks):
50 self.check(proc_ranks)
51 self.proc_ranks = proc_ranks
52 self.is_single_machine_cache = None
53 self.stream = _sh.get_next()
54
55 def check(self, proc_ranks):
56 assert _sd is not None, "please call init_process_group first"

Callers 2

__init__Method · 0.95
init_process_groupFunction · 0.45

Calls 2

checkMethod · 0.95
get_nextMethod · 0.80

Tested by

no test coverage detected