Searched defs:init_pg (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_spawn.py | 38 def _test_multiprocess(self, f, shared_tensors, init_pg, n_output): argument
72 cls, rank, filename, shared_tensors, world_size, init_pg, c2p, p2c argument
82 cls, rank, filename, shared_tensors, world_size, init_pg, c2p, p2c argument
92 cls, rank, filename, shared_tensors, world_size, init_pg, c2p, p2c argument
|
| H A D | test_c10d_spawn_nccl.py | 59 cls, rank, filename, shared_tensors, world_size, init_pg, c2p, p2c argument
|
| H A D | test_c10d_spawn_gloo.py | 97 cls, rank, filename, shared_tensor, world_size, init_pg, c2p, p2c argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_shard/sharded_tensor/ |
| H A D | __init__.py | 22 def init_pg(self, backend="nccl"): member in ShardedTensorTestBase
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_tensor/ |
| H A D | common_dtensor.py | 310 def init_pg(self) -> None: member in DTensorTestBase
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_distributed.py | 1229 def _dynamo_dist_per_rank_init(rank, world_size, init_pg=True, fake_pg=False): argument
|