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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bloqade-circuit
-Version: 0.7.11
+Version: 0.7.12
 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.7.11 → bloqade_circuit-0.7.12}/pyproject.toml

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

{bloqade_circuit-0.7.11 → bloqade_circuit-0.7.12}/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.12/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.12/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.12/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.12}/uv.lock

@@ -103,7 +103,7 @@ wheels = [
 
 [[package]]
 name = "bloqade-circuit"
-version = "0.7.11"
+version = "0.7.12"
 source = { editable = "." }
 dependencies = [
     { name = "kirin-toolchain" },