Searched refs:test_all_to_all_single (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/pytorch/test/distributed/ | ||
H A D | test_c10d_spawn_ucc.py | 127 def test_all_to_all_single(self): member in TestDistributedNNFunctionsUcc |
H A D | test_functional_api.py | 516 def test_all_to_all_single(self): member in TestCollectivesWithNCCL |
H A D | test_c10d_spawn_nccl.py | 145 def test_all_to_all_single(self): member in TestDistributedNNFunctionsNccl |
H A D | test_c10d_spawn_gloo.py | 269 def test_all_to_all_single(self): member in TestDistributedNNFunctionsGloo |
H A D | test_c10d_functional_native.py | 348 def test_all_to_all_single(self) -> None: member in TestWithNCCL |
H A D | test_c10d_gloo.py | 2437 def test_all_to_all_single(self): member in GlooProcessGroupWithDispatchedCollectivesTests |
H A D | test_c10d_nccl.py | 3129 def test_all_to_all_single(self): member in NcclProcessGroupWithDispatchedCollectivesTests |