Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- CallGraphSCCPass.h - Pass that operates BU on call graph -*- C++ -*-===// |
2 | // |
||
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
||
4 | // See https://llvm.org/LICENSE.txt for license information. |
||
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
||
6 | // |
||
7 | //===----------------------------------------------------------------------===// |
||
8 | // |
||
9 | // This file defines the CallGraphSCCPass class, which is used for passes which |
||
10 | // are implemented as bottom-up traversals on the call graph. Because there may |
||
11 | // be cycles in the call graph, passes of this type operate on the call-graph in |
||
12 | // SCC order: that is, they process function bottom-up, except for recursive |
||
13 | // functions, which they process all at once. |
||
14 | // |
||
15 | // These passes are inherently interprocedural, and are required to keep the |
||
16 | // call graph up-to-date if they do anything which could modify it. |
||
17 | // |
||
18 | //===----------------------------------------------------------------------===// |
||
19 | |||
20 | #ifndef LLVM_ANALYSIS_CALLGRAPHSCCPASS_H |
||
21 | #define LLVM_ANALYSIS_CALLGRAPHSCCPASS_H |
||
22 | |||
23 | #include "llvm/ADT/ArrayRef.h" |
||
24 | #include "llvm/Pass.h" |
||
25 | #include <vector> |
||
26 | |||
27 | namespace llvm { |
||
28 | |||
29 | class CallGraph; |
||
30 | class CallGraphNode; |
||
31 | class CallGraphSCC; |
||
32 | class PMStack; |
||
33 | |||
34 | class CallGraphSCCPass : public Pass { |
||
35 | public: |
||
36 | explicit CallGraphSCCPass(char &pid) : Pass(PT_CallGraphSCC, pid) {} |
||
37 | |||
38 | /// createPrinterPass - Get a pass that prints the Module |
||
39 | /// corresponding to a CallGraph. |
||
40 | Pass *createPrinterPass(raw_ostream &OS, |
||
41 | const std::string &Banner) const override; |
||
42 | |||
43 | using llvm::Pass::doInitialization; |
||
44 | using llvm::Pass::doFinalization; |
||
45 | |||
46 | /// doInitialization - This method is called before the SCC's of the program |
||
47 | /// has been processed, allowing the pass to do initialization as necessary. |
||
48 | virtual bool doInitialization(CallGraph &CG) { |
||
49 | return false; |
||
50 | } |
||
51 | |||
52 | /// runOnSCC - This method should be implemented by the subclass to perform |
||
53 | /// whatever action is necessary for the specified SCC. Note that |
||
54 | /// non-recursive (or only self-recursive) functions will have an SCC size of |
||
55 | /// 1, where recursive portions of the call graph will have SCC size > 1. |
||
56 | /// |
||
57 | /// SCC passes that add or delete functions to the SCC are required to update |
||
58 | /// the SCC list, otherwise stale pointers may be dereferenced. |
||
59 | virtual bool runOnSCC(CallGraphSCC &SCC) = 0; |
||
60 | |||
61 | /// doFinalization - This method is called after the SCC's of the program has |
||
62 | /// been processed, allowing the pass to do final cleanup as necessary. |
||
63 | virtual bool doFinalization(CallGraph &CG) { |
||
64 | return false; |
||
65 | } |
||
66 | |||
67 | /// Assign pass manager to manager this pass |
||
68 | void assignPassManager(PMStack &PMS, PassManagerType PMT) override; |
||
69 | |||
70 | /// Return what kind of Pass Manager can manage this pass. |
||
71 | PassManagerType getPotentialPassManagerType() const override { |
||
72 | return PMT_CallGraphPassManager; |
||
73 | } |
||
74 | |||
75 | /// getAnalysisUsage - For this class, we declare that we require and preserve |
||
76 | /// the call graph. If the derived class implements this method, it should |
||
77 | /// always explicitly call the implementation here. |
||
78 | void getAnalysisUsage(AnalysisUsage &Info) const override; |
||
79 | |||
80 | protected: |
||
81 | /// Optional passes call this function to check whether the pass should be |
||
82 | /// skipped. This is the case when optimization bisect is over the limit. |
||
83 | bool skipSCC(CallGraphSCC &SCC) const; |
||
84 | }; |
||
85 | |||
86 | /// CallGraphSCC - This is a single SCC that a CallGraphSCCPass is run on. |
||
87 | class CallGraphSCC { |
||
88 | const CallGraph &CG; // The call graph for this SCC. |
||
89 | void *Context; // The CGPassManager object that is vending this. |
||
90 | std::vector<CallGraphNode *> Nodes; |
||
91 | |||
92 | public: |
||
93 | CallGraphSCC(CallGraph &cg, void *context) : CG(cg), Context(context) {} |
||
94 | |||
95 | void initialize(ArrayRef<CallGraphNode *> NewNodes) { |
||
96 | Nodes.assign(NewNodes.begin(), NewNodes.end()); |
||
97 | } |
||
98 | |||
99 | bool isSingular() const { return Nodes.size() == 1; } |
||
100 | unsigned size() const { return Nodes.size(); } |
||
101 | |||
102 | /// ReplaceNode - This informs the SCC and the pass manager that the specified |
||
103 | /// Old node has been deleted, and New is to be used in its place. |
||
104 | void ReplaceNode(CallGraphNode *Old, CallGraphNode *New); |
||
105 | |||
106 | /// DeleteNode - This informs the SCC and the pass manager that the specified |
||
107 | /// Old node has been deleted. |
||
108 | void DeleteNode(CallGraphNode *Old); |
||
109 | |||
110 | using iterator = std::vector<CallGraphNode *>::const_iterator; |
||
111 | |||
112 | iterator begin() const { return Nodes.begin(); } |
||
113 | iterator end() const { return Nodes.end(); } |
||
114 | |||
115 | const CallGraph &getCallGraph() { return CG; } |
||
116 | }; |
||
117 | |||
118 | void initializeDummyCGSCCPassPass(PassRegistry &); |
||
119 | |||
120 | /// This pass is required by interprocedural register allocation. It forces |
||
121 | /// codegen to follow bottom up order on call graph. |
||
122 | class DummyCGSCCPass : public CallGraphSCCPass { |
||
123 | public: |
||
124 | static char ID; |
||
125 | |||
126 | DummyCGSCCPass() : CallGraphSCCPass(ID) { |
||
127 | PassRegistry &Registry = *PassRegistry::getPassRegistry(); |
||
128 | initializeDummyCGSCCPassPass(Registry); |
||
129 | } |
||
130 | |||
131 | bool runOnSCC(CallGraphSCC &SCC) override { return false; } |
||
132 | |||
133 | void getAnalysisUsage(AnalysisUsage &AU) const override { |
||
134 | AU.setPreservesAll(); |
||
135 | } |
||
136 | }; |
||
137 | |||
138 | } // end namespace llvm |
||
139 | |||
140 | #endif // LLVM_ANALYSIS_CALLGRAPHSCCPASS_H |