[ML Inliner] Fix inconsistancy between CallGraph and FunctionLevels

DataCorrupted · DataCorrupted · commit 7355c600342f · 2025-01-13T16:26:43.000-08:00
When building our internal code, one of the function annotated with `co_await` (c++20) will cause inconsistancy:
The function exists in CG but not FunctionLevels.

This patch fixes it by using FunctionLevels only.

Signed-off-by: Peter Rong &lt;PeterRong@meta.com&gt;
diff --git a/llvm/lib/Analysis/MLInlineAdvisor.cpp b/llvm/lib/Analysis/MLInlineAdvisor.cpp
@@ -189,7 +189,9 @@ MLInlineAdvisor::MLInlineAdvisor(
 }
 
 unsigned MLInlineAdvisor::getInitialFunctionLevel(const Function &F) const {
-  return CG.lookup(F) ? FunctionLevels.at(CG.lookup(F)) : 0;
+  return FunctionLevels.find(CG.lookup(F)) != FunctionLevels.end()
+             ? FunctionLevels.at(CG.lookup(F))
+             : 0;
 }
 
 void MLInlineAdvisor::onPassEntry(LazyCallGraph::SCC *CurSCC) {

Original file line number	Diff line number	Diff line change
`@@ -189,7 +189,9 @@ MLInlineAdvisor::MLInlineAdvisor(`
`189`	`189`	`}`
`190`	`190`
`191`	`191`	`unsigned MLInlineAdvisor::getInitialFunctionLevel(const Function &F) const {`
`192`		`- return CG.lookup(F) ? FunctionLevels.at(CG.lookup(F)) : 0;`
	`192`	`+ return FunctionLevels.find(CG.lookup(F)) != FunctionLevels.end()`
	`193`	`+ ? FunctionLevels.at(CG.lookup(F))`
	`194`	`+ : 0;`
`193`	`195`	`}`
`194`	`196`
`195`	`197`	`void MLInlineAdvisor::onPassEntry(LazyCallGraph::SCC *CurSCC) {`