[CINN} Limit merge non related group with large program (#72550)

huangjiyi · web-flow · commit 5883eefb5714 · 2025-05-08T14:34:14.000+08:00
diff --git a/paddle/fluid/pir/transforms/sub_graph_detector.cc b/paddle/fluid/pir/transforms/sub_graph_detector.cc
@@ -779,8 +779,10 @@ void SubgraphDetector::SubgraphFusion() {
     }
   }
 
-  VLOG(4) << "Merge non-related subgraphs";
   auto subgraph_list = GetSubgraphList();
+  VLOG(4) << "Merge non-related subgraphs (size=" << subgraph_list.size()
+          << ")";
+  if (subgraph_list.size() > 2048) return;
   for (size_t i = 0; i < subgraph_list.size(); ++i) {
     auto lhs = subgraph_list[i];
     if (!lhs->substitute) continue;

Original file line number	Diff line number	Diff line change
`@@ -779,8 +779,10 @@ void SubgraphDetector::SubgraphFusion() {`
`779`	`779`	`}`
`780`	`780`	`}`
`781`	`781`
`782`		`- VLOG(4) << "Merge non-related subgraphs";`
`783`	`782`	`auto subgraph_list = GetSubgraphList();`
	`783`	`+ VLOG(4) << "Merge non-related subgraphs (size=" << subgraph_list.size()`
	`784`	`+ << ")";`
	`785`	`+ if (subgraph_list.size() > 2048) return;`
`784`	`786`	`for (size_t i = 0; i < subgraph_list.size(); ++i) {`
`785`	`787`	`auto lhs = subgraph_list[i];`
`786`	`788`	`if (!lhs->substitute) continue;`