MCPcopy Create free account
hub / github.com/pytorch/pytorch / dist_init

Method dist_init

test/distributed/algorithms/test_join.py:170–177  ·  view source on GitHub ↗
(self, rank, world_size, backend=BACKEND)

Source from the content-addressed store, hash-verified

168 pass
169
170 def dist_init(self, rank, world_size, backend=BACKEND):
171 store = dist.FileStore(self.file_name, world_size)
172 return dist.init_process_group(
173 backend=backend,
174 store=store,
175 rank=rank,
176 world_size=world_size
177 )
178
179 def construct_uneven_inputs(self, base, offset, device=None):
180 r"""

Callers 1

_test_join_baseMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected