Searched refs:test_reduce_scatter (Results 1 – 6 of 6) sorted by relevance
423 test_reduce_scatter = partial(429 with patch_reduce_scatter(test_reduce_scatter, full_precision_param_dtype):592 test_reduce_scatter = partial(598 with patch_reduce_scatter(test_reduce_scatter, param_dtype):974 test_reduce_scatter = partial(981 with patch_reduce_scatter(test_reduce_scatter, torch.float32):
154 def test_reduce_scatter(self): member in TestDistributedNNFunctionsNccl
54 def test_reduce_scatter(self): member in TestFakePG
207 def test_reduce_scatter(self, device, dtype): member in TestNCCL
176 def test_reduce_scatter(self): member in TestCollectivesWithBaseClass
952 // CHECK-LABEL: func private @test_reduce_scatter954 %test_reduce_scatter {