Searched refs:test_all_reduce (Results 1 – 7 of 7) sorted by relevance
160 // CHECK-LABEL: func @test_all_reduce162 %test_all_reduce {211 // CHECK-LABEL: func @test_all_reduce213 %test_all_reduce {
36 def test_all_reduce(self): member in TestFakePG
113 def test_all_reduce(self, device, dtype): member in TestNCCL
364 def test_all_reduce(self): member in TestMetaCollectives379 def test_all_reduce(self): member in TestGradCollectives
197 def test_all_reduce(self): member in TestCollectivesWithBaseClass
255 def test_all_reduce(self): member in TestCommAnalysis
77 // CHECK-LABEL: func private @test_all_reduce79 %test_all_reduce {