[mlir][linalg][transform] Support dynamic tile sizes in TileToForeachThreadOp
authorMatthias Springer <springerm@google.com>
Mon, 22 Aug 2022 14:48:17 +0000 (16:48 +0200)
committerMatthias Springer <springerm@google.com>
Mon, 22 Aug 2022 14:48:45 +0000 (16:48 +0200)
commit416ba2256d2310abbfc185d1bce53715cf29092f
treea782426f63a450313a4b289504234e209785c9e2
parent45e80d980151252b2ba04c67a97d2def64d165c6
[mlir][linalg][transform] Support dynamic tile sizes in TileToForeachThreadOp

TileToForeachThreadOp now accepts mixed SSA value operands / index attributes for tile_sizes and num_threads. (Reusing OperandsOrIntegersSizesList.) In case of an operand, a PDL_Operation must be specified that is mapped to a payload op that returns the tile size or number of threads.

Differential Revision: https://reviews.llvm.org/D131949
mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td
mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp
mlir/test/Dialect/Linalg/tile-to-foreach-thread.mlir