Home
last modified time | relevance | path

Searched refs:test_all_to_all_single (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_spawn_ucc.py127 def test_all_to_all_single(self): member in TestDistributedNNFunctionsUcc
H A Dtest_functional_api.py516 def test_all_to_all_single(self): member in TestCollectivesWithNCCL
664 def test_all_to_all_single(self, compile: bool = True) -> None: member in TestFunctionalAutograd
825 def test_all_to_all_single(self) -> None: member in TestFunctionalAutogradWithNCCL
H A Dtest_c10d_spawn_nccl.py145 def test_all_to_all_single(self): member in TestDistributedNNFunctionsNccl
H A Dtest_c10d_spawn_gloo.py269 def test_all_to_all_single(self): member in TestDistributedNNFunctionsGloo
H A Dtest_c10d_functional_native.py348 def test_all_to_all_single(self) -> None: member in TestWithNCCL
H A Dtest_c10d_gloo.py2437 def test_all_to_all_single(self): member in GlooProcessGroupWithDispatchedCollectivesTests
H A Dtest_c10d_nccl.py3129 def test_all_to_all_single(self): member in NcclProcessGroupWithDispatchedCollectivesTests