[mlgo] Disable accounting upon ForceStop

[Northbadge]

Summary

Once ForceStop is set to true, we only return positive inlining advice when it is mandatory; There is no need for further node/edge accounting.

llvm/lib/Analysis/MLInlineAdvisor.cpp

MLInlineAdvisor::MLInlineAdvisor(Module &M, ModuleAnalysisManager &MAM,
   NodeCount = AllNodes.size();
 }
 
 unsigned MLInlineAdvisor::getInitialFunctionLevel(const Function &F) const {
   return CG.lookup(F) ? FunctionLevels.at(CG.lookup(F)) : 0;
 }
 
 void MLInlineAdvisor::onPassEntry() {
+  if (ForceStop)
+    return;
   FPICache.clear();
   // Function passes executed between InlinerPass runs may have changed the
   // module-wide features.
   // The cgscc pass manager rules are such that:
   // - if a pass leads to merging SCCs, then the pipeline is restarted on the
   // merged SCC
   // - if a pass leads to splitting the SCC, then we continue with one of the
   // splits
void MLInlineAdvisor::onPassEntry() {
 
   EdgeCount -= EdgesOfLastSeenNodes;
   EdgesOfLastSeenNodes = 0;
 }
 
 void MLInlineAdvisor::onPassExit(LazyCallGraph::SCC *LastSCC) {
   // No need to keep this around - function passes will invalidate it.
   FPICache.clear();
-  if (!LastSCC)
+  if (!LastSCC || ForceStop)
     return;
   // Keep track of the nodes and edges we last saw. Then, in onPassEntry,
   // we update the node count and edge count from the subset of these nodes that
   // survived.
   assert(NodesInLastSCC.empty());
   assert(NodeCount >= LastSCC->size());
   EdgesOfLastSeenNodes = 0;
   for (const auto &N : *LastSCC) {