Searched refs:ProcessGroupShareTensorTest (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/test/distributed/ |
H A D | test_c10d_spawn_ucc.py | 26 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 D | test_c10d_spawn_nccl.py | 25 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 D | test_c10d_spawn_gloo.py | 31 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,
|