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

bloqade/squin/cirq/__init__.py

@@ -157,6 +157,7 @@ def load_circuit(
 def emit_circuit(
     mt: ir.Method,
     qubits: Sequence[cirq.Qid] | None = None,
+    ignore_returns: bool = False,
 ) -> cirq.Circuit:
     """Converts a squin.kernel method to a cirq.Circuit object.
 
@@ -170,6 +171,9 @@ def emit_circuit(
             statement in the order they appear inside the kernel.
             **Note**: If a list of qubits is provided, make sure that there is a sufficient
             number of qubits for the resulting circuit.
+        ignore_returns (bool):
+            If `False`, emitting a circuit from a kernel that returns a value will error.
+            Set it to `True` in order to ignore the return value(s). Defaults to `False`.
 
     ## Examples:
 
@@ -228,11 +232,14 @@ def emit_circuit(
     and manipulate the qubits in other circuits directly written in cirq as well.
     """
 
-    if isinstance(mt.code, func.Function) and not mt.code.signature.output.is_subseteq(
-        types.NoneType
+    if (
+        not ignore_returns
+        and isinstance(mt.code, func.Function)
+        and not mt.code.signature.output.is_subseteq(types.NoneType)
     ):
         raise EmitError(
             "The method you are trying to convert to a circuit has a return value, but returning from a circuit is not supported."
+            " Set `ignore_returns = True` in order to simply ignore the return values and emit a circuit."
         )
 
     emitter = EmitCirq(qubits=qubits)
@@ -244,7 +251,12 @@ def emit_circuit(
     return emitter.run(mt_, args=())
 
 
-def dump_circuit(mt: ir.Method, qubits: Sequence[cirq.Qid] | None = None, **kwargs):
+def dump_circuit(
+    mt: ir.Method,
+    qubits: Sequence[cirq.Qid] | None = None,
+    ignore_returns: bool = False,
+    **kwargs,
+):
     """Converts a squin.kernel method to a cirq.Circuit object and dumps it as JSON.
 
     This just runs `emit_circuit` and calls the `cirq.to_json` function to emit a JSON.
@@ -259,7 +271,10 @@ def dump_circuit(mt: ir.Method, qubits: Sequence[cirq.Qid] | None = None, **kwar
             statement in the order they appear inside the kernel.
             **Note**: If a list of qubits is provided, make sure that there is a sufficient
             number of qubits for the resulting circuit.
+        ignore_returns (bool):
+            If `False`, emitting a circuit from a kernel that returns a value will error.
+            Set it to `True` in order to ignore the return value(s). Defaults to `False`.
 
     """
-    circuit = emit_circuit(mt, qubits=qubits)
+    circuit = emit_circuit(mt, qubits=qubits, ignore_returns=ignore_returns)
     return cirq.to_json(circuit, **kwargs)

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_circuit-0.6.1.dist-info → bloqade_circuit-0.6.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bloqade-circuit
-Version: 0.6.1
+Version: 0.6.3
 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.1.dist-info → bloqade_circuit-0.6.3.dist-info}/RECORD

@@ -13,12 +13,17 @@ 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
+bloqade/pyqrack/device.py,sha256=40vduanEgA26GAW3buHoRpyqPA0xUt2tONY3w5JeH5s,7524
 bloqade/pyqrack/reg.py,sha256=uTL07CT1R0xUsInLmwU9YuuNdV6lV0lCs1zhdUz1qIs,1660
 bloqade/pyqrack/target.py,sha256=c78VtLWAiDNp_0sXwvVzhaEoeFsr1fUVsupxWuo6p3s,3661
 bloqade/pyqrack/task.py,sha256=ydVso4wZQ1iIzgSBG0lDxClPTLcsZfkbnZk_1qFV95o,991
@@ -30,7 +35,7 @@ bloqade/pyqrack/qasm2/glob.py,sha256=EGe7sh9SuvpRW4V4rFcX6Gf7ot8iyThYbsdPeEBKzYM
 bloqade/pyqrack/qasm2/parallel.py,sha256=ITetuXOH2KUDpDOBuFnJoz2DhduvyBC72cOAOOixTaM,1606
 bloqade/pyqrack/qasm2/uop.py,sha256=bLZONsEK15ymFGIQwy7muQv-TX0mvLrECuMp1Y3XTfA,8612
 bloqade/pyqrack/squin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bloqade/pyqrack/squin/op.py,sha256=fHBPV6Q3sKj50K34KfnctacZoNQ8iFz4j5k3qTtRfnI,5123
+bloqade/pyqrack/squin/op.py,sha256=m9QiQgQnKy1tYvrieU6MjXf_b9i2fovj4r2novcQwIc,5535
 bloqade/pyqrack/squin/qubit.py,sha256=svQMbsLxv3yjiFMSRc4C7QGllzjtmlSWOsMY1mjTI8Q,2223
 bloqade/pyqrack/squin/runtime.py,sha256=BM_xRoTpsRipL_Vt-N5UO-x2Nsgiafw3nwiqiuOeH68,15401
 bloqade/pyqrack/squin/wire.py,sha256=rqlAeU-r_EHOwJMqHrEAxpZ_rKsvUpwGG7MP4BW75Nw,1658
@@ -134,8 +139,8 @@ bloqade/squin/analysis/nsites/__init__.py,sha256=RlQg7ivczXCXG5lMeL3ipYKj2oJKC4T
 bloqade/squin/analysis/nsites/analysis.py,sha256=rIe1RU1MZRItcE2aB8DYahLrv73HfD3IHCX3E_EGQ1c,1773
 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/__init__.py,sha256=qGTrEzDEHqIMOh031y23-Y1NsyKew3QCrBYD9P_i63E,9236
+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.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,,
+bloqade_circuit-0.6.3.dist-info/METADATA,sha256=K2Me5w0S2bHxND1_JeYHn5CBC7HjLOdzfpacoATbqWo,3849
+bloqade_circuit-0.6.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+bloqade_circuit-0.6.3.dist-info/licenses/LICENSE,sha256=S5GIJwR6QCixPA9wryYb44ZEek0Nz4rt_zLUqP05UbU,13160
+bloqade_circuit-0.6.3.dist-info/RECORD,,