up last_profiled_frame

Author: maurycy

Modules/_testinternalcapi/test_cases.c.h

@@ -681,6 +681,9 @@ _gen_throw(PyGenObject *gen, int close_on_genexit,
                'yield from' or awaiting on with 'await'. */
             ret = _gen_throw((PyGenObject *)yf, close_on_genexit,
                              typ, val, tb);
+            if (tstate->last_profiled_frame == frame) {
+                tstate->last_profiled_frame = prev;
+            }
             tstate->current_frame = prev;
             frame->previous = NULL;
         }
@@ -701,6 +704,9 @@ _gen_throw(PyGenObject *gen, int close_on_genexit,
             frame->previous = prev;
             tstate->current_frame = frame;
             ret = PyObject_CallFunctionObjArgs(meth, typ, val, tb, NULL);
+            if (tstate->last_profiled_frame == frame) {
+                tstate->last_profiled_frame = prev;
+            }
             tstate->current_frame = prev;
             frame->previous = NULL;
             Py_DECREF(meth);

Objects/genobject.c

@@ -1860,6 +1860,9 @@ dummy_func(
             gen->gi_exc_state.previous_item = NULL;
             _Py_LeaveRecursiveCallPy(tstate);
             _PyInterpreterFrame *gen_frame = frame;
+            if (tstate->last_profiled_frame == gen_frame) {
+                tstate->last_profiled_frame = gen_frame->previous;
+            }
             frame = tstate->current_frame = frame->previous;
             gen_frame->previous = NULL;
             ((_PyThreadStateImpl *)tstate)->generator_return_kind = GENERATOR_YIELD;
@@ -5874,6 +5877,9 @@ dummy_func(
             gen_frame->owner = FRAME_OWNED_BY_GENERATOR;
             _Py_LeaveRecursiveCallPy(tstate);
             _PyInterpreterFrame *prev = frame->previous;
+            if (tstate->last_profiled_frame == frame) {
+                tstate->last_profiled_frame = prev;
+            }
             _PyThreadState_PopFrame(tstate, frame);
             frame = tstate->current_frame = prev;
             LOAD_IP(frame->return_offset);

Python/bytecodes.c

@@ -2006,6 +2006,9 @@ _PyEvalFramePushAndInit(PyThreadState *tstate, _PyStackRef func,
     _PyFrame_Initialize(tstate, frame, func, locals, code, 0, previous);
     if (initialize_locals(tstate, func_obj, frame->localsplus, args, argcount, kwnames)) {
         assert(frame->owner == FRAME_OWNED_BY_THREAD);
+        if (tstate->last_profiled_frame == frame) {
+            tstate->last_profiled_frame = tstate->current_frame;
+        }
         clear_thread_frame(tstate, frame);
         return NULL;
     }