Skip to content

gh-124153: Fix unstable optimization of PyType_GetBaseByToken() and friends #124488

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
encukou merged 14 commits into from
Oct 10, 2024
Merged

gh-124153: Fix unstable optimization of PyType_GetBaseByToken() and friends #124488

Changes from 1 commit
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
3b5f12b
abandon if-else workaround for maintainability
neonene Sep 25, 2024
a0dae73
do not pressure PyType_GetModuleByDef() and others
neonene Oct 8, 2024
f1fa9f4
ready for moving get_base_by_token_from_mro()
neonene Oct 8, 2024
cae44aa
move get_base_by_token_from_mro() into get/check funcs
neonene Oct 8, 2024
230e06d
simplify check_base_by_token()
neonene Oct 8, 2024
55ca7b8
Merge branch 'main' into opttoken
neonene Oct 8, 2024
9285a99
Merge branch 'main' into opttoken
neonene Oct 8, 2024
75f600f
remove os-specific comments
neonene Oct 8, 2024
2e7bb29
remove redundant code
neonene Oct 8, 2024
d636fa8
remove check_base_by_token()
neonene Oct 9, 2024
b83b63a
reword
neonene Oct 9, 2024
c02f0a6
perf tuning
neonene Oct 9, 2024
07de01b
reduce gotos
neonene Oct 10, 2024
0fc1c90
add comment
neonene Oct 10, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
simplify check_base_by_token()
  • Loading branch information
@neonene
neonene committed Oct 8, 2024
commit 230e06dc46e51782bc40ef1d6109a49f8e45ffd4
16 changes: 6 additions & 10 deletions Objects/typeobject.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5315,7 +5315,8 @@ get_base_by_token_recursive(PyObject *bases, void *token)
}

static int
check_base_by_token(PyTypeObject *type, void *token) {
check_base_by_token(PyTypeObject *type, void *token)
{
if (token == NULL) {
PyErr_Format(PyExc_SystemError,
"PyType_GetBaseByToken called with token=NULL");
Expand All @@ -5337,22 +5338,17 @@ check_base_by_token(PyTypeObject *type, void *token) {
return get_base_by_token_recursive(lookup_tp_bases(type), token) ? 1 : 0;
}
else {
// mro_invoke() ensures that the type MRO cannot be empty.
assert(PyTuple_GET_SIZE(mro) >= 1);
// Also, the first item in the MRO is the type itself, which
// we already checked above. We skip it in the loop.
assert(PyTuple_GET_ITEM(mro, 0) == (PyObject *)type);
PyTypeObject *res = NULL;
Py_ssize_t n = PyTuple_GET_SIZE(mro);
for (Py_ssize_t i = 1; i < n; i++) {
// Find from the last when only checking
for (Py_ssize_t i = PyTuple_GET_SIZE(mro) - 1; i >= 1; i--) {
PyTypeObject *base = (PyTypeObject *)PyTuple_GET_ITEM(mro, i);
if (_PyType_HasFeature(base, Py_TPFLAGS_HEAPTYPE)
&& ((PyHeapTypeObject*)base)->ht_token == token) {
res = base;
break;
return 1;
}
}
return res ? 1 : 0;
return 0;
}
}

Expand Down