use control function

Author: Max191

mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp

@@ -974,6 +974,11 @@ class FoldPadWithProducerReshapeOpByExpansion
     if (!reshapeOp->hasOneUse())
       return failure();
 
+    if (!controlFoldingReshapes(&padOp.getSourceMutable())) {
+      return rewriter.notifyMatchFailure(padOp,
+                                         "fusion blocked by control function");
+    }
+
     ArrayRef<int64_t> low = padOp.getStaticLow();
     ArrayRef<int64_t> high = padOp.getStaticHigh();
     SmallVector<ReassociationIndices> reassociations =
@@ -1778,6 +1783,11 @@ class FoldPadWithProducerReshapeOpByCollapsing
     if (!reshapeOp->hasOneUse())
       return failure();
 
+    if (!controlFoldingReshapes(&padOp.getSourceMutable())) {
+      return rewriter.notifyMatchFailure(padOp,
+                                         "fusion blocked by control function");
+    }
+
     ArrayRef<int64_t> low = padOp.getStaticLow();
     ArrayRef<int64_t> high = padOp.getStaticHigh();
     SmallVector<ReassociationIndices> reassociations =