Searched refs:_test_rref_forward_synchronization (Results 1 – 1 of 1) sorted by relevance
6095 def _test_rref_forward_synchronization(self, local_device, remote_device): member in TensorPipeAgentCudaRpcTest6149 self._test_rref_forward_synchronization("cuda:0", "cuda:0")6153 self._test_rref_forward_synchronization("cuda:0", "cuda:1")6157 self._test_rref_forward_synchronization("cuda:1", "cuda:0")6161 self._test_rref_forward_synchronization("cuda:1", "cuda:1")