Add constructor in YieldOp

xdslproject · May 10, 2024 · 54e27ef · 54e27ef
1 parent 96a8fdf
commit 54e27ef
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/xdsl_pdl/dialects/irdl_extension.py b/xdsl_pdl/dialects/irdl_extension.py
@@ -1,4 +1,5 @@
 from __future__ import annotations
+from typing import Sequence
 
 from xdsl.irdl import (
     AttrSizedOperandSegments,
@@ -7,7 +8,7 @@
     region_def,
     var_operand_def,
 )
-from xdsl.ir import Dialect, Region
+from xdsl.ir import Dialect, Region, SSAValue
 
 from xdsl.dialects.irdl import AttributeType
 from xdsl.parser import DictionaryAttr, Parser
@@ -50,5 +51,15 @@ class YieldOp(IRDLOperation):
 
     assembly_format = "attr-dict $args"
 
+    def __init__(
+        self,
+        args: Sequence[SSAValue],
+        attr_dict: DictionaryAttr | None = None,
+    ):
+        super().__init__(
+            operands=args,
+            attributes=attr_dict.data if attr_dict is not None else None,
+        )
+
 
 IRDLExtension = Dialect("irdl_ext", [CheckSubsetOp, YieldOp])