bloqade-circuit 0.5.1__tar.gz → 0.5.3__tar.gz

{bloqade_circuit-0.5.1 → bloqade_circuit-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bloqade-circuit
-Version: 0.5.1
+Version: 0.5.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.5.1 → bloqade_circuit-0.5.3}/pyproject.toml

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

{bloqade_circuit-0.5.1 → bloqade_circuit-0.5.3}/src/bloqade/squin/cirq/lowering.py

@@ -281,7 +281,8 @@ class Squin(lowering.LoweringABC[CirqNode]):
             return state.current_frame.push(op.stmts.Z())
 
         # NOTE: just for the Z gate, an arbitrary exponent is equivalent to the ShiftOp
-        t = node.exponent
+        # up to a minus sign!
+        t = -node.exponent
         theta = state.current_frame.push(py.Constant(math.pi * t))
         return state.current_frame.push(op.stmts.ShiftOp(theta=theta.result))
 

{bloqade_circuit-0.5.1 → bloqade_circuit-0.5.3}/src/bloqade/squin/rewrite/U3_to_clifford.py

@@ -126,6 +126,14 @@ class SquinU3ToClifford(RewriteRule):
         if theta is None or phi is None or lam is None:
             return ()
 
+        # For U3(2*pi*n, phi, lam) = U3(0, 0, lam + phi) which is a Z rotation.
+        if np.isclose(np.mod(theta, math.tau), 0):
+            lam = lam + phi
+            phi = 0.0
+        elif np.isclose(np.mod(theta + np.pi, math.tau), 0):
+            lam = lam - phi
+            phi = 0.0
+
         theta_half_pi: int | None = self.resolve_angle(theta)
         phi_half_pi: int | None = self.resolve_angle(phi)
         lam_half_pi: int | None = self.resolve_angle(lam)

{bloqade_circuit-0.5.1 → bloqade_circuit-0.5.3}/test/squin/cirq/test_cirq_to_squin.py

@@ -291,3 +291,60 @@ def test_multiple_classical_controls(run_sim: bool = False):
 
     kernel = squin.cirq.load_circuit(circuit)
     kernel.print()
+
+
+def test_ghz_simulation():
+    q = cirq.LineQubit.range(2)
+
+    # NOTE: uses native gateset
+    circuit = cirq.Circuit(
+        cirq.PhasedXZGate(x_exponent=0.5, z_exponent=0.0, axis_phase_exponent=0.5).on(
+            q[0]
+        ),
+        cirq.PhasedXZGate(x_exponent=0.5, z_exponent=0.0, axis_phase_exponent=0.5).on(
+            q[1]
+        ),
+        cirq.CZ(*q),
+        cirq.PhasedXZGate(x_exponent=1.0, z_exponent=0.0, axis_phase_exponent=0.5).on(
+            q[0]
+        ),
+        cirq.PhasedXZGate(x_exponent=0.5, z_exponent=1.0, axis_phase_exponent=0.5).on(
+            q[1]
+        ),
+    )
+
+    print(circuit)
+
+    # manually written kernel
+    @squin.kernel
+    def manual():
+        q = squin.qubit.new(2)
+        s = squin.op.s()
+        s_adj = squin.op.adjoint(s)
+        squin.qubit.broadcast(s_adj, q)
+        x = squin.op.x()
+        xrot = squin.op.rot(x, math.pi / 2)
+        squin.qubit.broadcast(xrot, q)
+        squin.qubit.broadcast(s, q)
+        cz = squin.op.cz()
+        squin.qubit.apply(cz, q)
+        squin.qubit.broadcast(s_adj, q)
+        squin.qubit.apply(x, q[0])
+        squin.qubit.apply(xrot, q[1])
+        squin.qubit.broadcast(s, q)
+        z = squin.op.z()
+        squin.qubit.apply(z, q[1])
+
+    # lower from circuit
+    kernel = squin.cirq.load_circuit(circuit)
+    cirq_sim = cirq.Simulator().simulate(circuit)
+
+    sim = DynamicMemorySimulator()
+    ket_manual = sim.state_vector(manual)
+
+    ket_kernel = sim.state_vector(kernel)
+
+    for ket in (ket_manual, ket_kernel, cirq_sim.final_state_vector):
+        assert ket[1] == ket[2] == 0
+        assert math.isclose(abs(ket[0]) ** 2, 0.5, abs_tol=1e-5)
+        assert math.isclose(abs(ket[3]) ** 2, 0.5, abs_tol=1e-5)

{bloqade_circuit-0.5.1 → bloqade_circuit-0.5.3}/test/squin/rewrite/test_U3_to_clifford.py

@@ -59,6 +59,18 @@ def test_s():
     assert isinstance(get_stmt_at_idx(test_equiv, 5), op.stmts.S)
 
 
+def test_s_alternative():
+
+    @kernel
+    def test():
+        q = qubit.new(4)
+        oper = op.u(theta=0.0, phi=0.25 * math.tau, lam=0.0)
+        qubit.apply(oper, q[0])
+
+    SquinToCliffordTestPass(test.dialects)(test)
+    assert isinstance(get_stmt_at_idx(test, 5), op.stmts.S)
+
+
 def test_z():
 
     @kernel
@@ -81,6 +93,18 @@ def test_z():
     assert isinstance(get_stmt_at_idx(test, 15), op.stmts.Z)
 
 
+def test_z_alternative():
+
+    @kernel
+    def test():
+        q = qubit.new(4)
+        oper = op.u(theta=0.0, phi=0.5 * math.tau, lam=0.0)
+        qubit.apply(oper, q[0])
+
+    SquinToCliffordTestPass(test.dialects)(test)
+    assert isinstance(get_stmt_at_idx(test, 5), op.stmts.Z)
+
+
 def test_sdag():
 
     @kernel
@@ -104,6 +128,58 @@ def test_sdag():
     assert isinstance(get_stmt_at_idx(test_equiv, 6), op.stmts.Adjoint)
 
 
+def test_sdag_alternative_negative():
+
+    @kernel
+    def test():
+        q = qubit.new(4)
+        oper = op.u(theta=0.0, phi=-0.25 * math.tau, lam=0.0)
+        qubit.apply(oper, q[0])
+
+    SquinToCliffordTestPass(test.dialects)(test)
+    assert isinstance(get_stmt_at_idx(test, 5), op.stmts.S)
+    assert isinstance(get_stmt_at_idx(test, 6), op.stmts.Adjoint)
+
+
+def test_sdag_alternative():
+
+    @kernel
+    def test():
+        q = qubit.new(4)
+        oper = op.u(theta=0.0, phi=0.75 * math.tau, lam=0.0)
+        qubit.apply(oper, q[0])
+
+    SquinToCliffordTestPass(test.dialects)(test)
+    assert isinstance(get_stmt_at_idx(test, 5), op.stmts.S)
+    assert isinstance(get_stmt_at_idx(test, 6), op.stmts.Adjoint)
+
+
+def test_sdag_weird_case():
+
+    @kernel
+    def test():
+        q = qubit.new(4)
+        oper = op.u(theta=2 * math.tau, phi=0.7 * math.tau, lam=0.05 * math.tau)
+        qubit.apply(oper, q[0])
+
+    SquinToCliffordTestPass(test.dialects)(test)
+    assert isinstance(get_stmt_at_idx(test, 5), op.stmts.S)
+    assert isinstance(get_stmt_at_idx(test, 6), op.stmts.Adjoint)
+
+
+def test_sdag_weirder_case():
+
+    @kernel
+    def test():
+        q = qubit.new(4)
+        oper = op.u(theta=0.5 * math.tau, phi=0.05 * math.tau, lam=0.8 * math.tau)
+        qubit.apply(oper, q[0])
+
+    SquinToCliffordTestPass(test.dialects)(test)
+    assert isinstance(get_stmt_at_idx(test, 5), op.stmts.S)
+    assert isinstance(get_stmt_at_idx(test, 6), op.stmts.Adjoint)
+
+
 def test_sqrt_y():
 
     @kernel

{bloqade_circuit-0.5.1 → bloqade_circuit-0.5.3}/uv.lock

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