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
do not pressure PyType_GetModuleByDef() and others
  • Loading branch information
@neonene
neonene committed Oct 8, 2024
commit a0dae738f2c674ce8365ca266b18adf5567c420c
21 changes: 12 additions & 9 deletions Objects/typeobject.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5289,26 +5289,29 @@ _PyType_GetModuleByDef2(PyTypeObject *left, PyTypeObject *right,


static PyTypeObject *
get_base_by_token_recursive(PyTypeObject *type, void *token)
get_base_by_token_recursive(PyObject *bases, void *token)
{
assert(PyType_GetSlot(type, Py_tp_token) != token);
PyObject *bases = lookup_tp_bases(type);
assert(bases != NULL);
// MSVC: The result should be initialized outside the loop to avoid the c
// file being less PGO-optimized. Returning it from inside seems to be OK.
PyTypeObject *res = NULL;
Py_ssize_t n = PyTuple_GET_SIZE(bases);
for (Py_ssize_t i = 0; i < n; i++) {
PyTypeObject *base = _PyType_CAST(PyTuple_GET_ITEM(bases, i));
if (!_PyType_HasFeature(base, Py_TPFLAGS_HEAPTYPE)) {
continue;
}
if (((PyHeapTypeObject*)base)->ht_token == token) {
return base;
res = base;
break;
}
base = get_base_by_token_recursive(base, token);
base = get_base_by_token_recursive(lookup_tp_bases(base), token);
if (base != NULL) {
return base;
res = base;
break;
}
}
return NULL;
return res; // Prefer to return in one place
}

static inline PyTypeObject *
Expand Down Expand Up @@ -5361,7 +5364,7 @@ check_base_by_token(PyTypeObject *type, void *token) {
return get_base_by_token_from_mro(type, token) ? 1 : 0;
}
else {
return get_base_by_token_recursive(type, token) ? 1 : 0;
return get_base_by_token_recursive(lookup_tp_bases(type), token) ? 1 : 0;
}
}

Expand Down Expand Up @@ -5395,7 +5398,7 @@ PyType_GetBaseByToken(PyTypeObject *type, void *token, PyTypeObject **result)
base = get_base_by_token_from_mro(type, token);
}
else {
base = get_base_by_token_recursive(type, token);
base = get_base_by_token_recursive(lookup_tp_bases(type), token);
}
if (base != NULL) {
*result = (PyTypeObject *)Py_NewRef(base);
Expand Down