gh-144016: Fix bad stack assert in the JIT optimizer (GH-144019)

Fidget-Spinner · web-flow · commit ca99bfdefb70 · 2026-01-24T09:36:40.000Z
diff --git a/Python/optimizer_analysis.c b/Python/optimizer_analysis.c
@@ -507,7 +507,7 @@ optimize_uops(
             *(ctx->out_buffer.next++) = *this_instr;
         }
         assert(ctx->frame != NULL);
-        if (!CURRENT_FRAME_IS_INIT_SHIM()) {
+        if (!CURRENT_FRAME_IS_INIT_SHIM() && !ctx->done) {
             DPRINTF(3, " stack_level %d\n", STACK_LEVEL());
             ctx->frame->stack_pointer = stack_pointer;
             assert(STACK_LEVEL() >= 0);

Original file line number	Diff line number	Diff line change
`@@ -507,7 +507,7 @@ optimize_uops(`
`507`	`507`	`(ctx->out_buffer.next++) = this_instr;`
`508`	`508`	`}`
`509`	`509`	`assert(ctx->frame != NULL);`
`510`		`- if (!CURRENT_FRAME_IS_INIT_SHIM()) {`
	`510`	`+ if (!CURRENT_FRAME_IS_INIT_SHIM() && !ctx->done) {`
`511`	`511`	`DPRINTF(3, " stack_level %d\n", STACK_LEVEL());`
`512`	`512`	`ctx->frame->stack_pointer = stack_pointer;`
`513`	`513`	`assert(STACK_LEVEL() >= 0);`