mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-07-07 19:35:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

GH-131798: Type-propagate string/list/tuple slices (GH-134671)

Browse source
This commit is contained in:
Amit Lavon 2025-06-07 14:08:44 -07:00 • committed by GitHub
parent ac9c3431cc
commit 8fdbbf8b18
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 55 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
Python/optimizer_bytecodes.c
View file

@ -1237,6 +1237,20 @@ dummy_func(void) {
sym_set_const(callable, list_append);
}
op(_BINARY_SLICE, (container, start, stop -- res)) {
// Slicing a string/list/tuple always returns the same type.
PyTypeObject *type = sym_get_type(container);
if (type == &PyUnicode_Type ||
type == &PyList_Type ||
type == &PyTuple_Type)
{
res = sym_new_type(ctx, type);
}
else {
res = sym_new_not_null(ctx);
}
}
// END BYTECODES //
}

13
Python/optimizer_cases.c.h generated
View file

@ -568,8 +568,19 @@
}
case _BINARY_SLICE: {
JitOptSymbol *container;
JitOptSymbol *res;
res = sym_new_not_null(ctx);
container = stack_pointer[-3];
PyTypeObject *type = sym_get_type(container);
if (type == &PyUnicode_Type ||
type == &PyList_Type ||
type == &PyTuple_Type)
{
res = sym_new_type(ctx, type);
}
else {
res = sym_new_not_null(ctx);
}
stack_pointer[-3] = res;
stack_pointer += -2;
assert(WITHIN_STACK_BOUNDS());