Searched refs:_test_all_to_all_helper (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 3807 def _test_all_to_all_helper( member in DistributedTest._DistTestBase
3934 self._test_all_to_all_helper(group, group_id, rank)
3943 self._test_all_to_all_helper(group, group_id, rank, True, rank_to_GPU)
3950 self._test_all_to_all_helper(group, group_id, rank, dtype=torch.cfloat)
3959 self._test_all_to_all_helper(
4017 self._test_all_to_all_helper(group, group_id, rank)
4027 self._test_all_to_all_helper(group, group_id, rank, True, rank_to_GPU)
4078 self._test_all_to_all_helper(group, group_id, rank)
4087 self._test_all_to_all_helper(group, group_id, rank, True, rank_to_GPU)
|