Fix order of glob.u decorator yet again (to match original UGate class args)

Author: david-pl

src/bloqade/qasm2/dialects/glob.py

@@ -14,10 +14,10 @@
 class UGate(ir.Statement):
     name = "ugate"
     traits = frozenset({lowering.FromPythonCall()})
+    registers: ir.SSAValue = info.argument(ilist.IListType[QRegType])
     theta: ir.SSAValue = info.argument(types.Float)
     phi: ir.SSAValue = info.argument(types.Float)
     lam: ir.SSAValue = info.argument(types.Float)
-    registers: ir.SSAValue = info.argument(ilist.IListType[QRegType])
 
 
 @dialect.register(key="qasm2.schedule.dag")

src/bloqade/qasm2/glob.py

@@ -11,7 +11,7 @@
 
 @wraps(glob.UGate)
 def u(
-    theta: float, phi: float, lam: float, registers: ilist.IList[QReg, Any] | list
+    registers: ilist.IList[QReg, Any] | list, theta: float, phi: float, lam: float
 ) -> None:
     """Apply a U gate to all qubits in the input registers.
 

test/pyqrack/test_target.py

@@ -46,7 +46,7 @@ def global_h():
         q = qasm2.qreg(3)
 
         # rotate around Y by pi/2, i.e. perform a hadamard
-        qasm2.glob.u(math.pi / 2.0, 0, 0, [q])
+        qasm2.glob.u([q], math.pi / 2.0, 0, 0)
 
         return q
 
@@ -72,13 +72,28 @@ def multiple_registers():
         q3 = qasm2.qreg(2)
 
         # hadamard on first register
-        qasm2.glob.u(math.pi / 2.0, 0, 0, [q1])
+        qasm2.glob.u(
+            [q1],
+            math.pi / 2.0,
+            0,
+            0,
+        )
 
         # apply hadamard to the other two
-        qasm2.glob.u(math.pi / 2.0, 0, 0, [q2, q3])
+        qasm2.glob.u(
+            [q2, q3],
+            math.pi / 2.0,
+            0,
+            0,
+        )
 
         # rotate all of them back down
-        qasm2.glob.u(-math.pi / 2.0, 0, 0, [q1, q2, q3])
+        qasm2.glob.u(
+            [q1, q2, q3],
+            -math.pi / 2.0,
+            0,
+            0,
+        )
 
         return q1
 

test/qasm2/passes/test_heuristic_noise.py

@@ -265,7 +265,7 @@ def test_global_noise():
     def test_method():
         q0 = qasm2.qreg(1)
         q1 = qasm2.qreg(1)
-        glob.UGate([q0, q1], 0.1, 0.2, 0.3)
+        qasm2.glob.u([q0, q1], 0.1, 0.2, 0.3)
 
     px = 0.01
     py = 0.01