Ifs to stim

IfElseSimplification dataclass

IfElseSimplification()

contains_ifelse

contains_ifelse(stmt: IfElse) -> bool

Check if the IfElse statement contains another IfElse statement.

Source code in .venv/lib/python3.12/site-packages/bloqade/stim/rewrite/ifs_to_stim.py

def contains_ifelse(self, stmt: scf.IfElse) -> bool:
    """Check if the IfElse statement contains another IfElse statement."""
    for child in stmt.walk(include_self=False):
        if isinstance(child, scf.IfElse):
            return True
    return False

has_else_body

has_else_body(stmt: IfElse) -> bool

Check if the IfElse statement has an else body.

Source code in .venv/lib/python3.12/site-packages/bloqade/stim/rewrite/ifs_to_stim.py

def has_else_body(self, stmt: scf.IfElse) -> bool:
    """Check if the IfElse statement has an else body."""
    if stmt.else_body.blocks and not (
        len(stmt.else_body.blocks[0].stmts) == 1
        and isinstance(else_term := stmt.else_body.blocks[0].last_stmt, scf.Yield)
        and not else_term.values  # empty yield
    ):
        return True

    return False

is_nested_ifelse

is_nested_ifelse(stmt: IfElse) -> bool

Check if the IfElse statement is nested.

Source code in .venv/lib/python3.12/site-packages/bloqade/stim/rewrite/ifs_to_stim.py

def is_nested_ifelse(self, stmt: scf.IfElse) -> bool:
    """Check if the IfElse statement is nested."""
    if stmt.parent_stmt is not None:
        if isinstance(stmt.parent_stmt, scf.IfElse) or isinstance(
            stmt.parent_stmt.parent_stmt, scf.IfElse
        ):
            return True
        else:
            return False
    else:
        return False

IfToStim dataclass

IfToStim(measure_frame: MeasureIDFrame)

Bases: IfElseSimplification, RewriteRule

Rewrite if statements to stim equivalent statements.

StimSplitIfStmts dataclass

StimSplitIfStmts()

Bases: IfElseSimplification, SplitIfStmts

Splits the then body of an if-else statement into multiple if statements

Given an IfElse with multiple valid statements in the then-body:

if measure_result

squin.qubit.apply(op.X, q0) squin.qubit.apply(op.Y, q1)

this should be rewritten to:

if measure_result

squin.qubit.apply(op.X, q0)

if measure_result

squin.qubit.apply(op.Y, q1)