include/llvm/Transforms/IPO/InlinerPass.h

   1 //===- InlinerPass.h - Code common to all inliners --------------*- C++ -*-===//
   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 defines a simple policy-based bottom-up inliner.  This file
  11 // implements all of the boring mechanics of the bottom-up inlining, while the
  12 // subclass determines WHAT to inline, which is the much more interesting
  13 // component.
  14 //
  15 //===----------------------------------------------------------------------===//
  16
  17 #ifndef LLVM_TRANSFORMS_IPO_INLINERPASS_H
  18 #define LLVM_TRANSFORMS_IPO_INLINERPASS_H
  19
  20 #include "llvm/CallGraphSCCPass.h"
  21
  22 namespace llvm {
  23   class CallSite;
  24   class TargetData;
  25   class InlineCost;
  26   template<class PtrType, unsigned SmallSize>
  27   class SmallPtrSet;
  28
  29 /// Inliner - This class contains all of the helper code which is used to
  30 /// perform the inlining operations that do not depend on the policy.
  31 ///
  32 struct Inliner : public CallGraphSCCPass {
  33   explicit Inliner(void *ID);
  34   explicit Inliner(void *ID, int Threshold);
  35
  36   /// getAnalysisUsage - For this class, we declare that we require and preserve
  37   /// the call graph.  If the derived class implements this method, it should
  38   /// always explicitly call the implementation here.
  39   virtual void getAnalysisUsage(AnalysisUsage &Info) const;
  40
  41   // Main run interface method, this implements the interface required by the
  42   // Pass class.
  43   virtual bool runOnSCC(std::vector<CallGraphNode *> &SCC);
  44
  45   // doFinalization - Remove now-dead linkonce functions at the end of
  46   // processing to avoid breaking the SCC traversal.
  47   virtual bool doFinalization(CallGraph &CG);
  48
  49   /// This method returns the value specified by the -inline-threshold value,
  50   /// specified on the command line.  This is typically not directly needed.
  51   ///
  52   unsigned getInlineThreshold() const { return InlineThreshold; }
  53
  54   /// Calculate the inline threshold for given Caller. This threshold is lower
  55   /// if Caller is marked with OptimizeForSize and -inline-threshold is not
  56   /// given on the comand line.
  57   ///
  58   unsigned getInlineThreshold(Function* Caller) const;
  59
  60   /// getInlineCost - This method must be implemented by the subclass to
  61   /// determine the cost of inlining the specified call site.  If the cost
  62   /// returned is greater than the current inline threshold, the call site is
  63   /// not inlined.
  64   ///
  65   virtual InlineCost getInlineCost(CallSite CS) = 0;
  66
  67   // getInlineFudgeFactor - Return a > 1.0 factor if the inliner should use a
  68   // higher threshold to determine if the function call should be inlined.
  69   ///
  70   virtual float getInlineFudgeFactor(CallSite CS) = 0;
  71
  72   /// resetCachedCostInfo - erase any cached cost data from the derived class.
  73   /// If the derived class has no such data this can be empty.
  74   ///
  75   virtual void resetCachedCostInfo(Function* Caller) = 0;
  76
  77   /// removeDeadFunctions - Remove dead functions that are not included in
  78   /// DNR (Do Not Remove) list.
  79   bool removeDeadFunctions(CallGraph &CG,
  80                            SmallPtrSet<const Function *, 16> *DNR = NULL);
  81 private:
  82   // InlineThreshold - Cache the value here for easy access.
  83   unsigned InlineThreshold;
  84
  85   /// shouldInline - Return true if the inliner should attempt to
  86   /// inline at the given CallSite.
  87   bool shouldInline(CallSite CS);
  88 };
  89
  90 } // End llvm namespace
  91
  92 #endif