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