pytorch
dbe629c5 - [RPC Framework] Support creating a RemoteModule by RRef (#59242)

Commit
3 years ago
[RPC Framework] Support creating a RemoteModule by RRef (#59242) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/59242 #Oringal PR Issue: https://github.com/pytorch/pytorch/issues/58274 This can be a workaround: Instead of passing a script `RemoteModule` over RPC, pass its `module_rref` field over RPC, and then construct a new `RemoteModule` on the receiver end. ghstack-source-id: 130268018 Test Plan: buck test mode/dev-nosan caffe2/test/distributed/rpc:process_group_agent -- test_send_remote_module_over_the_wire_script_not_supported buck test mode/dev-nosan caffe2/test/distributed/rpc:process_group_agent -- test_remote_module_py_pickle_not_supported_script buck test mode/dev-nosan caffe2/test/distributed/rpc:process_group_agent -- test_create_remote_module_by_module_rref Reviewed By: vipannalla Differential Revision: D28794905 fbshipit-source-id: 1a677ff0d4b47c078ad47b50d7102a198a1fc39b
Author
Yi Wang
Parents
Loading