1//===- InlineSimple.cpp - Code to perform simple function inlining --------===//
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 bottom-up inlining of functions into callees.
11//
12//===----------------------------------------------------------------------===//
13
14#include "llvm/Analysis/AssumptionCache.h"
15#include "llvm/Analysis/CallGraph.h"
16#include "llvm/Analysis/InlineCost.h"
17#include "llvm/Analysis/ProfileSummaryInfo.h"
18#include "llvm/Analysis/TargetLibraryInfo.h"
19#include "llvm/Analysis/TargetTransformInfo.h"
20#include "llvm/IR/CallSite.h"
21#include "llvm/IR/CallingConv.h"
22#include "llvm/IR/DataLayout.h"
23#include "llvm/IR/Instructions.h"
24#include "llvm/IR/IntrinsicInst.h"
25#include "llvm/IR/Module.h"
26#include "llvm/IR/Type.h"
27#include "llvm/Transforms/IPO.h"
28#include "llvm/Transforms/IPO/InlinerPass.h"
29
30using namespace llvm;
31
32#define DEBUG_TYPE "inline"
33
34namespace {
35
36/// \brief Actual inliner pass implementation.
37///
38/// The common implementation of the inlining logic is shared between this
39/// inliner pass and the always inliner pass. The two passes use different cost
40/// analyses to determine when to inline.
41class SimpleInliner : public Inliner {
42  // This field is populated based on one of the following:
43  //  * optimization or size-optimization levels,
44  //  * the --inline-threshold flag, or
45  //  * a user specified value.
46  int DefaultThreshold;
47
48public:
49  SimpleInliner()
50      : Inliner(ID), DefaultThreshold(llvm::getDefaultInlineThreshold()) {
51    initializeSimpleInlinerPass(*PassRegistry::getPassRegistry());
52  }
53
54  explicit SimpleInliner(int Threshold)
55      : Inliner(ID), DefaultThreshold(Threshold) {
56    initializeSimpleInlinerPass(*PassRegistry::getPassRegistry());
57  }
58
59  static char ID; // Pass identification, replacement for typeid
60
61  InlineCost getInlineCost(CallSite CS) override {
62    Function *Callee = CS.getCalledFunction();
63    TargetTransformInfo &TTI = TTIWP->getTTI(*Callee);
64    return llvm::getInlineCost(CS, DefaultThreshold, TTI, ACT, PSI);
65  }
66
67  bool runOnSCC(CallGraphSCC &SCC) override;
68  void getAnalysisUsage(AnalysisUsage &AU) const override;
69
70private:
71  TargetTransformInfoWrapperPass *TTIWP;
72};
73
74} // end anonymous namespace
75
76char SimpleInliner::ID = 0;
77INITIALIZE_PASS_BEGIN(SimpleInliner, "inline",
78                "Function Integration/Inlining", false, false)
79INITIALIZE_PASS_DEPENDENCY(AssumptionCacheTracker)
80INITIALIZE_PASS_DEPENDENCY(CallGraphWrapperPass)
81INITIALIZE_PASS_DEPENDENCY(ProfileSummaryInfoWrapperPass)
82INITIALIZE_PASS_DEPENDENCY(TargetTransformInfoWrapperPass)
83INITIALIZE_PASS_DEPENDENCY(TargetLibraryInfoWrapperPass)
84INITIALIZE_PASS_END(SimpleInliner, "inline",
85                "Function Integration/Inlining", false, false)
86
87Pass *llvm::createFunctionInliningPass() { return new SimpleInliner(); }
88
89Pass *llvm::createFunctionInliningPass(int Threshold) {
90  return new SimpleInliner(Threshold);
91}
92
93Pass *llvm::createFunctionInliningPass(unsigned OptLevel,
94                                       unsigned SizeOptLevel) {
95  return new SimpleInliner(
96      llvm::computeThresholdFromOptLevels(OptLevel, SizeOptLevel));
97}
98
99bool SimpleInliner::runOnSCC(CallGraphSCC &SCC) {
100  TTIWP = &getAnalysis<TargetTransformInfoWrapperPass>();
101  return Inliner::runOnSCC(SCC);
102}
103
104void SimpleInliner::getAnalysisUsage(AnalysisUsage &AU) const {
105  AU.addRequired<TargetTransformInfoWrapperPass>();
106  Inliner::getAnalysisUsage(AU);
107}
108