[mlir][tensor] Fix transpose permutation in tensor.pack generalization pattern
authorQuinn Dawkins <quinn@nod-labs.com>
Mon, 20 Feb 2023 19:49:38 +0000 (14:49 -0500)
committerQuinn Dawkins <quinn@nod-labs.com>
Wed, 22 Feb 2023 19:49:49 +0000 (14:49 -0500)
commitbbf1d80d67db5076e4cd02caa754c0688239fe76
tree4be323db66fd3d5a6c9cd7ab1a7d5d96fc3a733d
parent5c1aabde675849498e5f3620aef1e8750ed58036
[mlir][tensor] Fix transpose permutation in tensor.pack generalization pattern

The generalization pattern for tensor.pack was inverting the
innerDimsPos permutation when normalizing. Thus, the transpose op
produced by the generalization would be incorrect.

Differential Revision: https://reviews.llvm.org/D144425
mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp
mlir/test/Dialect/Linalg/generalize-tensor-pack.mlir