updating the names of statement fields to match squin

Author: weinbe58

src/bloqade/native/dialects/gate/stmts.py

@@ -12,20 +12,20 @@
 @statement(dialect=dialect)
 class CZ(ir.Statement):
     traits = frozenset({lowering.FromPythonCall()})
-    ctrls: ir.SSAValue = info.argument(ilist.IListType[QubitType, N])
-    qargs: ir.SSAValue = info.argument(ilist.IListType[QubitType, N])
+    controls: ir.SSAValue = info.argument(ilist.IListType[QubitType, N])
+    targets: ir.SSAValue = info.argument(ilist.IListType[QubitType, N])
 
 
 @statement(dialect=dialect)
 class R(ir.Statement):
     traits = frozenset({lowering.FromPythonCall()})
-    inputs: ir.SSAValue = info.argument(ilist.IListType[QubitType, types.Any])
+    qubits: ir.SSAValue = info.argument(ilist.IListType[QubitType, types.Any])
     axis_angle: ir.SSAValue = info.argument(types.Float)
     rotation_angle: ir.SSAValue = info.argument(types.Float)
 
 
 @statement(dialect=dialect)
 class Rz(ir.Statement):
     traits = frozenset({lowering.FromPythonCall()})
-    inputs: ir.SSAValue = info.argument(ilist.IListType[QubitType, types.Any])
+    qubits: ir.SSAValue = info.argument(ilist.IListType[QubitType, types.Any])
     rotation_angle: ir.SSAValue = info.argument(types.Float)

src/bloqade/pyqrack/native.py

@@ -7,26 +7,26 @@
 from pyqrack import Pauli
 from bloqade.pyqrack import PyQrackQubit
 from bloqade.pyqrack.base import PyQrackInterpreter
-from bloqade.native.dialects import gates
+from bloqade.native.dialects import gate
 
 
-@gates.dialect.register(key="pyqrack")
+@gate.dialect.register(key="pyqrack")
 class NativeMethods(interp.MethodTable):
 
-    @interp.impl(gates.CZ)
-    def cz(self, _interp: PyQrackInterpreter, frame: interp.Frame, stmt: gates.CZ):
-        ctrls = frame.get_casted(stmt.ctrls, ilist.IList[PyQrackQubit, Any])
-        qargs = frame.get_casted(stmt.qargs, ilist.IList[PyQrackQubit, Any])
+    @interp.impl(gate.CZ)
+    def cz(self, _interp: PyQrackInterpreter, frame: interp.Frame, stmt: gate.CZ):
+        ctrls = frame.get_casted(stmt.controls, ilist.IList[PyQrackQubit, Any])
+        qargs = frame.get_casted(stmt.targets, ilist.IList[PyQrackQubit, Any])
 
         for ctrl, qarg in zip(ctrls, qargs):
             if ctrl.is_active() and qarg.is_active():
                 ctrl.sim_reg.mcz([ctrl.addr], qarg.addr)
 
         return ()
 
-    @interp.impl(gates.R)
-    def r(self, _interp: PyQrackInterpreter, frame: interp.Frame, stmt: gates.R):
-        inputs = frame.get_casted(stmt.inputs, ilist.IList[PyQrackQubit, Any])
+    @interp.impl(gate.R)
+    def r(self, _interp: PyQrackInterpreter, frame: interp.Frame, stmt: gate.R):
+        inputs = frame.get_casted(stmt.qubits, ilist.IList[PyQrackQubit, Any])
         rotation_angle = 2 * math.pi * frame.get_casted(stmt.rotation_angle, float)
         axis_angle = 2 * math.pi * frame.get_casted(stmt.axis_angle, float)
         for qubit in inputs:
@@ -37,9 +37,9 @@ def r(self, _interp: PyQrackInterpreter, frame: interp.Frame, stmt: gates.R):
 
         return ()
 
-    @interp.impl(gates.Rz)
-    def rz(self, _interp: PyQrackInterpreter, frame: interp.Frame, stmt: gates.Rz):
-        inputs = frame.get_casted(stmt.inputs, ilist.IList[PyQrackQubit, Any])
+    @interp.impl(gate.Rz)
+    def rz(self, _interp: PyQrackInterpreter, frame: interp.Frame, stmt: gate.Rz):
+        inputs = frame.get_casted(stmt.qubits, ilist.IList[PyQrackQubit, Any])
         rotation_angle = 2 * math.pi * frame.get_casted(stmt.rotation_angle, float)
 
         for qubit in inputs: