[Mlir-commits] [mlir] [MLIR][Python] Add structured.fuseop to generator. (PR #120601)

Hugo Trachino llvmlistbot at llvm.org
Thu Jan 2 03:54:46 PST 2025


================
@@ -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,
+    ):
+        ...
----------------
nujaa wrote:

Hi, sorry for the long silence, What do you mean by `you need to explicitly list both overloads` ? Do you mean adding a `@overload` under `...` ? I followed what was done in other examples. The last overload one never has @overload.

* Renamed arguments to `tile_sizes` and `tile_interchange`

https://github.com/llvm/llvm-project/pull/120601


More information about the Mlir-commits mailing list