Fix bad stack assert in the JIT optimizer

Fidget-Spinner · Fidget-Spinner · commit ee4548c020b3 · 2026-01-19T01:32:14.000Z
diff --git a/Python/optimizer_analysis.c b/Python/optimizer_analysis.c
@@ -468,7 +468,7 @@ optimize_uops(
             ctx->out_buffer[ctx->out_len++] = *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
`@@ -468,7 +468,7 @@ optimize_uops(`
`468`	`468`	`ctx->out_buffer[ctx->out_len++] = *this_instr;`
`469`	`469`	`}`
`470`	`470`	`assert(ctx->frame != NULL);`
`471`		`- if (!CURRENT_FRAME_IS_INIT_SHIM()) {`
	`471`	`+ if (!CURRENT_FRAME_IS_INIT_SHIM() && !ctx->done) {`
`472`	`472`	`DPRINTF(3, " stack_level %d\n", STACK_LEVEL());`
`473`	`473`	`ctx->frame->stack_pointer = stack_pointer;`
`474`	`474`	`assert(STACK_LEVEL() >= 0);`