wehub-resource-sync
131 lines
4.0 KiB
C
131 lines
4.0 KiB
C
/* Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License. */
|
|
|
|
#include "paddle/fluid/pybind/sot/frame_proxy.h"
|
|
#include "paddle/fluid/pybind/sot/cpython_internals.h"
|
|
|
|
#if SOT_IS_SUPPORTED
|
|
#include <Python.h>
|
|
#if PY_3_14_PLUS
|
|
#include <internal/pycore_interpframe.h>
|
|
#endif
|
|
|
|
#if PY_3_11_PLUS
|
|
|
|
#define DECLARE_PROXY_PROPERTY(name) \
|
|
static PyObject *PyInterpreterFrameProxy_property_##name( \
|
|
PyInterpreterFrameProxy *self, void *closure) { \
|
|
Py_XINCREF(self->frame->name); \
|
|
return (PyObject *)self->frame->name; \
|
|
}
|
|
|
|
// clang-format off
|
|
#define REGISTER_PROXY_PROPERTY(property_name, func_name) \
|
|
{ #property_name, (getter)PyInterpreterFrameProxy_property_##func_name, NULL, NULL, NULL }
|
|
// clang-format on
|
|
|
|
#if PY_3_14_PLUS
|
|
static PyObject *PyInterpreterFrameProxy_property_f_executable(
|
|
PyInterpreterFrameProxy *self, void *closure) {
|
|
return PyStackRef_AsPyObjectNew(self->frame->f_executable);
|
|
}
|
|
#elif PY_3_13_PLUS
|
|
DECLARE_PROXY_PROPERTY(f_executable)
|
|
#else
|
|
DECLARE_PROXY_PROPERTY(f_code)
|
|
#endif
|
|
#if PY_3_13_PLUS
|
|
static PyObject *PyInterpreterFrameProxy_property_f_locals(
|
|
PyInterpreterFrameProxy *self, void *closure) {
|
|
Py_XINCREF(self->locals);
|
|
return self->locals;
|
|
}
|
|
#else
|
|
DECLARE_PROXY_PROPERTY(f_locals)
|
|
#endif
|
|
DECLARE_PROXY_PROPERTY(f_globals)
|
|
DECLARE_PROXY_PROPERTY(f_builtins)
|
|
|
|
// Refer to
|
|
// https://github.com/python/cpython/blob/9414ddf91898892f3f6a672ae946931ee4b3ceb7/Objects/frameobject.c#L953-L961
|
|
static PyObject *PyInterpreterFrameProxy_method_repr(
|
|
PyInterpreterFrameProxy *self) {
|
|
#if PY_3_12_PLUS
|
|
int lineno = Internal_PyUnstable_InterpreterFrame_GetLine(self->frame);
|
|
#else
|
|
int lineno = Internal_PyInterpreterFrame_GetLine(self->frame);
|
|
#endif
|
|
PyCodeObject *code = PyFrame_GET_CODE(self->frame);
|
|
return PyUnicode_FromFormat(
|
|
"<PyInterpreterFrameProxy at %p, file %R, line %d, code %S>",
|
|
self,
|
|
code->co_filename,
|
|
lineno,
|
|
code->co_name);
|
|
}
|
|
|
|
static PyGetSetDef PyInterpreterFrameProxy_properties[] = {
|
|
#if PY_3_13_PLUS
|
|
REGISTER_PROXY_PROPERTY(f_code, f_executable),
|
|
#else
|
|
REGISTER_PROXY_PROPERTY(f_code, f_code),
|
|
#endif
|
|
REGISTER_PROXY_PROPERTY(f_locals, f_locals),
|
|
REGISTER_PROXY_PROPERTY(f_globals, f_globals),
|
|
REGISTER_PROXY_PROPERTY(f_builtins, f_builtins),
|
|
{NULL} /* Sentinel */
|
|
};
|
|
|
|
// clang-format off
|
|
static PyTypeObject PyInterpreterFrameProxyType = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
.tp_name = "paddle.framework.core.PyInterpreterFrameProxy",
|
|
.tp_doc = PyDoc_STR("A proxy object for _PyInterpreterFrame, "
|
|
"it's only define all properties we need."),
|
|
.tp_repr = (reprfunc)PyInterpreterFrameProxy_method_repr,
|
|
.tp_basicsize = sizeof(PyInterpreterFrameProxy),
|
|
.tp_itemsize = 0,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT,
|
|
.tp_getset = PyInterpreterFrameProxy_properties,
|
|
};
|
|
// clang-format on
|
|
|
|
PyInterpreterFrameProxy *PyInterpreterFrameProxy_New(
|
|
_PyInterpreterFrame *frame) {
|
|
PyTypeObject *type = &PyInterpreterFrameProxyType;
|
|
PyInterpreterFrameProxy *self =
|
|
(PyInterpreterFrameProxy *)type->tp_alloc(type, 0);
|
|
if (!self) {
|
|
// VLOG(7) << "Failed to allocate PyInterpreterFrameProxy";
|
|
return NULL;
|
|
}
|
|
self->frame = frame;
|
|
#if PY_3_13_PLUS
|
|
self->locals = NULL;
|
|
#endif
|
|
return self;
|
|
}
|
|
|
|
PyMODINIT_FUNC PyInit__frame_proxy() {
|
|
if (PyType_Ready(&PyInterpreterFrameProxyType) < 0) {
|
|
// VLOG(7) << "PyInterpreterFrameProxyType has not been ready!";
|
|
}
|
|
Py_INCREF(&PyInterpreterFrameProxyType);
|
|
return NULL;
|
|
}
|
|
|
|
#endif
|
|
|
|
#endif
|