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

bloqade/squin/cirq/lowering.py

@@ -253,19 +253,29 @@ class Squin(lowering.LoweringABC[CirqNode]):
         return state.current_frame.push(qubit.Apply(op_.result, qargs))
 
     def visit_HPowGate(self, state: lowering.State[CirqNode], node: cirq.HPowGate):
-        if node.exponent == 1:
+        if abs(node.exponent) == 1:
             return state.current_frame.push(op.stmts.H())
 
-        return state.lower(node.in_su2())
+        # NOTE: decompose into products of paulis for arbitrary exponents according to _decompose_ method
+        # can't use decompose directly since that method requires qubits to be passed in for some reason
+        y_rhs = state.lower(cirq.YPowGate(exponent=0.25)).expect_one()
+        x = state.lower(
+            cirq.XPowGate(exponent=node.exponent, global_shift=node.global_shift)
+        ).expect_one()
+        y_lhs = state.lower(cirq.YPowGate(exponent=-0.25)).expect_one()
+
+        # NOTE: reversed order since we're creating a mult stmt
+        m_lhs = state.current_frame.push(op.stmts.Mult(y_lhs, x))
+        return state.current_frame.push(op.stmts.Mult(m_lhs.result, y_rhs))
 
     def visit_XPowGate(self, state: lowering.State[CirqNode], node: cirq.XPowGate):
-        if node.exponent == 1:
+        if abs(node.exponent == 1):
             return state.current_frame.push(op.stmts.X())
 
         return self.visit(state, node.in_su2())
 
     def visit_YPowGate(self, state: lowering.State[CirqNode], node: cirq.YPowGate):
-        if node.exponent == 1:
+        if abs(node.exponent == 1):
             return state.current_frame.push(op.stmts.Y())
 
         return self.visit(state, node.in_su2())
@@ -277,7 +287,7 @@ class Squin(lowering.LoweringABC[CirqNode]):
         if node.exponent == 0.25:
             return state.current_frame.push(op.stmts.T())
 
-        if node.exponent == 1:
+        if abs(node.exponent == 1):
             return state.current_frame.push(op.stmts.Z())
 
         # NOTE: just for the Z gate, an arbitrary exponent is equivalent to the ShiftOp
@@ -390,3 +400,41 @@ class Squin(lowering.LoweringABC[CirqNode]):
         )
 
         return noise_channel
+
+    def visit_DepolarizingChannel(
+        self, state: lowering.State[CirqNode], node: cirq.DepolarizingChannel
+    ):
+        p = state.current_frame.push(py.Constant(node.p)).result
+        return state.current_frame.push(noise.stmts.Depolarize(p))
+
+    def visit_AsymmetricDepolarizingChannel(
+        self, state: lowering.State[CirqNode], node: cirq.AsymmetricDepolarizingChannel
+    ):
+        nqubits = node.num_qubits()
+        if nqubits > 2:
+            raise lowering.BuildError(
+                "AsymmetricDepolarizingChannel applied to more than 2 qubits is not supported!"
+            )
+
+        if nqubits == 1:
+            p_x = state.current_frame.push(py.Constant(node.p_x)).result
+            p_y = state.current_frame.push(py.Constant(node.p_y)).result
+            p_z = state.current_frame.push(py.Constant(node.p_z)).result
+            params = state.current_frame.push(ilist.New(values=(p_x, p_y, p_z))).result
+            return state.current_frame.push(noise.stmts.SingleQubitPauliChannel(params))
+
+        # NOTE: nqubits == 2
+        error_probs = node.error_probabilities
+        paulis = ("I", "X", "Y", "Z")
+        values = []
+        for p1 in paulis:
+            for p2 in paulis:
+                if p1 == p2 == "I":
+                    continue
+
+                p = error_probs.get(p1 + p2, 0.0)
+                p_ssa = state.current_frame.push(py.Constant(p)).result
+                values.append(p_ssa)
+
+        params = state.current_frame.push(ilist.New(values=values)).result
+        return state.current_frame.push(noise.stmts.TwoQubitPauliChannel(params))

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.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bloqade-circuit
-Version: 0.6.0
+Version: 0.6.2
 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.2.dist-info}/RECORD

@@ -13,9 +13,14 @@ bloqade/analysis/measure_id/__init__.py,sha256=J9I58iIyt4IjB36xK6_q9PV1B-40VJ1Gu
 bloqade/analysis/measure_id/analysis.py,sha256=93S3a_Wu7Bt1j217l1hZMY2tLlR00aUYN27o0IhZsqU,1748
 bloqade/analysis/measure_id/impls.py,sha256=ItS9bLMs8UdAkElYgrqD48Xlh9ZyoJq5xDPWcXjWH1c,4823
 bloqade/analysis/measure_id/lattice.py,sha256=WPrn0R79umCH909BFWsUJ64qx9n_3KYimIW5UaXNuGU,1891
-bloqade/cirq_utils/__init__.py,sha256=RZTqtfAviI64lUrA6_XX6Bf83RC8VnKxrFaVx0lcxik,223
+bloqade/cirq_utils/__init__.py,sha256=1DRDCF3PpgJCOr0z7iULdrn3dqm7GLpRGs9AlqE7XA8,280
 bloqade/cirq_utils/lineprog.py,sha256=JosrhfeOHI9FycUT_sYFj8TBzLpo97TL8zK-Ap2U4eQ,11021
-bloqade/cirq_utils/parallelize.py,sha256=jnvZdA22UOkcIWVVulfhlN95Ha5q2ES-b1ZuQz--gOY,13770
+bloqade/cirq_utils/parallelize.py,sha256=E2MsPm61Dkm3n0v4EFPJFGOc4B_Aw01yfhu-ViOKGiA,13812
+bloqade/cirq_utils/noise/__init__.py,sha256=Z7d293ibzsNZyfS0tbL2JrdoypsDm20KE4zwlY6QnVA,559
+bloqade/cirq_utils/noise/_two_zone_utils.py,sha256=3atgRbCMXZh0hu3zXAk7dThe4vljigKg0Z1bKPGM0s4,18293
+bloqade/cirq_utils/noise/conflict_graph.py,sha256=ZUwPWTknrb6SgtZUVPeICn3YA-nUeWQJDuKKX5jL9tE,7179
+bloqade/cirq_utils/noise/model.py,sha256=06Y_BLChOA-PhhAJcWLSgLVAAJoNjOrAujL1YCwcXA0,20590
+bloqade/cirq_utils/noise/transform.py,sha256=tvDt4WMLM8dKPME51y0_peSZk2-jKmjq0urOxm0lWuQ,2309
 bloqade/pyqrack/__init__.py,sha256=lonTS-luJkTVujCCtgdZRC12V7FQdoFcozAI-byXwN0,810
 bloqade/pyqrack/base.py,sha256=9z61PaaAFqCBBwkgsDZSr-qr9IQ5OJ_JUvltmJ7Bgls,4407
 bloqade/pyqrack/device.py,sha256=cOdyT1k0b73QbOsEIu5KqxHW2OAWP86fi_3XGPhaWGA,7134
@@ -37,7 +42,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 +86,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 +128,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
@@ -135,7 +140,7 @@ bloqade/squin/analysis/nsites/analysis.py,sha256=rIe1RU1MZRItcE2aB8DYahLrv73HfD3
 bloqade/squin/analysis/nsites/impls.py,sha256=bVqO4E2hDzfYWwSG0pfj2j8oT1m0C3b42LdSHr4HQlo,2874
 bloqade/squin/analysis/nsites/lattice.py,sha256=ruh0808SHtj3ecuT-C3AZTsLY2j3DRhtezGiTZvcuVs,942
 bloqade/squin/cirq/__init__.py,sha256=fxvBvwX5VNfDmqkeM4GHguLQh53k-PJVsz89Eu0wRXw,8552
-bloqade/squin/cirq/lowering.py,sha256=G3AQ86JMOQ_H4TYn5rDeHiV9nivG8lJU1RH_IqoX4WE,15691
+bloqade/squin/cirq/lowering.py,sha256=F0_skv9lORxUFrhbNaN2ZQpqGnhPyslHt5E92uta5BQ,17959
 bloqade/squin/cirq/emit/emit_circuit.py,sha256=7puJ3eCFwE9VdPb9NAiSdyRNkoQPwo_uVykz9Yv7c14,3761
 bloqade/squin/cirq/emit/noise.py,sha256=rESjGC_66s2Y4FwwYda4rY3mYHYjbqLlKE_vnqpZDYI,1534
 bloqade/squin/cirq/emit/op.py,sha256=z54NP5KqMxffXeFGWamEzvunpTNrxmYuluurk4j2-ps,4000
@@ -223,7 +228,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.2.dist-info/METADATA,sha256=jAkPyCEphiBvZ8PrjxLr8CBFybEdJl9wUY1TNVChGkc,3849
+bloqade_circuit-0.6.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+bloqade_circuit-0.6.2.dist-info/licenses/LICENSE,sha256=S5GIJwR6QCixPA9wryYb44ZEek0Nz4rt_zLUqP05UbU,13160
+bloqade_circuit-0.6.2.dist-info/RECORD,,