[mlir][Linalg] Add support for lowerPack on dynamic outer shapes.
authorHanhan Wang <hanchung@google.com>
Mon, 1 May 2023 18:23:02 +0000 (11:23 -0700)
committerHanhan Wang <hanchung@google.com>
Thu, 11 May 2023 17:47:19 +0000 (10:47 -0700)
commit9d3057c1cf11759720f4d71f34b4e0e14d273f57
tree7c07bd3a451eacd4d5717961d0ddf64ed3383162
parent1e0966cd6c80dffb084692f265377ef8fa5fbe96
[mlir][Linalg] Add support for lowerPack on dynamic outer shapes.

The revision adds support for tensor.pack op decomposition when all
inner tile sizes are static. The generated tensor.expand_shape op is
still valid because only one of the expanding dimension is dynamic.

Reviewed By: mravishankar

Differential Revision: https://reviews.llvm.org/D150233
mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp
mlir/test/Dialect/Linalg/transform-lower-pack.mlir