forked from OSchip/llvm-project
[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:
parent
9982f33e2c
commit
9b67096fe9
|
@ -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));
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue