bloqade-circuit 0.6.0__py3-none-any.whl → 0.6.1__py3-none-any.whl

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/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/squin/lowering.py

@@ -15,13 +15,40 @@ class ApplyAnyCallLowering(lowering.FromPythonCall["qubit.ApplyAny"]):
     def lower(
         self, stmt: type["qubit.ApplyAny"], state: lowering.State, node: ast.Call
     ):
-        if len(node.args) < 2:
+        if len(node.args) + len(node.keywords) < 2:
             raise lowering.BuildError(
                 "Apply requires at least one operator and one qubit as arguments!"
             )
-        op, *qubits = node.args
+
+        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.dist-info → bloqade_circuit-0.6.1.dist-info}/METADATA

@@ -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.dist-info → bloqade_circuit-0.6.1.dist-info}/RECORD

@@ -37,7 +37,7 @@ bloqade/pyqrack/squin/wire.py,sha256=rqlAeU-r_EHOwJMqHrEAxpZ_rKsvUpwGG7MP4BW75Nw
 bloqade/pyqrack/squin/noise/__init__.py,sha256=uXgRQPOrHNRp3k2ff2HD8mheUEaqxZPKEnwV-s4BiV4,31
 bloqade/pyqrack/squin/noise/native.py,sha256=KF4VGzU5Ps92DeLcIDIMsxQQtQ97z_3KUHqBPPkZFaM,2286
 bloqade/qasm2/__init__.py,sha256=W9dR4Qnvigc7e7Ay7puSJHAIuiQk8vWqY-W64SMu5oU,515
-bloqade/qasm2/_qasm_loading.py,sha256=1EFTt1YDkL8fsoSgSuqD1QcKO4EMFIGuBTX9HCnb6S0,4724
+bloqade/qasm2/_qasm_loading.py,sha256=dSfjlB6Bm1ednfxc9L42dEa48s5q4N5SGQ9Iw21sBZQ,4753
 bloqade/qasm2/_wrappers.py,sha256=4x3fldC4sV2K_XZ0FPZOorQKAbs_7pualListXtak4A,11148
 bloqade/qasm2/glob.py,sha256=dDZW2KYXi9e0JmEbpVJIJvQytVEr86G7eism9ghlABM,579
 bloqade/qasm2/groups.py,sha256=3-BGCVqJm6ZDgQeDapac65OLedoskJMVHX78YBKV7jY,2531
@@ -81,7 +81,7 @@ bloqade/qasm2/emit/impls/noise.py,sha256=-N9PmCbz8MwC6xtd55GOpjDoWMyJPJBMVDWT3G8
 bloqade/qasm2/parse/__init__.py,sha256=01tlLfrR015nAAPWw3i_Cs9IXsShpXMnJMagcJ_Vuik,986
 bloqade/qasm2/parse/ast.py,sha256=a48ssf0D_xaE-27PsyeBD5lBvwN2Dojj-RWIBhy7jJE,2904
 bloqade/qasm2/parse/build.py,sha256=2CibD1ZRX3_aknmhb5XvFQcI2sBOn97DlQHomb9CMEw,10621
-bloqade/qasm2/parse/lowering.py,sha256=aBIbvJWOdG5lauPiDhh1uNQ398814INtoWZrUZtpyEU,20879
+bloqade/qasm2/parse/lowering.py,sha256=bVAdctfa9LnWl0pQZZrHyTFwROpTXUfV7ktJPl6AzHg,21101
 bloqade/qasm2/parse/parser.py,sha256=fxqp65dv8NnXE-Ie7ryLESfSH3Xr0unx1EBQysctiHM,121
 bloqade/qasm2/parse/print.py,sha256=PaigQ5RbcfhOteWvDdQHoKsTE3tcNefpVfh1sp5eZEI,8973
 bloqade/qasm2/parse/qasm2.lark,sha256=IYrBydUoVLn1VCNDPP5uNN5BHDET3fQ2yG11cOy900k,2238
@@ -123,7 +123,7 @@ bloqade/rewrite/rules/split_ifs.py,sha256=Nm4lpEUHZcnCeewIld0tt7UuGO69LiBGl7Uybu
 bloqade/squin/__init__.py,sha256=MH7i5gR9DhTjLMI6vsP_NT7_yoaEowYiQwsYhrrUEX0,454
 bloqade/squin/_typeinfer.py,sha256=bilWfC6whTMwewFCqDgB6vDHZsgXPr3azNOYqqnvtB4,780
 bloqade/squin/groups.py,sha256=RXGJnNZUSXF_f5ljjhZ9At8UhaijayoxFoWvxEsUOWc,1310
-bloqade/squin/lowering.py,sha256=w-GyOKYZHHKCGA2slcgWNS97Q_znQU65PeYxEIkvChM,816
+bloqade/squin/lowering.py,sha256=SR6q-IfV8WHPKT97M7UFu5KoRgAojfDno8Bft1mUSKM,1736
 bloqade/squin/qubit.py,sha256=LgNJsm6qCyP7_O-lZg3YT8IiqzF5W5ff1VwQ79nXN4c,5148
 bloqade/squin/types.py,sha256=T3lkqid4HEWuAK_wRns_p-K5DbLDwlldoyZtVay7A3o,119
 bloqade/squin/wire.py,sha256=GZhF0EHCu7OU70zTV_N83yann-eQnYG_lM2u0QYFoAs,6596
@@ -223,7 +223,7 @@ bloqade/visual/animation/runtime/atoms.py,sha256=EmjxhujLiHHPS_HtH_B-7TiqeHgvW5u
 bloqade/visual/animation/runtime/ppoly.py,sha256=JB9IP53N1w6adBJEue6J5Nmj818Id9JvrlgrmiQTU1I,1385
 bloqade/visual/animation/runtime/qpustate.py,sha256=rlmxQeJSvaohXrTpXQL5y-NJcpvfW33xPaYM1slv7cc,4270
 bloqade/visual/animation/runtime/utils.py,sha256=ju9IzOWX-vKwfpqUjlUKu3Ssr_UFPFFq-tzH_Nqyo_c,1212
-bloqade_circuit-0.6.0.dist-info/METADATA,sha256=frsGcfoDPFTPVqO-q6wD0wMPtXtSTxyrERi4krz2x7w,3849
-bloqade_circuit-0.6.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-bloqade_circuit-0.6.0.dist-info/licenses/LICENSE,sha256=S5GIJwR6QCixPA9wryYb44ZEek0Nz4rt_zLUqP05UbU,13160
-bloqade_circuit-0.6.0.dist-info/RECORD,,
+bloqade_circuit-0.6.1.dist-info/METADATA,sha256=UulZPuoA-eaFH24p9sAdzlPQ2exC05Yvl7mdxTtuDJ0,3849
+bloqade_circuit-0.6.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+bloqade_circuit-0.6.1.dist-info/licenses/LICENSE,sha256=S5GIJwR6QCixPA9wryYb44ZEek0Nz4rt_zLUqP05UbU,13160
+bloqade_circuit-0.6.1.dist-info/RECORD,,