mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
de6353ba88
The pattern of a PyObject* bundled with a PyInterpreter* is pretty useful in many contexts (e.g., TorchDispatchTypeObject) so I have turned it into a dedicated class SafePyObject. In the process I fixed a bug with the old TorchDispatchTypeObject (copy constructor/assignment was not deleted), made the API more safe (retrieving the PyObject* pointer requires verification that the PyInterpreter* matches) and fixed some minor inefficiencies in C++ code. Signed-off-by: Edward Z. Yang <ezyangfb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/75142 Approved by: https://github.com/zou3519
12 lines
261 B
C++
12 lines
261 B
C++
#include <c10/core/SafePyObject.h>
|
|
#include <c10/core/TensorImpl.h>
|
|
|
|
namespace c10 {
|
|
|
|
PyObject* SafePyObject::ptr(const c10::impl::PyInterpreter* interpreter) const {
|
|
TORCH_INTERNAL_ASSERT(interpreter == pyinterpreter_);
|
|
return data_;
|
|
}
|
|
|
|
} // namespace c10
|