frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-24 15:44:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Reland "[RPC] Use qualified name str directly in RPC torch script code path" (#35489)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35489

Relanding https://github.com/pytorch/pytorch/pull/34733.

Fix is in https://github.com/pytorch/pytorch/pull/34988

Test Plan:
```
buck test mode/dev-nosan //caffe2/test/distributed/rpc:rpc_fork
```

```
buck test mode/dev-nosan //caffe2/test/distributed/rpc/jit:rpc_fork

buck build mode/dev-nosan //caffe2/test/distributed/rpc/jit:rpc_fork && \
buck-out/gen/caffe2/test/distributed/rpc/jit/rpc_fork\#binary.par \
-r test_return_local_script_class_rref_in_py_and_use_in_script

buck build mode/dev-nosan //caffe2/test/distributed/rpc/jit:rpc_fork && \
buck-out/gen/caffe2/test/distributed/rpc/jit/rpc_fork\#binary.par \
-r test_return_local_script_module_rref_in_py_and_use_in_script
```

Differential Revision: D20661748

fbshipit-source-id: d550daab8d689d0a9aa2450f3bdb7417ab79dae2
This commit is contained in:
Shihao Xu
2020-03-26 23:39:00 -07:00
committed by Facebook GitHub Bot
parent d2d40c45b6
commit ac639d927a
4 changed files with 12 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/distributed/rpc/api.py
View File

@ -475,7 +475,9 @@ def _invoke_rpc(to, func, rpc_type, args=None, kwargs=None):
if qualified_name is not None:
fut = _invoke_rpc_builtin(dst_worker_info, qualified_name, rf, *args, **kwargs)
elif isinstance(func, torch.jit.ScriptFunction):
fut = _invoke_rpc_torchscript(dst_worker_info.name, func, args, kwargs)
fut = _invoke_rpc_torchscript(
dst_worker_info.name, torch.jit._qualified_name(func), args, kwargs
)
else:
(pickled_python_udf, tensors) = _default_pickler.serialize(
PythonUDF(func, args, kwargs)