[mlir][linalg] Allow outer dims perm and untiled dims in pack/unpack generalization
Extends the pack/unpack generalization patterns to work for any packing op with only full tiles. This produces a combination of rank-reduced insert/extract slice ops paired with a transpose on the reduced shape, similar to what the pattern currently produces for fully tiled pack/unpacks. Note that only the outer dims are rank-reduced in this pattern, leaving the shape of the inner tile intact. Differential Revision: https://reviews.llvm.org/D147555
Loading
Please register or sign in to comment