[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