bloqade-circuit 0.7.11__tar.gz → 0.7.13__tar.gz

{bloqade_circuit-0.7.11 → bloqade_circuit-0.7.13}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: bloqade-circuit
-Version: 0.7.11
+Version: 0.7.13
 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
 Requires-Python: >=3.10
-Requires-Dist: kirin-toolchain~=0.17.23
+Requires-Dist: kirin-toolchain~=0.17.26
 Requires-Dist: numpy>=1.22.0
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: pydantic<2.11.0,>=1.3.0

{bloqade_circuit-0.7.11 → bloqade_circuit-0.7.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bloqade-circuit"
-version = "0.7.11"
+version = "0.7.13"
 description = "The software development toolkit for neutral atom arrays."
 readme = "README.md"
 authors = [
@@ -13,7 +13,7 @@ requires-python = ">=3.10"
 dependencies = [
     "numpy>=1.22.0",
     "scipy>=1.13.1",
-    "kirin-toolchain~=0.17.23",
+    "kirin-toolchain~=0.17.26",
     "rich>=13.9.4",
     "pydantic>=1.3.0,<2.11.0",
     "pandas>=2.2.3",

{bloqade_circuit-0.7.11 → bloqade_circuit-0.7.13}/src/bloqade/stim/emit/__init__.py

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

bloqade_circuit-0.7.13/src/bloqade/stim/emit/impls.py

@@ -0,0 +1,17 @@
+from kirin.emit import EmitStrFrame
+from kirin.interp import MethodTable, impl
+from kirin.dialects.debug import Info, dialect
+
+from bloqade.stim.emit.stim_str import EmitStimMain
+
+
+@dialect.register(key="emit.stim")
+class EmitStimDebugMethods(MethodTable):
+
+    @impl(Info)
+    def info(self, emit: EmitStimMain, frame: EmitStrFrame, stmt: Info):
+
+        msg: str = frame.get(stmt.msg)
+        emit.writeln(frame, f"# {msg}")
+
+        return ()

{bloqade_circuit-0.7.11 → bloqade_circuit-0.7.13}/src/bloqade/stim/rewrite/qubit_to_stim.py

@@ -3,7 +3,6 @@ from kirin.rewrite.abc import RewriteRule, RewriteResult
 
 from bloqade.squin import op, noise, qubit
 from bloqade.squin.rewrite import AddressAttribute
-from bloqade.stim.dialects import gate
 from bloqade.stim.rewrite.util import (
     SQUIN_STIM_OP_MAPPING,
     rewrite_Control,
@@ -40,20 +39,24 @@ class SquinQubitToStim(RewriteRule):
 
         assert isinstance(applied_op, op.stmts.Operator)
 
+        # Handle controlled gates with a separate procedure
         if isinstance(applied_op, op.stmts.Control):
             return rewrite_Control(stmt)
 
-        # need to handle Control through separate means
-
         # check if its adjoint, assume its canonicalized so no nested adjoints.
         is_conj = False
         if isinstance(applied_op, op.stmts.Adjoint):
-            if not applied_op.is_unitary:
+            # By default the Adjoint has is_unitary = False, so we need to check
+            # the inner applied operator to make sure its not just unitary,
+            # but something that has an equivalent stim representation with *_DAG format.
+            if isinstance(
+                applied_op.op.owner, (op.stmts.SqrtX, op.stmts.SqrtY, op.stmts.S)
+            ):
+                is_conj = True
+                applied_op = applied_op.op.owner
+            else:
                 return RewriteResult()
 
-            is_conj = True
-            applied_op = applied_op.op.owner
-
         stim_1q_op = SQUIN_STIM_OP_MAPPING.get(type(applied_op))
         if stim_1q_op is None:
             return RewriteResult()
@@ -71,13 +74,14 @@ class SquinQubitToStim(RewriteRule):
         if qubit_idx_ssas is None:
             return RewriteResult()
 
-        if isinstance(stim_1q_op, gate.stmts.Gate):
+        # At this point, we know for certain stim_1q_op must be SQRT_X, SQRT_Y, or S
+        # and has the option to set the dagger attribute. If is_conj is false,
+        # the rewrite would have terminated early so we know anything else has to be
+        # a non 1Q gate operation.
+        if is_conj:
             stim_1q_stmt = stim_1q_op(targets=tuple(qubit_idx_ssas), dagger=is_conj)
         else:
             stim_1q_stmt = stim_1q_op(targets=tuple(qubit_idx_ssas))
         stmt.replace_by(stim_1q_stmt)
 
         return RewriteResult(has_done_something=True)
-
-
-# put rewrites for measure statements in separate rule, then just have to dispatch

bloqade_circuit-0.7.13/test/stim/dialects/stim/emit/test_stim_debug.py

@@ -0,0 +1,16 @@
+from kirin.dialects import debug
+
+from bloqade import stim
+
+from .base import codegen
+
+
+def test_debug():
+
+    @stim.main
+    def test_debug_main():
+        debug.info("debug message")
+
+    test_debug_main.print()
+    out = codegen(test_debug_main)
+    assert out.strip() == "# debug message"

bloqade_circuit-0.7.13/test/stim/passes/stim_reference_programs/qubit/adjoint_rewrite.stim

@@ -0,0 +1,4 @@
+
+SQRT_X_DAG 0
+SQRT_Y_DAG 0
+S_DAG 0

bloqade_circuit-0.7.13/test/stim/passes/stim_reference_programs/qubit/u3_to_clifford.stim

@@ -0,0 +1,12 @@
+
+H 0
+S 0
+SQRT_Y 0
+S 0
+S_DAG 0
+SQRT_Y 0
+S 0
+S 0
+SQRT_Y 0
+S_DAG 0
+MZ(0.00000000) 0

bloqade_circuit-0.7.13/test/stim/passes/test_squin_debug_to_stim.py

@@ -0,0 +1,44 @@
+import os
+
+from kirin import ir
+from kirin.dialects import py, debug
+
+from bloqade.squin import kernel
+from bloqade.stim.emit import EmitStimMain
+from bloqade.stim.passes import SquinToStimPass
+
+
+# Taken gratuitously from Kai's unit test
+def codegen(mt: ir.Method):
+    # method should not have any arguments!
+    emit = EmitStimMain()
+    emit.initialize()
+    emit.run(mt=mt, args=())
+    return emit.get_output()
+
+
+def as_int(value: int):
+    return py.constant.Constant(value=value)
+
+
+def as_float(value: float):
+    return py.constant.Constant(value=value)
+
+
+def load_reference_program(filename):
+    path = os.path.join(
+        os.path.dirname(__file__), "stim_reference_programs", "debug", filename
+    )
+    with open(path, "r") as f:
+        return f.read()
+
+
+def test_info():
+    @kernel
+    def test():
+        debug.info("debug message")
+        return
+
+    SquinToStimPass(test.dialects)(test)
+    base_stim_prog = load_reference_program("debug.stim")
+    assert codegen(test) == base_stim_prog.rstrip()

{bloqade_circuit-0.7.11 → bloqade_circuit-0.7.13}/test/stim/passes/test_squin_qubit_to_stim.py

@@ -110,6 +110,26 @@ def test_qubit_loss():
     assert codegen(test) == base_stim_prog.rstrip()
 
 
+def test_adjoint_rewrite():
+
+    @squin.kernel
+    def test():
+        q = qubit.new(1)
+        sqrt_x_dag = op.adjoint(op.sqrt_x())
+        sqrt_y_dag = op.adjoint(op.sqrt_y())
+        sqrt_s_dag = op.adjoint(op.s())
+        qubit.apply(sqrt_x_dag, q[0])
+        qubit.apply(sqrt_y_dag, q[0])
+        qubit.apply(sqrt_s_dag, q[0])
+        return
+
+    SquinToStimPass(test.dialects)(test)
+
+    base_stim_prog = load_reference_program("adjoint_rewrite.stim")
+
+    assert codegen(test) == base_stim_prog.rstrip()
+
+
 def test_u3_to_clifford():
 
     @kernel
@@ -118,12 +138,24 @@ def test_u3_to_clifford():
         q = qubit.new(n_qubits)
         # apply U3 rotation that can be translated to a Clifford gate
         squin.qubit.apply(op.u(0.25 * math.tau, 0.0 * math.tau, 0.5 * math.tau), q[0])
+        # S @ SQRT_Y @ S = Z @ SQRT_X
+        squin.qubit.apply(
+            op.u(-0.25 * math.tau, -0.25 * math.tau, -0.25 * math.tau), q[0]
+        )
+        # S @ SQRT_Y @ S_DAG = SQRT_X_DAG
+        squin.qubit.apply(
+            op.u(-0.25 * math.tau, -0.25 * math.tau, 0.25 * math.tau), q[0]
+        )
+        # S_DAG @ SQRT_Y @ S = SQRT_X
+        squin.qubit.apply(
+            op.u(-0.25 * math.tau, 0.25 * math.tau, -0.25 * math.tau), q[0]
+        )
+
         # measure out
         squin.qubit.measure(q)
         return
 
     SquinToStimPass(test.dialects)(test)
-
     base_stim_prog = load_reference_program("u3_to_clifford.stim")
 
     assert codegen(test) == base_stim_prog.rstrip()

{bloqade_circuit-0.7.11 → bloqade_circuit-0.7.13}/uv.lock

@@ -103,7 +103,7 @@ wheels = [
 
 [[package]]
 name = "bloqade-circuit"
-version = "0.7.11"
+version = "0.7.13"
 source = { editable = "." }
 dependencies = [
     { name = "kirin-toolchain" },
@@ -168,7 +168,7 @@ requires-dist = [
     { name = "cirq-core", marker = "extra == 'cirq'", specifier = ">=1.4.1" },
     { name = "cirq-core", extras = ["contrib"], marker = "extra == 'cirq'", specifier = ">=1.4.1" },
     { name = "ffmpeg", marker = "extra == 'vis'", specifier = ">=1.4" },
-    { name = "kirin-toolchain", specifier = "~=0.17.23" },
+    { name = "kirin-toolchain", specifier = "~=0.17.26" },
     { name = "lark", marker = "extra == 'qasm2'", specifier = ">=1.2.2" },
     { name = "matplotlib", marker = "extra == 'vis'", specifier = ">=3.9.2" },
     { name = "numpy", specifier = ">=1.22.0" },
@@ -1072,16 +1072,16 @@ wheels = [
 
 [[package]]
 name = "kirin-toolchain"
-version = "0.17.25"
+version = "0.17.33"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "beartype" },
     { name = "rich" },
     { name = "typing-extensions" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/20/0f/ecbb7b88461247687d035ef9b11bc14e6138d745c2ebbc3776c31b597d60/kirin_toolchain-0.17.25.tar.gz", hash = "sha256:b91f12bda11db3a9b41dd0d0ca22797f93508f148f1c41e270b17a17e74c9126", size = 1183324 }
+sdist = { url = "https://files.pythonhosted.org/packages/69/1b/fe847bae90cafbf752ca450f4ef9b94c496514b9598fd9293ecbcae32792/kirin_toolchain-0.17.33.tar.gz", hash = "sha256:fb47db06c28fa1f833b4d84e2bdb1f332bf7a0312932a1cde377958fb705e4a9", size = 1203455 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/d5/f7/4124f0db718f12e2313c177bed83203cfdf320dd3afb1b4a25f909e19662/kirin_toolchain-0.17.25-py3-none-any.whl", hash = "sha256:f9ddfe6b925437e63ca476a6cb9d4b0025cade5f280b33300e19a71df9c38010", size = 230103 },
+    { url = "https://files.pythonhosted.org/packages/db/6b/b24d49ecbd347edd774fb8b0557bf8a05ce7635c03bfeaf66b087d80ed62/kirin_toolchain-0.17.33-py3-none-any.whl", hash = "sha256:b1605b94df594b6e8da7c2aa2ddfbcbde4b99773b9793fbbaa7c1a321cd829c5", size = 231964 },
 ]
 
 [[package]]

bloqade_circuit-0.7.11/test/stim/passes/stim_reference_programs/qubit/u3_to_clifford.stim

@@ -1,3 +0,0 @@
-
-H 0
-MZ(0.00000000) 0