[llvm-exegesis] Fix missing std::move.
[llvm-complete.git] / lib / Analysis / CallGraphSCCPass.cpp
blob80d0427529a5dfb432b8dfeaff3508f18d5d7ac3
1 //===- CallGraphSCCPass.cpp - Pass that operates BU on call graph ---------===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements the CallGraphSCCPass class, which is used for passes
11 // which are implemented as bottom-up traversals on the call graph. Because
12 // there may be cycles in the call graph, passes of this type operate on the
13 // call-graph in SCC order: that is, they process function bottom-up, except for
14 // recursive functions, which they process all at once.
16 //===----------------------------------------------------------------------===//
18 #include "llvm/Analysis/CallGraphSCCPass.h"
19 #include "llvm/ADT/DenseMap.h"
20 #include "llvm/ADT/SCCIterator.h"
21 #include "llvm/ADT/Statistic.h"
22 #include "llvm/Analysis/CallGraph.h"
23 #include "llvm/IR/CallSite.h"
24 #include "llvm/IR/Function.h"
25 #include "llvm/IR/IRPrintingPasses.h"
26 #include "llvm/IR/Intrinsics.h"
27 #include "llvm/IR/LLVMContext.h"
28 #include "llvm/IR/LegacyPassManagers.h"
29 #include "llvm/IR/Module.h"
30 #include "llvm/IR/OptBisect.h"
31 #include "llvm/IR/PassTimingInfo.h"
32 #include "llvm/Pass.h"
33 #include "llvm/Support/CommandLine.h"
34 #include "llvm/Support/Debug.h"
35 #include "llvm/Support/Timer.h"
36 #include "llvm/Support/raw_ostream.h"
37 #include <cassert>
38 #include <string>
39 #include <utility>
40 #include <vector>
42 using namespace llvm;
44 #define DEBUG_TYPE "cgscc-passmgr"
46 static cl::opt<unsigned>
47 MaxIterations("max-cg-scc-iterations", cl::ReallyHidden, cl::init(4));
49 STATISTIC(MaxSCCIterations, "Maximum CGSCCPassMgr iterations on one SCC");
51 //===----------------------------------------------------------------------===//
52 // CGPassManager
54 /// CGPassManager manages FPPassManagers and CallGraphSCCPasses.
56 namespace {
58 class CGPassManager : public ModulePass, public PMDataManager {
59 public:
60 static char ID;
62 explicit CGPassManager() : ModulePass(ID), PMDataManager() {}
64 /// Execute all of the passes scheduled for execution. Keep track of
65 /// whether any of the passes modifies the module, and if so, return true.
66 bool runOnModule(Module &M) override;
68 using ModulePass::doInitialization;
69 using ModulePass::doFinalization;
71 bool doInitialization(CallGraph &CG);
72 bool doFinalization(CallGraph &CG);
74 /// Pass Manager itself does not invalidate any analysis info.
75 void getAnalysisUsage(AnalysisUsage &Info) const override {
76 // CGPassManager walks SCC and it needs CallGraph.
77 Info.addRequired<CallGraphWrapperPass>();
78 Info.setPreservesAll();
81 StringRef getPassName() const override { return "CallGraph Pass Manager"; }
83 PMDataManager *getAsPMDataManager() override { return this; }
84 Pass *getAsPass() override { return this; }
86 // Print passes managed by this manager
87 void dumpPassStructure(unsigned Offset) override {
88 errs().indent(Offset*2) << "Call Graph SCC Pass Manager\n";
89 for (unsigned Index = 0; Index < getNumContainedPasses(); ++Index) {
90 Pass *P = getContainedPass(Index);
91 P->dumpPassStructure(Offset + 1);
92 dumpLastUses(P, Offset+1);
96 Pass *getContainedPass(unsigned N) {
97 assert(N < PassVector.size() && "Pass number out of range!");
98 return static_cast<Pass *>(PassVector[N]);
101 PassManagerType getPassManagerType() const override {
102 return PMT_CallGraphPassManager;
105 private:
106 bool RunAllPassesOnSCC(CallGraphSCC &CurSCC, CallGraph &CG,
107 bool &DevirtualizedCall);
109 bool RunPassOnSCC(Pass *P, CallGraphSCC &CurSCC,
110 CallGraph &CG, bool &CallGraphUpToDate,
111 bool &DevirtualizedCall);
112 bool RefreshCallGraph(const CallGraphSCC &CurSCC, CallGraph &CG,
113 bool IsCheckingMode);
116 } // end anonymous namespace.
118 char CGPassManager::ID = 0;
120 bool CGPassManager::RunPassOnSCC(Pass *P, CallGraphSCC &CurSCC,
121 CallGraph &CG, bool &CallGraphUpToDate,
122 bool &DevirtualizedCall) {
123 bool Changed = false;
124 PMDataManager *PM = P->getAsPMDataManager();
125 Module &M = CG.getModule();
127 if (!PM) {
128 CallGraphSCCPass *CGSP = (CallGraphSCCPass *)P;
129 if (!CallGraphUpToDate) {
130 DevirtualizedCall |= RefreshCallGraph(CurSCC, CG, false);
131 CallGraphUpToDate = true;
135 unsigned InstrCount, SCCCount = 0;
136 StringMap<std::pair<unsigned, unsigned>> FunctionToInstrCount;
137 bool EmitICRemark = M.shouldEmitInstrCountChangedRemark();
138 TimeRegion PassTimer(getPassTimer(CGSP));
139 if (EmitICRemark)
140 InstrCount = initSizeRemarkInfo(M, FunctionToInstrCount);
141 Changed = CGSP->runOnSCC(CurSCC);
143 if (EmitICRemark) {
144 // FIXME: Add getInstructionCount to CallGraphSCC.
145 SCCCount = M.getInstructionCount();
146 // Is there a difference in the number of instructions in the module?
147 if (SCCCount != InstrCount) {
148 // Yep. Emit a remark and update InstrCount.
149 int64_t Delta =
150 static_cast<int64_t>(SCCCount) - static_cast<int64_t>(InstrCount);
151 emitInstrCountChangedRemark(P, M, Delta, InstrCount,
152 FunctionToInstrCount);
153 InstrCount = SCCCount;
158 // After the CGSCCPass is done, when assertions are enabled, use
159 // RefreshCallGraph to verify that the callgraph was correctly updated.
160 #ifndef NDEBUG
161 if (Changed)
162 RefreshCallGraph(CurSCC, CG, true);
163 #endif
165 return Changed;
168 assert(PM->getPassManagerType() == PMT_FunctionPassManager &&
169 "Invalid CGPassManager member");
170 FPPassManager *FPP = (FPPassManager*)P;
172 // Run pass P on all functions in the current SCC.
173 for (CallGraphNode *CGN : CurSCC) {
174 if (Function *F = CGN->getFunction()) {
175 dumpPassInfo(P, EXECUTION_MSG, ON_FUNCTION_MSG, F->getName());
177 TimeRegion PassTimer(getPassTimer(FPP));
178 Changed |= FPP->runOnFunction(*F);
180 F->getContext().yield();
184 // The function pass(es) modified the IR, they may have clobbered the
185 // callgraph.
186 if (Changed && CallGraphUpToDate) {
187 LLVM_DEBUG(dbgs() << "CGSCCPASSMGR: Pass Dirtied SCC: " << P->getPassName()
188 << '\n');
189 CallGraphUpToDate = false;
191 return Changed;
194 /// Scan the functions in the specified CFG and resync the
195 /// callgraph with the call sites found in it. This is used after
196 /// FunctionPasses have potentially munged the callgraph, and can be used after
197 /// CallGraphSCC passes to verify that they correctly updated the callgraph.
199 /// This function returns true if it devirtualized an existing function call,
200 /// meaning it turned an indirect call into a direct call. This happens when
201 /// a function pass like GVN optimizes away stuff feeding the indirect call.
202 /// This never happens in checking mode.
203 bool CGPassManager::RefreshCallGraph(const CallGraphSCC &CurSCC, CallGraph &CG,
204 bool CheckingMode) {
205 DenseMap<Value*, CallGraphNode*> CallSites;
207 LLVM_DEBUG(dbgs() << "CGSCCPASSMGR: Refreshing SCC with " << CurSCC.size()
208 << " nodes:\n";
209 for (CallGraphNode *CGN
210 : CurSCC) CGN->dump(););
212 bool MadeChange = false;
213 bool DevirtualizedCall = false;
215 // Scan all functions in the SCC.
216 unsigned FunctionNo = 0;
217 for (CallGraphSCC::iterator SCCIdx = CurSCC.begin(), E = CurSCC.end();
218 SCCIdx != E; ++SCCIdx, ++FunctionNo) {
219 CallGraphNode *CGN = *SCCIdx;
220 Function *F = CGN->getFunction();
221 if (!F || F->isDeclaration()) continue;
223 // Walk the function body looking for call sites. Sync up the call sites in
224 // CGN with those actually in the function.
226 // Keep track of the number of direct and indirect calls that were
227 // invalidated and removed.
228 unsigned NumDirectRemoved = 0, NumIndirectRemoved = 0;
230 // Get the set of call sites currently in the function.
231 for (CallGraphNode::iterator I = CGN->begin(), E = CGN->end(); I != E; ) {
232 // If this call site is null, then the function pass deleted the call
233 // entirely and the WeakTrackingVH nulled it out.
234 if (!I->first ||
235 // If we've already seen this call site, then the FunctionPass RAUW'd
236 // one call with another, which resulted in two "uses" in the edge
237 // list of the same call.
238 CallSites.count(I->first) ||
240 // If the call edge is not from a call or invoke, or it is a
241 // instrinsic call, then the function pass RAUW'd a call with
242 // another value. This can happen when constant folding happens
243 // of well known functions etc.
244 !CallSite(I->first) ||
245 (CallSite(I->first).getCalledFunction() &&
246 CallSite(I->first).getCalledFunction()->isIntrinsic() &&
247 Intrinsic::isLeaf(
248 CallSite(I->first).getCalledFunction()->getIntrinsicID()))) {
249 assert(!CheckingMode &&
250 "CallGraphSCCPass did not update the CallGraph correctly!");
252 // If this was an indirect call site, count it.
253 if (!I->second->getFunction())
254 ++NumIndirectRemoved;
255 else
256 ++NumDirectRemoved;
258 // Just remove the edge from the set of callees, keep track of whether
259 // I points to the last element of the vector.
260 bool WasLast = I + 1 == E;
261 CGN->removeCallEdge(I);
263 // If I pointed to the last element of the vector, we have to bail out:
264 // iterator checking rejects comparisons of the resultant pointer with
265 // end.
266 if (WasLast)
267 break;
268 E = CGN->end();
269 continue;
272 assert(!CallSites.count(I->first) &&
273 "Call site occurs in node multiple times");
275 CallSite CS(I->first);
276 if (CS) {
277 Function *Callee = CS.getCalledFunction();
278 // Ignore intrinsics because they're not really function calls.
279 if (!Callee || !(Callee->isIntrinsic()))
280 CallSites.insert(std::make_pair(I->first, I->second));
282 ++I;
285 // Loop over all of the instructions in the function, getting the callsites.
286 // Keep track of the number of direct/indirect calls added.
287 unsigned NumDirectAdded = 0, NumIndirectAdded = 0;
289 for (BasicBlock &BB : *F)
290 for (Instruction &I : BB) {
291 CallSite CS(&I);
292 if (!CS) continue;
293 Function *Callee = CS.getCalledFunction();
294 if (Callee && Callee->isIntrinsic()) continue;
296 // If this call site already existed in the callgraph, just verify it
297 // matches up to expectations and remove it from CallSites.
298 DenseMap<Value*, CallGraphNode*>::iterator ExistingIt =
299 CallSites.find(CS.getInstruction());
300 if (ExistingIt != CallSites.end()) {
301 CallGraphNode *ExistingNode = ExistingIt->second;
303 // Remove from CallSites since we have now seen it.
304 CallSites.erase(ExistingIt);
306 // Verify that the callee is right.
307 if (ExistingNode->getFunction() == CS.getCalledFunction())
308 continue;
310 // If we are in checking mode, we are not allowed to actually mutate
311 // the callgraph. If this is a case where we can infer that the
312 // callgraph is less precise than it could be (e.g. an indirect call
313 // site could be turned direct), don't reject it in checking mode, and
314 // don't tweak it to be more precise.
315 if (CheckingMode && CS.getCalledFunction() &&
316 ExistingNode->getFunction() == nullptr)
317 continue;
319 assert(!CheckingMode &&
320 "CallGraphSCCPass did not update the CallGraph correctly!");
322 // If not, we either went from a direct call to indirect, indirect to
323 // direct, or direct to different direct.
324 CallGraphNode *CalleeNode;
325 if (Function *Callee = CS.getCalledFunction()) {
326 CalleeNode = CG.getOrInsertFunction(Callee);
327 // Keep track of whether we turned an indirect call into a direct
328 // one.
329 if (!ExistingNode->getFunction()) {
330 DevirtualizedCall = true;
331 LLVM_DEBUG(dbgs() << " CGSCCPASSMGR: Devirtualized call to '"
332 << Callee->getName() << "'\n");
334 } else {
335 CalleeNode = CG.getCallsExternalNode();
338 // Update the edge target in CGN.
339 CGN->replaceCallEdge(CS, CS, CalleeNode);
340 MadeChange = true;
341 continue;
344 assert(!CheckingMode &&
345 "CallGraphSCCPass did not update the CallGraph correctly!");
347 // If the call site didn't exist in the CGN yet, add it.
348 CallGraphNode *CalleeNode;
349 if (Function *Callee = CS.getCalledFunction()) {
350 CalleeNode = CG.getOrInsertFunction(Callee);
351 ++NumDirectAdded;
352 } else {
353 CalleeNode = CG.getCallsExternalNode();
354 ++NumIndirectAdded;
357 CGN->addCalledFunction(CS, CalleeNode);
358 MadeChange = true;
361 // We scanned the old callgraph node, removing invalidated call sites and
362 // then added back newly found call sites. One thing that can happen is
363 // that an old indirect call site was deleted and replaced with a new direct
364 // call. In this case, we have devirtualized a call, and CGSCCPM would like
365 // to iteratively optimize the new code. Unfortunately, we don't really
366 // have a great way to detect when this happens. As an approximation, we
367 // just look at whether the number of indirect calls is reduced and the
368 // number of direct calls is increased. There are tons of ways to fool this
369 // (e.g. DCE'ing an indirect call and duplicating an unrelated block with a
370 // direct call) but this is close enough.
371 if (NumIndirectRemoved > NumIndirectAdded &&
372 NumDirectRemoved < NumDirectAdded)
373 DevirtualizedCall = true;
375 // After scanning this function, if we still have entries in callsites, then
376 // they are dangling pointers. WeakTrackingVH should save us for this, so
377 // abort if
378 // this happens.
379 assert(CallSites.empty() && "Dangling pointers found in call sites map");
381 // Periodically do an explicit clear to remove tombstones when processing
382 // large scc's.
383 if ((FunctionNo & 15) == 15)
384 CallSites.clear();
387 LLVM_DEBUG(if (MadeChange) {
388 dbgs() << "CGSCCPASSMGR: Refreshed SCC is now:\n";
389 for (CallGraphNode *CGN : CurSCC)
390 CGN->dump();
391 if (DevirtualizedCall)
392 dbgs() << "CGSCCPASSMGR: Refresh devirtualized a call!\n";
393 } else {
394 dbgs() << "CGSCCPASSMGR: SCC Refresh didn't change call graph.\n";
396 (void)MadeChange;
398 return DevirtualizedCall;
401 /// Execute the body of the entire pass manager on the specified SCC.
402 /// This keeps track of whether a function pass devirtualizes
403 /// any calls and returns it in DevirtualizedCall.
404 bool CGPassManager::RunAllPassesOnSCC(CallGraphSCC &CurSCC, CallGraph &CG,
405 bool &DevirtualizedCall) {
406 bool Changed = false;
408 // Keep track of whether the callgraph is known to be up-to-date or not.
409 // The CGSSC pass manager runs two types of passes:
410 // CallGraphSCC Passes and other random function passes. Because other
411 // random function passes are not CallGraph aware, they may clobber the
412 // call graph by introducing new calls or deleting other ones. This flag
413 // is set to false when we run a function pass so that we know to clean up
414 // the callgraph when we need to run a CGSCCPass again.
415 bool CallGraphUpToDate = true;
417 // Run all passes on current SCC.
418 for (unsigned PassNo = 0, e = getNumContainedPasses();
419 PassNo != e; ++PassNo) {
420 Pass *P = getContainedPass(PassNo);
422 // If we're in -debug-pass=Executions mode, construct the SCC node list,
423 // otherwise avoid constructing this string as it is expensive.
424 if (isPassDebuggingExecutionsOrMore()) {
425 std::string Functions;
426 #ifndef NDEBUG
427 raw_string_ostream OS(Functions);
428 for (CallGraphSCC::iterator I = CurSCC.begin(), E = CurSCC.end();
429 I != E; ++I) {
430 if (I != CurSCC.begin()) OS << ", ";
431 (*I)->print(OS);
433 OS.flush();
434 #endif
435 dumpPassInfo(P, EXECUTION_MSG, ON_CG_MSG, Functions);
437 dumpRequiredSet(P);
439 initializeAnalysisImpl(P);
441 // Actually run this pass on the current SCC.
442 Changed |= RunPassOnSCC(P, CurSCC, CG,
443 CallGraphUpToDate, DevirtualizedCall);
445 if (Changed)
446 dumpPassInfo(P, MODIFICATION_MSG, ON_CG_MSG, "");
447 dumpPreservedSet(P);
449 verifyPreservedAnalysis(P);
450 removeNotPreservedAnalysis(P);
451 recordAvailableAnalysis(P);
452 removeDeadPasses(P, "", ON_CG_MSG);
455 // If the callgraph was left out of date (because the last pass run was a
456 // functionpass), refresh it before we move on to the next SCC.
457 if (!CallGraphUpToDate)
458 DevirtualizedCall |= RefreshCallGraph(CurSCC, CG, false);
459 return Changed;
462 /// Execute all of the passes scheduled for execution. Keep track of
463 /// whether any of the passes modifies the module, and if so, return true.
464 bool CGPassManager::runOnModule(Module &M) {
465 CallGraph &CG = getAnalysis<CallGraphWrapperPass>().getCallGraph();
466 bool Changed = doInitialization(CG);
468 // Walk the callgraph in bottom-up SCC order.
469 scc_iterator<CallGraph*> CGI = scc_begin(&CG);
471 CallGraphSCC CurSCC(CG, &CGI);
472 while (!CGI.isAtEnd()) {
473 // Copy the current SCC and increment past it so that the pass can hack
474 // on the SCC if it wants to without invalidating our iterator.
475 const std::vector<CallGraphNode *> &NodeVec = *CGI;
476 CurSCC.initialize(NodeVec);
477 ++CGI;
479 // At the top level, we run all the passes in this pass manager on the
480 // functions in this SCC. However, we support iterative compilation in the
481 // case where a function pass devirtualizes a call to a function. For
482 // example, it is very common for a function pass (often GVN or instcombine)
483 // to eliminate the addressing that feeds into a call. With that improved
484 // information, we would like the call to be an inline candidate, infer
485 // mod-ref information etc.
487 // Because of this, we allow iteration up to a specified iteration count.
488 // This only happens in the case of a devirtualized call, so we only burn
489 // compile time in the case that we're making progress. We also have a hard
490 // iteration count limit in case there is crazy code.
491 unsigned Iteration = 0;
492 bool DevirtualizedCall = false;
493 do {
494 LLVM_DEBUG(if (Iteration) dbgs()
495 << " SCCPASSMGR: Re-visiting SCC, iteration #" << Iteration
496 << '\n');
497 DevirtualizedCall = false;
498 Changed |= RunAllPassesOnSCC(CurSCC, CG, DevirtualizedCall);
499 } while (Iteration++ < MaxIterations && DevirtualizedCall);
501 if (DevirtualizedCall)
502 LLVM_DEBUG(dbgs() << " CGSCCPASSMGR: Stopped iteration after "
503 << Iteration
504 << " times, due to -max-cg-scc-iterations\n");
506 MaxSCCIterations.updateMax(Iteration);
508 Changed |= doFinalization(CG);
509 return Changed;
512 /// Initialize CG
513 bool CGPassManager::doInitialization(CallGraph &CG) {
514 bool Changed = false;
515 for (unsigned i = 0, e = getNumContainedPasses(); i != e; ++i) {
516 if (PMDataManager *PM = getContainedPass(i)->getAsPMDataManager()) {
517 assert(PM->getPassManagerType() == PMT_FunctionPassManager &&
518 "Invalid CGPassManager member");
519 Changed |= ((FPPassManager*)PM)->doInitialization(CG.getModule());
520 } else {
521 Changed |= ((CallGraphSCCPass*)getContainedPass(i))->doInitialization(CG);
524 return Changed;
527 /// Finalize CG
528 bool CGPassManager::doFinalization(CallGraph &CG) {
529 bool Changed = false;
530 for (unsigned i = 0, e = getNumContainedPasses(); i != e; ++i) {
531 if (PMDataManager *PM = getContainedPass(i)->getAsPMDataManager()) {
532 assert(PM->getPassManagerType() == PMT_FunctionPassManager &&
533 "Invalid CGPassManager member");
534 Changed |= ((FPPassManager*)PM)->doFinalization(CG.getModule());
535 } else {
536 Changed |= ((CallGraphSCCPass*)getContainedPass(i))->doFinalization(CG);
539 return Changed;
542 //===----------------------------------------------------------------------===//
543 // CallGraphSCC Implementation
544 //===----------------------------------------------------------------------===//
546 /// This informs the SCC and the pass manager that the specified
547 /// Old node has been deleted, and New is to be used in its place.
548 void CallGraphSCC::ReplaceNode(CallGraphNode *Old, CallGraphNode *New) {
549 assert(Old != New && "Should not replace node with self");
550 for (unsigned i = 0; ; ++i) {
551 assert(i != Nodes.size() && "Node not in SCC");
552 if (Nodes[i] != Old) continue;
553 Nodes[i] = New;
554 break;
557 // Update the active scc_iterator so that it doesn't contain dangling
558 // pointers to the old CallGraphNode.
559 scc_iterator<CallGraph*> *CGI = (scc_iterator<CallGraph*>*)Context;
560 CGI->ReplaceNode(Old, New);
563 //===----------------------------------------------------------------------===//
564 // CallGraphSCCPass Implementation
565 //===----------------------------------------------------------------------===//
567 /// Assign pass manager to manage this pass.
568 void CallGraphSCCPass::assignPassManager(PMStack &PMS,
569 PassManagerType PreferredType) {
570 // Find CGPassManager
571 while (!PMS.empty() &&
572 PMS.top()->getPassManagerType() > PMT_CallGraphPassManager)
573 PMS.pop();
575 assert(!PMS.empty() && "Unable to handle Call Graph Pass");
576 CGPassManager *CGP;
578 if (PMS.top()->getPassManagerType() == PMT_CallGraphPassManager)
579 CGP = (CGPassManager*)PMS.top();
580 else {
581 // Create new Call Graph SCC Pass Manager if it does not exist.
582 assert(!PMS.empty() && "Unable to create Call Graph Pass Manager");
583 PMDataManager *PMD = PMS.top();
585 // [1] Create new Call Graph Pass Manager
586 CGP = new CGPassManager();
588 // [2] Set up new manager's top level manager
589 PMTopLevelManager *TPM = PMD->getTopLevelManager();
590 TPM->addIndirectPassManager(CGP);
592 // [3] Assign manager to manage this new manager. This may create
593 // and push new managers into PMS
594 Pass *P = CGP;
595 TPM->schedulePass(P);
597 // [4] Push new manager into PMS
598 PMS.push(CGP);
601 CGP->add(this);
604 /// For this class, we declare that we require and preserve the call graph.
605 /// If the derived class implements this method, it should
606 /// always explicitly call the implementation here.
607 void CallGraphSCCPass::getAnalysisUsage(AnalysisUsage &AU) const {
608 AU.addRequired<CallGraphWrapperPass>();
609 AU.addPreserved<CallGraphWrapperPass>();
612 //===----------------------------------------------------------------------===//
613 // PrintCallGraphPass Implementation
614 //===----------------------------------------------------------------------===//
616 namespace {
618 /// PrintCallGraphPass - Print a Module corresponding to a call graph.
620 class PrintCallGraphPass : public CallGraphSCCPass {
621 std::string Banner;
622 raw_ostream &OS; // raw_ostream to print on.
624 public:
625 static char ID;
627 PrintCallGraphPass(const std::string &B, raw_ostream &OS)
628 : CallGraphSCCPass(ID), Banner(B), OS(OS) {}
630 void getAnalysisUsage(AnalysisUsage &AU) const override {
631 AU.setPreservesAll();
634 bool runOnSCC(CallGraphSCC &SCC) override {
635 bool BannerPrinted = false;
636 auto PrintBannerOnce = [&] () {
637 if (BannerPrinted)
638 return;
639 OS << Banner;
640 BannerPrinted = true;
642 for (CallGraphNode *CGN : SCC) {
643 if (Function *F = CGN->getFunction()) {
644 if (!F->isDeclaration() && isFunctionInPrintList(F->getName())) {
645 PrintBannerOnce();
646 F->print(OS);
648 } else if (isFunctionInPrintList("*")) {
649 PrintBannerOnce();
650 OS << "\nPrinting <null> Function\n";
653 return false;
656 StringRef getPassName() const override { return "Print CallGraph IR"; }
659 } // end anonymous namespace.
661 char PrintCallGraphPass::ID = 0;
663 Pass *CallGraphSCCPass::createPrinterPass(raw_ostream &OS,
664 const std::string &Banner) const {
665 return new PrintCallGraphPass(Banner, OS);
668 bool CallGraphSCCPass::skipSCC(CallGraphSCC &SCC) const {
669 return !SCC.getCallGraph().getModule()
670 .getContext()
671 .getOptPassGate()
672 .shouldRunPass(this, SCC);
675 char DummyCGSCCPass::ID = 0;
677 INITIALIZE_PASS(DummyCGSCCPass, "DummyCGSCCPass", "DummyCGSCCPass", false,
678 false)