1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
#include <torch/csrc/distributed/rpc/unpickled_python_call.h>
#include <c10/util/C++17.h>
#include <torch/csrc/distributed/rpc/python_rpc_handler.h>
namespace torch {
namespace distributed {
namespace rpc {
UnpickledPythonCall::UnpickledPythonCall(
const SerializedPyObj& serializedPyObj,
bool isAsyncExecution)
: isAsyncExecution_(isAsyncExecution) {
auto& pythonRpcHandler = PythonRpcHandler::getInstance();
pybind11::gil_scoped_acquire ag;
pythonUdf_ = pythonRpcHandler.deserialize(serializedPyObj);
}
UnpickledPythonCall::~UnpickledPythonCall() {
// explicitly setting PyObject* to nullptr to prevent py::object's dtor to
// decref on the PyObject again.
// See Note [Destructing py::object] in python_ivalue.h
py::gil_scoped_acquire acquire;
pythonUdf_.dec_ref();
pythonUdf_.ptr() = nullptr;
}
c10::intrusive_ptr<Message> UnpickledPythonCall::toMessageImpl() && {
TORCH_INTERNAL_ASSERT(
false, "UnpickledPythonCall does not support toMessage().");
}
const py::object& UnpickledPythonCall::pythonUdf() const {
return pythonUdf_;
}
} // namespace rpc
} // namespace distributed
} // namespace torch
|
