[Mlir-commits] [mlir] [mlir][python]Add sugared buider for transform.named_sequence (PR #71597)
Nicolas Vasilache
llvmlistbot at llvm.org
Wed Nov 8 00:49:56 PST 2023
================
@@ -165,6 +165,34 @@ def bodyExtraArgs(self) -> BlockArgumentList:
return self.body.arguments[1:]
+ at _ods_cext.register_operation(_Dialect, replace=True)
+class NamedSequenceOp(NamedSequenceOp):
+ def __init__(
+ self,
+ sym_name,
+ input_types: Sequence[Type],
+ result_types: Sequence[Type],
+ ):
+ function_type = FunctionType.get(input_types, result_types)
+ super().__init__(
+ sym_name=sym_name,
+ function_type=TypeAttr.get(function_type),
+ )
+ self.regions[0].blocks.append(*input_types)
+
+ @property
+ def body(self) -> Block:
+ return self.regions[0].blocks[0]
+
+ @property
+ def bodyTarget(self) -> Value:
----------------
nicolasvasilache wrote:
hmm I followed the existing here, I'll send a followup NFC that does the renaming more globally as I think it will break some downstream stuff..
https://github.com/llvm/llvm-project/pull/71597
More information about the Mlir-commits
mailing list