dialects: (qft) initialise dialect

inconspiquous/dialects/__init__.py

@@ -56,6 +56,11 @@ def get_qec():
 
         return QEC
 
+    def get_qft():
+        from inconspiquous.dialects.qft import QFT
+
+        return QFT
+
     def get_qref():
         from inconspiquous.dialects.qref import Qref
 
@@ -107,6 +112,7 @@ def get_varith():
         "measurement": get_measurement,
         "prob": get_prob,
         "qec": get_qec,
+        "qft": get_qft,
         "qref": get_qref,
         "qubit": get_qubit,
         "qssa": get_qssa,

inconspiquous/dialects/qft.py

@@ -0,0 +1,40 @@
+from xdsl.dialects.builtin import IntAttr
+from xdsl.ir import Dialect
+from xdsl.irdl import irdl_attr_definition, ParameterDef
+from xdsl.parser import AttrParser
+from xdsl.printer import Printer
+
+from inconspiquous.gates import GateAttr
+
+
+@irdl_attr_definition
+class QFTAttr(GateAttr):
+    name = "qft.n"
+
+    num_qubits_param: ParameterDef[IntAttr]
+
+    @property
+    def num_qubits(self) -> int:
+        return self.num_qubits_param.data
+
+    def __init__(self, num_qubits: int | IntAttr):
+        if not isinstance(num_qubits, IntAttr):
+            num_qubits = IntAttr(num_qubits)
+
+        super().__init__((num_qubits,))
+
+    @classmethod
+    def parse_parameters(cls, parser: AttrParser) -> tuple[IntAttr]:
+        with parser.in_angle_brackets():
+            num_qubits = parser.parse_integer(allow_negative=False, allow_boolean=False)
+        return (IntAttr.new(num_qubits),)
+
+    def print_parameters(self, printer: Printer) -> None:
+        return self.num_qubits_param.print_parameter(printer)
+
+
+QFT = Dialect(
+    "qft",
+    [],
+    [QFTAttr],
+)

tests/filecheck/dialects/qft/ops.mlir

@@ -0,0 +1,5 @@
+// RUN: QUOPT_ROUNDTRIP
+
+"test.op"() {gate = #qft.n<5>} : () -> ()
+
+// CHECK: "test.op"() {gate = #qft.n<5>} : () -> ()