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
6namespace torch {
7namespace distributed {
8namespace rpc {
9
10UnpickledPythonRemoteCall::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
19const RRefId& UnpickledPythonRemoteCall::rrefId() const {
20 return rrefId_;
21}
22
23const ForkId& UnpickledPythonRemoteCall::forkId() const {
24 return forkId_;
25}
26
27} // namespace rpc
28} // namespace distributed
29} // namespace torch
30