[Mlir-commits] [mlir] a529594 - Add memref.dim canonicalization patterns to TilingCanonicalizationPatterns

[Ahmed Taei]

Author: Ahmed Taei
Date: 2021-06-04T13:40:36-07:00
New Revision: a52959401d9544ac7037405aaf714bd08f071ea9

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

LOG: Add memref.dim canonicalization patterns to TilingCanonicalizationPatterns

Otherwise tiled and padded linalg op will be alive (after distribution).

Differential Revision: https://reviews.llvm.org/D103715

Added: 
    

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

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp b/mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
index b46ac20ec6b8..c88214cfbfd4 100644
--- a/mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
+++ b/mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
@@ -414,6 +414,7 @@ void mlir::linalg::populateLinalgTilingCanonicalizationPatterns(
   memref::SubViewOp::getCanonicalizationPatterns(patterns, ctx);
   tensor::CastOp::getCanonicalizationPatterns(patterns, ctx);
   memref::ViewOp::getCanonicalizationPatterns(patterns, ctx);
+  memref::DimOp::getCanonicalizationPatterns(patterns, ctx);
   CanonicalizationPatternList<
 #define GET_OP_LIST
 #include "mlir/Dialect/Linalg/IR/LinalgStructuredOps.cpp.inc"