bloqade-circuit 0.5.3__tar.gz → 0.6.0__tar.gz

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bloqade-circuit
-Version: 0.5.3
+Version: 0.6.0
 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

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/_typos.toml

@@ -13,3 +13,4 @@ Braket = "Braket"
 mch = "mch"
 IY = "IY"
 ket = "ket"
+typ = "typ"

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bloqade-circuit"
-version = "0.5.3"
+version = "0.6.0"
 description = "The software development toolkit for neutral atom arrays."
 readme = "README.md"
 authors = [

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/src/bloqade/squin/noise/__init__.py

@@ -4,6 +4,7 @@ from ._wrapper import (
     pp_error as pp_error,
     depolarize as depolarize,
     qubit_loss as qubit_loss,
+    depolarize2 as depolarize2,
     pauli_error as pauli_error,
     two_qubit_pauli_channel as two_qubit_pauli_channel,
     single_qubit_pauli_channel as single_qubit_pauli_channel,

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/src/bloqade/squin/noise/_wrapper.py

@@ -20,6 +20,10 @@ def pp_error(op: Op, p: float) -> Op: ...
 def depolarize(p: float) -> Op: ...
 
 
+@wraps(stmts.Depolarize2)
+def depolarize2(p: float) -> Op: ...
+
+
 @wraps(stmts.SingleQubitPauliChannel)
 def single_qubit_pauli_channel(
     params: ilist.IList[float, Literal[3]] | list[float] | tuple[float, float, float],

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/src/bloqade/squin/noise/stmts.py

@@ -33,7 +33,16 @@ class PPError(NoiseChannel):
 @statement(dialect=dialect)
 class Depolarize(NoiseChannel):
     """
-    Apply depolarize error to qubit
+    Apply depolarize error to single qubit
+    """
+
+    p: ir.SSAValue = info.argument(types.Float)
+
+
+@statement(dialect=dialect)
+class Depolarize2(NoiseChannel):
+    """
+    Apply correlated depolarize error to two qubit
     """
 
     p: ir.SSAValue = info.argument(types.Float)

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/src/bloqade/squin/op/stmts.py

@@ -103,7 +103,7 @@ class U3(PrimitiveOp):
     Note that we use the convention from the QASM2 specification, namely
 
     $$
-    U_3(\theta, \phi, \lambda) = R_z(\phi) R_y(\theta) R_z(\lambda)
+    U_3(\\theta, \\phi, \\lambda) = R_z(\\phi) R_y(\\theta) R_z(\\lambda)
     $$
     """
 
@@ -119,7 +119,7 @@ class PhaseOp(PrimitiveOp):
     A phase operator.
 
     $$
-    PhaseOp(\theta) = e^{i \theta} I
+    \\text{PhaseOp}(\\theta) = e^{i \\theta} I
     $$
     """
 
@@ -133,7 +133,7 @@ class ShiftOp(PrimitiveOp):
     A phase shift operator.
 
     $$
-    Shift(\theta) = \\begin{bmatrix} 1 & 0 \\\\ 0 & e^{i \\theta} \\end{bmatrix}
+    \\text{Shift}(\\theta) = \\begin{bmatrix} 1 & 0 \\\\ 0 & e^{i \\theta} \\end{bmatrix}
     $$
     """
 
@@ -144,7 +144,7 @@ class ShiftOp(PrimitiveOp):
 @statement(dialect=dialect)
 class Reset(PrimitiveOp):
     """
-    Reset operator for qubits or wires.
+    Reset operator for qubits and wires.
     """
 
     traits = frozenset({ir.Pure(), lowering.FromPythonCall(), FixedSites(1)})

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/src/bloqade/squin/rewrite/desugar.py

@@ -54,7 +54,9 @@ class ApplyDesugarRule(RewriteRule):
         qubits = node.qubits
 
         if len(qubits) > 1 and all(q.type.is_subseteq(QubitType) for q in qubits):
-            (qubits_ilist_stmt := ilist.New(qubits)).insert_before(node)
+            (qubits_ilist_stmt := ilist.New(qubits)).insert_before(
+                node
+            )  # qubits is just a tuple of SSAValues
             qubits_ilist = qubits_ilist_stmt.result
 
         elif len(qubits) == 1 and qubits[0].type.is_subseteq(QubitType):
@@ -76,34 +78,44 @@ class ApplyDesugarRule(RewriteRule):
                 return RewriteResult()
 
             is_ilist = isinstance(qbit_stmt := qubits[0].stmt, ilist.New)
+
             if is_ilist:
-                if len(qbit_stmt.values) != 1:
-                    return RewriteResult()
 
-                if not isinstance(
-                    qbit_getindex_result := qbit_stmt.values[0], ir.ResultValue
+                if not all(
+                    isinstance(qbit_getindex_result, ir.ResultValue)
+                    for qbit_getindex_result in qbit_stmt.values
                 ):
                     return RewriteResult()
 
-                qbit_getindex = qbit_getindex_result.stmt
+                # Get the parent statement that the qubit came from
+                # (should be a GetItem instance, see logic below)
+                qbit_getindices = [
+                    qbit_getindex_result.stmt
+                    for qbit_getindex_result in qbit_stmt.values
+                ]
             else:
-                qbit_getindex = qubits[0].stmt
+                qbit_getindices = [qubit.stmt for qubit in qubits]
 
-            if not isinstance(qbit_getindex, py.indexing.GetItem):
+            if any(
+                not isinstance(qbit_getindex, py.indexing.GetItem)
+                for qbit_getindex in qbit_getindices
+            ):
                 return RewriteResult()
 
-            if not qbit_getindex.obj.type.is_subseteq(
-                ilist.IListType[QubitType, types.Any]
+            # The GetItem should have been applied on something that returns an IList of Qubits
+            if any(
+                not qbit_getindex.obj.type.is_subseteq(
+                    ilist.IListType[QubitType, types.Any]
+                )
+                for qbit_getindex in qbit_getindices
             ):
                 return RewriteResult()
 
             if is_ilist:
-                values = qbit_stmt.values
+                qubits_ilist = qbit_stmt.result
             else:
-                values = [qubits[0]]
-
-            (qubits_ilist_stmt := ilist.New(values=values)).insert_before(node)
-            qubits_ilist = qubits_ilist_stmt.result
+                (qubits_ilist_stmt := ilist.New(values=[qubits[0]])).insert_before(node)
+                qubits_ilist = qubits_ilist_stmt.result
         else:
             return RewriteResult()
 

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/src/bloqade/stim/passes/squin_to_stim.py

@@ -11,7 +11,6 @@ from kirin.rewrite import (
     DeadCodeElimination,
     CommonSubexpressionElimination,
 )
-from kirin.analysis import const
 from kirin.dialects import scf, ilist
 from kirin.ir.method import Method
 from kirin.passes.abc import Pass
@@ -44,11 +43,6 @@ class SquinToStimPass(Pass):
 
     def unsafe_run(self, mt: Method) -> RewriteResult:
 
-        cp_frame, _ = const.Propagate(dialects=mt.dialects).run_analysis(mt)
-        cp_results = cp_frame.entries
-
-        # Assume that address analysis and
-        # wrapping has been done before this pass!
         # inline aggressively:
         rewrite_result = InlinePass(
             dialects=mt.dialects, no_raise=self.no_raise
@@ -117,11 +111,7 @@ class SquinToStimPass(Pass):
         )
 
         # Rewrite the noise statements first.
-        rewrite_result = (
-            Walk(SquinNoiseToStim(cp_results=cp_results))
-            .rewrite(mt.code)
-            .join(rewrite_result)
-        )
+        rewrite_result = Walk(SquinNoiseToStim()).rewrite(mt.code).join(rewrite_result)
 
         # Wrap Rewrite + SquinToStim can happen w/ standard walk
         rewrite_result = Walk(SquinU3ToClifford()).rewrite(mt.code).join(rewrite_result)

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/src/bloqade/stim/rewrite/squin_noise.py

@@ -1,14 +1,14 @@
-from typing import Dict, Tuple
+from typing import Tuple
 from dataclasses import dataclass
 
 from kirin.ir import SSAValue, Statement
-from kirin.analysis import const
-from kirin.dialects import py
+from kirin.dialects import py, ilist
 from kirin.rewrite.abc import RewriteRule, RewriteResult
 
-from bloqade.squin import wire, noise as squin_noise, qubit
+from bloqade.squin import op, wire, noise as squin_noise, qubit
 from bloqade.stim.dialects import noise as stim_noise
 from bloqade.stim.rewrite.util import (
+    get_const_value,
     create_wire_passthrough,
     insert_qubit_idx_after_apply,
 )
@@ -17,8 +17,6 @@ from bloqade.stim.rewrite.util import (
 @dataclass
 class SquinNoiseToStim(RewriteRule):
 
-    cp_results: Dict[SSAValue, const.Result]
-
     def rewrite_Statement(self, node: Statement) -> RewriteResult:
         match node:
             case qubit.Apply() | qubit.Broadcast():
@@ -34,17 +32,17 @@ class SquinNoiseToStim(RewriteRule):
         # this is an SSAValue, need it to be the actual operator
         applied_op = stmt.operator.owner
 
+        if isinstance(applied_op, squin_noise.stmts.QubitLoss):
+            return RewriteResult()
+
         if isinstance(applied_op, squin_noise.stmts.NoiseChannel):
 
             qubit_idx_ssas = insert_qubit_idx_after_apply(stmt=stmt)
             if qubit_idx_ssas is None:
                 return RewriteResult()
 
-            stim_stmt = None
-            if isinstance(applied_op, squin_noise.stmts.SingleQubitPauliChannel):
-                stim_stmt = self.rewrite_SingleQubitPauliChannel(stmt, qubit_idx_ssas)
-            elif isinstance(applied_op, squin_noise.stmts.TwoQubitPauliChannel):
-                stim_stmt = self.rewrite_TwoQubitPauliChannel(stmt, qubit_idx_ssas)
+            rewrite_method = getattr(self, f"rewrite_{type(applied_op).__name__}")
+            stim_stmt = rewrite_method(stmt, qubit_idx_ssas)
 
             if isinstance(stmt, (wire.Apply, wire.Broadcast)):
                 create_wire_passthrough(stmt)
@@ -57,6 +55,29 @@ class SquinNoiseToStim(RewriteRule):
             return RewriteResult(has_done_something=True)
         return RewriteResult()
 
+    def rewrite_PauliError(
+        self,
+        stmt: qubit.Apply | qubit.Broadcast | wire.Broadcast | wire.Apply,
+        qubit_idx_ssas: Tuple[SSAValue],
+    ) -> Statement:
+        """Rewrite squin.noise.PauliError to XError, YError, ZError."""
+        squin_channel = stmt.operator.owner
+        assert isinstance(squin_channel, squin_noise.stmts.PauliError)
+        basis = squin_channel.basis.owner
+        assert isinstance(basis, op.stmts.PauliOp)
+        p = get_const_value(float, squin_channel.p)
+
+        p_stmt = py.Constant(p)
+        p_stmt.insert_before(stmt)
+
+        if isinstance(basis, op.stmts.X):
+            stim_stmt = stim_noise.XError(targets=qubit_idx_ssas, p=p_stmt.result)
+        elif isinstance(basis, op.stmts.Y):
+            stim_stmt = stim_noise.YError(targets=qubit_idx_ssas, p=p_stmt.result)
+        else:
+            stim_stmt = stim_noise.ZError(targets=qubit_idx_ssas, p=p_stmt.result)
+        return stim_stmt
+
     def rewrite_SingleQubitPauliChannel(
         self,
         stmt: qubit.Apply | qubit.Broadcast | wire.Broadcast | wire.Apply,
@@ -67,7 +88,7 @@ class SquinNoiseToStim(RewriteRule):
         squin_channel = stmt.operator.owner
         assert isinstance(squin_channel, squin_noise.stmts.SingleQubitPauliChannel)
 
-        params = self.cp_results.get(squin_channel.params).data
+        params = get_const_value(ilist.IList, squin_channel.params)
         new_stmts = [
             p_x := py.Constant(params[0]),
             p_y := py.Constant(params[1]),
@@ -94,7 +115,7 @@ class SquinNoiseToStim(RewriteRule):
         squin_channel = stmt.operator.owner
         assert isinstance(squin_channel, squin_noise.stmts.TwoQubitPauliChannel)
 
-        params = self.cp_results.get(squin_channel.params).data
+        params = get_const_value(ilist.IList, squin_channel.params)
         param_stmts = [py.Constant(p) for p in params]
         for param_stmt in param_stmts:
             param_stmt.insert_before(stmt)
@@ -118,3 +139,37 @@ class SquinNoiseToStim(RewriteRule):
             pzz=param_stmts[14].result,
         )
         return stim_stmt
+
+    def rewrite_Depolarize2(
+        self,
+        stmt: qubit.Apply | qubit.Broadcast | wire.Broadcast | wire.Apply,
+        qubit_idx_ssas: Tuple[SSAValue],
+    ) -> Statement:
+        """Rewrite squin.noise.Depolarize2 to stim.Depolarize2."""
+
+        squin_channel = stmt.operator.owner
+        assert isinstance(squin_channel, squin_noise.stmts.Depolarize2)
+
+        p = get_const_value(float, squin_channel.p)
+        p_stmt = py.Constant(p)
+        p_stmt.insert_before(stmt)
+
+        stim_stmt = stim_noise.Depolarize2(targets=qubit_idx_ssas, p=p_stmt.result)
+        return stim_stmt
+
+    def rewrite_Depolarize(
+        self,
+        stmt: qubit.Apply | qubit.Broadcast | wire.Broadcast | wire.Apply,
+        qubit_idx_ssas: Tuple[SSAValue],
+    ) -> Statement:
+        """Rewrite squin.noise.Depolarize to stim.Depolarize1."""
+
+        squin_channel = stmt.operator.owner
+        assert isinstance(squin_channel, squin_noise.stmts.Depolarize)
+
+        p = get_const_value(float, squin_channel.p)
+        p_stmt = py.Constant(p)
+        p_stmt.insert_before(stmt)
+
+        stim_stmt = stim_noise.Depolarize1(targets=qubit_idx_ssas, p=p_stmt.result)
+        return stim_stmt

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/src/bloqade/stim/rewrite/util.py

@@ -1,4 +1,7 @@
-from kirin import ir
+from typing import TypeVar
+
+from kirin import ir, interp
+from kirin.analysis import const
 from kirin.dialects import py
 from kirin.rewrite.abc import RewriteResult
 
@@ -201,3 +204,19 @@ def is_measure_result_used(
     Check if the result of a measure statement is used in the program.
     """
     return bool(stmt.result.uses)
+
+
+T = TypeVar("T")
+
+
+def get_const_value(typ: type[T], value: ir.SSAValue) -> T:
+    if isinstance(hint := value.hints.get("const"), const.Value):
+        data = hint.data
+        if isinstance(data, typ):
+            return hint.data
+        raise interp.InterpreterError(
+            f"Expected constant value <type = {typ}>, got {data}"
+        )
+    raise interp.InterpreterError(
+        f"Expected constant value <type = {typ}>, got {value}"
+    )

{bloqade_circuit-0.5.3 → bloqade_circuit-0.6.0}/test/squin/test_sugar.py

@@ -46,7 +46,7 @@ def test_apply_sugar():
         h = squin.op.h()
         x = squin.op.x()
 
-        # test applying to lest with getindex
+        # test applying to list with getindex
         squin.qubit.apply(x, [q[0]])
 
         # test apply with ast.Name
@@ -89,3 +89,18 @@ def test_apply_in_for_loop():
 
     assert math.isclose(abs(ket[0]) ** 2, 1, abs_tol=1e-7)
     assert ket[1] == ket[2] == ket[3] == 0
+
+
+def test_apply_in_for_loop_index_multiple_index():
+
+    @squin.kernel
+    def main():
+        q = squin.qubit.new(3)
+        squin.qubit.apply(squin.op.h(), q[0])
+        cx = squin.op.cx()
+        for i in range(2):
+            squin.qubit.apply(cx, [q[i], q[i + 1]])
+
+    sim = StackMemorySimulator(min_qubits=3)
+    ket = sim.state_vector(main)
+    assert math.isclose(abs(ket[0]) ** 2, 0.5, abs_tol=1e-5)

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/apply_depolarize1.stim

@@ -0,0 +1 @@
+DEPOLARIZE1(0.01000000) 0

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/apply_pauli_channel_1.stim

@@ -0,0 +1 @@
+PAULI_CHANNEL_1(0.01000000, 0.02000000, 0.03000000) 0

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/broadcast_depolarize1.stim

@@ -0,0 +1 @@
+DEPOLARIZE1(0.01000000) 0 1 2 3

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/broadcast_depolarize2.stim

@@ -0,0 +1 @@
+DEPOLARIZE2(0.01500000) 0 1

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/broadcast_iid_bit_flip_channel.stim

@@ -0,0 +1 @@
+X_ERROR(0.01000000) 0 1 2 3

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/broadcast_iid_phase_flip_channel.stim

@@ -0,0 +1 @@
+Z_ERROR(0.01000000) 0 1 2 3

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/broadcast_iid_y_flip_channel.stim

@@ -0,0 +1 @@
+Y_ERROR(0.01000000) 0 1 2 3

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/broadcast_pauli_channel_1.stim

@@ -0,0 +1 @@
+PAULI_CHANNEL_1(0.01000000, 0.02000000, 0.03000000) 0

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/broadcast_pauli_channel_1_many_qubits.stim

@@ -0,0 +1 @@
+PAULI_CHANNEL_1(0.01000000, 0.02000000, 0.03000000) 0 1

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/broadcast_pauli_channel_1_reuse.stim

@@ -0,0 +1,3 @@
+PAULI_CHANNEL_1(0.01000000, 0.02000000, 0.03000000) 0
+PAULI_CHANNEL_1(0.01000000, 0.02000000, 0.03000000) 0
+PAULI_CHANNEL_1(0.01000000, 0.02000000, 0.03000000) 0

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/broadcast_pauli_channel_2.stim

@@ -0,0 +1 @@
+PAULI_CHANNEL_2(0.00100000, 0.00200000, 0.00300000, 0.00400000, 0.00500000, 0.00600000, 0.00700000, 0.00800000, 0.00900000, 0.01000000, 0.01100000, 0.01200000, 0.01300000, 0.01400000, 0.01500000) 0 1

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/noise/broadcast_pauli_channel_2_reuse_on_4_qubits.stim

@@ -0,0 +1,2 @@
+PAULI_CHANNEL_2(0.00100000, 0.00200000, 0.00300000, 0.00400000, 0.00500000, 0.00600000, 0.00700000, 0.00800000, 0.00900000, 0.01000000, 0.01100000, 0.01200000, 0.01300000, 0.01400000, 0.01500000) 0 1
+PAULI_CHANNEL_2(0.00100000, 0.00200000, 0.00300000, 0.00400000, 0.00500000, 0.00600000, 0.00700000, 0.00800000, 0.00900000, 0.01000000, 0.01100000, 0.01200000, 0.01300000, 0.01400000, 0.01500000) 2 3

bloqade_circuit-0.6.0/test/stim/passes/stim_reference_programs/qubit/for_loop.stim

@@ -0,0 +1,4 @@
+
+H 0
+CX 0 1
+CX 1 2