[Mlir-commits] [mlir] [MLIR][Python] Add structured.fuseop to generator. (PR #120601)
Oleksandr Alex Zinenko
llvmlistbot at llvm.org
Fri Dec 20 00:45:47 PST 2024
================
@@ -140,6 +140,57 @@ def __init__(
)
+ at _ods_cext.register_operation(_Dialect, replace=True)
+class FuseOp(FuseOp):
+ """Specialization for FuseOp class."""
+
+ @overload
+ def __init__(
+ self,
+ target: Union[Operation, Value, OpView],
+ *,
+ sizes: Optional[Union[DynamicIndexList, ArrayAttr]] = None,
+ interchange: OptionalIntList = None,
+ loc=None,
+ ip=None,
+ ):
+ ...
+
+ def __init__(
+ self,
+ loop_types_or_target: Union[Type, List[Type], Operation, Value],
----------------
ftynse wrote:
```suggestion
loop_types_or_target: Union[Type, Sequence[Type], Operation, OpView, Value],
```
https://github.com/llvm/llvm-project/pull/120601
More information about the Mlir-commits
mailing list