Add cirq emit methods for new and missing statements (#372)

Implements part of #352

Author: david-pl

src/bloqade/squin/cirq/emit/op.py

@@ -9,6 +9,7 @@
     SnRuntime,
     SpRuntime,
     U3Runtime,
+    RotRuntime,
     KronRuntime,
     MultRuntime,
     ScaleRuntime,
@@ -123,3 +124,21 @@ def pauli_string(
         self, emit: EmitCirq, frame: EmitCirqFrame, stmt: op.stmts.PauliString
     ):
         return (PauliStringRuntime(stmt.string),)
+
+    @impl(op.stmts.Rot)
+    def rot(self, emit: EmitCirq, frame: EmitCirqFrame, stmt: op.stmts.Rot):
+        axis_op: HermitianRuntime = frame.get(stmt.axis)
+        angle = frame.get(stmt.angle)
+
+        axis_name = str(axis_op.gate).lower()
+        return (RotRuntime(axis=axis_name, angle=angle),)
+
+    @impl(op.stmts.SqrtX)
+    def sqrt_x(self, emit: EmitCirq, frame: EmitCirqFrame, stmt: op.stmts.SqrtX):
+        cirq_op = cirq.XPowGate(exponent=0.5)
+        return (UnitaryRuntime(cirq_op),)
+
+    @impl(op.stmts.SqrtY)
+    def sqrt_y(self, emit: EmitCirq, frame: EmitCirqFrame, stmt: op.stmts.SqrtY):
+        cirq_op = cirq.YPowGate(exponent=0.5)
+        return (UnitaryRuntime(cirq_op),)

src/bloqade/squin/cirq/emit/runtime.py

@@ -240,3 +240,18 @@ def unsafe_apply(
             qbit: pauli_label for (qbit, pauli_label) in zip(qubits, self.string)
         }
         return [cirq.PauliString(pauli_mapping)]
+
+
+@dataclass
+class RotRuntime(OperatorRuntimeABC):
+    axis: str
+    angle: float
+
+    def num_qubits(self) -> int:
+        return 1
+
+    def unsafe_apply(
+        self, qubits: Sequence[cirq.Qid], adjoint: bool = False
+    ) -> list[cirq.Operation]:
+        rot = getattr(cirq, "R" + self.axis.lower())(rads=self.angle)
+        return [rot(*qubits)]

test/squin/cirq/test_squin_to_cirq.py

@@ -349,6 +349,32 @@ def main():
     assert len(target._cached_circuit_operations) == 2
 
 
+def test_additional_stmts():
+    @squin.kernel
+    def main():
+        x = squin.op.x()
+        r = squin.op.rot(x, 0.123)
+        q = squin.qubit.new(3)
+        squin.qubit.apply(r, q[0])
+        sqrt_x = squin.op.sqrt_x()
+        sqrt_y = squin.op.sqrt_y()
+        squin.qubit.apply(sqrt_x, q[1])
+        squin.qubit.apply(sqrt_y, q[2])
+
+    main.print()
+
+    circuit = squin.cirq.emit_circuit(main)
+
+    q = cirq.LineQubit.range(3)
+    expected_circuit = cirq.Circuit(
+        cirq.Rx(rads=0.123).on(q[0]),
+        cirq.X(q[1]) ** 0.5,
+        cirq.Y(q[2]) ** 0.5,
+    )
+
+    assert circuit == expected_circuit
+
+
 def test_return_measurement():
 
     @squin.kernel