Fixed other issues with IZ->IS conversions...

Author: Jason Ludmir

src/bloqade/pyqrack/squin/gate/gate.py

@@ -24,6 +24,7 @@
     Rz,
     SqrtX,
     SqrtY,
+    PhasedXZ,
 )
 
 
@@ -134,3 +135,26 @@ def u3(self, interp: PyQrackInterpreter, frame: interp.Frame, stmt: U3):
                 continue
 
             qbit.sim_reg.u(qbit.addr, theta, phi, lam)
+
+    @interp.impl(PhasedXZ)
+    def phased_xz(
+        self, interp: PyQrackInterpreter, frame: interp.Frame, stmt: PhasedXZ
+    ):
+        x_exponent = frame.get(stmt.x_exponent)
+        z_exponent = frame.get(stmt.z_exponent)
+        axis_phase_exponent = frame.get(stmt.axis_phase_exponent)
+        qubits: ilist.IList[PyQrackQubit, Any] = frame.get(stmt.qubits)
+
+        # Match cirq.PhasedXZGate decomposition:
+        # Rz(-axis_phase) * Rx(x_exponent) * Rz(axis_phase + z_exponent)
+        angle_rz_pre = -axis_phase_exponent * math.pi
+        angle_rx = x_exponent * math.pi
+        angle_rz_post = (axis_phase_exponent + z_exponent) * math.pi
+
+        for qbit in qubits:
+            if not qbit.is_active():
+                continue
+
+            qbit.sim_reg.r(Pauli.PauliZ, angle_rz_pre, qbit.addr)
+            qbit.sim_reg.r(Pauli.PauliX, angle_rx, qbit.addr)
+            qbit.sim_reg.r(Pauli.PauliZ, angle_rz_post, qbit.addr)