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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
#include <torch/csrc/lazy/python/python_util.h>
#include <Python.h>
#include <frameobject.h>
#include <pybind11/pybind11.h>
#include <torch/csrc/lazy/core/debug_util.h>
#include <torch/csrc/utils/pybind.h>
#include <torch/csrc/utils/python_compat.h>
#include <torch/csrc/utils/python_strings.h>
namespace torch::lazy {
std::optional<SourceLocation> GetPythonFrameTop() {
if (!Py_IsInitialized()) {
return std::nullopt;
}
pybind11::gil_scoped_acquire gil;
PyFrameObject* frame = PyEval_GetFrame();
if (frame == nullptr) {
return std::nullopt;
}
SourceLocation loc;
auto code = THPCodeObjectPtr(PyFrame_GetCode(frame));
loc.line = PyFrame_GetLineNumber(frame);
loc.file = THPUtils_unpackString(code->co_filename);
loc.function = THPUtils_unpackString(code->co_name);
return loc;
}
std::vector<SourceLocation> GetPythonFrames() {
std::vector<SourceLocation> frames;
if (Py_IsInitialized()) {
pybind11::gil_scoped_acquire gil;
PyFrameObject* frame = PyEval_GetFrame();
if (frame != nullptr) {
Py_INCREF(frame);
}
while (frame != nullptr) {
SourceLocation loc;
auto code = THPCodeObjectPtr(PyFrame_GetCode(frame));
loc.line = PyFrame_GetLineNumber(frame);
loc.file = THPUtils_unpackString(code->co_filename);
loc.function = THPUtils_unpackString(code->co_name);
frames.push_back(std::move(loc));
auto new_frame = PyFrame_GetBack(frame);
Py_DECREF(frame);
frame = new_frame;
}
}
return frames;
}
} // namespace torch::lazy
|
