mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
05ee8114f8
This pr gets rid of the pyobj_interpreter_ variable from PyObjectSlot and saves a word in the process Gonna ask for review from @huydhn as there are some changes to CI. Pull Request resolved: https://github.com/pytorch/pytorch/pull/162659 Approved by: https://github.com/albanD, https://github.com/huydhn
57 lines
1.7 KiB
C++
57 lines
1.7 KiB
C++
#include <c10/core/impl/PyObjectSlot.h>
|
|
|
|
namespace c10::impl {
|
|
|
|
PyObjectSlot::PyObjectSlot() : pyobj_(nullptr) {}
|
|
|
|
PyObjectSlot::~PyObjectSlot() {
|
|
maybe_destroy_pyobj();
|
|
}
|
|
|
|
void PyObjectSlot::maybe_destroy_pyobj() {
|
|
if (owns_pyobj()) {
|
|
TORCH_INTERNAL_ASSERT(getGlobalPyInterpreter() != nullptr);
|
|
TORCH_INTERNAL_ASSERT(pyobj_ != nullptr);
|
|
(*getGlobalPyInterpreter())
|
|
->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 getGlobalPyInterpreter();
|
|
}
|
|
|
|
PyObject* PyObjectSlot::_unchecked_untagged_pyobj() const {
|
|
// NOLINTNEXTLINE(performance-no-int-to-ptr)
|
|
return reinterpret_cast<PyObject*>(
|
|
reinterpret_cast<uintptr_t>(pyobj_) & ~0x1ULL);
|
|
}
|
|
|
|
PyInterpreter& PyObjectSlot::load_pyobj_interpreter() const {
|
|
auto interpreter = getGlobalPyInterpreter();
|
|
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
|