quantum -> gate

inconspiquous/dialects/__init__.py

@@ -31,10 +31,10 @@ def get_linalg():
 
         return Linalg
 
-    def get_quantum():
-        from inconspiquous.dialects.quantum import Quantum
+    def get_gate():
+        from inconspiquous.dialects.gate import Gate
 
-        return Quantum
+        return Gate
 
     def get_scf():
         from xdsl.dialects.scf import Scf
@@ -61,8 +61,8 @@ def get_test():
         "builtin": get_builtin,
         "cf": get_cf,
         "func": get_func,
+        "gate": get_gate,
         "linalg": get_linalg,
-        "quantum": get_quantum,
         "scf": get_scf,
         "stim": get_stim,
         "tensor": get_tensor,

inconspiquous/dialects/quantum.py → inconspiquous/dialects/gate.py

@@ -17,7 +17,7 @@ class AngleAttr(ParametrizedAttribute):
     [0, 2) and implicitly multiplying by pi
     """
 
-    name = "quantum.angle"
+    name = "gate.angle"
     data: ParameterDef[FloatAttr[Float64Type]]
 
     def __init__(self, f: float):
@@ -71,37 +71,37 @@ def __neg__(self) -> AngleAttr:
 
 @irdl_attr_definition
 class HadamardGate(SingleQubitGate):
-    name = "quantum.h"
+    name = "gate.h"
 
 
 @irdl_attr_definition
 class XGate(SingleQubitGate):
-    name = "quantum.x"
+    name = "gate.x"
 
 
 @irdl_attr_definition
 class YGate(SingleQubitGate):
-    name = "quantum.y"
+    name = "gate.y"
 
 
 @irdl_attr_definition
 class ZGate(SingleQubitGate):
-    name = "quantum.z"
+    name = "gate.z"
 
 
 @irdl_attr_definition
 class PhaseGate(SingleQubitGate):
-    name = "quantum.s"
+    name = "gate.s"
 
 
 @irdl_attr_definition
 class TGate(SingleQubitGate):
-    name = "quantum.t"
+    name = "gate.t"
 
 
 @irdl_attr_definition
 class RZGate(SingleQubitGate):
-    name = "quantum.rz"
+    name = "gate.rz"
 
     angle: ParameterDef[AngleAttr]
 
@@ -121,24 +121,24 @@ def print_parameters(self, printer: Printer) -> None:
 
 @irdl_attr_definition
 class CNotGate(TwoQubitGate):
-    name = "quantum.cnot"
+    name = "gate.cnot"
 
 
 @irdl_attr_definition
 class CZGate(TwoQubitGate):
-    name = "quantum.cz"
+    name = "gate.cz"
 
 
 @irdl_attr_definition
 class ToffoliGate(GateAttr):
-    name = "quantum.toffoli"
+    name = "gate.toffoli"
 
     def num_qubits(self) -> int:
         return 3
 
 
-Quantum = Dialect(
-    "quantum",
+Gate = Dialect(
+    "gate",
     [],
     [
         AngleAttr,

tests/filecheck/dialects/gate.mlir

@@ -0,0 +1,41 @@
+// RUN: QUOPT_ROUNDTRIP
+
+"test.op"() {"angle" = #gate.angle<0>} : () -> ()
+
+// CHECK: "test.op"() {"angle" = #gate.angle<0>} : () -> ()
+
+"test.op"() {"angle" = #gate.angle<pi>} : () -> ()
+
+// CHECK-NEXT: "test.op"() {"angle" = #gate.angle<pi>} : () -> ()
+
+"test.op"() {"angle" = #gate.angle<2pi>} : () -> ()
+
+// CHECK-NEXT: "test.op"() {"angle" = #gate.angle<0>} : () -> ()
+
+"test.op"() {"angle" = #gate.angle<0.5pi>} : () -> ()
+
+// CHECK-NEXT: "test.op"() {"angle" = #gate.angle<0.5pi>} : () -> ()
+
+"test.op"() {"angle" = #gate.angle<1.5pi>} : () -> ()
+
+// CHECK-NEXT: "test.op"() {"angle" = #gate.angle<1.5pi>} : () -> ()
+
+"test.op"() {"angle" = #gate.angle<2.5pi>} : () -> ()
+
+// CHECK-NEXT: "test.op"() {"angle" = #gate.angle<0.5pi>} : () -> ()
+
+"test.op"() {"angle" = #gate.angle<-0.5pi>} : () -> ()
+
+// CHECK-NEXT: "test.op"() {"angle" = #gate.angle<1.5pi>} : () -> ()
+
+"test.op"() {"angle" = #gate.angle<-pi>} : () -> ()
+
+// CHECK-NEXT: "test.op"() {"angle" = #gate.angle<pi>} : () -> ()
+
+"test.op"() {"gate" = #gate.h} : () -> ()
+
+// CHECK-NEXT: "test.op"() {"gate" = #gate.h} : () -> ()
+
+"test.op"() {"gate" = #gate.rz<pi>} : () -> ()
+
+// CHECK-NEXT: "test.op"() {"gate" = #gate.rz<pi>} : () -> ()