pytorch/torch/distributed/rpc
Rohan Varma c3f2f3294e [RPC] Add option to make rref.get_type not block. (#50977)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/50977

Adds a `blocking` flag that can be set to False to make this API return a `Future` to the type. This is to make this function non-blocking, mostly for a future change that will allow `rref.rpc_async()` to be completely non-blocking (it currently calls and waits for this function that issues an RPC in-line).
ghstack-source-id: 121021433

Test Plan: Modified UT

Reviewed By: mrshenli

Differential Revision: D25944582

fbshipit-source-id: e3b48a52af2d4578551a30ba6838927b489b1c03
2021-02-04 20:18:50 -08:00
..
_testing Fix type annotation errors in torch.distributed.* directory (#47531) 2020-11-16 23:23:13 -08:00
__init__.py Reorder torch.distributed.rpc.init_rpc docstring arguments (#50419) 2021-01-14 07:58:09 -08:00
api.py [RPC] Add option to make rref.get_type not block. (#50977) 2021-02-04 20:18:50 -08:00
backend_registry.py Remove workaround for TensorPipe failing to get device of CUDA ptr (#50580) 2021-01-19 12:18:00 -08:00
constants.py Add type annotations to torch._C._distributed_rpc module. (#46624) 2020-11-06 01:28:51 -08:00
functions.py Add type annotations to torch._C._distributed_rpc module. (#46624) 2020-11-06 01:28:51 -08:00
internal.py [*.py] Rename "Arguments:" to "Args:" (#49736) 2020-12-28 09:34:47 -08:00
options.py [*.py] Rename "Arguments:" to "Args:" (#49736) 2020-12-28 09:34:47 -08:00
rref_proxy.py [RPC] Support timeout for RRef proxy functions (#50499) 2021-01-15 13:23:23 -08:00
server_process_global_profiler.py Remove flops warnings from the default profiler use case (#49896) 2020-12-30 23:49:29 -08:00
utils.py [*.py] Rename "Arguments:" to "Args:" (#49736) 2020-12-28 09:34:47 -08:00