GH-126892: Reset warmup counters when JIT compiling code (GH-126893)

1 files changed, 8 insertions, 5 deletions

diff --git a/Python/generated_cases.c.h b/Python/generated_cases.c.h
index 15308d6..4d89753 100644
--- a/Python/generated_cases.c.h
+++ b/Python/generated_cases.c.h
@@ -5092,15 +5092,18 @@
                     _PyFrame_SetStackPointer(frame, stack_pointer);
                     int optimized = _PyOptimizer_Optimize(frame, start, stack_pointer, &executor, 0);
                     stack_pointer = _PyFrame_GetStackPointer(frame);
-                    if (optimized < 0) goto error;
-                    if (optimized) {
+                    if (optimized <= 0) {
+                        this_instr[1].counter = restart_backoff_counter(counter);
+                        if (optimized < 0) goto error;
+                    }
+                    else {
+                        _PyFrame_SetStackPointer(frame, stack_pointer);
+                        this_instr[1].counter = initial_jump_backoff_counter();
+                        stack_pointer = _PyFrame_GetStackPointer(frame);
                         assert(tstate->previous_executor == NULL);
                         tstate->previous_executor = Py_None;
                         GOTO_TIER_TWO(executor);
                     }
-                    else {
-                        this_instr[1].counter = restart_backoff_counter(counter);
-                    }
                 }
                 else {
                     ADVANCE_ADAPTIVE_COUNTER(this_instr[1].counter);