[CINN] Add input output maximun constraint in anchorfusion (#70873)

huangjiyi · web-flow · commit 4cee31952226 · 2025-01-17T17:03:58.000+08:00
diff --git a/paddle/cinn/operator_fusion/pattern_graph.cc b/paddle/cinn/operator_fusion/pattern_graph.cc
@@ -231,17 +231,19 @@ void PatternGraph::LimitedAnchorFusion() {
       ->DisableStrategy(ItersTransformType::ReuseIters)
       ->DisableStrategy(ItersTransformType::AppendIters);
 
-  GraphTransformer<ReverseTopoNodePairPattern,
-                   CanFuseItersPermutationMatcher,
-                   FuseItersPermutatioOperation>(this);
+  GraphTransformer<
+      ReverseTopoNodePairPattern,
+      And<CanFuseItersPermutationMatcher, InputOutputMaximumConstrain>,
+      FuseItersPermutatioOperation>(this);
 }
 
 void PatternGraph::ItersPermutationFusion() {
   iters_fusion_policy()->EnableAllStrategies();
 
-  GraphTransformer<ReverseTopoNodePairPattern,
-                   CanFuseItersPermutationMatcher,
-                   FuseItersPermutatioOperation>(this);
+  GraphTransformer<
+      ReverseTopoNodePairPattern,
+      And<CanFuseItersPermutationMatcher, InputOutputMaximumConstrain>,
+      FuseItersPermutatioOperation>(this);
 }
 
 void PatternGraph::SplitRecomputePattern() {
