gh-101928: fix crash in compiler on multi-line lambda in function call (#101933)

diff --git a/Lib/test/test_compile.py b/Lib/test/test_compile.py
index 90b067bcf3091276c606faf765283f86b53c45fa..a77742c0cfa6fcb3af0add9cd54f48881de0dc48 100644 (file)
--- a/Lib/test/test_compile.py
+++ b/Lib/test/test_compile.py
@@ -1155,6 +1155,17 @@ if 1:
             with self.subTest(expr=expr):
                 compile(expr, "<single>", "exec")
 
+    def test_multi_line_lambda_as_argument(self):
+        # See gh-101928
+        compile("""
+def foo(param, lambda_exp):
+    pass
+
+foo(param=0,
+    lambda_exp=lambda:
+    1)
+        """, "<test>", "exec")
+
 
 @requires_debug_ranges()
 class TestSourcePositions(unittest.TestCase):

diff --git a/Python/compile.c b/Python/compile.c
index 0534b536e3d12eafc0d8ef9f72bca93da00933b8..c3b344c7af2a7f3f0c79906aa48fa41174fc6fa4 100644 (file)
--- a/Python/compile.c
+++ b/Python/compile.c
@@ -9085,8 +9085,8 @@ optimize_basic_block(PyObject *const_cache, basicblock *bb, PyObject *consts)
                         Py_DECREF(cnt);
                         break;
                     case RETURN_VALUE:
-                        INSTR_SET_OP1(inst, RETURN_CONST, oparg);
-                        INSTR_SET_OP0(&bb->b_instr[i + 1], NOP);
+                        INSTR_SET_OP0(inst, NOP);
+                        INSTR_SET_OP1(&bb->b_instr[++i], RETURN_CONST, oparg);
                         break;
                 }
                 break;