bloqade-circuit 0.6.4__py3-none-any.whl → 0.6.5__py3-none-any.whl

bloqade/pyqrack/base.py

@@ -48,7 +48,7 @@ def _default_pyqrack_args() -> PyQrackOptions:
         isSchmidtDecomposeMulti=True,
         isSchmidtDecompose=True,
         isStabilizerHybrid=False,
-        isBinaryDecisionTree=True,
+        isBinaryDecisionTree=False,
         isPaged=True,
         isCpuGpuHybrid=True,
         isOpenCL=True,

bloqade/squin/__init__.py

@@ -9,6 +9,10 @@ from . import (
 )
 from .groups import wired as wired, kernel as kernel
 
+# NOTE: it's important to keep these imports here since they import squin.kernel
+# we skip isort here
+from . import gate as gate, parallel as parallel  # isort: skip
+
 try:
     # NOTE: make sure optional cirq dependency is installed
     import cirq as cirq_package  # noqa: F401

bloqade/squin/gate.py

@@ -0,0 +1,193 @@
+from bloqade.types import Qubit
+
+from . import op as _op, qubit as _qubit
+from .groups import kernel
+
+
+@kernel
+def x(qubit: Qubit) -> None:
+    """x gate applied to qubit."""
+    op = _op.x()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def y(qubit: Qubit) -> None:
+    """y gate applied to qubit."""
+    op = _op.y()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def z(qubit: Qubit) -> None:
+    """z gate applied to qubit."""
+    op = _op.z()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def sqrt_x(qubit: Qubit) -> None:
+    """Square root x gate applied to qubit."""
+    op = _op.sqrt_x()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def sqrt_x_adj(qubit: Qubit) -> None:
+    """Adjoint sqrt_x gate applied to qubit."""
+    op = _op.sqrt_x()
+    _qubit.apply(_op.adjoint(op), qubit)
+
+
+@kernel
+def sqrt_y(qubit: Qubit) -> None:
+    """Square root y gate applied to qubit."""
+    op = _op.sqrt_y()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def sqrt_y_adj(qubit: Qubit) -> None:
+    """Adjoint sqrt_y gate applied to qubit."""
+    op = _op.sqrt_y()
+    _qubit.apply(_op.adjoint(op), qubit)
+
+
+@kernel
+def sqrt_z(qubit: Qubit) -> None:
+    """Square root z gate applied to qubit."""
+    op = _op.s()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def sqrt_z_adj(qubit: Qubit) -> None:
+    """Adjoint square root z gate applied to qubit."""
+    op = _op.s()
+    _qubit.apply(_op.adjoint(op), qubit)
+
+
+@kernel
+def h(qubit: Qubit) -> None:
+    """Hadamard gate applied to qubit."""
+    op = _op.h()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def s(qubit: Qubit) -> None:
+    """s gate applied to qubit."""
+    op = _op.s()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def s_adj(qubit: Qubit) -> None:
+    """Adjoint s gate applied to qubit."""
+    op = _op.s()
+    _qubit.apply(_op.adjoint(op), qubit)
+
+
+@kernel
+def t(qubit: Qubit) -> None:
+    """t gate applied to qubit."""
+    op = _op.t()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def t_adj(qubit: Qubit) -> None:
+    """Adjoint t gate applied to qubit."""
+    op = _op.t()
+    _qubit.apply(_op.adjoint(op), qubit)
+
+
+@kernel
+def p0(qubit: Qubit) -> None:
+    """Projector on 0 applied to qubit."""
+    op = _op.p0()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def p1(qubit: Qubit) -> None:
+    """Projector on 1 applied to qubit."""
+    op = _op.p1()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def spin_n(qubit: Qubit) -> None:
+    """Spin lowering gate applied to qubit."""
+    op = _op.spin_n()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def spin_p(qubit: Qubit) -> None:
+    """Spin raising gate applied to qubit."""
+    op = _op.spin_p()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def reset(qubit: Qubit) -> None:
+    """Reset qubit to 0."""
+    op = _op.reset()
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def cx(control: Qubit, target: Qubit) -> None:
+    """Controlled x gate applied to control and target"""
+    op = _op.cx()
+    _qubit.apply(op, control, target)
+
+
+@kernel
+def cy(control: Qubit, target: Qubit) -> None:
+    """Controlled y gate applied to control and target"""
+    op = _op.cy()
+    _qubit.apply(op, control, target)
+
+
+@kernel
+def cz(control: Qubit, target: Qubit) -> None:
+    """Controlled z gate applied to control and target"""
+    op = _op.cz()
+    _qubit.apply(op, control, target)
+
+
+@kernel
+def ch(control: Qubit, target: Qubit) -> None:
+    """Controlled Hadamard gate applied to control and target"""
+    op = _op.ch()
+    _qubit.apply(op, control, target)
+
+
+@kernel
+def u(theta: float, phi: float, lam: float, qubit: Qubit) -> None:
+    """3D rotation gate applied to control and target"""
+    op = _op.u(theta, phi, lam)
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def rx(theta: float, qubit: Qubit) -> None:
+    """Rotation X gate applied to qubit."""
+    op = _op.rot(_op.x(), theta)
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def ry(theta: float, qubit: Qubit) -> None:
+    """Rotation Y gate applied to qubit."""
+    op = _op.rot(_op.y(), theta)
+    _qubit.apply(op, qubit)
+
+
+@kernel
+def rz(theta: float, qubit: Qubit) -> None:
+    """Rotation Z gate applied to qubit."""
+    op = _op.rot(_op.z(), theta)
+    _qubit.apply(op, qubit)

bloqade/squin/parallel.py

@@ -0,0 +1,200 @@
+from typing import Any, TypeVar
+
+from kirin.dialects import ilist
+
+from bloqade.types import Qubit
+
+from . import op as _op, qubit as _qubit
+from .groups import kernel
+
+
+@kernel
+def x(qubits: ilist.IList[Qubit, Any]) -> None:
+    """x gate applied to qubits in parallel."""
+    op = _op.x()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def y(qubits: ilist.IList[Qubit, Any]) -> None:
+    """y gate applied to qubits in parallel."""
+    op = _op.y()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def z(qubits: ilist.IList[Qubit, Any]) -> None:
+    """z gate applied to qubits in parallel."""
+    op = _op.z()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def sqrt_x(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Square root x gate applied to qubits in parallel."""
+    op = _op.sqrt_x()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def sqrt_y(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Square root y gate applied to qubits in parallel."""
+    op = _op.sqrt_y()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def sqrt_z(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Square root gate applied to qubits in parallel."""
+    op = _op.s()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def h(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Hadamard gate applied to qubits in parallel."""
+    op = _op.h()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def s(qubits: ilist.IList[Qubit, Any]) -> None:
+    """s gate applied to qubits in parallel."""
+    op = _op.s()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def t(qubits: ilist.IList[Qubit, Any]) -> None:
+    """t gate applied to qubits in parallel."""
+    op = _op.t()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def p0(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Projector on 0 applied to qubits in parallel."""
+    op = _op.p0()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def p1(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Projector on 1 applied to qubits in parallel."""
+    op = _op.p1()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def spin_n(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Spin lowering gate applied to qubits in parallel."""
+    op = _op.spin_n()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def spin_p(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Spin raising gate applied to qubits in parallel."""
+    op = _op.spin_p()
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def reset(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Reset qubit to 0."""
+    op = _op.reset()
+    _qubit.broadcast(op, qubits)
+
+
+N = TypeVar("N")
+
+
+@kernel
+def cx(controls: ilist.IList[Qubit, N], targets: ilist.IList[Qubit, N]) -> None:
+    """Controlled x gate applied to controls and targets in parallel."""
+    op = _op.cx()
+    _qubit.broadcast(op, controls, targets)
+
+
+@kernel
+def cy(controls: ilist.IList[Qubit, N], targets: ilist.IList[Qubit, N]) -> None:
+    """Controlled y gate applied to controls and targets in parallel."""
+    op = _op.cy()
+    _qubit.broadcast(op, controls, targets)
+
+
+@kernel
+def cz(controls: ilist.IList[Qubit, N], targets: ilist.IList[Qubit, N]) -> None:
+    """Controlled z gate applied to controls and targets in parallel."""
+    op = _op.cz()
+    _qubit.broadcast(op, controls, targets)
+
+
+@kernel
+def ch(controls: ilist.IList[Qubit, N], targets: ilist.IList[Qubit, N]) -> None:
+    """Controlled Hadamard gate applied to controls and targets in parallel."""
+    op = _op.ch()
+    _qubit.broadcast(op, controls, targets)
+
+
+@kernel
+def u(theta: float, phi: float, lam: float, qubits: ilist.IList[Qubit, Any]) -> None:
+    """3D rotation gate applied to controls and targets in parallel."""
+    op = _op.u(theta, phi, lam)
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def rx(theta: float, qubits: ilist.IList[Qubit, Any]) -> None:
+    """Rotation X gate applied to qubits in parallel."""
+    op = _op.rot(_op.x(), theta)
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def ry(theta: float, qubits: ilist.IList[Qubit, Any]) -> None:
+    """Rotation Y gate applied to qubits in parallel."""
+    op = _op.rot(_op.y(), theta)
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def rz(theta: float, qubits: ilist.IList[Qubit, Any]) -> None:
+    """Rotation Z gate applied to qubits in parallel."""
+    op = _op.rot(_op.z(), theta)
+    _qubit.broadcast(op, qubits)
+
+
+@kernel
+def sqrt_x_adj(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Adjoint sqrt_x gate applied to qubits in parallel."""
+    op = _op.sqrt_x()
+    _qubit.broadcast(_op.adjoint(op), qubits)
+
+
+@kernel
+def sqrt_y_adj(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Adjoint sqrt_y gate applied to qubits in parallel."""
+    op = _op.sqrt_y()
+    _qubit.broadcast(_op.adjoint(op), qubits)
+
+
+@kernel
+def sqrt_z_adj(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Adjoint square root z gate applied to qubits in parallel."""
+    op = _op.s()
+    _qubit.broadcast(_op.adjoint(op), qubits)
+
+
+@kernel
+def s_adj(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Adjoint s gate applied to qubits in parallel."""
+    op = _op.s()
+    _qubit.broadcast(_op.adjoint(op), qubits)
+
+
+@kernel
+def t_adj(qubits: ilist.IList[Qubit, Any]) -> None:
+    """Adjoint t gate applied to qubits in parallel."""
+    op = _op.t()
+    _qubit.broadcast(_op.adjoint(op), qubits)

{bloqade_circuit-0.6.4.dist-info → bloqade_circuit-0.6.5.dist-info}/METADATA

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

@@ -22,7 +22,7 @@ bloqade/cirq_utils/noise/conflict_graph.py,sha256=ZUwPWTknrb6SgtZUVPeICn3YA-nUeW
 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/base.py,sha256=g0GRlEgyJ_P8z-lR8RK2CAuRTj6KPfglKX0iwrgg4DM,4408
 bloqade/pyqrack/device.py,sha256=40vduanEgA26GAW3buHoRpyqPA0xUt2tONY3w5JeH5s,7524
 bloqade/pyqrack/reg.py,sha256=uTL07CT1R0xUsInLmwU9YuuNdV6lV0lCs1zhdUz1qIs,1660
 bloqade/pyqrack/target.py,sha256=c78VtLWAiDNp_0sXwvVzhaEoeFsr1fUVsupxWuo6p3s,3661
@@ -125,10 +125,12 @@ bloqade/rewrite/rules/__init__.py,sha256=3e1Z5T3INqNtP6OU0Vivu_SdMOR_2KDixeA0Yjo
 bloqade/rewrite/rules/flatten_ilist.py,sha256=QoIxMaBXSlatpWzi5s_MAPnV3bV3GeoWc31RBw0WQ3s,1465
 bloqade/rewrite/rules/inline_getitem_ilist.py,sha256=uIXQRCsr3_GPMciDT4ghI-ezhQmkDcGcC6pguABPUVw,875
 bloqade/rewrite/rules/split_ifs.py,sha256=Nm4lpEUHZcnCeewIld0tt7UuGO69LiBGl7Uybuwissw,2119
-bloqade/squin/__init__.py,sha256=MH7i5gR9DhTjLMI6vsP_NT7_yoaEowYiQwsYhrrUEX0,454
+bloqade/squin/__init__.py,sha256=b7ZD69ql9GriIPxN6JhWxANJVzuIJh_r-gC24wsW1mM,621
 bloqade/squin/_typeinfer.py,sha256=bilWfC6whTMwewFCqDgB6vDHZsgXPr3azNOYqqnvtB4,780
+bloqade/squin/gate.py,sha256=tCnjfrSVsXHX7VxkEulZ2SQS5ydtmON8QlcGibM6c2I,4028
 bloqade/squin/groups.py,sha256=RXGJnNZUSXF_f5ljjhZ9At8UhaijayoxFoWvxEsUOWc,1310
 bloqade/squin/lowering.py,sha256=SR6q-IfV8WHPKT97M7UFu5KoRgAojfDno8Bft1mUSKM,1736
+bloqade/squin/parallel.py,sha256=X6Ps9kQIgnFMlZO14y2ntdxvivqbIP28PAWF8KmxByM,5172
 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
@@ -228,7 +230,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.4.dist-info/METADATA,sha256=DXODLKpcteMafxszRYcrC1WliONCL2gKY-9F-g4WftQ,3849
-bloqade_circuit-0.6.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-bloqade_circuit-0.6.4.dist-info/licenses/LICENSE,sha256=S5GIJwR6QCixPA9wryYb44ZEek0Nz4rt_zLUqP05UbU,13160
-bloqade_circuit-0.6.4.dist-info/RECORD,,
+bloqade_circuit-0.6.5.dist-info/METADATA,sha256=8Zb_1qZrO5V9v3JR32KxFQk9malYU1JPPJpkf1U9RPc,3849
+bloqade_circuit-0.6.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+bloqade_circuit-0.6.5.dist-info/licenses/LICENSE,sha256=S5GIJwR6QCixPA9wryYb44ZEek0Nz4rt_zLUqP05UbU,13160
+bloqade_circuit-0.6.5.dist-info/RECORD,,