[Mlir-commits] [mlir] 9b67096 - [mlir] Linalg: add vector transfer lowering patterns to the contraction lowering

[Eugene Zhulenev]

Author: Eugene Zhulenev
Date: 2021-05-03T16:21:51-07:00
New Revision: 9b67096fe952250d081bf0127b357d41dd332b03

URL: https://github.com/llvm/llvm-project/commit/9b67096fe952250d081bf0127b357d41dd332b03
DIFF: https://github.com/llvm/llvm-project/commit/9b67096fe952250d081bf0127b357d41dd332b03.diff

LOG: [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

Added: 
    

Modified: 
    mlir/lib/Dialect/Linalg/Transforms/CodegenStrategy.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Linalg/Transforms/CodegenStrategy.cpp b/mlir/lib/Dialect/Linalg/Transforms/CodegenStrategy.cpp
index e31a6b5210e3..93a8b475a94c 100644
--- a/mlir/lib/Dialect/Linalg/Transforms/CodegenStrategy.cpp
+++ b/mlir/lib/Dialect/Linalg/Transforms/CodegenStrategy.cpp
@@ -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));
   }