Rafaelha/adjoint gates squin to sim (#559)

Author: rafaelha

src/bloqade/stim/rewrite/qubit_to_stim.py

@@ -80,7 +80,13 @@ def rewrite_SingleQubitGate(
         stim_stmt_cls = getattr(stim_gate.stmts, stmt_name, None)
         if stim_stmt_cls is None:
             return RewriteResult()
-        stim_stmt = stim_stmt_cls(targets=tuple(qubit_idx_ssas))
+
+        if isinstance(stmt, gate.stmts.SingleQubitNonHermitianGate):
+            stim_stmt = stim_stmt_cls(
+                targets=tuple(qubit_idx_ssas), dagger=stmt.adjoint
+            )
+        else:
+            stim_stmt = stim_stmt_cls(targets=tuple(qubit_idx_ssas))
         stmt.replace_by(stim_stmt)
 
         return RewriteResult(has_done_something=True)

test/stim/passes/stim_reference_programs/qubit/u3_gates.stim

@@ -0,0 +1,10 @@
+
+S 0
+SQRT_Y 0
+S 0
+S_DAG 0
+SQRT_Y 0
+S 0
+S 0
+SQRT_Y 0
+S_DAG 0

test/stim/passes/test_squin_qubit_to_stim.py

@@ -1,5 +1,6 @@
 import os
 import math
+from math import pi
 
 from kirin import ir
 from kirin.dialects import py
@@ -151,6 +152,37 @@ def test():
     assert codegen(test).strip() == "SQRT_Y 0"
 
 
+def test_adjoint_gates_rewrite():
+
+    @sq.kernel
+    def test():
+        q = sq.qalloc(4)
+        sq.s_adj(q[0])
+        sq.sqrt_x_adj(q[1])
+        sq.sqrt_y_adj(q[2])
+        sq.sqrt_z_adj(q[3])  # same as S_DAG
+        return
+
+    SquinToStimPass(test.dialects)(test)
+    assert codegen(test).strip() == "S_DAG 0\nSQRT_X_DAG 1\nSQRT_Y_DAG 2\nS_DAG 3"
+
+
+def test_u3_rewrite():
+
+    @sq.kernel
+    def test():
+        q = sq.qalloc(1)
+
+        sq.u3(-pi / 2, -pi / 2, -pi / 2, q[0])  # S @ SQRT_Y @ S = Z @ SQRT_X
+        sq.u3(-pi / 2, -pi / 2, pi / 2, q[0])  # S @ SQRT_Y @ S_DAG = SQRT_X_DAG
+        sq.u3(-pi / 2, pi / 2, -pi / 2, q[0])  # S_DAG @ SQRT_Y @ S = SQRT_X
+        return
+
+    SquinToStimPass(test.dialects)(test)
+    base_stim_prog = load_reference_program("u3_gates.stim")
+    assert codegen(test) == base_stim_prog.rstrip()
+
+
 def test_for_loop_nontrivial_index_rewrite():
 
     @sq.kernel