Use the Python frame safely in _pythonCallstack (#89997)

pytorch · Dec 6, 2022 · 59b4f3b · 59b4f3b
1 parent be6ac74
commit 59b4f3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/csrc/jit/python/python_tracer.cpp b/torch/csrc/jit/python/python_tracer.cpp
@@ -27,7 +27,7 @@ namespace tracer {
 std::vector<StackEntry> _pythonCallstack() {
   pybind11::gil_scoped_acquire gil;
   PyFrameObject* frame = PyEval_GetFrame();
-  Py_INCREF(frame);
+  Py_XINCREF(frame);
   std::vector<StackEntry> entries;
 
   while (nullptr != frame) {