bloqade-circuit 0.2.3__py3-none-any.whl → 0.4.0__py3-none-any.whl

bloqade/stim/dialects/noise/emit.py

@@ -1,7 +1,7 @@
 from kirin.emit import EmitStrFrame
 from kirin.interp import MethodTable, impl
 
-from bloqade.stim.emit.stim import EmitStimMain
+from bloqade.stim.emit.stim_str import EmitStimMain
 
 from . import stmts
 from ._dialect import dialect
@@ -44,7 +44,7 @@ class EmitStimNoiseMethods(MethodTable):
         px: str = frame.get(stmt.px)
         py: str = frame.get(stmt.py)
         pz: str = frame.get(stmt.pz)
-        res = f"PAULI_CHANNEL_1({px},{py},{pz}) " + " ".join(targets)
+        res = f"PAULI_CHANNEL_1({px}, {py}, {pz}) " + " ".join(targets)
         emit.writeln(frame, res)
 
         return ()
@@ -64,3 +64,34 @@ class EmitStimNoiseMethods(MethodTable):
         emit.writeln(frame, res)
 
         return ()
+
+    @impl(stmts.TrivialError)
+    def non_stim_error(
+        self, emit: EmitStimMain, frame: EmitStrFrame, stmt: stmts.TrivialError
+    ):
+
+        targets: tuple[str, ...] = frame.get_values(stmt.targets)
+        prob: tuple[str, ...] = frame.get_values(stmt.probs)
+        prob_str: str = ", ".join(prob)
+
+        res = f"I_ERROR[{stmt.name}]({prob_str}) " + " ".join(targets)
+        emit.writeln(frame, res)
+
+        return ()
+
+    @impl(stmts.TrivialCorrelatedError)
+    def non_stim_corr_error(
+        self,
+        emit: EmitStimMain,
+        frame: EmitStrFrame,
+        stmt: stmts.TrivialCorrelatedError,
+    ):
+
+        targets: tuple[str, ...] = frame.get_values(stmt.targets)
+        prob: tuple[str, ...] = frame.get_values(stmt.probs)
+        prob_str: str = ", ".join(prob)
+
+        res = f"I_ERROR[{stmt.name}:{stmt.nonce}]({prob_str}) " + " ".join(targets)
+        emit.writeln(frame, res)
+
+        return ()

bloqade/stim/dialects/noise/stmts.py

@@ -75,3 +75,32 @@ class ZError(ir.Statement):
     traits = frozenset({lowering.FromPythonCall()})
     p: ir.SSAValue = info.argument(types.Float)
     targets: tuple[ir.SSAValue, ...] = info.argument(types.Int)
+
+
+@statement
+class NonStimError(ir.Statement):
+    name = "NonStimError"
+    traits = frozenset({lowering.FromPythonCall()})
+    probs: tuple[ir.SSAValue, ...] = info.argument(types.Float)
+    targets: tuple[ir.SSAValue, ...] = info.argument(types.Int)
+
+
+@statement
+class NonStimCorrelatedError(ir.Statement):
+    name = "NonStimCorrelatedError"
+    traits = frozenset({lowering.FromPythonCall()})
+    nonce: int = (
+        info.attribute()
+    )  # Must be a unique value, otherwise stim might merge two correlated errors with equal probabilities
+    probs: tuple[ir.SSAValue, ...] = info.argument(types.Float)
+    targets: tuple[ir.SSAValue, ...] = info.argument(types.Int)
+
+
+@statement(dialect=dialect)
+class TrivialCorrelatedError(NonStimCorrelatedError):
+    name = "TRIV_CORR_ERROR"
+
+
+@statement(dialect=dialect)
+class TrivialError(NonStimError):
+    name = "TRIV_ERROR"

bloqade/stim/emit/__init__.py

@@ -1 +1 @@
-from .stim import FuncEmit as FuncEmit, EmitStimMain as EmitStimMain
+from .stim_str import FuncEmit as FuncEmit, EmitStimMain as EmitStimMain

bloqade/stim/groups.py

@@ -2,10 +2,12 @@ from kirin import ir
 from kirin.passes import Fold, TypeInfer
 from kirin.dialects import func, lowering
 
-from .dialects import aux, gate, noise, collapse
+from .dialects import gate, noise, collapse, auxiliary
 
 
-@ir.dialect_group([noise, gate, aux, collapse, func, lowering.func, lowering.call])
+@ir.dialect_group(
+    [noise, gate, auxiliary, collapse, func, lowering.func, lowering.call]
+)
 def main(self):
     typeinfer_pass = TypeInfer(self)
     fold_pass = Fold(self)

bloqade/stim/parse/__init__.py

@@ -0,0 +1 @@
+from .lowering import loads as loads, loadfile as loadfile