diff options
| author | Irit Katriel <1055913+iritkatriel@users.noreply.github.com> | 2025-05-05 16:46:56 (GMT) |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-05-05 16:46:56 (GMT) |
| commit | 082dbf77884264a8470b1ea132cc458d0b5bf08a (patch) | |
| tree | 3d441ff384d05c127bc67c3a9e4a8df8c8a2f6ad /Python/generated_cases.c.h | |
| parent | 07f416a3f063db6b91b8b99ff61a51b64b0503f1 (diff) | |
| download | cpython-082dbf77884264a8470b1ea132cc458d0b5bf08a.zip cpython-082dbf77884264a8470b1ea132cc458d0b5bf08a.tar.gz cpython-082dbf77884264a8470b1ea132cc458d0b5bf08a.tar.bz2 | |
gh-133395: add option for extension modules to specialize BINARY_OP/SUBSCR, apply to arrays (#133396)
Diffstat (limited to 'Python/generated_cases.c.h')
| -rw-r--r-- | Python/generated_cases.c.h | 29 |
1 files changed, 24 insertions, 5 deletions
diff --git a/Python/generated_cases.c.h b/Python/generated_cases.c.h index ee54b38..072951d 100644 --- a/Python/generated_cases.c.h +++ b/Python/generated_cases.c.h @@ -283,14 +283,30 @@ PyObject *right_o = PyStackRef_AsPyObjectBorrow(right); _PyBinaryOpSpecializationDescr *d = (_PyBinaryOpSpecializationDescr*)descr; assert(INLINE_CACHE_ENTRIES_BINARY_OP == 5); - assert(d && d->guard); + assert(d); + assert(d->guard); _PyFrame_SetStackPointer(frame, stack_pointer); int res = d->guard(left_o, right_o); stack_pointer = _PyFrame_GetStackPointer(frame); - if (!res) { - UPDATE_MISS_STATS(BINARY_OP); - assert(_PyOpcode_Deopt[opcode] == (BINARY_OP)); - JUMP_TO_PREDICTED(BINARY_OP); + if (res < 0) { + JUMP_TO_LABEL(error); + } + if (res == 0) { + if (d->free) { + _PyFrame_SetStackPointer(frame, stack_pointer); + d->free(d); + stack_pointer = _PyFrame_GetStackPointer(frame); + } + _PyBinaryOpCache *cache = (_PyBinaryOpCache *)(this_instr+1); + _PyFrame_SetStackPointer(frame, stack_pointer); + write_ptr(cache->external_cache, NULL); + stack_pointer = _PyFrame_GetStackPointer(frame); + this_instr->op.code = BINARY_OP; + if (true) { + UPDATE_MISS_STATS(BINARY_OP); + assert(_PyOpcode_Deopt[opcode] == (BINARY_OP)); + JUMP_TO_PREDICTED(BINARY_OP); + } } } /* Skip -4 cache entry */ @@ -315,6 +331,9 @@ stack_pointer = _PyFrame_GetStackPointer(frame); stack_pointer += -2; assert(WITHIN_STACK_BOUNDS()); + if (res_o == NULL) { + JUMP_TO_LABEL(error); + } res = PyStackRef_FromPyObjectSteal(res_o); } stack_pointer[0] = res; |