[Mlir-commits] [mlir] 93f1b48 - [mlir] Set pattern that resolves tensor dimensions as having bounded recursion.

[Mahesh Ravishankar]

Author: Mahesh Ravishankar
Date: 2022-10-20T16:59:40Z
New Revision: 93f1b48cbbeae0a04be560a98daa16350a3da069

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

LOG: [mlir] Set pattern that resolves tensor dimensions as having bounded recursion.

Reviewed By: springerm

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

Added: 
    

Modified: 
    mlir/lib/Dialect/MemRef/Transforms/ResolveShapedTypeResultDims.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/MemRef/Transforms/ResolveShapedTypeResultDims.cpp b/mlir/lib/Dialect/MemRef/Transforms/ResolveShapedTypeResultDims.cpp
index 5ce1ce5bf602d..3d52b1319957a 100644
--- a/mlir/lib/Dialect/MemRef/Transforms/ResolveShapedTypeResultDims.cpp
+++ b/mlir/lib/Dialect/MemRef/Transforms/ResolveShapedTypeResultDims.cpp
@@ -76,6 +76,8 @@ template <typename OpTy>
 struct DimOfReifyRankedShapedTypeOpInterface : public OpRewritePattern<OpTy> {
   using OpRewritePattern<OpTy>::OpRewritePattern;
 
+  void initialize() { OpRewritePattern<OpTy>::setHasBoundedRewriteRecursion(); }
+
   LogicalResult matchAndRewrite(OpTy dimOp,
                                 PatternRewriter &rewriter) const override {
     OpResult dimValue = dimOp.getSource().template dyn_cast<OpResult>();