[mlir][linalg] Stage application of pad tensor op vectoriztaion.
authorgysit <gysit@google.com>
Mon, 13 Dec 2021 19:49:16 +0000 (19:49 +0000)
committergysit <gysit@google.com>
Mon, 13 Dec 2021 19:49:35 +0000 (19:49 +0000)
commit8fc0525a159b3be3c83f8d80c03585237ea40a73
tree8c16fa8e493f1814c6c707a342c260d822ba6827
parent515afe8b13a659a7a979280ed87891e74cc7915c
[mlir][linalg] Stage application of pad tensor op vectoriztaion.

Adapt the LinalgStrategyVectorizationPattern pass to apply the vectorization patterns in two stages. The change ensures the generic pad tensor op vectorization pattern does not run too early. Additionally, the revision adds the transfer op canonicalization patterns to the set of applied patterns, since they are needed to enable efficient vectorization for rank-reduced convolutions.

Reviewed By: nicolasvasilache

Differential Revision: https://reviews.llvm.org/D115627
mlir/lib/Dialect/Linalg/Transforms/LinalgStrategyPasses.cpp
mlir/test/Dialect/Linalg/codegen-strategy.mlir
mlir/test/lib/Dialect/Linalg/TestLinalgCodegenStrategy.cpp