Jul 8, 2025 · Jun 19, 2025 · Jun 19, 2025 · Jul 7, 2025 · Jul 7, 2025 · Jul 7, 2025
diff --git a/mypyc/irbuild/generator.py b/mypyc/irbuild/generator.py
    Unreachable,
    Value,
 )
 from mypyc.ir.rtypes import RInstance, int32_rprimitive, object_rprimitive, object_pointer_rprimitive
 from mypyc.ir.rtypes import (
    RInstance,
    int32_rprimitive,
    object_pointer_rprimitive,
    object_rprimitive,
 )
 from mypyc.irbuild.builder import IRBuilder, calculate_arg_defaults, gen_arg_defaults
 from mypyc.irbuild.context import FuncInfo, GeneratorClass
 from mypyc.irbuild.env_class import (
        result = builder.add(
            Call(
                fn_decl,
                [builder.self(), none_reg, none_reg, none_reg, none_reg, Integer(0, object_pointer_rprimitive)],
                [
                    builder.self(),
                    none_reg,
                    none_reg,
                    none_reg,
                    none_reg,
                    Integer(0, object_pointer_rprimitive),
                ],
                fn_info.fitem.line,
            )
        )
        result = builder.add(
            Call(
                fn_decl,
                [builder.self(), none_reg, none_reg, none_reg, builder.read(arg), Integer(0, object_pointer_rprimitive)],
                [
                    builder.self(),
                    none_reg,
                    none_reg,
                    none_reg,
                    builder.read(arg),
                    Integer(0, object_pointer_rprimitive),
                ],
                fn_info.fitem.line,
            )
        )
        result = builder.add(
            Call(
                fn_decl,
                [builder.self(), builder.read(typ), builder.read(val), builder.read(tb), none_reg, Integer(0, object_pointer_rprimitive)],
                [
                    builder.self(),
                    builder.read(typ),
                    builder.read(val),
                    builder.read(tb),
                    none_reg,
                    Integer(0, object_pointer_rprimitive),
                ],
                fn_info.fitem.line,
            )
        )

    # Parameter that can used to pass a pointer which can used instead of
    # raising StopIteration(value). If the value is NULL, this won't be used.
    stop_iter_value_arg = builder.add_local(Var("stop_iter_ptr"), object_pointer_rprimitive, is_arg=True)
    stop_iter_value_arg = builder.add_local(
        Var("stop_iter_ptr"), object_pointer_rprimitive, is_arg=True
    )

    cls.exc_regs = (exc_type, exc_val, exc_tb)
    cls.send_arg_reg = exc_arg
diff --git a/mypyc/irbuild/nonlocalcontrol.py b/mypyc/irbuild/nonlocalcontrol.py
    Branch,
    Goto,
    Integer,
    LoadErrorValue,
    Register,
    Return,
    SetMem,
        builder.add(SetMem(object_rprimitive, stop_iter_reg, value))
        builder.add(Return(Integer(0, object_rprimitive)))


 class CleanupNonlocalControl(NonlocalControl):
    """Abstract nonlocal control that runs some cleanup code."""

diff --git a/mypyc/irbuild/prepare.py b/mypyc/irbuild/prepare.py
    RType,
    dict_rprimitive,
    none_rprimitive,
    object_pointer_rprimitive,
    object_rprimitive,
    tuple_rprimitive,
 )
 )
 from mypyc.options import CompilerOptions
 from mypyc.sametype import is_same_type
 from mypyc.ir.rtypes import object_rprimitive, object_pointer_rprimitive


 def build_type_map(
Original file line number	Diff line number	Diff line change
Expand Up		@@ -32,7 +32,12 @@
		Unreachable,
		Value,
		)
		from mypyc.ir.rtypes import RInstance, int32_rprimitive, object_rprimitive, object_pointer_rprimitive
		from mypyc.ir.rtypes import (
		RInstance,
		int32_rprimitive,
		object_pointer_rprimitive,
		object_rprimitive,
		)
		from mypyc.irbuild.builder import IRBuilder, calculate_arg_defaults, gen_arg_defaults
		from mypyc.irbuild.context import FuncInfo, GeneratorClass
		from mypyc.irbuild.env_class import (
Expand DownExpand Up		@@ -256,7 +261,14 @@ def add_next_to_generator_class(builder: IRBuilder, fn_info: FuncInfo, fn_decl:
		result = builder.add(
		Call(
		fn_decl,
		[builder.self(), none_reg, none_reg, none_reg, none_reg, Integer(0, object_pointer_rprimitive)],
		[
		builder.self(),
		none_reg,
		none_reg,
		none_reg,
		none_reg,
		Integer(0, object_pointer_rprimitive),
		],
		fn_info.fitem.line,
		)
		)
Expand All		@@ -272,7 +284,14 @@ def add_send_to_generator_class(builder: IRBuilder, fn_info: FuncInfo, fn_decl:
		result = builder.add(
		Call(
		fn_decl,
		[builder.self(), none_reg, none_reg, none_reg, builder.read(arg), Integer(0, object_pointer_rprimitive)],
		[
		builder.self(),
		none_reg,
		none_reg,
		none_reg,
		builder.read(arg),
		Integer(0, object_pointer_rprimitive),
		],
		fn_info.fitem.line,
		)
		)
Expand All		@@ -297,7 +316,14 @@ def add_throw_to_generator_class(builder: IRBuilder, fn_info: FuncInfo, fn_decl:
		result = builder.add(
		Call(
		fn_decl,
		[builder.self(), builder.read(typ), builder.read(val), builder.read(tb), none_reg, Integer(0, object_pointer_rprimitive)],
		[
		builder.self(),
		builder.read(typ),
		builder.read(val),
		builder.read(tb),
		none_reg,
		Integer(0, object_pointer_rprimitive),
		],
		fn_info.fitem.line,
		)
		)
Expand DownExpand Up		@@ -379,7 +405,9 @@ def setup_env_for_generator_class(builder: IRBuilder) -> None:

		# Parameter that can used to pass a pointer which can used instead of
		# raising StopIteration(value). If the value is NULL, this won't be used.
		stop_iter_value_arg = builder.add_local(Var("stop_iter_ptr"), object_pointer_rprimitive, is_arg=True)
		stop_iter_value_arg = builder.add_local(
		Var("stop_iter_ptr"), object_pointer_rprimitive, is_arg=True
		)

		cls.exc_regs = (exc_type, exc_val, exc_tb)
		cls.send_arg_reg = exc_arg
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -14,7 +14,6 @@
		Branch,
		Goto,
		Integer,
		LoadErrorValue,
		Register,
		Return,
		SetMem,
Expand DownExpand Up		@@ -131,6 +130,7 @@ def gen_return(self, builder: IRBuilder, value: Value, line: int) -> None:
		builder.add(SetMem(object_rprimitive, stop_iter_reg, value))
		builder.add(Return(Integer(0, object_rprimitive)))


		class CleanupNonlocalControl(NonlocalControl):
		"""Abstract nonlocal control that runs some cleanup code."""

Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -56,6 +56,7 @@
		RType,
		dict_rprimitive,
		none_rprimitive,
		object_pointer_rprimitive,
		object_rprimitive,
		tuple_rprimitive,
		)
Expand All		@@ -69,7 +70,6 @@
		)
		from mypyc.options import CompilerOptions
		from mypyc.sametype import is_same_type
		from mypyc.ir.rtypes import object_rprimitive, object_pointer_rprimitive


		def build_type_map(
Expand Down