xref: /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/unpickled_python_remote_call.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <torch/csrc/distributed/rpc/unpickled_python_remote_call.h>
2 
3 #include <torch/csrc/distributed/rpc/python_rpc_handler.h>
4 
5 namespace torch::distributed::rpc {
6 
UnpickledPythonRemoteCall(const SerializedPyObj & serializedPyObj,const at::IValue & rrefId,const at::IValue & forkId,bool isAsyncExecution)7 UnpickledPythonRemoteCall::UnpickledPythonRemoteCall(
8     const SerializedPyObj& serializedPyObj,
9     const at::IValue& rrefId,
10     const at::IValue& forkId,
11     bool isAsyncExecution)
12     : UnpickledPythonCall(serializedPyObj, isAsyncExecution),
13       rrefId_(RRefId::fromIValue(rrefId)),
14       forkId_(ForkId::fromIValue(forkId)) {}
15 
rrefId() const16 const RRefId& UnpickledPythonRemoteCall::rrefId() const {
17   return rrefId_;
18 }
19 
forkId() const20 const ForkId& UnpickledPythonRemoteCall::forkId() const {
21   return forkId_;
22 }
23 
24 } // namespace torch::distributed::rpc
25