bloqade-circuit 0.7.7__tar.gz → 0.7.9__tar.gz

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/.github/workflows/ci.yml

@@ -30,7 +30,7 @@ jobs:
       - name: Set up Python ${{ matrix.python-version }}
         run: uv python install ${{ matrix.python-version }}
       - name: Install the project
-        run: uv sync --dev --all-extras --no-extra pyqrack-cuda
+        run: uv sync --dev --all-extras --no-extra pyqrack-cuda --no-extra pyqrack-opencl
       - name: Run tests
         # For example, using `pytest`
         run: uv run just coverage

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bloqade-circuit
-Version: 0.7.7
+Version: 0.7.9
 Summary: The software development toolkit for neutral atom arrays.
 Author-email: Roger-luo <rluo@quera.com>, kaihsin <khwu@quera.com>, weinbe58 <pweinberg@quera.com>, johnzl-777 <jlong@quera.com>
 License-File: LICENSE
@@ -9,8 +9,7 @@ Requires-Dist: kirin-toolchain~=0.17.23
 Requires-Dist: numpy>=1.22.0
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: pydantic<2.11.0,>=1.3.0
-Requires-Dist: pyqrack-cpu>=1.69.0; sys_platform != 'darwin-arm64'
-Requires-Dist: pyqrack>=1.69.0; sys_platform == 'darwin-arm64'
+Requires-Dist: pyqrack-cpu>=1.69.1
 Requires-Dist: rich>=13.9.4
 Requires-Dist: scipy>=1.13.1
 Provides-Extra: cirq
@@ -18,9 +17,9 @@ Requires-Dist: cirq-core>=1.4.1; extra == 'cirq'
 Requires-Dist: cirq-core[contrib]>=1.4.1; extra == 'cirq'
 Requires-Dist: qpsolvers[clarabel]>=4.7.0; extra == 'cirq'
 Provides-Extra: pyqrack-cuda
-Requires-Dist: pyqrack-cuda>=1.69.0; extra == 'pyqrack-cuda'
+Requires-Dist: pyqrack-cuda>=1.69.1; extra == 'pyqrack-cuda'
 Provides-Extra: pyqrack-opencl
-Requires-Dist: pyqrack>=1.69.0; (sys_platform == 'darwin-arm64') and extra == 'pyqrack-opencl'
+Requires-Dist: pyqrack>=1.69.1; extra == 'pyqrack-opencl'
 Provides-Extra: qasm2
 Requires-Dist: lark>=1.2.2; extra == 'qasm2'
 Provides-Extra: qbraid

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bloqade-circuit"
-version = "0.7.7"
+version = "0.7.9"
 description = "The software development toolkit for neutral atom arrays."
 readme = "README.md"
 authors = [
@@ -17,8 +17,7 @@ dependencies = [
     "rich>=13.9.4",
     "pydantic>=1.3.0,<2.11.0",
     "pandas>=2.2.3",
-    "pyqrack>=1.69.0; sys_platform == 'darwin-arm64'",
-    "pyqrack-cpu>=1.69.0; sys_platform != 'darwin-arm64'",
+    "pyqrack-cpu>=1.69.1",
 ]
 
 [project.optional-dependencies]
@@ -41,10 +40,10 @@ cirq = [
     "qpsolvers[clarabel]>=4.7.0",
 ]
 pyqrack-opencl = [
-    "pyqrack>=1.69.0; sys_platform == 'darwin-arm64'",
+    "pyqrack>=1.69.1",
 ]
 pyqrack-cuda = [
-    "pyqrack-cuda>=1.69.0",
+    "pyqrack-cuda>=1.69.1",
 ]
 stim = [
     "stim>=1.15.0",

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/src/bloqade/cirq_utils/__init__.py

@@ -2,7 +2,7 @@ from . import noise as noise
 from .parallelize import (
     transpile as transpile,
     parallelize as parallelize,
-    no_similarity as no_similarity,
+    remove_tags as remove_tags,
     auto_similarity as auto_similarity,
     block_similarity as block_similarity,
     moment_similarity as moment_similarity,

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/src/bloqade/cirq_utils/noise/model.py

@@ -357,9 +357,10 @@ class GeminiOneZoneNoiseModelConflictGraphMoves(GeminiOneZoneNoiseModel):
     def noisy_moment(self, moment, system_qubits):
         # Moment with original ops
         original_moment = moment
-        assert np.all(
-            [isinstance(q, cirq.GridQubit) for q in system_qubits]
-        ), "Found a qubit that is not a GridQubit."
+        assert np.all([isinstance(q, cirq.GridQubit) for q in system_qubits]), (
+            "Found a qubit that is not a GridQubit. In order for the conflict graph to know the qubit geometry, "
+            "all qubits in the circuit must be defined as cirq.GridQubit objects."
+        )
         # Check if the moment is empty
         if len(moment.operations) == 0:
             move_moments = []

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/src/bloqade/cirq_utils/parallelize.py

@@ -136,7 +136,7 @@ def auto_similarity(
     return cirq.Circuit(flattened_circuit), weights
 
 
-def no_similarity(circuit: cirq.Circuit) -> cirq.Circuit:
+def remove_tags(circuit: cirq.Circuit) -> cirq.Circuit:
     """
     Removes all tags from the circuit
 
@@ -146,10 +146,11 @@ def no_similarity(circuit: cirq.Circuit) -> cirq.Circuit:
     Returns:
     [0] - cirq.Circuit - the circuit with all tags removed.
     """
-    new_moments = []
-    for moment in circuit.moments:
-        new_moments.append([gate.untagged for gate in moment.operations])
-    return cirq.Circuit(new_moments)
+
+    def remove_tag(op: cirq.Operation, _):
+        return op.untagged
+
+    return cirq.map_operations(circuit, remove_tag)
 
 
 def to_dag_circuit(circuit: cirq.Circuit, can_reorder=None) -> nx.DiGraph:
@@ -399,4 +400,6 @@ def parallelize(
     )
     # Convert the epochs to a cirq circuit.
     moments = map(cirq.Moment, epochs)
-    return cirq.Circuit(moments)
+    circuit = cirq.Circuit(moments)
+
+    return remove_tags(circuit)

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/src/bloqade/pyqrack/base.py

@@ -146,7 +146,13 @@ class PyQrackInterpreter(Interpreter, typing.Generic[MemoryType]):
     loss_m_result: Measurement = field(default=Measurement.One, kw_only=True)
     """The value of a measurement result when a qubit is lost."""
 
+    global_measurement_id: int = field(init=False, default=0)
+
     def initialize(self) -> Self:
         super().initialize()
         self.memory.reset()  # reset allocated qubits
         return self
+
+    def set_global_measurement_id(self, m: Measurement):
+        m.measurement_id = self.global_measurement_id
+        self.global_measurement_id += 1

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/src/bloqade/pyqrack/reg.py

@@ -3,14 +3,19 @@ from typing import TYPE_CHECKING
 from dataclasses import dataclass
 
 from bloqade.qasm2.types import Qubit
+from bloqade.squin.types import MeasurementResult
 
 if TYPE_CHECKING:
     from pyqrack import QrackSimulator
 
 
-class Measurement(enum.IntEnum):
+class Measurement(MeasurementResult, enum.IntEnum):
     """Enumeration of measurement results."""
 
+    def __init__(self, measurement_id: int = 0) -> None:
+        super().__init__()
+        self.measurement_id = measurement_id
+
     Zero = 0
     One = 1
     Lost = enum.auto()

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/src/bloqade/pyqrack/squin/noise/native.py

@@ -46,7 +46,7 @@ class QubitLossRuntime(OperatorRuntimeABC):
         return 1
 
     def apply(self, qubit: PyQrackQubit, adjoint: bool = False) -> None:
-        if random.uniform(0.0, 1.0) < self.p:
+        if random.uniform(0.0, 1.0) <= self.p:
             qubit.state = QubitState.Lost
 
 

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/src/bloqade/pyqrack/squin/qubit.py

@@ -4,7 +4,7 @@ from kirin import interp
 from kirin.dialects import ilist
 
 from bloqade.squin import qubit
-from bloqade.pyqrack.reg import QubitState, PyQrackQubit
+from bloqade.pyqrack.reg import QubitState, Measurement, PyQrackQubit
 from bloqade.pyqrack.base import PyQrackInterpreter
 
 from .runtime import OperatorRuntimeABC
@@ -41,9 +41,12 @@ class PyQrackMethods(interp.MethodTable):
 
     def _measure_qubit(self, qbit: PyQrackQubit, interp: PyQrackInterpreter):
         if qbit.is_active():
-            return bool(qbit.sim_reg.m(qbit.addr))
+            m = Measurement(bool(qbit.sim_reg.m(qbit.addr)))
         else:
-            return interp.loss_m_result
+            m = Measurement(interp.loss_m_result)
+
+        interp.set_global_measurement_id(m)
+        return m
 
     @interp.impl(qubit.MeasureQubitList)
     def measure_qubit_list(
@@ -63,3 +66,17 @@ class PyQrackMethods(interp.MethodTable):
         qbit: PyQrackQubit = frame.get(stmt.qubit)
         result = self._measure_qubit(qbit, interp)
         return (result,)
+
+    @interp.impl(qubit.QubitId)
+    def qubit_id(
+        self, interp: PyQrackInterpreter, frame: interp.Frame, stmt: qubit.QubitId
+    ):
+        qbit: PyQrackQubit = frame.get(stmt.qubit)
+        return (qbit.addr,)
+
+    @interp.impl(qubit.MeasurementId)
+    def measurement_id(
+        self, interp: PyQrackInterpreter, frame: interp.Frame, stmt: qubit.MeasurementId
+    ):
+        measurement: Measurement = frame.get(stmt.measurement)
+        return (measurement.measurement_id,)

bloqade_circuit-0.7.9/src/bloqade/qasm2/passes/fold.py

@@ -0,0 +1,41 @@
+from dataclasses import field, dataclass
+
+from kirin import ir
+from kirin.passes import Pass
+from kirin.ir.method import Method
+from kirin.rewrite.abc import RewriteResult
+
+from bloqade.qasm2.dialects import expr
+from bloqade.rewrite.passes import AggressiveUnroll
+
+from .unroll_if import UnrollIfs
+
+
+@dataclass
+class QASM2Fold(Pass):
+    """Fold pass for qasm2.extended"""
+
+    inline_gate_subroutine: bool = True
+    unroll_ifs: bool = True
+    aggressive_unroll: AggressiveUnroll = field(init=False)
+
+    def __post_init__(self):
+        def inline_simple(node: ir.Statement):
+            if isinstance(node, expr.GateFunction):
+                return self.inline_gate_subroutine
+
+            return True
+
+        self.aggressive_unroll = AggressiveUnroll(
+            self.dialects, inline_simple, no_raise=self.no_raise
+        )
+
+    def unsafe_run(self, mt: Method) -> RewriteResult:
+        result = RewriteResult()
+
+        if self.unroll_ifs:
+            result = UnrollIfs(mt.dialects).unsafe_run(mt).join(result)
+
+        result = self.aggressive_unroll.unsafe_run(mt).join(result)
+
+        return result

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/src/bloqade/rewrite/passes/__init__.py

@@ -1 +1,2 @@
+from .aggressive_unroll import AggressiveUnroll as AggressiveUnroll
 from .canonicalize_ilist import CanonicalizeIList as CanonicalizeIList

bloqade_circuit-0.7.9/src/bloqade/rewrite/passes/aggressive_unroll.py

@@ -0,0 +1,93 @@
+from typing import Callable
+from dataclasses import field, dataclass
+
+from kirin import ir
+from kirin.passes import Pass, HintConst, TypeInfer
+from kirin.rewrite import (
+    Walk,
+    Chain,
+    Inline,
+    Fixpoint,
+    Call2Invoke,
+    ConstantFold,
+    CFGCompactify,
+    InlineGetItem,
+    InlineGetField,
+    DeadCodeElimination,
+)
+from kirin.dialects import scf, ilist
+from kirin.ir.method import Method
+from kirin.rewrite.abc import RewriteResult
+from kirin.rewrite.cse import CommonSubexpressionElimination
+from kirin.passes.aggressive import UnrollScf
+
+
+@dataclass
+class Fold(Pass):
+    hint_const: HintConst = field(init=False)
+
+    def __post_init__(self):
+        self.hint_const = HintConst(self.dialects, no_raise=self.no_raise)
+
+    def unsafe_run(self, mt: Method) -> RewriteResult:
+        result = RewriteResult()
+        result = self.hint_const.unsafe_run(mt).join(result)
+        rule = Chain(
+            ConstantFold(),
+            Call2Invoke(),
+            InlineGetField(),
+            InlineGetItem(),
+            ilist.rewrite.InlineGetItem(),
+            ilist.rewrite.HintLen(),
+        )
+        result = Fixpoint(Walk(rule)).rewrite(mt.code).join(result)
+
+        return result
+
+
+@dataclass
+class AggressiveUnroll(Pass):
+    """A pass to unroll structured control flow"""
+
+    additional_inline_heuristic: Callable[[ir.Statement], bool] = lambda node: True
+
+    fold: Fold = field(init=False)
+    typeinfer: TypeInfer = field(init=False)
+    scf_unroll: UnrollScf = field(init=False)
+
+    def __post_init__(self):
+        self.fold = Fold(self.dialects, no_raise=self.no_raise)
+        self.typeinfer = TypeInfer(self.dialects, no_raise=self.no_raise)
+        self.scf_unroll = UnrollScf(self.dialects, no_raise=self.no_raise)
+
+    def unsafe_run(self, mt: Method) -> RewriteResult:
+        result = RewriteResult()
+        result = self.scf_unroll.unsafe_run(mt).join(result)
+        result = (
+            Walk(Chain(ilist.rewrite.ConstList2IList(), ilist.rewrite.Unroll()))
+            .rewrite(mt.code)
+            .join(result)
+        )
+        result = self.typeinfer.unsafe_run(mt).join(result)
+        result = self.fold.unsafe_run(mt).join(result)
+        result = Walk(Inline(self.inline_heuristic)).rewrite(mt.code).join(result)
+        result = Walk(Fixpoint(CFGCompactify())).rewrite(mt.code).join(result)
+
+        rule = Chain(
+            CommonSubexpressionElimination(),
+            DeadCodeElimination(),
+        )
+        result = Fixpoint(Walk(rule)).rewrite(mt.code).join(result)
+
+        return result
+
+    def inline_heuristic(self, node: ir.Statement) -> bool:
+        """The heuristic to decide whether to inline a function call or not.
+        inside loops and if-else, only inline simple functions, i.e.
+        functions with a single block
+        """
+        return not isinstance(
+            node.parent_stmt, (scf.For, scf.IfElse)
+        ) and self.additional_inline_heuristic(
+            node
+        )  # always inline calls outside of loops and if-else

bloqade_circuit-0.7.9/src/bloqade/rewrite/passes/canonicalize_ilist.py

@@ -0,0 +1,32 @@
+from dataclasses import field, dataclass
+
+from kirin import ir, passes
+from kirin.rewrite import (
+    Walk,
+    Chain,
+    Fixpoint,
+)
+from kirin.dialects.ilist import rewrite
+
+
+@dataclass
+class CanonicalizeIList(passes.Pass):
+
+    fold_pass: passes.Fold = field(init=False)
+
+    def __post_init__(self):
+        self.fold_pass = passes.Fold(self.dialects, no_raise=self.no_raise)
+
+    def unsafe_run(self, mt: ir.Method):
+        result = Fixpoint(
+            Walk(
+                Chain(
+                    rewrite.InlineGetItem(),
+                    rewrite.FlattenAdd(),
+                    rewrite.HintLen(),
+                )
+            )
+        ).rewrite(mt.code)
+
+        result = self.fold_pass(mt).join(result)
+        return result

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/src/bloqade/squin/cirq/lowering.py

@@ -144,6 +144,11 @@ class Squin(lowering.LoweringABC[CirqNode]):
         qbits = self.lower_qubit_getindices(state, node.qubits)
         return state.current_frame.push(qubit.Apply(operator=op_, qubits=qbits))
 
+    def visit_TaggedOperation(
+        self, state: lowering.State[CirqNode], node: cirq.TaggedOperation
+    ):
+        state.lower(node.untagged)
+
     def lower_measurement(
         self, state: lowering.State[CirqNode], node: cirq.GateOperation
     ):

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/src/bloqade/squin/qubit.py

@@ -79,6 +79,20 @@ class MeasureQubitList(ir.Statement):
     result: ir.ResultValue = info.result(ilist.IListType[MeasurementResultType])
 
 
+@statement(dialect=dialect)
+class QubitId(ir.Statement):
+    traits = frozenset({lowering.FromPythonCall(), ir.Pure()})
+    qubit: ir.SSAValue = info.argument(QubitType)
+    result: ir.ResultValue = info.result(types.Int)
+
+
+@statement(dialect=dialect)
+class MeasurementId(ir.Statement):
+    traits = frozenset({lowering.FromPythonCall(), ir.Pure()})
+    measurement: ir.SSAValue = info.argument(MeasurementResultType)
+    result: ir.ResultValue = info.result(types.Int)
+
+
 # NOTE: no dependent types in Python, so we have to mark it Any...
 @wraps(New)
 def new(n_qubits: int) -> ilist.IList[Qubit, Any]:
@@ -127,9 +141,10 @@ def measure(input: Any) -> Any:
         input: A qubit or a list of qubits to measure.
 
     Returns:
-        bool | list[bool]: The result of the measurement. If a single qubit is measured,
-            a single boolean is returned. If a list of qubits is measured, a list of booleans
+        MeasurementResult | list[MeasurementResult]: The result of the measurement. If a single qubit is measured,
+            a single result is returned. If a list of qubits is measured, a list of results
             is returned.
+            A MeasurementResult can represent both 0 and 1, but also atoms that are lost.
     """
     ...
 
@@ -169,3 +184,11 @@ def broadcast(operator: Op, *qubits: ilist.IList[Qubit, OpSize] | list[Qubit]) -
         None
     """
     ...
+
+
+@wraps(QubitId)
+def get_qubit_id(qubit: Qubit) -> int: ...
+
+
+@wraps(MeasurementId)
+def get_measurement_id(measurement: MeasurementResult) -> int: ...

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/test/analysis/measure_id/test_measure_id.py

@@ -4,7 +4,6 @@ from kirin.dialects import scf
 from bloqade.squin import op, qubit, kernel
 from bloqade.analysis.measure_id import MeasurementIDAnalysis
 from bloqade.analysis.measure_id.lattice import (
-    NotMeasureId,
     MeasureIdBool,
     MeasureIdTuple,
     InvalidMeasureId,
@@ -111,10 +110,15 @@ def test_scf_cond_true():
     HintConst(dialects=test.dialects).unsafe_run(test)
     frame, _ = MeasurementIDAnalysis(test.dialects).run_analysis(test)
 
-    assert [frame.entries[result] for result in results_at(test, 0, 7)] == [
-        NotMeasureId(),
-        MeasureIdTuple((MeasureIdBool(idx=1),)),
+    # MeasureIdTuple(data=MeasureIdBool(idx=1),) should occur twice:
+    # First from the measurement in the true branch, then
+    # the result of the scf.IfElse itself
+    analysis_results = [
+        val
+        for val in frame.entries.values()
+        if val == MeasureIdTuple(data=(MeasureIdBool(idx=1),))
     ]
+    assert len(analysis_results) == 2
 
 
 def test_scf_cond_false():
@@ -136,10 +140,13 @@ def test_scf_cond_false():
     HintConst(dialects=test.dialects).unsafe_run(test)
     frame, _ = MeasurementIDAnalysis(test.dialects).run_analysis(test)
 
-    assert [frame.entries[result] for result in results_at(test, 0, 7)] == [
-        NotMeasureId(),
-        MeasureIdBool(idx=1),
+    # MeasureIdBool(idx=1) should occur twice:
+    # First from the measurement in the false branch, then
+    # the result of the scf.IfElse itself
+    analysis_results = [
+        val for val in frame.entries.values() if val == MeasureIdBool(idx=1)
     ]
+    assert len(analysis_results) == 2
 
 
 def test_slice():

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/test/cirq_utils/test_parallelize.py

@@ -6,7 +6,7 @@ import pytest
 
 from bloqade.cirq_utils import (
     parallelize,
-    no_similarity,
+    remove_tags,
     block_similarity,
     moment_similarity,
 )
@@ -28,7 +28,7 @@ def test1():
     circuit_m, _ = moment_similarity(circuit, weight=1.0)
     # print(circuit_m)
     circuit_b, _ = block_similarity(circuit, weight=1.0, block_id=1)
-    circuit_m2 = no_similarity(circuit_m)
+    circuit_m2 = remove_tags(circuit_m)
     print(circuit_m2)
     circuit2 = parallelize(circuit)
     # print(circuit2)

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/test/qasm2/emit/test_qasm2_emit.py

@@ -288,3 +288,30 @@ def test_if():
 
     target = qasm2.emit.QASM2()
     target.emit(nested_kernel)
+
+
+def test_loop_unroll():
+    n_qubits = 4
+
+    @qasm2.extended
+    def ghz_linear():
+        q = qasm2.qreg(n_qubits)
+        qasm2.h(q[0])
+        for i in range(1, n_qubits):
+            qasm2.cx(q[i - 1], q[i])
+
+    target = qasm2.emit.QASM2(
+        allow_parallel=True,
+    )
+    qasm2_str = target.emit_str(ghz_linear)
+
+    assert qasm2_str == (
+        """KIRIN {func,lowering.call,lowering.func,py.ilist,qasm2.core,qasm2.expr,qasm2.indexing,qasm2.noise,qasm2.parallel,qasm2.uop,scf};
+include "qelib1.inc";
+qreg q[4];
+h q[0];
+CX q[0], q[1];
+CX q[1], q[2];
+CX q[2], q[3];
+"""
+    )

{bloqade_circuit-0.7.7 → bloqade_circuit-0.7.9}/test/squin/cirq/test_cirq_to_squin.py

@@ -34,6 +34,11 @@ def basic_circuit():
     )
 
 
+def tagged_circuit():
+    q = cirq.LineQubit.range(2)
+    return cirq.Circuit(cirq.H(q[0]).with_tags("FOO"), cirq.CX(*q).with_tags("BAR"))
+
+
 def controlled_gates():
     q0 = cirq.NamedQubit("q0")
     q1 = cirq.NamedQubit("q1")
@@ -166,6 +171,7 @@ def nested_circuit():
     "circuit_f",
     [
         basic_circuit,
+        tagged_circuit,
         controlled_gates,
         parity_gate_circuit,
         phased_gates,

bloqade_circuit-0.7.9/test/squin/test_qubit.py

@@ -0,0 +1,48 @@
+from kirin import types
+
+from bloqade import squin
+from bloqade.pyqrack import StackMemorySimulator
+from bloqade.pyqrack.reg import Measurement
+
+
+def test_get_ids():
+    @squin.kernel
+    def main():
+        q = squin.qubit.new(3)
+
+        m = squin.qubit.measure(q)
+
+        qid = squin.qubit.get_qubit_id(q[0])
+        mid = squin.qubit.get_measurement_id(m[1])
+        return mid + qid
+
+    main.print()
+    assert main.return_type.is_subseteq(types.Int)
+
+    @squin.kernel
+    def main2():
+        q = squin.qubit.new(2)
+
+        qid = squin.qubit.get_qubit_id(q[0])
+        m1 = squin.qubit.measure(q[qid])
+
+        squin.gate.x(q[qid])
+        m2 = squin.qubit.measure(q[qid])
+
+        m1_id = squin.qubit.get_measurement_id(m1)
+        m2_id = squin.qubit.get_measurement_id(m2)
+
+        if m1_id != 0:
+            # do something that errors
+            squin.gate.x(q[4])
+
+        if m2_id != 1:
+            squin.gate.x(q[4])
+
+        return squin.qubit.measure(q)
+
+    sim = StackMemorySimulator(min_qubits=2)
+    result = sim.run(main2)
+    for i, res in enumerate(result):
+        assert isinstance(res, Measurement)
+        assert res.measurement_id == i + 2