diff options
| author | Irit Katriel <1055913+iritkatriel@users.noreply.github.com> | 2023-11-03 10:01:36 (GMT) |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-03 10:01:36 (GMT) |
| commit | d49aba5a7a3c695213810a9f82715809c6332df2 (patch) | |
| tree | f949b9af90ff9d697ed2836efaea5b3b3e06e34c /Python/bytecodes.c | |
| parent | 7810b6981ac663b77bc9ee9dc4b1960ec6845ea7 (diff) | |
| download | cpython-d49aba5a7a3c695213810a9f82715809c6332df2.zip cpython-d49aba5a7a3c695213810a9f82715809c6332df2.tar.gz cpython-d49aba5a7a3c695213810a9f82715809c6332df2.tar.bz2 | |
gh-111354: Simplify _PyGen_yf by moving some of its work to the compiler and frame state (#111648)
Diffstat (limited to 'Python/bytecodes.c')
| -rw-r--r-- | Python/bytecodes.c | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/Python/bytecodes.c b/Python/bytecodes.c index ddaae32..f487e95 100644 --- a/Python/bytecodes.c +++ b/Python/bytecodes.c @@ -1049,7 +1049,9 @@ dummy_func( assert(frame != &entry_frame); frame->instr_ptr = next_instr; PyGenObject *gen = _PyFrame_GetGenerator(frame); - gen->gi_frame_state = FRAME_SUSPENDED; + assert(FRAME_SUSPENDED_YIELD_FROM == FRAME_SUSPENDED + 1); + assert(oparg == 0 || oparg == 1); + gen->gi_frame_state = FRAME_SUSPENDED + oparg; _PyFrame_SetStackPointer(frame, stack_pointer - 1); int err = _Py_call_instrumentation_arg( tstate, PY_MONITORING_EVENT_PY_YIELD, @@ -1075,7 +1077,9 @@ dummy_func( assert(frame != &entry_frame); frame->instr_ptr = next_instr; PyGenObject *gen = _PyFrame_GetGenerator(frame); - gen->gi_frame_state = FRAME_SUSPENDED; + assert(FRAME_SUSPENDED_YIELD_FROM == FRAME_SUSPENDED + 1); + assert(oparg == 0 || oparg == 1); + gen->gi_frame_state = FRAME_SUSPENDED + oparg; _PyFrame_SetStackPointer(frame, stack_pointer - 1); tstate->exc_info = gen->gi_exc_state.previous_item; gen->gi_exc_state.previous_item = NULL; |