1 | #include <torch/csrc/distributed/rpc/unpickled_python_remote_call.h> |
---|---|
2 | |
3 | #include <c10/util/C++17.h> |
4 | #include <torch/csrc/distributed/rpc/python_rpc_handler.h> |
5 | |
6 | namespace torch { |
7 | namespace distributed { |
8 | namespace rpc { |
9 | |
10 | UnpickledPythonRemoteCall::UnpickledPythonRemoteCall( |
11 | const SerializedPyObj& serializedPyObj, |
12 | const at::IValue& rrefId, |
13 | const at::IValue& forkId, |
14 | bool isAsyncExecution) |
15 | : UnpickledPythonCall(serializedPyObj, isAsyncExecution), |
16 | rrefId_(RRefId::fromIValue(rrefId)), |
17 | forkId_(ForkId::fromIValue(forkId)) {} |
18 | |
19 | const RRefId& UnpickledPythonRemoteCall::rrefId() const { |
20 | return rrefId_; |
21 | } |
22 | |
23 | const ForkId& UnpickledPythonRemoteCall::forkId() const { |
24 | return forkId_; |
25 | } |
26 | |
27 | } // namespace rpc |
28 | } // namespace distributed |
29 | } // namespace torch |
30 |