LLVM  14.0.0git
CallGraphUpdater.h
Go to the documentation of this file.
1 //===- CallGraphUpdater.h - A (lazy) call graph update helper ---*- 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 /// \file
9 ///
10 /// This file provides interfaces used to manipulate a call graph, regardless
11 /// if it is a "old style" CallGraph or an "new style" LazyCallGraph.
12 ///
13 //===----------------------------------------------------------------------===//
14 
15 #ifndef LLVM_TRANSFORMS_UTILS_CALLGRAPHUPDATER_H
16 #define LLVM_TRANSFORMS_UTILS_CALLGRAPHUPDATER_H
17 
22 
23 namespace llvm {
24 
25 /// Wrapper to unify "old style" CallGraph and "new style" LazyCallGraph. This
26 /// simplifies the interface and the call sites, e.g., new and old pass manager
27 /// passes can share the same code.
29  /// Containers for functions which we did replace or want to delete when
30  /// `finalize` is called. This can happen explicitly or as part of the
31  /// destructor. Dead functions in comdat sections are tracked separately
32  /// because a function with discardable linakage in a COMDAT should only
33  /// be dropped if the entire COMDAT is dropped, see git ac07703842cf.
34  ///{
35  SmallPtrSet<Function *, 16> ReplacedFunctions;
36  SmallVector<Function *, 16> DeadFunctions;
37  SmallVector<Function *, 16> DeadFunctionsInComdats;
38  ///}
39 
40  /// Old PM variables
41  ///{
42  CallGraph *CG = nullptr;
43  CallGraphSCC *CGSCC = nullptr;
44  ///}
45 
46  /// New PM variables
47  ///{
48  LazyCallGraph *LCG = nullptr;
49  LazyCallGraph::SCC *SCC = nullptr;
50  CGSCCAnalysisManager *AM = nullptr;
51  CGSCCUpdateResult *UR = nullptr;
52  FunctionAnalysisManager *FAM = nullptr;
53  ///}
54 
55 public:
58 
59  /// Initializers for usage outside of a CGSCC pass, inside a CGSCC pass in
60  /// the old and new pass manager (PM).
61  ///{
62  void initialize(CallGraph &CG, CallGraphSCC &SCC) {
63  this->CG = &CG;
64  this->CGSCC = &SCC;
65  }
68  this->LCG = &LCG;
69  this->SCC = &SCC;
70  this->AM = &AM;
71  this->UR = &UR;
72  FAM =
73  &AM.getResult<FunctionAnalysisManagerCGSCCProxy>(SCC, LCG).getManager();
74  }
75  ///}
76 
77  /// Finalizer that will trigger actions like function removal from the CG.
78  bool finalize();
79 
80  /// Remove \p Fn from the call graph.
81  void removeFunction(Function &Fn);
82 
83  /// After an CGSCC pass changes a function in ways that affect the call
84  /// graph, this method can be called to update it.
85  void reanalyzeFunction(Function &Fn);
86 
87  /// If a new function was created by outlining, this method can be called
88  /// to update the call graph for the new function. Note that the old one
89  /// still needs to be re-analyzed or manually updated.
90  void registerOutlinedFunction(Function &OriginalFn, Function &NewFn);
91 
92  /// Replace \p OldFn in the call graph (and SCC) with \p NewFn. The uses
93  /// outside the call graph and the function \p OldFn are not modified.
94  /// Note that \p OldFn is also removed from the call graph
95  /// (\see removeFunction).
96  void replaceFunctionWith(Function &OldFn, Function &NewFn);
97 
98  /// Remove the call site \p CS from the call graph.
99  void removeCallSite(CallBase &CS);
100 
101  /// Replace \p OldCS with the new call site \p NewCS.
102  /// \return True if the replacement was successful, otherwise False. In the
103  /// latter case the parent function of \p OldCB needs to be re-analyzed.
104  bool replaceCallSite(CallBase &OldCS, CallBase &NewCS);
105 };
106 
107 } // end namespace llvm
108 
109 #endif // LLVM_TRANSFORMS_UTILS_CALLGRAPHUPDATER_H
llvm::CallGraphUpdater::CallGraphUpdater
CallGraphUpdater()
}
Definition: CallGraphUpdater.h:56
llvm::CallGraphUpdater
Wrapper to unify "old style" CallGraph and "new style" LazyCallGraph.
Definition: CallGraphUpdater.h:28
llvm
This file implements support for optimizing divisions by a constant.
Definition: AllocatorList.h:23
llvm::AnalysisManager::getResult
PassT::Result & getResult(IRUnitT &IR, ExtraArgTs... ExtraArgs)
Get the result of an analysis pass for a given IR unit.
Definition: PassManager.h:783
llvm::Function
Definition: Function.h:62
llvm::CallGraphUpdater::replaceCallSite
bool replaceCallSite(CallBase &OldCS, CallBase &NewCS)
Replace OldCS with the new call site NewCS.
Definition: CallGraphUpdater.cpp:143
llvm::SmallVector
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
Definition: SmallVector.h:1168
llvm::CallGraph
The basic data container for the call graph of a Module of IR.
Definition: CallGraph.h:73
FAM
FunctionAnalysisManager FAM
Definition: PassBuilderBindings.cpp:59
llvm::SmallPtrSet
SmallPtrSet - This class implements a set which is optimized for holding SmallSize or less elements.
Definition: SmallPtrSet.h:449
llvm::CallGraphUpdater::finalize
bool finalize()
}
Definition: CallGraphUpdater.cpp:21
llvm::CallGraphSCC
CallGraphSCC - This is a single SCC that a CallGraphSCCPass is run on.
Definition: CallGraphSCCPass.h:87
llvm::LazyCallGraph::SCC
An SCC of the call graph.
Definition: LazyCallGraph.h:422
llvm::CallGraphUpdater::removeCallSite
void removeCallSite(CallBase &CS)
Remove the call site CS from the call graph.
Definition: CallGraphUpdater.cpp:160
LazyCallGraph.h
llvm::CallGraphUpdater::~CallGraphUpdater
~CallGraphUpdater()
Definition: CallGraphUpdater.h:57
CGSCCPassManager.h
llvm::CallGraphUpdater::registerOutlinedFunction
void registerOutlinedFunction(Function &OriginalFn, Function &NewFn)
If a new function was created by outlining, this method can be called to update the call graph for th...
Definition: CallGraphUpdater.cpp:99
CallGraphSCCPass.h
llvm::CGSCCUpdateResult
Support structure for SCC passes to communicate updates the call graph back to the CGSCC pass manager...
Definition: CGSCCPassManager.h:238
llvm::CallGraphUpdater::replaceFunctionWith
void replaceFunctionWith(Function &OldFn, Function &NewFn)
Replace OldFn in the call graph (and SCC) with NewFn.
Definition: CallGraphUpdater.cpp:123
CallGraph.h
llvm::CallGraphUpdater::removeFunction
void removeFunction(Function &Fn)
Remove Fn from the call graph.
Definition: CallGraphUpdater.cpp:107
llvm::CallGraphUpdater::initialize
void initialize(LazyCallGraph &LCG, LazyCallGraph::SCC &SCC, CGSCCAnalysisManager &AM, CGSCCUpdateResult &UR)
Definition: CallGraphUpdater.h:66
llvm::CallBase
Base class for all callable instructions (InvokeInst and CallInst) Holds everything related to callin...
Definition: InstrTypes.h:1161
llvm::AnalysisManager
A container for analyses that lazily runs them and caches their results.
Definition: InstructionSimplify.h:44
llvm::LazyCallGraph
A lazily constructed view of the call graph of a module.
Definition: LazyCallGraph.h:113
llvm::CallGraphUpdater::initialize
void initialize(CallGraph &CG, CallGraphSCC &SCC)
Initializers for usage outside of a CGSCC pass, inside a CGSCC pass in the old and new pass manager (...
Definition: CallGraphUpdater.h:62
llvm::CallGraphUpdater::reanalyzeFunction
void reanalyzeFunction(Function &Fn)
After an CGSCC pass changes a function in ways that affect the call graph, this method can be called ...
Definition: CallGraphUpdater.cpp:87
llvm::FunctionAnalysisManagerCGSCCProxy
A proxy from a FunctionAnalysisManager to an SCC.
Definition: CGSCCPassManager.h:408