[Mlir-commits] [mlir] [mlir][linalg] Implement TilingInterface for winograd operators (PR #96184)

[Hsiangkai Wang]

Hsiangkai wrote:

> @Hsiangkai I mentioned it on your first PR, but now that the pieces will have landed for this layout, do you have cycles to work on extending this to work for NCHW convolutions as well? There is a big performance benefit from having the convolutions in NCHW layout with winograd.
> 
> You will need to add some fields to the ops that indicate the layout of the convolution. In IREE we did it by adding a list of indices for where the image dims are: https://github.com/iree-org/iree/blob/b144e905701572d6bd89ea1d4bf55f817b9d3a2b/compiler/src/iree/compiler/Dialect/LinalgExt/IR/LinalgExtOps.td#L1187
> 
> And then you can tell which layout it is based on these dims: https://github.com/iree-org/iree/blob/b144e905701572d6bd89ea1d4bf55f817b9d3a2b/compiler/src/iree/compiler/Dialect/LinalgExt/IR/LinalgExtOps.td#L1251-L1258
> 
> This isn't the only solution (and not necessarily the cleanest either), but just an idea of how it could work.
> 
> Let me know if you would like to work on this :-)

It's not in the plan of my team. I have been assigned to another task. I can try to do it when I have free time but I cannot promise to you.

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