Home
last modified time | relevance | path

Searched refs:_test_owner_rref_forward_synchronization (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Drpc_test.py6163 def _test_owner_rref_forward_synchronization(self, local_device, remote_device): member in TensorPipeAgentCudaRpcTest
6190 self._test_owner_rref_forward_synchronization("cuda:0", "cuda:0")
6194 self._test_owner_rref_forward_synchronization("cuda:0", "cuda:1")
6198 self._test_owner_rref_forward_synchronization("cuda:1", "cuda:0")
6202 self._test_owner_rref_forward_synchronization("cuda:1", "cuda:1")