From ff96b81d78c4a52fb1eb8384300af3dd0dd2db0d Mon Sep 17 00:00:00 2001 From: Ken Jin Date: Sat, 2 Mar 2024 03:40:04 +0800 Subject: gh-115480: Type propagate _BINARY_OP_ADD_UNICODE (GH-115710) --- Python/optimizer_bytecodes.c | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'Python/optimizer_bytecodes.c') diff --git a/Python/optimizer_bytecodes.c b/Python/optimizer_bytecodes.c index 2b47381ec76..786d884fc5a 100644 --- a/Python/optimizer_bytecodes.c +++ b/Python/optimizer_bytecodes.c @@ -254,6 +254,22 @@ dummy_func(void) { } } + op(_BINARY_OP_ADD_UNICODE, (left, right -- res)) { + if (sym_is_const(left) && sym_is_const(right) && + sym_matches_type(left, &PyUnicode_Type) && sym_matches_type(right, &PyUnicode_Type)) { + PyObject *temp = PyUnicode_Concat(sym_get_const(left), sym_get_const(right)); + if (temp == NULL) { + goto error; + } + res = sym_new_const(ctx, temp); + Py_DECREF(temp); + OUT_OF_SPACE_IF_NULL(res); + } + else { + OUT_OF_SPACE_IF_NULL(res = sym_new_type(ctx, &PyUnicode_Type)); + } + } + op(_TO_BOOL, (value -- res)) { (void)value; res = sym_new_type(ctx, &PyBool_Type); -- cgit v1.2.3