Home
last modified time | relevance | path

Searched refs:ProcessGroupShareTensorTest (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_spawn_ucc.py26 class ProcessGroupShareTensorTest( class
43 ProcessGroupShareTensorTest._test_broadcast_process,
45 ProcessGroupShareTensorTest._init_pg_ucc,
55 ProcessGroupShareTensorTest._test_allreduce_process,
57 ProcessGroupShareTensorTest._init_pg_ucc,
67 ProcessGroupShareTensorTest._test_allgather_process,
69 ProcessGroupShareTensorTest._init_pg_ucc,
H A Dtest_c10d_spawn_nccl.py25 class ProcessGroupShareTensorTest( class
39 ProcessGroupShareTensorTest._test_broadcast_process,
41 ProcessGroupShareTensorTest._init_pg_nccl,
51 ProcessGroupShareTensorTest._test_allreduce_process,
53 ProcessGroupShareTensorTest._init_pg_nccl,
76 ProcessGroupShareTensorTest._test_reduce_process,
78 ProcessGroupShareTensorTest._init_pg_nccl,
88 ProcessGroupShareTensorTest._test_allgather_process,
90 ProcessGroupShareTensorTest._init_pg_nccl,
H A Dtest_c10d_spawn_gloo.py31 class ProcessGroupShareTensorTest( class
46 store, rank, world_size, ProcessGroupShareTensorTest.opts()
67 ProcessGroupShareTensorTest._test_broadcast_process,
69 ProcessGroupShareTensorTest._init_pg_gloo,
78 ProcessGroupShareTensorTest._test_allreduce_process,
80 ProcessGroupShareTensorTest._init_pg_gloo,
89 ProcessGroupShareTensorTest._test_allgather_process,
91 ProcessGroupShareTensorTest._init_pg_gloo,
113 ProcessGroupShareTensorTest._test_allgather_chunk_process,
115 ProcessGroupShareTensorTest._init_pg_gloo,