Home
last modified time | relevance | path

Searched refs:rpc_async (Results 1 – 25 of 31) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/jit/
H A Drpc_test.py157 fut1 = rpc.rpc_async(dst_worker_name, script_add_ones, (t, ))
161 fut2 = rpc.rpc_async(dst_worker_name, script_add_ones, (t, ))
203 fut = rpc.rpc_async(dst_worker_name, one_arg, (torch.tensor(1),))
210 …fut = rpc.rpc_async(dst_worker_name, script_add_ones_with_record_function, (torch.tensor(1), block…
301 ret_fut = rpc.rpc_async(worker_name(dst_rank), two_args_two_kwargs, args=inputs)
314 return rpc.rpc_async(
333 fut = rpc.rpc_async(dst_worker_name, torch.add, (input_0, input_1), {})
439 fut = rpc.rpc_async(
469 fut = rpc.rpc_async(
523 fut = rpc.rpc_async(dst_worker_name, two_args_two_kwargs, args, kwargs)
[all …]
H A Drpc_test_faulty.py33 fut = rpc.rpc_async(dst_worker_name, two_args_two_kwargs, args, kwargs)
45 fut = rpc.rpc_async(dst_worker_name, two_args_two_kwargs, args, kwargs, timeout)
57 fut = rpc.rpc_async(dst_worker_name, two_args_two_kwargs, args, kwargs, timeout)
65 fut = rpc.rpc_async(dst_worker_name, two_args_two_kwargs, args, kwargs)
78 fut = rpc.rpc_async(dst_worker_name, rref_to_here, args)
H A Ddist_autograd_test.py9 from torch.distributed.rpc import rpc_async
24 return rpc_async(dst, local_add, (t1, t2)).wait()
98 res1_fut = rpc.rpc_async(dst_worker_name, local_add, (t1, t1))
103 res2_fut = rpc.rpc_async(dst_worker_name, local_add, (t2, t2))
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Drpc_test.py314 rpc.rpc_async(
450 fut = rpc.rpc_async(to, torch.add, args=(x, y))
461 fut = rpc.rpc_async(to, torch.add, args=(x, y))
470 fut2 = rpc.rpc_async(to, torch.add, args=(fut1.wait(), z))
473 fut1 = rpc.rpc_async(to, torch.add, args=(x, y))
494 return rpc.rpc_async(to, torch.add, args=(x, y))
506 return rpc.rpc_async(to, slow_add, args=(x, y, device))
512 rpc.rpc_async(to, torch.add, args=(x, y)).then(
520 return rpc.rpc_async(to, torch.add, args=(x, y)).then(
527 fut = rpc.rpc_async(to, torch.add, args=(x, 0))
[all …]
H A Dfaulty_agent_rpc_test.py88 rpc.rpc_async(dst_worker, add_rref_to_value, args=(rref, 1))
263 fut = rpc.rpc_async(
271 fut = rpc.rpc_async(
278 fut = rpc.rpc_async(
285 fut = rpc.rpc_async(
300 fut = rpc.rpc_async(dst_worker, my_script_func, args=(torch.tensor(1),), timeout=1)
306 fut = rpc.rpc_async(
313 fut = rpc.rpc_async(
321 fut = rpc.rpc_async(
H A Ddist_optimizer_test.py89 return rpc.rpc_async(
257 output1 = remote_module1.rpc_async().forward(t2)
258 output2 = remote_module2.rpc_async().forward(output1.wait())
264 new_w1 = remote_module1.rpc_async().get_w().wait()
265 new_w2 = remote_module2.rpc_async().get_w().wait()
H A Ddist_autograd_test.py228 fut = rpc.rpc_async(worker_name(dst), method, args=(args))
722 rpc.rpc_async(
2776 futs.append(remote_layer.rpc_async().gradients(context_id))
/aosp_15_r20/external/pytorch/docs/source/rpc/
H A Drref.rst32 :meth:`~torch.distributed.rpc.rpc_async` or
51 :meth:`~torch.distributed.rpc.rpc_async` or
81 :meth:`~torch.distributed.rpc.rpc_async`, or
239 rpc.rpc_async('B', func, args=(rref, ))
282 dist.rpc_async('C', func, args=(rref, ))
290 scope after the rpc_async call, it will not be deleted, because internally
313 rpc.rpc_async('C', func, args=(rref, ))
/aosp_15_r20/external/pytorch/torch/distributed/rpc/
H A Drref_proxy.py7 from . import functions, rpc_async
43 if rpc_api != rpc_async:
H A Dapi.py258 fut = rpc_async(
467 fut = rpc_async(rref.owner(), _rref_typeof_on_owner, args=(rref,), timeout=timeout)
832 def rpc_async(to, func, args=None, kwargs=None, timeout=UNSET_RPC_TIMEOUT): function
/aosp_15_r20/external/pytorch/torch/distributed/optim/
H A Doptimizer.py216 remote_optim_rref_fut = rpc.rpc_async(
251 rpc.rpc_async(
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/examples/
H A Dreinforcement_learning_rpc_test.py15 from torch.distributed.rpc import RRef, rpc_sync, rpc_async, remote
174 rpc_async(
H A Dparameter_server_test.py113 rpc.rpc_async(trainer, run_trainer, args=(ps_rref,))
/aosp_15_r20/external/pytorch/docs/source/
H A Drpc.rst38 :meth:`~torch.distributed.rpc.rpc_async` (asynchronous), and
50 :meth:`~torch.distributed.rpc.rpc_async` are no longer appropriate, as they
112 .. autofunction:: rpc_async
196 :meth:`~torch.distributed.rpc.rpc_async` and
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Doperator.cpp216 prim::unchecked_cast, prim::tolist, prim::rpc_async, in printerHasSpecialCaseFor()
333 prim::rpc_async, in aliasAnalysisHasSpecialCaseFor()
H A Dregister_distributed_ops.cpp264 prim::rpc_async,
/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/
H A Dutils.py50 fut = cref.server_rref.rpc_async().average_gradient(*server_args)
H A Dlauncher.py181 fut = rpc.rpc_async(
/aosp_15_r20/external/pytorch/torch/distributed/nn/api/
H A Dremote_module.py242 fut = rpc.rpc_async(
253 fut = rpc.rpc_async(
/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/rl/
H A Dcoordinator.py80 ob_rref.rpc_async().run_ob_episode(self.agent_rref, n_steps)
/aosp_15_r20/external/pytorch/torch/distributed/benchmarks/
H A Dbenchmark_ddp_rpc.py241 fut = rpc.rpc_async(
/aosp_15_r20/external/pytorch/torch/_C/
H A D_distributed_rpc.pyi60 def rpc_async(self, timeout: float = ...) -> Any: ...
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Dinterned_strings.h183 _(prim, rpc_async) \
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dddp_under_dist_autograd_test.py89 return rpc.rpc_async(rref.owner(), _call_method, args=args_tup, kwargs=kwargs)
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/
H A Dapi.py356 fut = rpc.rpc_async(

12