Searched defs:test_reduce_scatter (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ | ||
| H A D | test_nccl.py | 207 def test_reduce_scatter(self, device, dtype): member in TestNCCL |
| H A D | test_fake_pg.py | 54 def test_reduce_scatter(self): member in TestFakePG |
| H A D | test_c10d_spawn_nccl.py | 154 def test_reduce_scatter(self): member in TestDistributedNNFunctionsNccl |
| H A D | test_multi_threaded_pg.py | 176 def test_reduce_scatter(self): member in TestCollectivesWithBaseClass |