[all-commits] [llvm/llvm-project] ea7551: [mlir][TilingInterface] Enable tile and fuse using...
MaheshRavishankar via All-commits
all-commits at lists.llvm.org
Tue Jun 21 09:47:32 PDT 2022
Author: Mahesh Ravishankar <ravishankarm at google.com>
Date: 2022-06-21 (Tue, 21 Jun 2022)
[mlir][TilingInterface] Enable tile and fuse using TilingInterface.
This patch implements tile and fuse transformation for ops that
implement the tiling interface. To do so,
- `TilingInterface` needs a new method that generates a tiled
implementation of the operation based on the tile of the result
- A pattern is added that replaces a `tensor.extract_slice` whose
source is defined by an operation that implements the
`TilingInterface` with a tiled implementation that produces the
extracted slice in-place (using the method added to
- A pattern is added that takes a sequence of operations that
implement the `TilingInterface` (for now `LinalgOp`s), tiles the
consumer, and greedily fuses its producers iteratively.
Differential Revision: https://reviews.llvm.org/D127809
More information about the All-commits