Searched refs:_test_rref_synchronization (Results 1 – 1 of 1) sorted by relevance
5984 def _test_rref_synchronization(self, local_device, remote_device): member in TensorPipeAgentCudaRpcTest6014 self._test_rref_synchronization("cuda:0", "cuda:0")6018 self._test_rref_synchronization("cuda:1", "cuda:0")6022 self._test_rref_synchronization("cuda:1", "cuda:1")6026 self._test_rref_synchronization("cuda:0", "cuda:1")