[mlir][linalg] Extend linalg vectorization to support non-identity input maps
authorthomasraoux <thomasraoux@google.com>
Tue, 16 Mar 2021 21:14:51 +0000 (14:14 -0700)
committerthomasraoux <thomasraoux@google.com>
Thu, 18 Mar 2021 19:32:35 +0000 (12:32 -0700)
commit16947650d5ca602d63d5cd64e68bb0bb0f3674b7
tree8f08e8606ff629f6af3ce61fd2adf0519064b9e2
parent3614df3537f9d699fe0835baf6fc0ddd5c9d699d
[mlir][linalg] Extend linalg vectorization to support non-identity input maps

This propagates the affine map to transfer_read op in case it is not a
minor identity map.

Differential Revision: https://reviews.llvm.org/D98523
mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
mlir/lib/Dialect/Vector/VectorOps.cpp
mlir/lib/IR/AffineMap.cpp
mlir/test/Dialect/Linalg/vectorization.mlir