https://github.com/python/cpython/commit/685fd74f81e673dc0430120f50009636b0e437a5 commit: 685fd74f81e673dc0430120f50009636b0e437a5 branch: main author: Amit Lavon <amitlavon1.s...@gmail.com> committer: brandtbucher <brandtbuc...@gmail.com> date: 2025-03-30T16:07:25-07:00 summary:
GH-131798: Remove type checks for _TO_BOOL_STR (GH-131816) files: A Misc/NEWS.d/next/Core_and_Builtins/2025-03-28-19-02-55.gh-issue-131798.fNZ5-2.rst M Include/internal/pycore_opcode_metadata.h M Include/internal/pycore_uop_ids.h M Include/internal/pycore_uop_metadata.h M Lib/test/test_capi/test_opt.py M Python/bytecodes.c M Python/executor_cases.c.h M Python/generated_cases.c.h M Python/optimizer_bytecodes.c M Python/optimizer_cases.c.h diff --git a/Include/internal/pycore_opcode_metadata.h b/Include/internal/pycore_opcode_metadata.h index 9c101302591872..32bafaf09ce3da 100644 --- a/Include/internal/pycore_opcode_metadata.h +++ b/Include/internal/pycore_opcode_metadata.h @@ -1465,7 +1465,7 @@ _PyOpcode_macro_expansion[256] = { [TO_BOOL_INT] = { .nuops = 1, .uops = { { _TO_BOOL_INT, OPARG_SIMPLE, 3 } } }, [TO_BOOL_LIST] = { .nuops = 1, .uops = { { _TO_BOOL_LIST, OPARG_SIMPLE, 3 } } }, [TO_BOOL_NONE] = { .nuops = 1, .uops = { { _TO_BOOL_NONE, OPARG_SIMPLE, 3 } } }, - [TO_BOOL_STR] = { .nuops = 1, .uops = { { _TO_BOOL_STR, OPARG_SIMPLE, 3 } } }, + [TO_BOOL_STR] = { .nuops = 2, .uops = { { _GUARD_TOS_UNICODE, OPARG_SIMPLE, 0 }, { _TO_BOOL_STR, OPARG_SIMPLE, 3 } } }, [UNARY_INVERT] = { .nuops = 1, .uops = { { _UNARY_INVERT, OPARG_SIMPLE, 0 } } }, [UNARY_NEGATIVE] = { .nuops = 1, .uops = { { _UNARY_NEGATIVE, OPARG_SIMPLE, 0 } } }, [UNARY_NOT] = { .nuops = 1, .uops = { { _UNARY_NOT, OPARG_SIMPLE, 0 } } }, diff --git a/Include/internal/pycore_uop_ids.h b/Include/internal/pycore_uop_ids.h index fb5486171ef4ca..1d9c2bef4cedda 100644 --- a/Include/internal/pycore_uop_ids.h +++ b/Include/internal/pycore_uop_ids.h @@ -135,17 +135,18 @@ extern "C" { #define _GUARD_NOT_EXHAUSTED_TUPLE 381 #define _GUARD_TOS_FLOAT 382 #define _GUARD_TOS_INT 383 -#define _GUARD_TYPE_VERSION 384 -#define _GUARD_TYPE_VERSION_AND_LOCK 385 +#define _GUARD_TOS_UNICODE 384 +#define _GUARD_TYPE_VERSION 385 +#define _GUARD_TYPE_VERSION_AND_LOCK 386 #define _IMPORT_FROM IMPORT_FROM #define _IMPORT_NAME IMPORT_NAME -#define _INIT_CALL_BOUND_METHOD_EXACT_ARGS 386 -#define _INIT_CALL_PY_EXACT_ARGS 387 -#define _INIT_CALL_PY_EXACT_ARGS_0 388 -#define _INIT_CALL_PY_EXACT_ARGS_1 389 -#define _INIT_CALL_PY_EXACT_ARGS_2 390 -#define _INIT_CALL_PY_EXACT_ARGS_3 391 -#define _INIT_CALL_PY_EXACT_ARGS_4 392 +#define _INIT_CALL_BOUND_METHOD_EXACT_ARGS 387 +#define _INIT_CALL_PY_EXACT_ARGS 388 +#define _INIT_CALL_PY_EXACT_ARGS_0 389 +#define _INIT_CALL_PY_EXACT_ARGS_1 390 +#define _INIT_CALL_PY_EXACT_ARGS_2 391 +#define _INIT_CALL_PY_EXACT_ARGS_3 392 +#define _INIT_CALL_PY_EXACT_ARGS_4 393 #define _INSTRUMENTED_FOR_ITER INSTRUMENTED_FOR_ITER #define _INSTRUMENTED_INSTRUCTION INSTRUMENTED_INSTRUCTION #define _INSTRUMENTED_JUMP_FORWARD INSTRUMENTED_JUMP_FORWARD @@ -155,153 +156,153 @@ extern "C" { #define _INSTRUMENTED_POP_JUMP_IF_NONE INSTRUMENTED_POP_JUMP_IF_NONE #define _INSTRUMENTED_POP_JUMP_IF_NOT_NONE INSTRUMENTED_POP_JUMP_IF_NOT_NONE #define _INSTRUMENTED_POP_JUMP_IF_TRUE INSTRUMENTED_POP_JUMP_IF_TRUE -#define _IS_NONE 393 +#define _IS_NONE 394 #define _IS_OP IS_OP -#define _ITER_CHECK_LIST 394 -#define _ITER_CHECK_RANGE 395 -#define _ITER_CHECK_TUPLE 396 -#define _ITER_JUMP_LIST 397 -#define _ITER_JUMP_RANGE 398 -#define _ITER_JUMP_TUPLE 399 -#define _ITER_NEXT_LIST 400 -#define _ITER_NEXT_LIST_TIER_TWO 401 -#define _ITER_NEXT_RANGE 402 -#define _ITER_NEXT_TUPLE 403 -#define _JUMP_TO_TOP 404 +#define _ITER_CHECK_LIST 395 +#define _ITER_CHECK_RANGE 396 +#define _ITER_CHECK_TUPLE 397 +#define _ITER_JUMP_LIST 398 +#define _ITER_JUMP_RANGE 399 +#define _ITER_JUMP_TUPLE 400 +#define _ITER_NEXT_LIST 401 +#define _ITER_NEXT_LIST_TIER_TWO 402 +#define _ITER_NEXT_RANGE 403 +#define _ITER_NEXT_TUPLE 404 +#define _JUMP_TO_TOP 405 #define _LIST_APPEND LIST_APPEND #define _LIST_EXTEND LIST_EXTEND -#define _LOAD_ATTR 405 -#define _LOAD_ATTR_CLASS 406 +#define _LOAD_ATTR 406 +#define _LOAD_ATTR_CLASS 407 #define _LOAD_ATTR_GETATTRIBUTE_OVERRIDDEN LOAD_ATTR_GETATTRIBUTE_OVERRIDDEN -#define _LOAD_ATTR_INSTANCE_VALUE 407 -#define _LOAD_ATTR_METHOD_LAZY_DICT 408 -#define _LOAD_ATTR_METHOD_NO_DICT 409 -#define _LOAD_ATTR_METHOD_WITH_VALUES 410 -#define _LOAD_ATTR_MODULE 411 -#define _LOAD_ATTR_NONDESCRIPTOR_NO_DICT 412 -#define _LOAD_ATTR_NONDESCRIPTOR_WITH_VALUES 413 -#define _LOAD_ATTR_PROPERTY_FRAME 414 -#define _LOAD_ATTR_SLOT 415 -#define _LOAD_ATTR_WITH_HINT 416 +#define _LOAD_ATTR_INSTANCE_VALUE 408 +#define _LOAD_ATTR_METHOD_LAZY_DICT 409 +#define _LOAD_ATTR_METHOD_NO_DICT 410 +#define _LOAD_ATTR_METHOD_WITH_VALUES 411 +#define _LOAD_ATTR_MODULE 412 +#define _LOAD_ATTR_NONDESCRIPTOR_NO_DICT 413 +#define _LOAD_ATTR_NONDESCRIPTOR_WITH_VALUES 414 +#define _LOAD_ATTR_PROPERTY_FRAME 415 +#define _LOAD_ATTR_SLOT 416 +#define _LOAD_ATTR_WITH_HINT 417 #define _LOAD_BUILD_CLASS LOAD_BUILD_CLASS -#define _LOAD_BYTECODE 417 +#define _LOAD_BYTECODE 418 #define _LOAD_COMMON_CONSTANT LOAD_COMMON_CONSTANT #define _LOAD_CONST LOAD_CONST #define _LOAD_CONST_IMMORTAL LOAD_CONST_IMMORTAL -#define _LOAD_CONST_INLINE 418 -#define _LOAD_CONST_INLINE_BORROW 419 +#define _LOAD_CONST_INLINE 419 +#define _LOAD_CONST_INLINE_BORROW 420 #define _LOAD_CONST_MORTAL LOAD_CONST_MORTAL #define _LOAD_DEREF LOAD_DEREF -#define _LOAD_FAST 420 -#define _LOAD_FAST_0 421 -#define _LOAD_FAST_1 422 -#define _LOAD_FAST_2 423 -#define _LOAD_FAST_3 424 -#define _LOAD_FAST_4 425 -#define _LOAD_FAST_5 426 -#define _LOAD_FAST_6 427 -#define _LOAD_FAST_7 428 +#define _LOAD_FAST 421 +#define _LOAD_FAST_0 422 +#define _LOAD_FAST_1 423 +#define _LOAD_FAST_2 424 +#define _LOAD_FAST_3 425 +#define _LOAD_FAST_4 426 +#define _LOAD_FAST_5 427 +#define _LOAD_FAST_6 428 +#define _LOAD_FAST_7 429 #define _LOAD_FAST_AND_CLEAR LOAD_FAST_AND_CLEAR #define _LOAD_FAST_CHECK LOAD_FAST_CHECK #define _LOAD_FAST_LOAD_FAST LOAD_FAST_LOAD_FAST #define _LOAD_FROM_DICT_OR_DEREF LOAD_FROM_DICT_OR_DEREF #define _LOAD_FROM_DICT_OR_GLOBALS LOAD_FROM_DICT_OR_GLOBALS -#define _LOAD_GLOBAL 429 -#define _LOAD_GLOBAL_BUILTINS 430 -#define _LOAD_GLOBAL_MODULE 431 +#define _LOAD_GLOBAL 430 +#define _LOAD_GLOBAL_BUILTINS 431 +#define _LOAD_GLOBAL_MODULE 432 #define _LOAD_LOCALS LOAD_LOCALS #define _LOAD_NAME LOAD_NAME -#define _LOAD_SMALL_INT 432 -#define _LOAD_SMALL_INT_0 433 -#define _LOAD_SMALL_INT_1 434 -#define _LOAD_SMALL_INT_2 435 -#define _LOAD_SMALL_INT_3 436 +#define _LOAD_SMALL_INT 433 +#define _LOAD_SMALL_INT_0 434 +#define _LOAD_SMALL_INT_1 435 +#define _LOAD_SMALL_INT_2 436 +#define _LOAD_SMALL_INT_3 437 #define _LOAD_SPECIAL LOAD_SPECIAL #define _LOAD_SUPER_ATTR_ATTR LOAD_SUPER_ATTR_ATTR #define _LOAD_SUPER_ATTR_METHOD LOAD_SUPER_ATTR_METHOD -#define _MAKE_CALLARGS_A_TUPLE 437 +#define _MAKE_CALLARGS_A_TUPLE 438 #define _MAKE_CELL MAKE_CELL #define _MAKE_FUNCTION MAKE_FUNCTION -#define _MAKE_WARM 438 +#define _MAKE_WARM 439 #define _MAP_ADD MAP_ADD #define _MATCH_CLASS MATCH_CLASS #define _MATCH_KEYS MATCH_KEYS #define _MATCH_MAPPING MATCH_MAPPING #define _MATCH_SEQUENCE MATCH_SEQUENCE -#define _MAYBE_EXPAND_METHOD 439 -#define _MAYBE_EXPAND_METHOD_KW 440 -#define _MONITOR_CALL 441 -#define _MONITOR_CALL_KW 442 -#define _MONITOR_JUMP_BACKWARD 443 -#define _MONITOR_RESUME 444 +#define _MAYBE_EXPAND_METHOD 440 +#define _MAYBE_EXPAND_METHOD_KW 441 +#define _MONITOR_CALL 442 +#define _MONITOR_CALL_KW 443 +#define _MONITOR_JUMP_BACKWARD 444 +#define _MONITOR_RESUME 445 #define _NOP NOP #define _POP_EXCEPT POP_EXCEPT -#define _POP_JUMP_IF_FALSE 445 -#define _POP_JUMP_IF_TRUE 446 +#define _POP_JUMP_IF_FALSE 446 +#define _POP_JUMP_IF_TRUE 447 #define _POP_TOP POP_TOP -#define _POP_TOP_LOAD_CONST_INLINE 447 -#define _POP_TOP_LOAD_CONST_INLINE_BORROW 448 -#define _POP_TWO_LOAD_CONST_INLINE_BORROW 449 +#define _POP_TOP_LOAD_CONST_INLINE 448 +#define _POP_TOP_LOAD_CONST_INLINE_BORROW 449 +#define _POP_TWO_LOAD_CONST_INLINE_BORROW 450 #define _PUSH_EXC_INFO PUSH_EXC_INFO -#define _PUSH_FRAME 450 +#define _PUSH_FRAME 451 #define _PUSH_NULL PUSH_NULL -#define _PUSH_NULL_CONDITIONAL 451 -#define _PY_FRAME_GENERAL 452 -#define _PY_FRAME_KW 453 -#define _QUICKEN_RESUME 454 -#define _REPLACE_WITH_TRUE 455 +#define _PUSH_NULL_CONDITIONAL 452 +#define _PY_FRAME_GENERAL 453 +#define _PY_FRAME_KW 454 +#define _QUICKEN_RESUME 455 +#define _REPLACE_WITH_TRUE 456 #define _RESUME_CHECK RESUME_CHECK #define _RETURN_GENERATOR RETURN_GENERATOR #define _RETURN_VALUE RETURN_VALUE -#define _SAVE_RETURN_OFFSET 456 -#define _SEND 457 -#define _SEND_GEN_FRAME 458 +#define _SAVE_RETURN_OFFSET 457 +#define _SEND 458 +#define _SEND_GEN_FRAME 459 #define _SETUP_ANNOTATIONS SETUP_ANNOTATIONS #define _SET_ADD SET_ADD #define _SET_FUNCTION_ATTRIBUTE SET_FUNCTION_ATTRIBUTE #define _SET_UPDATE SET_UPDATE -#define _START_EXECUTOR 459 -#define _STORE_ATTR 460 -#define _STORE_ATTR_INSTANCE_VALUE 461 -#define _STORE_ATTR_SLOT 462 -#define _STORE_ATTR_WITH_HINT 463 +#define _START_EXECUTOR 460 +#define _STORE_ATTR 461 +#define _STORE_ATTR_INSTANCE_VALUE 462 +#define _STORE_ATTR_SLOT 463 +#define _STORE_ATTR_WITH_HINT 464 #define _STORE_DEREF STORE_DEREF -#define _STORE_FAST 464 -#define _STORE_FAST_0 465 -#define _STORE_FAST_1 466 -#define _STORE_FAST_2 467 -#define _STORE_FAST_3 468 -#define _STORE_FAST_4 469 -#define _STORE_FAST_5 470 -#define _STORE_FAST_6 471 -#define _STORE_FAST_7 472 +#define _STORE_FAST 465 +#define _STORE_FAST_0 466 +#define _STORE_FAST_1 467 +#define _STORE_FAST_2 468 +#define _STORE_FAST_3 469 +#define _STORE_FAST_4 470 +#define _STORE_FAST_5 471 +#define _STORE_FAST_6 472 +#define _STORE_FAST_7 473 #define _STORE_FAST_LOAD_FAST STORE_FAST_LOAD_FAST #define _STORE_FAST_STORE_FAST STORE_FAST_STORE_FAST #define _STORE_GLOBAL STORE_GLOBAL #define _STORE_NAME STORE_NAME -#define _STORE_SLICE 473 -#define _STORE_SUBSCR 474 +#define _STORE_SLICE 474 +#define _STORE_SUBSCR 475 #define _STORE_SUBSCR_DICT STORE_SUBSCR_DICT #define _STORE_SUBSCR_LIST_INT STORE_SUBSCR_LIST_INT #define _SWAP SWAP -#define _TIER2_RESUME_CHECK 475 -#define _TO_BOOL 476 +#define _TIER2_RESUME_CHECK 476 +#define _TO_BOOL 477 #define _TO_BOOL_BOOL TO_BOOL_BOOL #define _TO_BOOL_INT TO_BOOL_INT #define _TO_BOOL_LIST TO_BOOL_LIST #define _TO_BOOL_NONE TO_BOOL_NONE -#define _TO_BOOL_STR TO_BOOL_STR +#define _TO_BOOL_STR 478 #define _UNARY_INVERT UNARY_INVERT #define _UNARY_NEGATIVE UNARY_NEGATIVE #define _UNARY_NOT UNARY_NOT #define _UNPACK_EX UNPACK_EX -#define _UNPACK_SEQUENCE 477 +#define _UNPACK_SEQUENCE 479 #define _UNPACK_SEQUENCE_LIST UNPACK_SEQUENCE_LIST #define _UNPACK_SEQUENCE_TUPLE UNPACK_SEQUENCE_TUPLE #define _UNPACK_SEQUENCE_TWO_TUPLE UNPACK_SEQUENCE_TWO_TUPLE #define _WITH_EXCEPT_START WITH_EXCEPT_START #define _YIELD_VALUE YIELD_VALUE -#define MAX_UOP_ID 477 +#define MAX_UOP_ID 479 #ifdef __cplusplus } diff --git a/Include/internal/pycore_uop_metadata.h b/Include/internal/pycore_uop_metadata.h index 59572fca3753d6..4f5f6bbde2571a 100644 --- a/Include/internal/pycore_uop_metadata.h +++ b/Include/internal/pycore_uop_metadata.h @@ -64,7 +64,8 @@ const uint16_t _PyUop_Flags[MAX_UOP_ID+1] = { [_TO_BOOL_INT] = HAS_EXIT_FLAG | HAS_ESCAPES_FLAG, [_TO_BOOL_LIST] = HAS_EXIT_FLAG, [_TO_BOOL_NONE] = HAS_EXIT_FLAG, - [_TO_BOOL_STR] = HAS_EXIT_FLAG | HAS_ESCAPES_FLAG, + [_GUARD_TOS_UNICODE] = HAS_EXIT_FLAG, + [_TO_BOOL_STR] = HAS_ESCAPES_FLAG, [_REPLACE_WITH_TRUE] = HAS_ESCAPES_FLAG, [_UNARY_INVERT] = HAS_ERROR_FLAG | HAS_ESCAPES_FLAG, [_GUARD_BOTH_INT] = HAS_EXIT_FLAG, @@ -413,6 +414,7 @@ const char *const _PyOpcode_uop_name[MAX_UOP_ID+1] = { [_GUARD_NOT_EXHAUSTED_TUPLE] = "_GUARD_NOT_EXHAUSTED_TUPLE", [_GUARD_TOS_FLOAT] = "_GUARD_TOS_FLOAT", [_GUARD_TOS_INT] = "_GUARD_TOS_INT", + [_GUARD_TOS_UNICODE] = "_GUARD_TOS_UNICODE", [_GUARD_TYPE_VERSION] = "_GUARD_TYPE_VERSION", [_GUARD_TYPE_VERSION_AND_LOCK] = "_GUARD_TYPE_VERSION_AND_LOCK", [_IMPORT_FROM] = "_IMPORT_FROM", @@ -649,6 +651,8 @@ int _PyUop_num_popped(int opcode, int oparg) return 1; case _TO_BOOL_NONE: return 1; + case _GUARD_TOS_UNICODE: + return 0; case _TO_BOOL_STR: return 1; case _REPLACE_WITH_TRUE: diff --git a/Lib/test/test_capi/test_opt.py b/Lib/test/test_capi/test_opt.py index cc6b4c88fe0fce..feab3b8b84f566 100644 --- a/Lib/test/test_capi/test_opt.py +++ b/Lib/test/test_capi/test_opt.py @@ -1581,6 +1581,24 @@ def testfunc(n): self.assertNotIn("_COMPARE_OP_INT", uops) self.assertIn("_POP_TWO_LOAD_CONST_INLINE_BORROW", uops) + def test_remove_guard_for_known_type_str(self): + def f(n): + for i in range(n): + false = i == TIER2_THRESHOLD + empty = "X"[:false] + empty += "" # Make JIT realize this is a string. + if empty: + return 1 + return 0 + + res, ex = self._run_with_optimizer(f, TIER2_THRESHOLD) + self.assertEqual(res, 0) + self.assertIsNotNone(ex) + uops = get_opnames(ex) + self.assertIn("_TO_BOOL_STR", uops) + self.assertNotIn("_GUARD_TOS_UNICODE", uops) + + def global_identity(x): return x diff --git a/Misc/NEWS.d/next/Core_and_Builtins/2025-03-28-19-02-55.gh-issue-131798.fNZ5-2.rst b/Misc/NEWS.d/next/Core_and_Builtins/2025-03-28-19-02-55.gh-issue-131798.fNZ5-2.rst new file mode 100644 index 00000000000000..69c63ffd7fe986 --- /dev/null +++ b/Misc/NEWS.d/next/Core_and_Builtins/2025-03-28-19-02-55.gh-issue-131798.fNZ5-2.rst @@ -0,0 +1 @@ +Allow JIT to omit str guard in truthiness test when str type is known. diff --git a/Python/bytecodes.c b/Python/bytecodes.c index 63367141c6951b..2d18f658702fe7 100644 --- a/Python/bytecodes.c +++ b/Python/bytecodes.c @@ -512,10 +512,14 @@ dummy_func( res = PyStackRef_False; } - inst(TO_BOOL_STR, (unused/1, unused/2, value -- res)) { + op(_GUARD_TOS_UNICODE, (value -- value)) { PyObject *value_o = PyStackRef_AsPyObjectBorrow(value); EXIT_IF(!PyUnicode_CheckExact(value_o)); + } + + op(_TO_BOOL_STR, (value -- res)) { STAT_INC(TO_BOOL, hit); + PyObject *value_o = PyStackRef_AsPyObjectBorrow(value); if (value_o == &_Py_STR(empty)) { assert(_Py_IsImmortal(value_o)); DEAD(value); @@ -528,6 +532,9 @@ dummy_func( } } + macro(TO_BOOL_STR) = + _GUARD_TOS_UNICODE + unused/1 + unused/2 + _TO_BOOL_STR; + op(_REPLACE_WITH_TRUE, (value -- res)) { PyStackRef_CLOSE(value); res = PyStackRef_True; diff --git a/Python/executor_cases.c.h b/Python/executor_cases.c.h index bf21c7bfcef393..29c3a270a27526 100644 --- a/Python/executor_cases.c.h +++ b/Python/executor_cases.c.h @@ -653,16 +653,23 @@ break; } - case _TO_BOOL_STR: { + case _GUARD_TOS_UNICODE: { _PyStackRef value; - _PyStackRef res; value = stack_pointer[-1]; PyObject *value_o = PyStackRef_AsPyObjectBorrow(value); if (!PyUnicode_CheckExact(value_o)) { UOP_STAT_INC(uopcode, miss); JUMP_TO_JUMP_TARGET(); } + break; + } + + case _TO_BOOL_STR: { + _PyStackRef value; + _PyStackRef res; + value = stack_pointer[-1]; STAT_INC(TO_BOOL, hit); + PyObject *value_o = PyStackRef_AsPyObjectBorrow(value); if (value_o == &_Py_STR(empty)) { assert(_Py_IsImmortal(value_o)); res = PyStackRef_False; diff --git a/Python/generated_cases.c.h b/Python/generated_cases.c.h index 4e37c68e2bd8bc..416035ae29d71c 100644 --- a/Python/generated_cases.c.h +++ b/Python/generated_cases.c.h @@ -11616,30 +11616,37 @@ static_assert(INLINE_CACHE_ENTRIES_TO_BOOL == 3, "incorrect cache size"); _PyStackRef value; _PyStackRef res; + // _GUARD_TOS_UNICODE + { + value = stack_pointer[-1]; + PyObject *value_o = PyStackRef_AsPyObjectBorrow(value); + if (!PyUnicode_CheckExact(value_o)) { + UPDATE_MISS_STATS(TO_BOOL); + assert(_PyOpcode_Deopt[opcode] == (TO_BOOL)); + JUMP_TO_PREDICTED(TO_BOOL); + } + } /* Skip 1 cache entry */ /* Skip 2 cache entries */ - value = stack_pointer[-1]; - PyObject *value_o = PyStackRef_AsPyObjectBorrow(value); - if (!PyUnicode_CheckExact(value_o)) { - UPDATE_MISS_STATS(TO_BOOL); - assert(_PyOpcode_Deopt[opcode] == (TO_BOOL)); - JUMP_TO_PREDICTED(TO_BOOL); - } - STAT_INC(TO_BOOL, hit); - if (value_o == &_Py_STR(empty)) { - assert(_Py_IsImmortal(value_o)); - res = PyStackRef_False; - } - else { - assert(Py_SIZE(value_o)); - stack_pointer += -1; - assert(WITHIN_STACK_BOUNDS()); - _PyFrame_SetStackPointer(frame, stack_pointer); - PyStackRef_CLOSE(value); - stack_pointer = _PyFrame_GetStackPointer(frame); - res = PyStackRef_True; - stack_pointer += 1; - assert(WITHIN_STACK_BOUNDS()); + // _TO_BOOL_STR + { + STAT_INC(TO_BOOL, hit); + PyObject *value_o = PyStackRef_AsPyObjectBorrow(value); + if (value_o == &_Py_STR(empty)) { + assert(_Py_IsImmortal(value_o)); + res = PyStackRef_False; + } + else { + assert(Py_SIZE(value_o)); + stack_pointer += -1; + assert(WITHIN_STACK_BOUNDS()); + _PyFrame_SetStackPointer(frame, stack_pointer); + PyStackRef_CLOSE(value); + stack_pointer = _PyFrame_GetStackPointer(frame); + res = PyStackRef_True; + stack_pointer += 1; + assert(WITHIN_STACK_BOUNDS()); + } } stack_pointer[-1] = res; DISPATCH(); diff --git a/Python/optimizer_bytecodes.c b/Python/optimizer_bytecodes.c index 2ccfd0a152f547..8d9e94a2445182 100644 --- a/Python/optimizer_bytecodes.c +++ b/Python/optimizer_bytecodes.c @@ -414,10 +414,16 @@ dummy_func(void) { } } + op(_GUARD_TOS_UNICODE, (value -- value)) { + if (sym_matches_type(value, &PyUnicode_Type)) { + REPLACE_OP(this_instr, _NOP, 0, 0); + } + sym_set_type(value, &PyUnicode_Type); + } + op(_TO_BOOL_STR, (value -- res)) { if (!optimize_to_bool(this_instr, ctx, value, &res)) { res = sym_new_truthiness(ctx, value, true); - sym_set_type(value, &PyUnicode_Type); } } diff --git a/Python/optimizer_cases.c.h b/Python/optimizer_cases.c.h index 85fa8f4a2a7910..6a8ac75b63eb0e 100644 --- a/Python/optimizer_cases.c.h +++ b/Python/optimizer_cases.c.h @@ -208,13 +208,22 @@ break; } + case _GUARD_TOS_UNICODE: { + JitOptSymbol *value; + value = stack_pointer[-1]; + if (sym_matches_type(value, &PyUnicode_Type)) { + REPLACE_OP(this_instr, _NOP, 0, 0); + } + sym_set_type(value, &PyUnicode_Type); + break; + } + case _TO_BOOL_STR: { JitOptSymbol *value; JitOptSymbol *res; value = stack_pointer[-1]; if (!optimize_to_bool(this_instr, ctx, value, &res)) { res = sym_new_truthiness(ctx, value, true); - sym_set_type(value, &PyUnicode_Type); } stack_pointer[-1] = res; break; _______________________________________________ Python-checkins mailing list -- python-checkins@python.org To unsubscribe send an email to python-checkins-le...@python.org https://mail.python.org/mailman3/lists/python-checkins.python.org/ Member address: arch...@mail-archive.com