Searched refs:_test_allreduce_basics (Results 1 – 2 of 2) sorted by relevance
207 def _test_allreduce_basics(self, fn): member in ProcessGroupUCCTest235 self._test_allreduce_basics(lambda t: t.clone())
419 def _test_allreduce_basics(self, fn): member in ProcessGroupGlooTest460 self._test_allreduce_basics(lambda t: t.clone())465 self._test_allreduce_basics(lambda t: t.clone().cuda())