[mlir] Linalg: add vector transfer lowering patterns to the contraction lowering

This fixes a performance regression in vec-mat vectorization

Reviewed By: asaadaldien

Differential Revision: https://reviews.llvm.org/D101795
This commit is contained in:
Eugene Zhulenev 2021-05-03 16:06:43 -07:00
parent 9982f33e2c
commit 9b67096fe9
1 changed files with 2 additions and 0 deletions

View File

@ -89,6 +89,8 @@ void mlir::linalg::CodegenStrategy::transform(FuncOp func) const {
.add<ContractionOpToOuterProductOpLowering,
ContractionOpToMatmulOpLowering, ContractionOpLowering>(
vectorTransformsOptions, context);
vector::populateVectorTransferLoweringPatterns(
vectorContractLoweringPatterns);
(void)applyPatternsAndFoldGreedily(
func, std::move(vectorContractLoweringPatterns));
}