[Mlir-commits] [mlir] [MLIR][SCF] Add an API to fuse consumer to a producer within scf loop (PR #88712)

[Abhishek Varma]

================
@@ -131,6 +150,42 @@ def TilingInterface : OpInterface<"TilingInterface"> {
           return failure();
         }]
       >,
+      InterfaceMethod<
+        /*desc=*/[{
+          Method to generate the tiled implementation of an operation from
+          operand tile position.
+
+          Generates the IR that computes the tiled implementation of an
+          operation from operand tile.  The `offsets` and `sizes`
+          describe the tile of the operand required. This is different from
+          `getTiledImplementation` which generates the tiled
+          implementation of the operation given a tile of the
+          iteration space. This method generates a tiled
+          implementation of the operation based on the tile of the
+          operand required. This method enables consumer fusion by using
+          tile and fuse. The method returns failure if the operation
+          can't be tiled to generate the operand tile. In practical terms
+          this implies it cannot be tiled and fused with its producers.
+
+          - `offsets` provides the offset of the tile in the coordinate system
+            of the original iteration space, i.e., if an iteration space
+            dimension had non-zero offset, it must be included in the offset
+            provided here (as opposed to zero-based offset "relative" to the
+            iteration space).
----------------
Abhishek-Varma wrote:

> Please update the description then. Somebody will be reading it and implementing it.

Done.

> > From what I understand the non-zero offsets perhaps mean to account for, as an example, the following :
%iv0 and %iv1 loop induction variables in %source [%iv0, %iv1] [32, 32] [1, 1].

> Is there a test for this case?

Hi @ftynse - here is the [test case](https://github.com/llvm/llvm-project/blob/611cb4e750c1975455f30c7865199140ba687fb0/mlir/test/Interfaces/TilingInterface/tile-and-fuse-consumer.mlir#L67).

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