diff --git a/mypyc/codegen/emit.py b/mypyc/codegen/emit.py index 56ce9637307b..a2e3d9849dca 100644 --- a/mypyc/codegen/emit.py +++ b/mypyc/codegen/emit.py @@ -229,6 +229,9 @@ def emit_label(self, label: BasicBlock | str) -> None: if isinstance(label, str): text = label else: + if label.label == 0 or not label.referenced: + return + text = self.label(label) # Extra semicolon prevents an error when the next line declares a tempvar self.fragments.append(f"{text}: ;\n") diff --git a/mypyc/codegen/emitfunc.py b/mypyc/codegen/emitfunc.py index f2406ff1a257..0dd8efac640f 100644 --- a/mypyc/codegen/emitfunc.py +++ b/mypyc/codegen/emitfunc.py @@ -23,6 +23,7 @@ CallC, Cast, ComparisonOp, + ControlOp, DecRef, Extend, Float, @@ -123,6 +124,28 @@ def generate_native_function( for i, block in enumerate(blocks): block.label = i + # Find blocks that are never jumped to or are only jumped to from the + # block directly above it. This allows for more labels and gotos to be + # eliminated during code generation. + for block in fn.blocks: + terminator = block.terminator + assert isinstance(terminator, ControlOp) + + for target in terminator.targets(): + is_next_block = target.label == block.label + 1 + + # Always emit labels for GetAttr error checks since the emit code that + # generates them will add instructions between the branch and the + # next label, causing the label to be wrongly removed. A better + # solution would be to change the IR so that it adds a basic block + # inbetween the calls. + is_problematic_op = isinstance(terminator, Branch) and any( + isinstance(s, GetAttr) for s in terminator.sources() + ) + + if not is_next_block or is_problematic_op: + fn.blocks[target.label].referenced = True + common = frequently_executed_blocks(fn.blocks[0]) for i in range(len(blocks)): @@ -216,7 +239,8 @@ def visit_branch(self, op: Branch) -> None: if false is self.next_block: if op.traceback_entry is None: - self.emit_line(f"if ({cond}) goto {self.label(true)};") + if true is not self.next_block: + self.emit_line(f"if ({cond}) goto {self.label(true)};") else: self.emit_line(f"if ({cond}) {{") self.emit_traceback(op) @@ -224,9 +248,11 @@ def visit_branch(self, op: Branch) -> None: else: self.emit_line(f"if ({cond}) {{") self.emit_traceback(op) - self.emit_lines( - "goto %s;" % self.label(true), "} else", " goto %s;" % self.label(false) - ) + + if true is not self.next_block: + self.emit_line("goto %s;" % self.label(true)) + + self.emit_lines("} else", " goto %s;" % self.label(false)) def visit_return(self, op: Return) -> None: value_str = self.reg(op.value) diff --git a/mypyc/ir/ops.py b/mypyc/ir/ops.py index 351f7c01efe2..6007f8a4ce04 100644 --- a/mypyc/ir/ops.py +++ b/mypyc/ir/ops.py @@ -81,6 +81,7 @@ def __init__(self, label: int = -1) -> None: self.label = label self.ops: list[Op] = [] self.error_handler: BasicBlock | None = None + self.referenced = False @property def terminated(self) -> bool: diff --git a/mypyc/test/test_emitfunc.py b/mypyc/test/test_emitfunc.py index d7dcf3be532b..ab1586bb22a8 100644 --- a/mypyc/test/test_emitfunc.py +++ b/mypyc/test/test_emitfunc.py @@ -894,12 +894,7 @@ def test_simple(self) -> None: generate_native_function(fn, emitter, "prog.py", "prog") result = emitter.fragments assert_string_arrays_equal( - [ - "CPyTagged CPyDef_myfunc(CPyTagged cpy_r_arg) {\n", - "CPyL0: ;\n", - " return cpy_r_arg;\n", - "}\n", - ], + ["CPyTagged CPyDef_myfunc(CPyTagged cpy_r_arg) {\n", " return cpy_r_arg;\n", "}\n"], result, msg="Generated code invalid", ) @@ -922,7 +917,6 @@ def test_register(self) -> None: [ "PyObject *CPyDef_myfunc(CPyTagged cpy_r_arg) {\n", " CPyTagged cpy_r_r0;\n", - "CPyL0: ;\n", " cpy_r_r0 = 10;\n", " CPy_Unreachable();\n", "}\n",