Home
last modified time | relevance | path

Searched refs:init_process_group (Results 1 – 25 of 65) sorted by relevance

123

/external/pytorch/test/distributed/
Dtest_fake_pg.py38 dist.init_process_group(backend="fake", rank=1, world_size=2, store=store)
46 dist.init_process_group(backend="fake", rank=1, world_size=2, store=store)
56 dist.init_process_group(backend="fake", rank=1, world_size=2, store=store)
67 dist.init_process_group(backend="fake", rank=0, world_size=2, store=store)
73 dist.init_process_group(backend="fake", rank=0, world_size=2, store=store)
90 dist.init_process_group(backend="fake", rank=0, world_size=2, store=store)
102 dist.init_process_group(backend="fake", rank=0, world_size=2, store=store)
116 dist.init_process_group(backend="fake", rank=0, world_size=2, store=store)
131 dist.init_process_group(backend="fake", rank=0, world_size=2, store=store)
142 dist.init_process_group(backend="fake", rank=0, world_size=2, store=store)
[all …]
Dtest_c10d_spawn.py128 c10d.init_process_group(
148 c10d.init_process_group(
169 c10d.init_process_group(
188 c10d.init_process_group(
208 c10d.init_process_group(
232 c10d.init_process_group(
Dtest_c10d_common.py82 dist.init_process_group(
1173 dist.init_process_group(
1187 dist.init_process_group(
1199 dist.init_process_group(
1214 dist.init_process_group(
1231 dist.init_process_group(
1260 dist.init_process_group(
1285 dist.init_process_group(
1342 dist.init_process_group(
1364 dist.init_process_group(
[all …]
Dtest_collective_utils.py29 c10d.init_process_group(
82 c10d.init_process_group(
Dtest_c10d_nccl.py121 c10d.init_process_group(backend="nccl", world_size=1)
131 c10d.init_process_group(backend="nccl", rank=0)
139 c10d.init_process_group(backend="nccl", rank=0, world_size=1)
145 c10d.init_process_group(backend="nccl")
215 c10d.init_process_group(
388 c10d.init_process_group(
413 c10d.init_process_group(
554 dist.init_process_group(
581 dist.init_process_group(**base_opts)
587 dist.init_process_group(**base_opts, timeout=new_timeout)
[all …]
Dtest_c10d_spawn_gloo.py49 c10d.init_process_group(
134 c10d.init_process_group(
282 c10d.init_process_group(
313 c10d.init_process_group(
Dtest_c10d_spawn_nccl.py158 c10d.init_process_group(
190 c10d.init_process_group(
219 c10d.init_process_group(
Dtest_c10d_gloo.py197 c10d.init_process_group(backend="gloo", init_method="env://")
592 c10d.init_process_group(
669 dist.init_process_group(
682 dist.init_process_group(
704 dist.init_process_group(
728 dist.init_process_group(
1277 c10d.init_process_group(
1493 c10d.init_process_group(
1502 c10d.init_process_group(
1786 pg = dist.init_process_group(
[all …]
Dtest_c10d_spawn_ucc.py32 c10d.init_process_group(
Dtest_c10d_ucc.py128 c10d.init_process_group(backend="ucc", init_method="env://")
351 c10d.init_process_group(
664 dist.init_process_group(
1020 c10d.init_process_group(
1064 dist.init_process_group(
/external/pytorch/test/distributed/_composable/
Dtest_replicate_with_compiler.py108 dist.init_process_group(
270 def _test_bucketing(self, init_process_group=True, loop=1): argument
271 if init_process_group:
272 dist.init_process_group(
324 code = self._test_bucketing(init_process_group=False, loop=2)
359 code = self._test_bucketing(init_process_group=False, loop=2)
378 dist.init_process_group(
Dtest_replicate.py44 dist.init_process_group(
104 dist.init_process_group(
/external/pytorch/test/distributed/algorithms/quantization/
Dtest_quantization.py84 dist.init_process_group(
100 dist.init_process_group(
118 dist.init_process_group(
143 dist.init_process_group(
167 dist.init_process_group(
191 dist.init_process_group(
/external/pytorch/torch/testing/_internal/distributed/
Dddp_under_dist_autograd_test.py331 dist.init_process_group(
353 dist.init_process_group(
370 dist.init_process_group(
519 dist.init_process_group(
587 dist.init_process_group(
620 dist.init_process_group(
672 dist.init_process_group(
Ddistributed_utils.py35 dist.init_process_group(
/external/pytorch/docs/source/
Ddistributed.rst63 If the `init_method` argument of :func:`init_process_group` points to a file it must adhere
179 The package needs to be initialized using the :func:`torch.distributed.init_process_group`
185 .. autofunction:: init_process_group
219 dist.init_process_group(backend, init_method='tcp://10.1.1.20:23456',
231 :func:`init_process_group` call on the same file path/name.
250 if you plan to call :func:`init_process_group` multiple times on the same file name.
252 :func:`init_process_group` again on that file, failures are expected.
254 empty every time :func:`init_process_group` is called.
261 dist.init_process_group(backend, init_method='file:///mnt/nfs/sharedfile',
284 Once :func:`torch.distributed.init_process_group` was run, the following functions can be used. To
[all …]
/external/pytorch/test/distributed/launcher/bin/
Dtest_script_init_method.py48 dist.init_process_group(
/external/pytorch/torch/testing/_internal/
Dcommon_distributed.py940 c10d.init_process_group(
1080 c10d.init_process_group(
1239 c10d.init_process_group(
1246 c10d.init_process_group("nccl", rank=rank, world_size=world_size)
1282 c10d.init_process_group("nccl", rank=cls.rank, world_size=1)
1373 c10d.init_process_group(
Ddist_utils.py177 dist.init_process_group(
/external/pytorch/torch/distributed/checkpoint/examples/
Dfsdp_checkpoint_example.py65 dist.init_process_group("cpu:gloo,cuda:nccl", rank=rank, world_size=world_size)
Dstateful_example.py78 dist.init_process_group("cpu:gloo,cuda:nccl", rank=rank, world_size=world_size)
/external/pytorch/test/fx/
Dtest_dce_pass.py257 torch.distributed.init_process_group(
287 torch.distributed.init_process_group(
/external/pytorch/torch/testing/_internal/distributed/_shard/sharded_tensor/
D__init__.py26 dist.init_process_group(
/external/pytorch/docs/source/elastic/
Dtrain_script.rst40 torch.distributed.init_process_group(backend=args.backend)
/external/pytorch/benchmarks/dynamo/
Ddist_util.py33 dist.init_process_group("nccl")

123