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

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

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

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

{bloqade_circuit-0.6.0 → bloqade_circuit-0.6.1}/src/bloqade/qasm2/_qasm_loading.py

@@ -82,7 +82,7 @@ def loads(
         body=body,
     )
 
-    return ir.Method(
+    mt = ir.Method(
         mod=None,
         py_func=None,
         sym_name=kernel_name,
@@ -91,6 +91,9 @@ def loads(
         code=code,
     )
 
+    mt.verify()
+    return mt
+
 
 def loadfile(
     qasm_file: str | pathlib.Path,

{bloqade_circuit-0.6.0 → bloqade_circuit-0.6.1}/src/bloqade/qasm2/parse/lowering.py

@@ -202,7 +202,13 @@ class QASM2(lowering.LoweringABC[ast.Node]):
 
             then_body = if_frame.curr_region
 
-        state.current_frame.push(scf.IfElse(cond, then_body=then_body))
+        # NOTE: create empty else body
+        else_body = ir.Block(stmts=[scf.Yield()])
+        else_body.args.append_from(types.Bool)
+
+        state.current_frame.push(
+            scf.IfElse(cond, then_body=then_body, else_body=else_body)
+        )
 
     def branch_next_if_not_terminated(self, frame: lowering.Frame):
         """Branch to the next block if the current block is not terminated.
@@ -381,6 +387,8 @@ class QASM2(lowering.LoweringABC[ast.Node]):
             QubitType for _ in node.qparams
         ]
 
+        self_name = node.name + "_self"
+
         with state.frame(
             stmts=node.body,
             finalize_next=False,
@@ -390,7 +398,7 @@ class QASM2(lowering.LoweringABC[ast.Node]):
                 types.Generic(
                     ir.Method, types.Tuple.where(tuple(arg_types)), types.NoneType
                 ),
-                name=node.name + "_self",
+                name=self_name,
             )
 
             for arg_type, arg_name in zip(arg_types, arg_names):
@@ -422,7 +430,7 @@ class QASM2(lowering.LoweringABC[ast.Node]):
             py_func=None,
             sym_name=node.name,
             dialects=self.dialects,
-            arg_names=[*node.cparams, *node.qparams],
+            arg_names=[self_name, *node.cparams, *node.qparams],
             code=gate_func,
         )
         state.current_frame.globals[node.name] = mt

bloqade_circuit-0.6.1/src/bloqade/squin/lowering.py

@@ -0,0 +1,54 @@
+import ast
+from dataclasses import dataclass
+
+from kirin import lowering
+
+from . import qubit
+
+
+@dataclass(frozen=True)
+class ApplyAnyCallLowering(lowering.FromPythonCall["qubit.ApplyAny"]):
+    """
+    Custom lowering for ApplyAny that collects vararg qubits into a single tuple argument
+    """
+
+    def lower(
+        self, stmt: type["qubit.ApplyAny"], state: lowering.State, node: ast.Call
+    ):
+        if len(node.args) + len(node.keywords) < 2:
+            raise lowering.BuildError(
+                "Apply requires at least one operator and one qubit as arguments!"
+            )
+
+        op, qubits = self.unpack_arguments(node)
+
+        op_ssa = state.lower(op).expect_one()
+        qubits_lowered = [state.lower(qbit).expect_one() for qbit in qubits]
+
+        s = stmt(op_ssa, tuple(qubits_lowered))
+        return state.current_frame.push(s)
+
+    def unpack_arguments(self, node: ast.Call) -> tuple[ast.expr, list[ast.expr]]:
+        if len(node.keywords) == 0:
+            op, *qubits = node.args
+            return op, qubits
+
+        kwargs = {kw.arg: kw.value for kw in node.keywords}
+        if len(kwargs) > 2 or "qubits" not in kwargs:
+            raise lowering.BuildError(f"Got unsupported keyword argument {kwargs}")
+
+        qubits = kwargs["qubits"]
+        if len(kwargs) == 1:
+            if len(node.args) != 1:
+                raise lowering.BuildError("Missing operator argument")
+            op = node.args[0]
+        else:
+            try:
+                op = kwargs["operator"]
+            except KeyError:
+                raise lowering.BuildError(f"Got unsupported keyword argument {kwargs}")
+
+        if isinstance(qubits, ast.List):
+            return op, qubits.elts
+
+        return op, [qubits]

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

@@ -104,3 +104,15 @@ def test_apply_in_for_loop_index_multiple_index():
     sim = StackMemorySimulator(min_qubits=3)
     ket = sim.state_vector(main)
     assert math.isclose(abs(ket[0]) ** 2, 0.5, abs_tol=1e-5)
+
+
+def test_apply_with_named_args():
+    @squin.kernel
+    def main():
+        q = squin.qubit.new(2)
+        h = squin.op.h()
+        squin.qubit.apply(h, qubits=[q[0]])
+        cx = squin.op.cx()
+        squin.qubit.apply(qubits=q, operator=cx)
+
+    main.print()

{bloqade_circuit-0.6.0 → bloqade_circuit-0.6.1}/uv.lock

@@ -102,7 +102,7 @@ wheels = [
 
 [[package]]
 name = "bloqade-circuit"
-version = "0.6.0"
+version = "0.6.1"
 source = { editable = "." }
 dependencies = [
     { name = "kirin-toolchain" },

bloqade_circuit-0.6.0/src/bloqade/squin/lowering.py

@@ -1,27 +0,0 @@
-import ast
-from dataclasses import dataclass
-
-from kirin import lowering
-
-from . import qubit
-
-
-@dataclass(frozen=True)
-class ApplyAnyCallLowering(lowering.FromPythonCall["qubit.ApplyAny"]):
-    """
-    Custom lowering for ApplyAny that collects vararg qubits into a single tuple argument
-    """
-
-    def lower(
-        self, stmt: type["qubit.ApplyAny"], state: lowering.State, node: ast.Call
-    ):
-        if len(node.args) < 2:
-            raise lowering.BuildError(
-                "Apply requires at least one operator and one qubit as arguments!"
-            )
-        op, *qubits = node.args
-        op_ssa = state.lower(op).expect_one()
-        qubits_lowered = [state.lower(qbit).expect_one() for qbit in qubits]
-
-        s = stmt(op_ssa, tuple(qubits_lowered))
-        return state.current_frame.push(s)