Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-11-04 07:31:38 +00:00
Code Issues Projects Releases Packages Wiki Activity

GH-133231: Add JIT utilities in sys._jit (GH-133233)

Browse source
This commit is contained in:
Brandt Bucher 2025-05-05 15:25:22 -07:00 • committed by GitHub
parent f9b22bb79d
commit b1aa515bd6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 296 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

75
Python/sysmodule.c
View file

@ -3986,6 +3986,71 @@ _PySys_SetPreliminaryStderr(PyObject *sysdict)
PyObject *_Py_CreateMonitoringObject(void);
/*[clinic input]
module _jit
[clinic start generated code]*/
/*[clinic end generated code: output=da39a3ee5e6b4b0d input=10952f74d7bbd972]*/
PyDoc_STRVAR(_jit_doc, "Utilities for observing just-in-time compilation.");
/*[clinic input]
_jit.is_available -> bool
Return True if the current Python executable supports JIT compilation, and False otherwise.
[clinic start generated code]*/
static int
_jit_is_available_impl(PyObject *module)
/*[clinic end generated code: output=6849a9cd2ff4aac9 input=03add84aa8347cf1]*/
{
(void)module;
#ifdef _Py_TIER2
return true;
#else
return false;
#endif
}
/*[clinic input]
_jit.is_enabled -> bool
Return True if JIT compilation is enabled for the current Python process (implies sys._jit.is_available()), and False otherwise.
[clinic start generated code]*/
static int
_jit_is_enabled_impl(PyObject *module)
/*[clinic end generated code: output=55865f8de993fe42 input=02439394da8e873f]*/
{
(void)module;
return _PyInterpreterState_GET()->jit;
}
/*[clinic input]
_jit.is_active -> bool
Return True if the topmost Python frame is currently executing JIT code (implies sys._jit.is_enabled()), and False otherwise.
[clinic start generated code]*/
static int
_jit_is_active_impl(PyObject *module)
/*[clinic end generated code: output=7facca06b10064d4 input=be2fcd8a269d9b72]*/
{
(void)module;
return _PyThreadState_GET()->current_executor != NULL;
}
static PyMethodDef _jit_methods[] = {
_JIT_IS_AVAILABLE_METHODDEF
_JIT_IS_ENABLED_METHODDEF
_JIT_IS_ACTIVE_METHODDEF
{NULL}
};
static struct PyModuleDef _jit_module = {
PyModuleDef_HEAD_INIT,
.m_name = "sys._jit",
.m_doc = _jit_doc,
.m_size = -1,
.m_methods = _jit_methods,
};
/* Create sys module without all attributes.
_PySys_UpdateConfig() should be called later to add remaining attributes. */
PyStatus
@ -4047,6 +4112,16 @@ _PySys_Create(PyThreadState *tstate, PyObject **sysmod_p)
goto error;
}
PyObject *_jit = _PyModule_CreateInitialized(&_jit_module, PYTHON_API_VERSION);
if (_jit == NULL) {
goto error;
}
err = PyDict_SetItemString(sysdict, "_jit", _jit);
Py_DECREF(_jit);
if (err) {
goto error;
}
assert(!_PyErr_Occurred(tstate));
*sysmod_p = sysmod;