mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
255c0545e7
This PR makes some less risky changes to PyObjectSlot as there is a lot of stuff we do not need since there is only one interpreter. Specifically `check_interpreter` and `has_pyobj_nonhermetic` are removed Pull Request resolved: https://github.com/pytorch/pytorch/pull/158407 Approved by: https://github.com/albanD ghstack dependencies: #158288, #158290, #158291
62 lines
2.0 KiB
C++
62 lines
2.0 KiB
C++
#include <c10/core/impl/PyObjectSlot.h>
|
|
|
|
namespace c10::impl {
|
|
|
|
PyObjectSlot::PyObjectSlot() : pyobj_interpreter_(nullptr), pyobj_(nullptr) {}
|
|
|
|
PyObjectSlot::~PyObjectSlot() {
|
|
maybe_destroy_pyobj();
|
|
}
|
|
|
|
void PyObjectSlot::maybe_destroy_pyobj() {
|
|
if (owns_pyobj()) {
|
|
TORCH_INTERNAL_ASSERT(pyobj_interpreter_ != nullptr);
|
|
TORCH_INTERNAL_ASSERT(pyobj_ != nullptr);
|
|
(*pyobj_interpreter_.load(std::memory_order_acquire))
|
|
->decref(_unchecked_untagged_pyobj(), /*has_pyobj_slot*/ true);
|
|
// NB: this destructor can only be entered when there are no
|
|
// references to this C++ object (obviously), NOR any references
|
|
// to the PyObject (if there are references to the PyObject,
|
|
// then the PyObject holds an owning reference to the tensor).
|
|
// So it is OK to clear pyobj_ here as it is impossible for it to
|
|
// be used again (modulo weak reference races)
|
|
pyobj_ = nullptr; // for safety
|
|
}
|
|
}
|
|
|
|
PyInterpreter* PyObjectSlot::pyobj_interpreter() {
|
|
return pyobj_interpreter_.load(std::memory_order_acquire);
|
|
}
|
|
|
|
PyObject* PyObjectSlot::_unchecked_untagged_pyobj() const {
|
|
// NOLINTNEXTLINE(performance-no-int-to-ptr)
|
|
return reinterpret_cast<PyObject*>(
|
|
reinterpret_cast<uintptr_t>(pyobj_) & ~0x1ULL);
|
|
}
|
|
|
|
void PyObjectSlot::unchecked_clear_pyobj(PyInterpreter* interpreter) {
|
|
TORCH_INTERNAL_ASSERT_DEBUG_ONLY(interpreter == pyobj_interpreter_.load());
|
|
pyobj_ = nullptr;
|
|
}
|
|
|
|
PyInterpreter& PyObjectSlot::load_pyobj_interpreter() const {
|
|
auto interpreter = pyobj_interpreter_.load(std::memory_order_acquire);
|
|
if (interpreter) {
|
|
return *interpreter;
|
|
}
|
|
TORCH_CHECK(false, "cannot access PyObject for Tensor - no interpreter set");
|
|
}
|
|
|
|
bool PyObjectSlot::owns_pyobj() {
|
|
// NOLINTNEXTLINE(performance-no-int-to-ptr)
|
|
return reinterpret_cast<uintptr_t>(pyobj_) & 1;
|
|
}
|
|
|
|
void PyObjectSlot::set_owns_pyobj(bool b) {
|
|
// NOLINTNEXTLINE(performance-no-int-to-ptr)
|
|
pyobj_ = reinterpret_cast<PyObject*>(
|
|
reinterpret_cast<uintptr_t>(_unchecked_untagged_pyobj()) | b);
|
|
}
|
|
|
|
} // namespace c10::impl
|