17adc3a2bSChandler Carruth //===- InlineCost.cpp - Cost analysis for inliner -------------------------===//
27adc3a2bSChandler Carruth //
37adc3a2bSChandler Carruth //                     The LLVM Compiler Infrastructure
47adc3a2bSChandler Carruth //
57adc3a2bSChandler Carruth // This file is distributed under the University of Illinois Open Source
67adc3a2bSChandler Carruth // License. See LICENSE.TXT for details.
77adc3a2bSChandler Carruth //
87adc3a2bSChandler Carruth //===----------------------------------------------------------------------===//
97adc3a2bSChandler Carruth //
107adc3a2bSChandler Carruth // This file implements inline cost analysis.
117adc3a2bSChandler Carruth //
127adc3a2bSChandler Carruth //===----------------------------------------------------------------------===//
137adc3a2bSChandler Carruth 
147adc3a2bSChandler Carruth #include "llvm/Analysis/InlineCost.h"
157adc3a2bSChandler Carruth #include "llvm/ADT/STLExtras.h"
167adc3a2bSChandler Carruth #include "llvm/ADT/SetVector.h"
177adc3a2bSChandler Carruth #include "llvm/ADT/SmallPtrSet.h"
187adc3a2bSChandler Carruth #include "llvm/ADT/SmallVector.h"
197adc3a2bSChandler Carruth #include "llvm/ADT/Statistic.h"
207adc3a2bSChandler Carruth #include "llvm/Analysis/AssumptionCache.h"
217adc3a2bSChandler Carruth #include "llvm/Analysis/CodeMetrics.h"
227adc3a2bSChandler Carruth #include "llvm/Analysis/ConstantFolding.h"
237adc3a2bSChandler Carruth #include "llvm/Analysis/InstructionSimplify.h"
247adc3a2bSChandler Carruth #include "llvm/Analysis/TargetTransformInfo.h"
257adc3a2bSChandler Carruth #include "llvm/IR/CallSite.h"
267adc3a2bSChandler Carruth #include "llvm/IR/CallingConv.h"
277adc3a2bSChandler Carruth #include "llvm/IR/DataLayout.h"
287adc3a2bSChandler Carruth #include "llvm/IR/GetElementPtrTypeIterator.h"
297adc3a2bSChandler Carruth #include "llvm/IR/GlobalAlias.h"
307adc3a2bSChandler Carruth #include "llvm/IR/InstVisitor.h"
317adc3a2bSChandler Carruth #include "llvm/IR/IntrinsicInst.h"
327adc3a2bSChandler Carruth #include "llvm/IR/Operator.h"
337adc3a2bSChandler Carruth #include "llvm/Support/Debug.h"
347adc3a2bSChandler Carruth #include "llvm/Support/raw_ostream.h"
357adc3a2bSChandler Carruth 
367adc3a2bSChandler Carruth using namespace llvm;
377adc3a2bSChandler Carruth 
387adc3a2bSChandler Carruth #define DEBUG_TYPE "inline-cost"
397adc3a2bSChandler Carruth 
407adc3a2bSChandler Carruth STATISTIC(NumCallsAnalyzed, "Number of call sites analyzed");
417adc3a2bSChandler Carruth 
42f4bb2f0dSEaswaran Raman // Threshold to use when optsize is specified (and there is no
43f4bb2f0dSEaswaran Raman // -inline-threshold).
44f4bb2f0dSEaswaran Raman const int OptSizeThreshold = 75;
45f4bb2f0dSEaswaran Raman 
46f4bb2f0dSEaswaran Raman // Threshold to use when -Oz is specified (and there is no -inline-threshold).
47f4bb2f0dSEaswaran Raman const int OptMinSizeThreshold = 25;
48f4bb2f0dSEaswaran Raman 
49f4bb2f0dSEaswaran Raman // Threshold to use when -O[34] is specified (and there is no
50f4bb2f0dSEaswaran Raman // -inline-threshold).
51f4bb2f0dSEaswaran Raman const int OptAggressiveThreshold = 275;
52f4bb2f0dSEaswaran Raman 
53f4bb2f0dSEaswaran Raman static cl::opt<int> DefaultInlineThreshold(
54f4bb2f0dSEaswaran Raman     "inline-threshold", cl::Hidden, cl::init(225), cl::ZeroOrMore,
55f4bb2f0dSEaswaran Raman     cl::desc("Control the amount of inlining to perform (default = 225)"));
56f4bb2f0dSEaswaran Raman 
57f4bb2f0dSEaswaran Raman static cl::opt<int> HintThreshold(
58f4bb2f0dSEaswaran Raman     "inlinehint-threshold", cl::Hidden, cl::init(325),
59f4bb2f0dSEaswaran Raman     cl::desc("Threshold for inlining functions with inline hint"));
60f4bb2f0dSEaswaran Raman 
61f4bb2f0dSEaswaran Raman // We introduce this threshold to help performance of instrumentation based
62f4bb2f0dSEaswaran Raman // PGO before we actually hook up inliner with analysis passes such as BPI and
63f4bb2f0dSEaswaran Raman // BFI.
64f4bb2f0dSEaswaran Raman static cl::opt<int> ColdThreshold(
65f4bb2f0dSEaswaran Raman     "inlinecold-threshold", cl::Hidden, cl::init(225),
66f4bb2f0dSEaswaran Raman     cl::desc("Threshold for inlining functions with cold attribute"));
67f4bb2f0dSEaswaran Raman 
687adc3a2bSChandler Carruth namespace {
697adc3a2bSChandler Carruth 
707adc3a2bSChandler Carruth class CallAnalyzer : public InstVisitor<CallAnalyzer, bool> {
717adc3a2bSChandler Carruth   typedef InstVisitor<CallAnalyzer, bool> Base;
727adc3a2bSChandler Carruth   friend class InstVisitor<CallAnalyzer, bool>;
737adc3a2bSChandler Carruth 
747adc3a2bSChandler Carruth   /// The TargetTransformInfo available for this compilation.
757adc3a2bSChandler Carruth   const TargetTransformInfo &TTI;
767adc3a2bSChandler Carruth 
777adc3a2bSChandler Carruth   /// The cache of @llvm.assume intrinsics.
787adc3a2bSChandler Carruth   AssumptionCacheTracker *ACT;
797adc3a2bSChandler Carruth 
807adc3a2bSChandler Carruth   // The called function.
817adc3a2bSChandler Carruth   Function &F;
827adc3a2bSChandler Carruth 
837adc3a2bSChandler Carruth   // The candidate callsite being analyzed. Please do not use this to do
847adc3a2bSChandler Carruth   // analysis in the caller function; we want the inline cost query to be
857adc3a2bSChandler Carruth   // easily cacheable. Instead, use the cover function paramHasAttr.
867adc3a2bSChandler Carruth   CallSite CandidateCS;
877adc3a2bSChandler Carruth 
887adc3a2bSChandler Carruth   int Threshold;
897adc3a2bSChandler Carruth   int Cost;
907adc3a2bSChandler Carruth 
917adc3a2bSChandler Carruth   bool IsCallerRecursive;
927adc3a2bSChandler Carruth   bool IsRecursiveCall;
937adc3a2bSChandler Carruth   bool ExposesReturnsTwice;
947adc3a2bSChandler Carruth   bool HasDynamicAlloca;
957adc3a2bSChandler Carruth   bool ContainsNoDuplicateCall;
967adc3a2bSChandler Carruth   bool HasReturn;
977adc3a2bSChandler Carruth   bool HasIndirectBr;
987adc3a2bSChandler Carruth   bool HasFrameEscape;
997adc3a2bSChandler Carruth 
1007adc3a2bSChandler Carruth   /// Number of bytes allocated statically by the callee.
1017adc3a2bSChandler Carruth   uint64_t AllocatedSize;
1027adc3a2bSChandler Carruth   unsigned NumInstructions, NumVectorInstructions;
1037adc3a2bSChandler Carruth   int FiftyPercentVectorBonus, TenPercentVectorBonus;
1047adc3a2bSChandler Carruth   int VectorBonus;
1057adc3a2bSChandler Carruth 
1067adc3a2bSChandler Carruth   // While we walk the potentially-inlined instructions, we build up and
1077adc3a2bSChandler Carruth   // maintain a mapping of simplified values specific to this callsite. The
1087adc3a2bSChandler Carruth   // idea is to propagate any special information we have about arguments to
1097adc3a2bSChandler Carruth   // this call through the inlinable section of the function, and account for
1107adc3a2bSChandler Carruth   // likely simplifications post-inlining. The most important aspect we track
1117adc3a2bSChandler Carruth   // is CFG altering simplifications -- when we prove a basic block dead, that
1127adc3a2bSChandler Carruth   // can cause dramatic shifts in the cost of inlining a function.
1137adc3a2bSChandler Carruth   DenseMap<Value *, Constant *> SimplifiedValues;
1147adc3a2bSChandler Carruth 
1157adc3a2bSChandler Carruth   // Keep track of the values which map back (through function arguments) to
1167adc3a2bSChandler Carruth   // allocas on the caller stack which could be simplified through SROA.
1177adc3a2bSChandler Carruth   DenseMap<Value *, Value *> SROAArgValues;
1187adc3a2bSChandler Carruth 
1197adc3a2bSChandler Carruth   // The mapping of caller Alloca values to their accumulated cost savings. If
1207adc3a2bSChandler Carruth   // we have to disable SROA for one of the allocas, this tells us how much
1217adc3a2bSChandler Carruth   // cost must be added.
1227adc3a2bSChandler Carruth   DenseMap<Value *, int> SROAArgCosts;
1237adc3a2bSChandler Carruth 
1247adc3a2bSChandler Carruth   // Keep track of values which map to a pointer base and constant offset.
1257adc3a2bSChandler Carruth   DenseMap<Value *, std::pair<Value *, APInt>> ConstantOffsetPtrs;
1267adc3a2bSChandler Carruth 
1277adc3a2bSChandler Carruth   // Custom simplification helper routines.
1287adc3a2bSChandler Carruth   bool isAllocaDerivedArg(Value *V);
1297adc3a2bSChandler Carruth   bool lookupSROAArgAndCost(Value *V, Value *&Arg,
1307adc3a2bSChandler Carruth                             DenseMap<Value *, int>::iterator &CostIt);
1317adc3a2bSChandler Carruth   void disableSROA(DenseMap<Value *, int>::iterator CostIt);
1327adc3a2bSChandler Carruth   void disableSROA(Value *V);
1337adc3a2bSChandler Carruth   void accumulateSROACost(DenseMap<Value *, int>::iterator CostIt,
1347adc3a2bSChandler Carruth                           int InstructionCost);
1357adc3a2bSChandler Carruth   bool isGEPOffsetConstant(GetElementPtrInst &GEP);
1367adc3a2bSChandler Carruth   bool accumulateGEPOffset(GEPOperator &GEP, APInt &Offset);
1377adc3a2bSChandler Carruth   bool simplifyCallSite(Function *F, CallSite CS);
1387adc3a2bSChandler Carruth   ConstantInt *stripAndComputeInBoundsConstantOffsets(Value *&V);
1397adc3a2bSChandler Carruth 
1407adc3a2bSChandler Carruth   /// Return true if the given argument to the function being considered for
1417adc3a2bSChandler Carruth   /// inlining has the given attribute set either at the call site or the
1427adc3a2bSChandler Carruth   /// function declaration.  Primarily used to inspect call site specific
1437adc3a2bSChandler Carruth   /// attributes since these can be more precise than the ones on the callee
1447adc3a2bSChandler Carruth   /// itself.
1457adc3a2bSChandler Carruth   bool paramHasAttr(Argument *A, Attribute::AttrKind Attr);
1467adc3a2bSChandler Carruth 
1477adc3a2bSChandler Carruth   /// Return true if the given value is known non null within the callee if
1487adc3a2bSChandler Carruth   /// inlined through this particular callsite.
1497adc3a2bSChandler Carruth   bool isKnownNonNullInCallee(Value *V);
1507adc3a2bSChandler Carruth 
151f4bb2f0dSEaswaran Raman   /// Update Threshold based on callsite properties such as callee
152f4bb2f0dSEaswaran Raman   /// attributes and callee hotness for PGO builds. The Callee is explicitly
153f4bb2f0dSEaswaran Raman   /// passed to support analyzing indirect calls whose target is inferred by
154f4bb2f0dSEaswaran Raman   /// analysis.
155f4bb2f0dSEaswaran Raman   void updateThreshold(CallSite CS, Function &Callee);
156f4bb2f0dSEaswaran Raman 
1579a3fc17aSEaswaran Raman   /// Return true if size growth is allowed when inlining the callee at CS.
1589a3fc17aSEaswaran Raman   bool allowSizeGrowth(CallSite CS);
1599a3fc17aSEaswaran Raman 
1607adc3a2bSChandler Carruth   // Custom analysis routines.
1617adc3a2bSChandler Carruth   bool analyzeBlock(BasicBlock *BB, SmallPtrSetImpl<const Value *> &EphValues);
1627adc3a2bSChandler Carruth 
1637adc3a2bSChandler Carruth   // Disable several entry points to the visitor so we don't accidentally use
1647adc3a2bSChandler Carruth   // them by declaring but not defining them here.
165567556aaSChad Rosier   void visit(Module *);
166567556aaSChad Rosier   void visit(Module &);
167567556aaSChad Rosier   void visit(Function *);
168567556aaSChad Rosier   void visit(Function &);
169567556aaSChad Rosier   void visit(BasicBlock *);
170567556aaSChad Rosier   void visit(BasicBlock &);
1717adc3a2bSChandler Carruth 
1727adc3a2bSChandler Carruth   // Provide base case for our instruction visit.
1737adc3a2bSChandler Carruth   bool visitInstruction(Instruction &I);
1747adc3a2bSChandler Carruth 
1757adc3a2bSChandler Carruth   // Our visit overrides.
1767adc3a2bSChandler Carruth   bool visitAlloca(AllocaInst &I);
1777adc3a2bSChandler Carruth   bool visitPHI(PHINode &I);
1787adc3a2bSChandler Carruth   bool visitGetElementPtr(GetElementPtrInst &I);
1797adc3a2bSChandler Carruth   bool visitBitCast(BitCastInst &I);
1807adc3a2bSChandler Carruth   bool visitPtrToInt(PtrToIntInst &I);
1817adc3a2bSChandler Carruth   bool visitIntToPtr(IntToPtrInst &I);
1827adc3a2bSChandler Carruth   bool visitCastInst(CastInst &I);
1837adc3a2bSChandler Carruth   bool visitUnaryInstruction(UnaryInstruction &I);
1847adc3a2bSChandler Carruth   bool visitCmpInst(CmpInst &I);
1857adc3a2bSChandler Carruth   bool visitSub(BinaryOperator &I);
1867adc3a2bSChandler Carruth   bool visitBinaryOperator(BinaryOperator &I);
1877adc3a2bSChandler Carruth   bool visitLoad(LoadInst &I);
1887adc3a2bSChandler Carruth   bool visitStore(StoreInst &I);
1897adc3a2bSChandler Carruth   bool visitExtractValue(ExtractValueInst &I);
1907adc3a2bSChandler Carruth   bool visitInsertValue(InsertValueInst &I);
1917adc3a2bSChandler Carruth   bool visitCallSite(CallSite CS);
1927adc3a2bSChandler Carruth   bool visitReturnInst(ReturnInst &RI);
1937adc3a2bSChandler Carruth   bool visitBranchInst(BranchInst &BI);
1947adc3a2bSChandler Carruth   bool visitSwitchInst(SwitchInst &SI);
1957adc3a2bSChandler Carruth   bool visitIndirectBrInst(IndirectBrInst &IBI);
1967adc3a2bSChandler Carruth   bool visitResumeInst(ResumeInst &RI);
1977adc3a2bSChandler Carruth   bool visitCleanupReturnInst(CleanupReturnInst &RI);
1987adc3a2bSChandler Carruth   bool visitCatchReturnInst(CatchReturnInst &RI);
1997adc3a2bSChandler Carruth   bool visitUnreachableInst(UnreachableInst &I);
2007adc3a2bSChandler Carruth 
2017adc3a2bSChandler Carruth public:
2027adc3a2bSChandler Carruth   CallAnalyzer(const TargetTransformInfo &TTI, AssumptionCacheTracker *ACT,
203b1bd398cSEaswaran Raman                Function &Callee, int Threshold, CallSite CSArg)
204b1bd398cSEaswaran Raman       : TTI(TTI), ACT(ACT), F(Callee), CandidateCS(CSArg), Threshold(Threshold),
205b1bd398cSEaswaran Raman         Cost(0), IsCallerRecursive(false), IsRecursiveCall(false),
206b1bd398cSEaswaran Raman         ExposesReturnsTwice(false), HasDynamicAlloca(false),
207b1bd398cSEaswaran Raman         ContainsNoDuplicateCall(false), HasReturn(false), HasIndirectBr(false),
208b1bd398cSEaswaran Raman         HasFrameEscape(false), AllocatedSize(0), NumInstructions(0),
209b1bd398cSEaswaran Raman         NumVectorInstructions(0), FiftyPercentVectorBonus(0),
210b1bd398cSEaswaran Raman         TenPercentVectorBonus(0), VectorBonus(0), NumConstantArgs(0),
211b1bd398cSEaswaran Raman         NumConstantOffsetPtrArgs(0), NumAllocaArgs(0), NumConstantPtrCmps(0),
212b1bd398cSEaswaran Raman         NumConstantPtrDiffs(0), NumInstructionsSimplified(0),
213b1bd398cSEaswaran Raman         SROACostSavings(0), SROACostSavingsLost(0) {}
2147adc3a2bSChandler Carruth 
2157adc3a2bSChandler Carruth   bool analyzeCall(CallSite CS);
2167adc3a2bSChandler Carruth 
2177adc3a2bSChandler Carruth   int getThreshold() { return Threshold; }
2187adc3a2bSChandler Carruth   int getCost() { return Cost; }
2197adc3a2bSChandler Carruth 
2207adc3a2bSChandler Carruth   // Keep a bunch of stats about the cost savings found so we can print them
2217adc3a2bSChandler Carruth   // out when debugging.
2227adc3a2bSChandler Carruth   unsigned NumConstantArgs;
2237adc3a2bSChandler Carruth   unsigned NumConstantOffsetPtrArgs;
2247adc3a2bSChandler Carruth   unsigned NumAllocaArgs;
2257adc3a2bSChandler Carruth   unsigned NumConstantPtrCmps;
2267adc3a2bSChandler Carruth   unsigned NumConstantPtrDiffs;
2277adc3a2bSChandler Carruth   unsigned NumInstructionsSimplified;
2287adc3a2bSChandler Carruth   unsigned SROACostSavings;
2297adc3a2bSChandler Carruth   unsigned SROACostSavingsLost;
2307adc3a2bSChandler Carruth 
2317adc3a2bSChandler Carruth   void dump();
2327adc3a2bSChandler Carruth };
2337adc3a2bSChandler Carruth 
2347adc3a2bSChandler Carruth } // namespace
2357adc3a2bSChandler Carruth 
2367adc3a2bSChandler Carruth /// \brief Test whether the given value is an Alloca-derived function argument.
2377adc3a2bSChandler Carruth bool CallAnalyzer::isAllocaDerivedArg(Value *V) {
2387adc3a2bSChandler Carruth   return SROAArgValues.count(V);
2397adc3a2bSChandler Carruth }
2407adc3a2bSChandler Carruth 
2417adc3a2bSChandler Carruth /// \brief Lookup the SROA-candidate argument and cost iterator which V maps to.
2427adc3a2bSChandler Carruth /// Returns false if V does not map to a SROA-candidate.
2437adc3a2bSChandler Carruth bool CallAnalyzer::lookupSROAArgAndCost(
2447adc3a2bSChandler Carruth     Value *V, Value *&Arg, DenseMap<Value *, int>::iterator &CostIt) {
2457adc3a2bSChandler Carruth   if (SROAArgValues.empty() || SROAArgCosts.empty())
2467adc3a2bSChandler Carruth     return false;
2477adc3a2bSChandler Carruth 
2487adc3a2bSChandler Carruth   DenseMap<Value *, Value *>::iterator ArgIt = SROAArgValues.find(V);
2497adc3a2bSChandler Carruth   if (ArgIt == SROAArgValues.end())
2507adc3a2bSChandler Carruth     return false;
2517adc3a2bSChandler Carruth 
2527adc3a2bSChandler Carruth   Arg = ArgIt->second;
2537adc3a2bSChandler Carruth   CostIt = SROAArgCosts.find(Arg);
2547adc3a2bSChandler Carruth   return CostIt != SROAArgCosts.end();
2557adc3a2bSChandler Carruth }
2567adc3a2bSChandler Carruth 
2577adc3a2bSChandler Carruth /// \brief Disable SROA for the candidate marked by this cost iterator.
2587adc3a2bSChandler Carruth ///
2597adc3a2bSChandler Carruth /// This marks the candidate as no longer viable for SROA, and adds the cost
2607adc3a2bSChandler Carruth /// savings associated with it back into the inline cost measurement.
2617adc3a2bSChandler Carruth void CallAnalyzer::disableSROA(DenseMap<Value *, int>::iterator CostIt) {
2627adc3a2bSChandler Carruth   // If we're no longer able to perform SROA we need to undo its cost savings
2637adc3a2bSChandler Carruth   // and prevent subsequent analysis.
2647adc3a2bSChandler Carruth   Cost += CostIt->second;
2657adc3a2bSChandler Carruth   SROACostSavings -= CostIt->second;
2667adc3a2bSChandler Carruth   SROACostSavingsLost += CostIt->second;
2677adc3a2bSChandler Carruth   SROAArgCosts.erase(CostIt);
2687adc3a2bSChandler Carruth }
2697adc3a2bSChandler Carruth 
2707adc3a2bSChandler Carruth /// \brief If 'V' maps to a SROA candidate, disable SROA for it.
2717adc3a2bSChandler Carruth void CallAnalyzer::disableSROA(Value *V) {
2727adc3a2bSChandler Carruth   Value *SROAArg;
2737adc3a2bSChandler Carruth   DenseMap<Value *, int>::iterator CostIt;
2747adc3a2bSChandler Carruth   if (lookupSROAArgAndCost(V, SROAArg, CostIt))
2757adc3a2bSChandler Carruth     disableSROA(CostIt);
2767adc3a2bSChandler Carruth }
2777adc3a2bSChandler Carruth 
2787adc3a2bSChandler Carruth /// \brief Accumulate the given cost for a particular SROA candidate.
2797adc3a2bSChandler Carruth void CallAnalyzer::accumulateSROACost(DenseMap<Value *, int>::iterator CostIt,
2807adc3a2bSChandler Carruth                                       int InstructionCost) {
2817adc3a2bSChandler Carruth   CostIt->second += InstructionCost;
2827adc3a2bSChandler Carruth   SROACostSavings += InstructionCost;
2837adc3a2bSChandler Carruth }
2847adc3a2bSChandler Carruth 
2857adc3a2bSChandler Carruth /// \brief Check whether a GEP's indices are all constant.
2867adc3a2bSChandler Carruth ///
2877adc3a2bSChandler Carruth /// Respects any simplified values known during the analysis of this callsite.
2887adc3a2bSChandler Carruth bool CallAnalyzer::isGEPOffsetConstant(GetElementPtrInst &GEP) {
2897adc3a2bSChandler Carruth   for (User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end(); I != E; ++I)
2907adc3a2bSChandler Carruth     if (!isa<Constant>(*I) && !SimplifiedValues.lookup(*I))
2917adc3a2bSChandler Carruth       return false;
2927adc3a2bSChandler Carruth 
2937adc3a2bSChandler Carruth   return true;
2947adc3a2bSChandler Carruth }
2957adc3a2bSChandler Carruth 
2967adc3a2bSChandler Carruth /// \brief Accumulate a constant GEP offset into an APInt if possible.
2977adc3a2bSChandler Carruth ///
2987adc3a2bSChandler Carruth /// Returns false if unable to compute the offset for any reason. Respects any
2997adc3a2bSChandler Carruth /// simplified values known during the analysis of this callsite.
3007adc3a2bSChandler Carruth bool CallAnalyzer::accumulateGEPOffset(GEPOperator &GEP, APInt &Offset) {
3017adc3a2bSChandler Carruth   const DataLayout &DL = F.getParent()->getDataLayout();
3027adc3a2bSChandler Carruth   unsigned IntPtrWidth = DL.getPointerSizeInBits();
3037adc3a2bSChandler Carruth   assert(IntPtrWidth == Offset.getBitWidth());
3047adc3a2bSChandler Carruth 
3057adc3a2bSChandler Carruth   for (gep_type_iterator GTI = gep_type_begin(GEP), GTE = gep_type_end(GEP);
3067adc3a2bSChandler Carruth        GTI != GTE; ++GTI) {
3077adc3a2bSChandler Carruth     ConstantInt *OpC = dyn_cast<ConstantInt>(GTI.getOperand());
3087adc3a2bSChandler Carruth     if (!OpC)
3097adc3a2bSChandler Carruth       if (Constant *SimpleOp = SimplifiedValues.lookup(GTI.getOperand()))
3107adc3a2bSChandler Carruth         OpC = dyn_cast<ConstantInt>(SimpleOp);
3117adc3a2bSChandler Carruth     if (!OpC)
3127adc3a2bSChandler Carruth       return false;
313567556aaSChad Rosier     if (OpC->isZero())
314567556aaSChad Rosier       continue;
3157adc3a2bSChandler Carruth 
3167adc3a2bSChandler Carruth     // Handle a struct index, which adds its field offset to the pointer.
3177adc3a2bSChandler Carruth     if (StructType *STy = dyn_cast<StructType>(*GTI)) {
3187adc3a2bSChandler Carruth       unsigned ElementIdx = OpC->getZExtValue();
3197adc3a2bSChandler Carruth       const StructLayout *SL = DL.getStructLayout(STy);
3207adc3a2bSChandler Carruth       Offset += APInt(IntPtrWidth, SL->getElementOffset(ElementIdx));
3217adc3a2bSChandler Carruth       continue;
3227adc3a2bSChandler Carruth     }
3237adc3a2bSChandler Carruth 
3247adc3a2bSChandler Carruth     APInt TypeSize(IntPtrWidth, DL.getTypeAllocSize(GTI.getIndexedType()));
3257adc3a2bSChandler Carruth     Offset += OpC->getValue().sextOrTrunc(IntPtrWidth) * TypeSize;
3267adc3a2bSChandler Carruth   }
3277adc3a2bSChandler Carruth   return true;
3287adc3a2bSChandler Carruth }
3297adc3a2bSChandler Carruth 
3307adc3a2bSChandler Carruth bool CallAnalyzer::visitAlloca(AllocaInst &I) {
3317adc3a2bSChandler Carruth   // Check whether inlining will turn a dynamic alloca into a static
332*0f153424SSanjay Patel   // alloca and handle that case.
3337adc3a2bSChandler Carruth   if (I.isArrayAllocation()) {
334*0f153424SSanjay Patel     Constant *Size = SimplifiedValues.lookup(I.getArraySize());
335*0f153424SSanjay Patel     if (auto *AllocSize = dyn_cast_or_null<ConstantInt>(Size)) {
3367adc3a2bSChandler Carruth       Type *Ty = I.getAllocatedType();
337*0f153424SSanjay Patel       // FIXME: This can't be right. AllocatedSize is in *bytes*.
3387adc3a2bSChandler Carruth       AllocatedSize += Ty->getPrimitiveSizeInBits() * AllocSize->getZExtValue();
3397adc3a2bSChandler Carruth       return Base::visitAlloca(I);
3407adc3a2bSChandler Carruth     }
3417adc3a2bSChandler Carruth   }
3427adc3a2bSChandler Carruth 
3437adc3a2bSChandler Carruth   // Accumulate the allocated size.
3447adc3a2bSChandler Carruth   if (I.isStaticAlloca()) {
3457adc3a2bSChandler Carruth     const DataLayout &DL = F.getParent()->getDataLayout();
3467adc3a2bSChandler Carruth     Type *Ty = I.getAllocatedType();
3477adc3a2bSChandler Carruth     AllocatedSize += DL.getTypeAllocSize(Ty);
3487adc3a2bSChandler Carruth   }
3497adc3a2bSChandler Carruth 
3507adc3a2bSChandler Carruth   // We will happily inline static alloca instructions.
3517adc3a2bSChandler Carruth   if (I.isStaticAlloca())
3527adc3a2bSChandler Carruth     return Base::visitAlloca(I);
3537adc3a2bSChandler Carruth 
3547adc3a2bSChandler Carruth   // FIXME: This is overly conservative. Dynamic allocas are inefficient for
3557adc3a2bSChandler Carruth   // a variety of reasons, and so we would like to not inline them into
3567adc3a2bSChandler Carruth   // functions which don't currently have a dynamic alloca. This simply
3577adc3a2bSChandler Carruth   // disables inlining altogether in the presence of a dynamic alloca.
3587adc3a2bSChandler Carruth   HasDynamicAlloca = true;
3597adc3a2bSChandler Carruth   return false;
3607adc3a2bSChandler Carruth }
3617adc3a2bSChandler Carruth 
3627adc3a2bSChandler Carruth bool CallAnalyzer::visitPHI(PHINode &I) {
3637adc3a2bSChandler Carruth   // FIXME: We should potentially be tracking values through phi nodes,
3647adc3a2bSChandler Carruth   // especially when they collapse to a single value due to deleted CFG edges
3657adc3a2bSChandler Carruth   // during inlining.
3667adc3a2bSChandler Carruth 
3677adc3a2bSChandler Carruth   // FIXME: We need to propagate SROA *disabling* through phi nodes, even
3687adc3a2bSChandler Carruth   // though we don't want to propagate it's bonuses. The idea is to disable
3697adc3a2bSChandler Carruth   // SROA if it *might* be used in an inappropriate manner.
3707adc3a2bSChandler Carruth 
3717adc3a2bSChandler Carruth   // Phi nodes are always zero-cost.
3727adc3a2bSChandler Carruth   return true;
3737adc3a2bSChandler Carruth }
3747adc3a2bSChandler Carruth 
3757adc3a2bSChandler Carruth bool CallAnalyzer::visitGetElementPtr(GetElementPtrInst &I) {
3767adc3a2bSChandler Carruth   Value *SROAArg;
3777adc3a2bSChandler Carruth   DenseMap<Value *, int>::iterator CostIt;
378567556aaSChad Rosier   bool SROACandidate =
379567556aaSChad Rosier       lookupSROAArgAndCost(I.getPointerOperand(), SROAArg, CostIt);
3807adc3a2bSChandler Carruth 
3817adc3a2bSChandler Carruth   // Try to fold GEPs of constant-offset call site argument pointers. This
3827adc3a2bSChandler Carruth   // requires target data and inbounds GEPs.
3837adc3a2bSChandler Carruth   if (I.isInBounds()) {
3847adc3a2bSChandler Carruth     // Check if we have a base + offset for the pointer.
3857adc3a2bSChandler Carruth     Value *Ptr = I.getPointerOperand();
3867adc3a2bSChandler Carruth     std::pair<Value *, APInt> BaseAndOffset = ConstantOffsetPtrs.lookup(Ptr);
3877adc3a2bSChandler Carruth     if (BaseAndOffset.first) {
3887adc3a2bSChandler Carruth       // Check if the offset of this GEP is constant, and if so accumulate it
3897adc3a2bSChandler Carruth       // into Offset.
3907adc3a2bSChandler Carruth       if (!accumulateGEPOffset(cast<GEPOperator>(I), BaseAndOffset.second)) {
3917adc3a2bSChandler Carruth         // Non-constant GEPs aren't folded, and disable SROA.
3927adc3a2bSChandler Carruth         if (SROACandidate)
3937adc3a2bSChandler Carruth           disableSROA(CostIt);
3947adc3a2bSChandler Carruth         return false;
3957adc3a2bSChandler Carruth       }
3967adc3a2bSChandler Carruth 
3977adc3a2bSChandler Carruth       // Add the result as a new mapping to Base + Offset.
3987adc3a2bSChandler Carruth       ConstantOffsetPtrs[&I] = BaseAndOffset;
3997adc3a2bSChandler Carruth 
4007adc3a2bSChandler Carruth       // Also handle SROA candidates here, we already know that the GEP is
4017adc3a2bSChandler Carruth       // all-constant indexed.
4027adc3a2bSChandler Carruth       if (SROACandidate)
4037adc3a2bSChandler Carruth         SROAArgValues[&I] = SROAArg;
4047adc3a2bSChandler Carruth 
4057adc3a2bSChandler Carruth       return true;
4067adc3a2bSChandler Carruth     }
4077adc3a2bSChandler Carruth   }
4087adc3a2bSChandler Carruth 
4097adc3a2bSChandler Carruth   if (isGEPOffsetConstant(I)) {
4107adc3a2bSChandler Carruth     if (SROACandidate)
4117adc3a2bSChandler Carruth       SROAArgValues[&I] = SROAArg;
4127adc3a2bSChandler Carruth 
4137adc3a2bSChandler Carruth     // Constant GEPs are modeled as free.
4147adc3a2bSChandler Carruth     return true;
4157adc3a2bSChandler Carruth   }
4167adc3a2bSChandler Carruth 
4177adc3a2bSChandler Carruth   // Variable GEPs will require math and will disable SROA.
4187adc3a2bSChandler Carruth   if (SROACandidate)
4197adc3a2bSChandler Carruth     disableSROA(CostIt);
4207adc3a2bSChandler Carruth   return false;
4217adc3a2bSChandler Carruth }
4227adc3a2bSChandler Carruth 
4237adc3a2bSChandler Carruth bool CallAnalyzer::visitBitCast(BitCastInst &I) {
4247adc3a2bSChandler Carruth   // Propagate constants through bitcasts.
4257adc3a2bSChandler Carruth   Constant *COp = dyn_cast<Constant>(I.getOperand(0));
4267adc3a2bSChandler Carruth   if (!COp)
4277adc3a2bSChandler Carruth     COp = SimplifiedValues.lookup(I.getOperand(0));
4287adc3a2bSChandler Carruth   if (COp)
4297adc3a2bSChandler Carruth     if (Constant *C = ConstantExpr::getBitCast(COp, I.getType())) {
4307adc3a2bSChandler Carruth       SimplifiedValues[&I] = C;
4317adc3a2bSChandler Carruth       return true;
4327adc3a2bSChandler Carruth     }
4337adc3a2bSChandler Carruth 
4347adc3a2bSChandler Carruth   // Track base/offsets through casts
435567556aaSChad Rosier   std::pair<Value *, APInt> BaseAndOffset =
436567556aaSChad Rosier       ConstantOffsetPtrs.lookup(I.getOperand(0));
4377adc3a2bSChandler Carruth   // Casts don't change the offset, just wrap it up.
4387adc3a2bSChandler Carruth   if (BaseAndOffset.first)
4397adc3a2bSChandler Carruth     ConstantOffsetPtrs[&I] = BaseAndOffset;
4407adc3a2bSChandler Carruth 
4417adc3a2bSChandler Carruth   // Also look for SROA candidates here.
4427adc3a2bSChandler Carruth   Value *SROAArg;
4437adc3a2bSChandler Carruth   DenseMap<Value *, int>::iterator CostIt;
4447adc3a2bSChandler Carruth   if (lookupSROAArgAndCost(I.getOperand(0), SROAArg, CostIt))
4457adc3a2bSChandler Carruth     SROAArgValues[&I] = SROAArg;
4467adc3a2bSChandler Carruth 
4477adc3a2bSChandler Carruth   // Bitcasts are always zero cost.
4487adc3a2bSChandler Carruth   return true;
4497adc3a2bSChandler Carruth }
4507adc3a2bSChandler Carruth 
4517adc3a2bSChandler Carruth bool CallAnalyzer::visitPtrToInt(PtrToIntInst &I) {
4527adc3a2bSChandler Carruth   // Propagate constants through ptrtoint.
4537adc3a2bSChandler Carruth   Constant *COp = dyn_cast<Constant>(I.getOperand(0));
4547adc3a2bSChandler Carruth   if (!COp)
4557adc3a2bSChandler Carruth     COp = SimplifiedValues.lookup(I.getOperand(0));
4567adc3a2bSChandler Carruth   if (COp)
4577adc3a2bSChandler Carruth     if (Constant *C = ConstantExpr::getPtrToInt(COp, I.getType())) {
4587adc3a2bSChandler Carruth       SimplifiedValues[&I] = C;
4597adc3a2bSChandler Carruth       return true;
4607adc3a2bSChandler Carruth     }
4617adc3a2bSChandler Carruth 
4627adc3a2bSChandler Carruth   // Track base/offset pairs when converted to a plain integer provided the
4637adc3a2bSChandler Carruth   // integer is large enough to represent the pointer.
4647adc3a2bSChandler Carruth   unsigned IntegerSize = I.getType()->getScalarSizeInBits();
4657adc3a2bSChandler Carruth   const DataLayout &DL = F.getParent()->getDataLayout();
4667adc3a2bSChandler Carruth   if (IntegerSize >= DL.getPointerSizeInBits()) {
467567556aaSChad Rosier     std::pair<Value *, APInt> BaseAndOffset =
468567556aaSChad Rosier         ConstantOffsetPtrs.lookup(I.getOperand(0));
4697adc3a2bSChandler Carruth     if (BaseAndOffset.first)
4707adc3a2bSChandler Carruth       ConstantOffsetPtrs[&I] = BaseAndOffset;
4717adc3a2bSChandler Carruth   }
4727adc3a2bSChandler Carruth 
4737adc3a2bSChandler Carruth   // This is really weird. Technically, ptrtoint will disable SROA. However,
4747adc3a2bSChandler Carruth   // unless that ptrtoint is *used* somewhere in the live basic blocks after
4757adc3a2bSChandler Carruth   // inlining, it will be nuked, and SROA should proceed. All of the uses which
4767adc3a2bSChandler Carruth   // would block SROA would also block SROA if applied directly to a pointer,
4777adc3a2bSChandler Carruth   // and so we can just add the integer in here. The only places where SROA is
4787adc3a2bSChandler Carruth   // preserved either cannot fire on an integer, or won't in-and-of themselves
4797adc3a2bSChandler Carruth   // disable SROA (ext) w/o some later use that we would see and disable.
4807adc3a2bSChandler Carruth   Value *SROAArg;
4817adc3a2bSChandler Carruth   DenseMap<Value *, int>::iterator CostIt;
4827adc3a2bSChandler Carruth   if (lookupSROAArgAndCost(I.getOperand(0), SROAArg, CostIt))
4837adc3a2bSChandler Carruth     SROAArgValues[&I] = SROAArg;
4847adc3a2bSChandler Carruth 
4857adc3a2bSChandler Carruth   return TargetTransformInfo::TCC_Free == TTI.getUserCost(&I);
4867adc3a2bSChandler Carruth }
4877adc3a2bSChandler Carruth 
4887adc3a2bSChandler Carruth bool CallAnalyzer::visitIntToPtr(IntToPtrInst &I) {
4897adc3a2bSChandler Carruth   // Propagate constants through ptrtoint.
4907adc3a2bSChandler Carruth   Constant *COp = dyn_cast<Constant>(I.getOperand(0));
4917adc3a2bSChandler Carruth   if (!COp)
4927adc3a2bSChandler Carruth     COp = SimplifiedValues.lookup(I.getOperand(0));
4937adc3a2bSChandler Carruth   if (COp)
4947adc3a2bSChandler Carruth     if (Constant *C = ConstantExpr::getIntToPtr(COp, I.getType())) {
4957adc3a2bSChandler Carruth       SimplifiedValues[&I] = C;
4967adc3a2bSChandler Carruth       return true;
4977adc3a2bSChandler Carruth     }
4987adc3a2bSChandler Carruth 
4997adc3a2bSChandler Carruth   // Track base/offset pairs when round-tripped through a pointer without
5007adc3a2bSChandler Carruth   // modifications provided the integer is not too large.
5017adc3a2bSChandler Carruth   Value *Op = I.getOperand(0);
5027adc3a2bSChandler Carruth   unsigned IntegerSize = Op->getType()->getScalarSizeInBits();
5037adc3a2bSChandler Carruth   const DataLayout &DL = F.getParent()->getDataLayout();
5047adc3a2bSChandler Carruth   if (IntegerSize <= DL.getPointerSizeInBits()) {
5057adc3a2bSChandler Carruth     std::pair<Value *, APInt> BaseAndOffset = ConstantOffsetPtrs.lookup(Op);
5067adc3a2bSChandler Carruth     if (BaseAndOffset.first)
5077adc3a2bSChandler Carruth       ConstantOffsetPtrs[&I] = BaseAndOffset;
5087adc3a2bSChandler Carruth   }
5097adc3a2bSChandler Carruth 
5107adc3a2bSChandler Carruth   // "Propagate" SROA here in the same manner as we do for ptrtoint above.
5117adc3a2bSChandler Carruth   Value *SROAArg;
5127adc3a2bSChandler Carruth   DenseMap<Value *, int>::iterator CostIt;
5137adc3a2bSChandler Carruth   if (lookupSROAArgAndCost(Op, SROAArg, CostIt))
5147adc3a2bSChandler Carruth     SROAArgValues[&I] = SROAArg;
5157adc3a2bSChandler Carruth 
5167adc3a2bSChandler Carruth   return TargetTransformInfo::TCC_Free == TTI.getUserCost(&I);
5177adc3a2bSChandler Carruth }
5187adc3a2bSChandler Carruth 
5197adc3a2bSChandler Carruth bool CallAnalyzer::visitCastInst(CastInst &I) {
5207adc3a2bSChandler Carruth   // Propagate constants through ptrtoint.
5217adc3a2bSChandler Carruth   Constant *COp = dyn_cast<Constant>(I.getOperand(0));
5227adc3a2bSChandler Carruth   if (!COp)
5237adc3a2bSChandler Carruth     COp = SimplifiedValues.lookup(I.getOperand(0));
5247adc3a2bSChandler Carruth   if (COp)
5257adc3a2bSChandler Carruth     if (Constant *C = ConstantExpr::getCast(I.getOpcode(), COp, I.getType())) {
5267adc3a2bSChandler Carruth       SimplifiedValues[&I] = C;
5277adc3a2bSChandler Carruth       return true;
5287adc3a2bSChandler Carruth     }
5297adc3a2bSChandler Carruth 
5307adc3a2bSChandler Carruth   // Disable SROA in the face of arbitrary casts we don't whitelist elsewhere.
5317adc3a2bSChandler Carruth   disableSROA(I.getOperand(0));
5327adc3a2bSChandler Carruth 
5337adc3a2bSChandler Carruth   return TargetTransformInfo::TCC_Free == TTI.getUserCost(&I);
5347adc3a2bSChandler Carruth }
5357adc3a2bSChandler Carruth 
5367adc3a2bSChandler Carruth bool CallAnalyzer::visitUnaryInstruction(UnaryInstruction &I) {
5377adc3a2bSChandler Carruth   Value *Operand = I.getOperand(0);
5387adc3a2bSChandler Carruth   Constant *COp = dyn_cast<Constant>(Operand);
5397adc3a2bSChandler Carruth   if (!COp)
5407adc3a2bSChandler Carruth     COp = SimplifiedValues.lookup(Operand);
5417adc3a2bSChandler Carruth   if (COp) {
5427adc3a2bSChandler Carruth     const DataLayout &DL = F.getParent()->getDataLayout();
543e902459cSManuel Jacob     if (Constant *C = ConstantFoldInstOperands(&I, COp, DL)) {
5447adc3a2bSChandler Carruth       SimplifiedValues[&I] = C;
5457adc3a2bSChandler Carruth       return true;
5467adc3a2bSChandler Carruth     }
5477adc3a2bSChandler Carruth   }
5487adc3a2bSChandler Carruth 
5497adc3a2bSChandler Carruth   // Disable any SROA on the argument to arbitrary unary operators.
5507adc3a2bSChandler Carruth   disableSROA(Operand);
5517adc3a2bSChandler Carruth 
5527adc3a2bSChandler Carruth   return false;
5537adc3a2bSChandler Carruth }
5547adc3a2bSChandler Carruth 
5557adc3a2bSChandler Carruth bool CallAnalyzer::paramHasAttr(Argument *A, Attribute::AttrKind Attr) {
5567adc3a2bSChandler Carruth   unsigned ArgNo = A->getArgNo();
5577adc3a2bSChandler Carruth   return CandidateCS.paramHasAttr(ArgNo + 1, Attr);
5587adc3a2bSChandler Carruth }
5597adc3a2bSChandler Carruth 
5607adc3a2bSChandler Carruth bool CallAnalyzer::isKnownNonNullInCallee(Value *V) {
5617adc3a2bSChandler Carruth   // Does the *call site* have the NonNull attribute set on an argument?  We
5627adc3a2bSChandler Carruth   // use the attribute on the call site to memoize any analysis done in the
5637adc3a2bSChandler Carruth   // caller. This will also trip if the callee function has a non-null
5647adc3a2bSChandler Carruth   // parameter attribute, but that's a less interesting case because hopefully
5657adc3a2bSChandler Carruth   // the callee would already have been simplified based on that.
5667adc3a2bSChandler Carruth   if (Argument *A = dyn_cast<Argument>(V))
5677adc3a2bSChandler Carruth     if (paramHasAttr(A, Attribute::NonNull))
5687adc3a2bSChandler Carruth       return true;
5697adc3a2bSChandler Carruth 
5707adc3a2bSChandler Carruth   // Is this an alloca in the caller?  This is distinct from the attribute case
5717adc3a2bSChandler Carruth   // above because attributes aren't updated within the inliner itself and we
5727adc3a2bSChandler Carruth   // always want to catch the alloca derived case.
5737adc3a2bSChandler Carruth   if (isAllocaDerivedArg(V))
5747adc3a2bSChandler Carruth     // We can actually predict the result of comparisons between an
5757adc3a2bSChandler Carruth     // alloca-derived value and null. Note that this fires regardless of
5767adc3a2bSChandler Carruth     // SROA firing.
5777adc3a2bSChandler Carruth     return true;
5787adc3a2bSChandler Carruth 
5797adc3a2bSChandler Carruth   return false;
5807adc3a2bSChandler Carruth }
5817adc3a2bSChandler Carruth 
5829a3fc17aSEaswaran Raman bool CallAnalyzer::allowSizeGrowth(CallSite CS) {
5839a3fc17aSEaswaran Raman   // If the normal destination of the invoke or the parent block of the call
5849a3fc17aSEaswaran Raman   // site is unreachable-terminated, there is little point in inlining this
5859a3fc17aSEaswaran Raman   // unless there is literally zero cost.
5869a3fc17aSEaswaran Raman   // FIXME: Note that it is possible that an unreachable-terminated block has a
5879a3fc17aSEaswaran Raman   // hot entry. For example, in below scenario inlining hot_call_X() may be
5889a3fc17aSEaswaran Raman   // beneficial :
5899a3fc17aSEaswaran Raman   // main() {
5909a3fc17aSEaswaran Raman   //   hot_call_1();
5919a3fc17aSEaswaran Raman   //   ...
5929a3fc17aSEaswaran Raman   //   hot_call_N()
5939a3fc17aSEaswaran Raman   //   exit(0);
5949a3fc17aSEaswaran Raman   // }
5959a3fc17aSEaswaran Raman   // For now, we are not handling this corner case here as it is rare in real
5969a3fc17aSEaswaran Raman   // code. In future, we should elaborate this based on BPI and BFI in more
5979a3fc17aSEaswaran Raman   // general threshold adjusting heuristics in updateThreshold().
5989a3fc17aSEaswaran Raman   Instruction *Instr = CS.getInstruction();
5999a3fc17aSEaswaran Raman   if (InvokeInst *II = dyn_cast<InvokeInst>(Instr)) {
6009a3fc17aSEaswaran Raman     if (isa<UnreachableInst>(II->getNormalDest()->getTerminator()))
6019a3fc17aSEaswaran Raman       return false;
6029a3fc17aSEaswaran Raman   } else if (isa<UnreachableInst>(Instr->getParent()->getTerminator()))
6039a3fc17aSEaswaran Raman     return false;
6049a3fc17aSEaswaran Raman 
6059a3fc17aSEaswaran Raman   return true;
6069a3fc17aSEaswaran Raman }
6079a3fc17aSEaswaran Raman 
608f4bb2f0dSEaswaran Raman void CallAnalyzer::updateThreshold(CallSite CS, Function &Callee) {
6099a3fc17aSEaswaran Raman   // If no size growth is allowed for this inlining, set Threshold to 0.
6109a3fc17aSEaswaran Raman   if (!allowSizeGrowth(CS)) {
6119a3fc17aSEaswaran Raman     Threshold = 0;
6129a3fc17aSEaswaran Raman     return;
6139a3fc17aSEaswaran Raman   }
6149a3fc17aSEaswaran Raman 
61530a93c18SEaswaran Raman   // If -inline-threshold is not given, listen to the optsize and minsize
61630a93c18SEaswaran Raman   // attributes when they would decrease the threshold.
617f4bb2f0dSEaswaran Raman   Function *Caller = CS.getCaller();
618f4bb2f0dSEaswaran Raman 
61930a93c18SEaswaran Raman   if (!(DefaultInlineThreshold.getNumOccurrences() > 0)) {
62030a93c18SEaswaran Raman     if (Caller->optForMinSize() && OptMinSizeThreshold < Threshold)
62130a93c18SEaswaran Raman       Threshold = OptMinSizeThreshold;
62230a93c18SEaswaran Raman     else if (Caller->optForSize() && OptSizeThreshold < Threshold)
623f4bb2f0dSEaswaran Raman       Threshold = OptSizeThreshold;
62430a93c18SEaswaran Raman   }
625f4bb2f0dSEaswaran Raman 
626f4bb2f0dSEaswaran Raman   // If profile information is available, use that to adjust threshold of hot
627f4bb2f0dSEaswaran Raman   // and cold functions.
628f4bb2f0dSEaswaran Raman   // FIXME: The heuristic used below for determining hotness and coldness are
629f4bb2f0dSEaswaran Raman   // based on preliminary SPEC tuning and may not be optimal. Replace this with
630f4bb2f0dSEaswaran Raman   // a well-tuned heuristic based on *callsite* hotness and not callee hotness.
631f4bb2f0dSEaswaran Raman   uint64_t FunctionCount = 0, MaxFunctionCount = 0;
632f4bb2f0dSEaswaran Raman   bool HasPGOCounts = false;
633d09f15eaSEric Liu   if (Callee.getEntryCount() && Callee.getParent()->getMaximumFunctionCount()) {
634f4bb2f0dSEaswaran Raman     HasPGOCounts = true;
635f4bb2f0dSEaswaran Raman     FunctionCount = Callee.getEntryCount().getValue();
636d09f15eaSEric Liu     MaxFunctionCount = Callee.getParent()->getMaximumFunctionCount().getValue();
637f4bb2f0dSEaswaran Raman   }
638f4bb2f0dSEaswaran Raman 
639f4bb2f0dSEaswaran Raman   // Listen to the inlinehint attribute or profile based hotness information
640f4bb2f0dSEaswaran Raman   // when it would increase the threshold and the caller does not need to
641f4bb2f0dSEaswaran Raman   // minimize its size.
642f4bb2f0dSEaswaran Raman   bool InlineHint =
643f4bb2f0dSEaswaran Raman       Callee.hasFnAttribute(Attribute::InlineHint) ||
644f4bb2f0dSEaswaran Raman       (HasPGOCounts &&
645f4bb2f0dSEaswaran Raman        FunctionCount >= (uint64_t)(0.3 * (double)MaxFunctionCount));
646f4bb2f0dSEaswaran Raman   if (InlineHint && HintThreshold > Threshold && !Caller->optForMinSize())
647f4bb2f0dSEaswaran Raman     Threshold = HintThreshold;
648f4bb2f0dSEaswaran Raman 
649f4bb2f0dSEaswaran Raman   // Listen to the cold attribute or profile based coldness information
650f4bb2f0dSEaswaran Raman   // when it would decrease the threshold.
651f4bb2f0dSEaswaran Raman   bool ColdCallee =
652f4bb2f0dSEaswaran Raman       Callee.hasFnAttribute(Attribute::Cold) ||
653f4bb2f0dSEaswaran Raman       (HasPGOCounts &&
654f4bb2f0dSEaswaran Raman        FunctionCount <= (uint64_t)(0.01 * (double)MaxFunctionCount));
655f4bb2f0dSEaswaran Raman   // Command line argument for DefaultInlineThreshold will override the default
656f4bb2f0dSEaswaran Raman   // ColdThreshold. If we have -inline-threshold but no -inlinecold-threshold,
657f4bb2f0dSEaswaran Raman   // do not use the default cold threshold even if it is smaller.
658f4bb2f0dSEaswaran Raman   if ((DefaultInlineThreshold.getNumOccurrences() == 0 ||
659f4bb2f0dSEaswaran Raman        ColdThreshold.getNumOccurrences() > 0) &&
660f4bb2f0dSEaswaran Raman       ColdCallee && ColdThreshold < Threshold)
661f4bb2f0dSEaswaran Raman     Threshold = ColdThreshold;
6628650a4daSJustin Lebar 
6638650a4daSJustin Lebar   // Finally, take the target-specific inlining threshold multiplier into
6648650a4daSJustin Lebar   // account.
6658650a4daSJustin Lebar   Threshold *= TTI.getInliningThresholdMultiplier();
666f4bb2f0dSEaswaran Raman }
667f4bb2f0dSEaswaran Raman 
6687adc3a2bSChandler Carruth bool CallAnalyzer::visitCmpInst(CmpInst &I) {
6697adc3a2bSChandler Carruth   Value *LHS = I.getOperand(0), *RHS = I.getOperand(1);
6707adc3a2bSChandler Carruth   // First try to handle simplified comparisons.
6717adc3a2bSChandler Carruth   if (!isa<Constant>(LHS))
6727adc3a2bSChandler Carruth     if (Constant *SimpleLHS = SimplifiedValues.lookup(LHS))
6737adc3a2bSChandler Carruth       LHS = SimpleLHS;
6747adc3a2bSChandler Carruth   if (!isa<Constant>(RHS))
6757adc3a2bSChandler Carruth     if (Constant *SimpleRHS = SimplifiedValues.lookup(RHS))
6767adc3a2bSChandler Carruth       RHS = SimpleRHS;
6777adc3a2bSChandler Carruth   if (Constant *CLHS = dyn_cast<Constant>(LHS)) {
6787adc3a2bSChandler Carruth     if (Constant *CRHS = dyn_cast<Constant>(RHS))
679567556aaSChad Rosier       if (Constant *C =
680567556aaSChad Rosier               ConstantExpr::getCompare(I.getPredicate(), CLHS, CRHS)) {
6817adc3a2bSChandler Carruth         SimplifiedValues[&I] = C;
6827adc3a2bSChandler Carruth         return true;
6837adc3a2bSChandler Carruth       }
6847adc3a2bSChandler Carruth   }
6857adc3a2bSChandler Carruth 
6867adc3a2bSChandler Carruth   if (I.getOpcode() == Instruction::FCmp)
6877adc3a2bSChandler Carruth     return false;
6887adc3a2bSChandler Carruth 
6897adc3a2bSChandler Carruth   // Otherwise look for a comparison between constant offset pointers with
6907adc3a2bSChandler Carruth   // a common base.
6917adc3a2bSChandler Carruth   Value *LHSBase, *RHSBase;
6927adc3a2bSChandler Carruth   APInt LHSOffset, RHSOffset;
6937adc3a2bSChandler Carruth   std::tie(LHSBase, LHSOffset) = ConstantOffsetPtrs.lookup(LHS);
6947adc3a2bSChandler Carruth   if (LHSBase) {
6957adc3a2bSChandler Carruth     std::tie(RHSBase, RHSOffset) = ConstantOffsetPtrs.lookup(RHS);
6967adc3a2bSChandler Carruth     if (RHSBase && LHSBase == RHSBase) {
6977adc3a2bSChandler Carruth       // We have common bases, fold the icmp to a constant based on the
6987adc3a2bSChandler Carruth       // offsets.
6997adc3a2bSChandler Carruth       Constant *CLHS = ConstantInt::get(LHS->getContext(), LHSOffset);
7007adc3a2bSChandler Carruth       Constant *CRHS = ConstantInt::get(RHS->getContext(), RHSOffset);
7017adc3a2bSChandler Carruth       if (Constant *C = ConstantExpr::getICmp(I.getPredicate(), CLHS, CRHS)) {
7027adc3a2bSChandler Carruth         SimplifiedValues[&I] = C;
7037adc3a2bSChandler Carruth         ++NumConstantPtrCmps;
7047adc3a2bSChandler Carruth         return true;
7057adc3a2bSChandler Carruth       }
7067adc3a2bSChandler Carruth     }
7077adc3a2bSChandler Carruth   }
7087adc3a2bSChandler Carruth 
7097adc3a2bSChandler Carruth   // If the comparison is an equality comparison with null, we can simplify it
7107adc3a2bSChandler Carruth   // if we know the value (argument) can't be null
7117adc3a2bSChandler Carruth   if (I.isEquality() && isa<ConstantPointerNull>(I.getOperand(1)) &&
7127adc3a2bSChandler Carruth       isKnownNonNullInCallee(I.getOperand(0))) {
7137adc3a2bSChandler Carruth     bool IsNotEqual = I.getPredicate() == CmpInst::ICMP_NE;
7147adc3a2bSChandler Carruth     SimplifiedValues[&I] = IsNotEqual ? ConstantInt::getTrue(I.getType())
7157adc3a2bSChandler Carruth                                       : ConstantInt::getFalse(I.getType());
7167adc3a2bSChandler Carruth     return true;
7177adc3a2bSChandler Carruth   }
7187adc3a2bSChandler Carruth   // Finally check for SROA candidates in comparisons.
7197adc3a2bSChandler Carruth   Value *SROAArg;
7207adc3a2bSChandler Carruth   DenseMap<Value *, int>::iterator CostIt;
7217adc3a2bSChandler Carruth   if (lookupSROAArgAndCost(I.getOperand(0), SROAArg, CostIt)) {
7227adc3a2bSChandler Carruth     if (isa<ConstantPointerNull>(I.getOperand(1))) {
7237adc3a2bSChandler Carruth       accumulateSROACost(CostIt, InlineConstants::InstrCost);
7247adc3a2bSChandler Carruth       return true;
7257adc3a2bSChandler Carruth     }
7267adc3a2bSChandler Carruth 
7277adc3a2bSChandler Carruth     disableSROA(CostIt);
7287adc3a2bSChandler Carruth   }
7297adc3a2bSChandler Carruth 
7307adc3a2bSChandler Carruth   return false;
7317adc3a2bSChandler Carruth }
7327adc3a2bSChandler Carruth 
7337adc3a2bSChandler Carruth bool CallAnalyzer::visitSub(BinaryOperator &I) {
7347adc3a2bSChandler Carruth   // Try to handle a special case: we can fold computing the difference of two
7357adc3a2bSChandler Carruth   // constant-related pointers.
7367adc3a2bSChandler Carruth   Value *LHS = I.getOperand(0), *RHS = I.getOperand(1);
7377adc3a2bSChandler Carruth   Value *LHSBase, *RHSBase;
7387adc3a2bSChandler Carruth   APInt LHSOffset, RHSOffset;
7397adc3a2bSChandler Carruth   std::tie(LHSBase, LHSOffset) = ConstantOffsetPtrs.lookup(LHS);
7407adc3a2bSChandler Carruth   if (LHSBase) {
7417adc3a2bSChandler Carruth     std::tie(RHSBase, RHSOffset) = ConstantOffsetPtrs.lookup(RHS);
7427adc3a2bSChandler Carruth     if (RHSBase && LHSBase == RHSBase) {
7437adc3a2bSChandler Carruth       // We have common bases, fold the subtract to a constant based on the
7447adc3a2bSChandler Carruth       // offsets.
7457adc3a2bSChandler Carruth       Constant *CLHS = ConstantInt::get(LHS->getContext(), LHSOffset);
7467adc3a2bSChandler Carruth       Constant *CRHS = ConstantInt::get(RHS->getContext(), RHSOffset);
7477adc3a2bSChandler Carruth       if (Constant *C = ConstantExpr::getSub(CLHS, CRHS)) {
7487adc3a2bSChandler Carruth         SimplifiedValues[&I] = C;
7497adc3a2bSChandler Carruth         ++NumConstantPtrDiffs;
7507adc3a2bSChandler Carruth         return true;
7517adc3a2bSChandler Carruth       }
7527adc3a2bSChandler Carruth     }
7537adc3a2bSChandler Carruth   }
7547adc3a2bSChandler Carruth 
7557adc3a2bSChandler Carruth   // Otherwise, fall back to the generic logic for simplifying and handling
7567adc3a2bSChandler Carruth   // instructions.
7577adc3a2bSChandler Carruth   return Base::visitSub(I);
7587adc3a2bSChandler Carruth }
7597adc3a2bSChandler Carruth 
7607adc3a2bSChandler Carruth bool CallAnalyzer::visitBinaryOperator(BinaryOperator &I) {
7617adc3a2bSChandler Carruth   Value *LHS = I.getOperand(0), *RHS = I.getOperand(1);
7627adc3a2bSChandler Carruth   const DataLayout &DL = F.getParent()->getDataLayout();
7637adc3a2bSChandler Carruth   if (!isa<Constant>(LHS))
7647adc3a2bSChandler Carruth     if (Constant *SimpleLHS = SimplifiedValues.lookup(LHS))
7657adc3a2bSChandler Carruth       LHS = SimpleLHS;
7667adc3a2bSChandler Carruth   if (!isa<Constant>(RHS))
7677adc3a2bSChandler Carruth     if (Constant *SimpleRHS = SimplifiedValues.lookup(RHS))
7687adc3a2bSChandler Carruth       RHS = SimpleRHS;
7697adc3a2bSChandler Carruth   Value *SimpleV = nullptr;
7707adc3a2bSChandler Carruth   if (auto FI = dyn_cast<FPMathOperator>(&I))
7717adc3a2bSChandler Carruth     SimpleV =
7727adc3a2bSChandler Carruth         SimplifyFPBinOp(I.getOpcode(), LHS, RHS, FI->getFastMathFlags(), DL);
7737adc3a2bSChandler Carruth   else
7747adc3a2bSChandler Carruth     SimpleV = SimplifyBinOp(I.getOpcode(), LHS, RHS, DL);
7757adc3a2bSChandler Carruth 
7767adc3a2bSChandler Carruth   if (Constant *C = dyn_cast_or_null<Constant>(SimpleV)) {
7777adc3a2bSChandler Carruth     SimplifiedValues[&I] = C;
7787adc3a2bSChandler Carruth     return true;
7797adc3a2bSChandler Carruth   }
7807adc3a2bSChandler Carruth 
7817adc3a2bSChandler Carruth   // Disable any SROA on arguments to arbitrary, unsimplified binary operators.
7827adc3a2bSChandler Carruth   disableSROA(LHS);
7837adc3a2bSChandler Carruth   disableSROA(RHS);
7847adc3a2bSChandler Carruth 
7857adc3a2bSChandler Carruth   return false;
7867adc3a2bSChandler Carruth }
7877adc3a2bSChandler Carruth 
7887adc3a2bSChandler Carruth bool CallAnalyzer::visitLoad(LoadInst &I) {
7897adc3a2bSChandler Carruth   Value *SROAArg;
7907adc3a2bSChandler Carruth   DenseMap<Value *, int>::iterator CostIt;
7917adc3a2bSChandler Carruth   if (lookupSROAArgAndCost(I.getPointerOperand(), SROAArg, CostIt)) {
7927adc3a2bSChandler Carruth     if (I.isSimple()) {
7937adc3a2bSChandler Carruth       accumulateSROACost(CostIt, InlineConstants::InstrCost);
7947adc3a2bSChandler Carruth       return true;
7957adc3a2bSChandler Carruth     }
7967adc3a2bSChandler Carruth 
7977adc3a2bSChandler Carruth     disableSROA(CostIt);
7987adc3a2bSChandler Carruth   }
7997adc3a2bSChandler Carruth 
8007adc3a2bSChandler Carruth   return false;
8017adc3a2bSChandler Carruth }
8027adc3a2bSChandler Carruth 
8037adc3a2bSChandler Carruth bool CallAnalyzer::visitStore(StoreInst &I) {
8047adc3a2bSChandler Carruth   Value *SROAArg;
8057adc3a2bSChandler Carruth   DenseMap<Value *, int>::iterator CostIt;
8067adc3a2bSChandler Carruth   if (lookupSROAArgAndCost(I.getPointerOperand(), SROAArg, CostIt)) {
8077adc3a2bSChandler Carruth     if (I.isSimple()) {
8087adc3a2bSChandler Carruth       accumulateSROACost(CostIt, InlineConstants::InstrCost);
8097adc3a2bSChandler Carruth       return true;
8107adc3a2bSChandler Carruth     }
8117adc3a2bSChandler Carruth 
8127adc3a2bSChandler Carruth     disableSROA(CostIt);
8137adc3a2bSChandler Carruth   }
8147adc3a2bSChandler Carruth 
8157adc3a2bSChandler Carruth   return false;
8167adc3a2bSChandler Carruth }
8177adc3a2bSChandler Carruth 
8187adc3a2bSChandler Carruth bool CallAnalyzer::visitExtractValue(ExtractValueInst &I) {
8197adc3a2bSChandler Carruth   // Constant folding for extract value is trivial.
8207adc3a2bSChandler Carruth   Constant *C = dyn_cast<Constant>(I.getAggregateOperand());
8217adc3a2bSChandler Carruth   if (!C)
8227adc3a2bSChandler Carruth     C = SimplifiedValues.lookup(I.getAggregateOperand());
8237adc3a2bSChandler Carruth   if (C) {
8247adc3a2bSChandler Carruth     SimplifiedValues[&I] = ConstantExpr::getExtractValue(C, I.getIndices());
8257adc3a2bSChandler Carruth     return true;
8267adc3a2bSChandler Carruth   }
8277adc3a2bSChandler Carruth 
8287adc3a2bSChandler Carruth   // SROA can look through these but give them a cost.
8297adc3a2bSChandler Carruth   return false;
8307adc3a2bSChandler Carruth }
8317adc3a2bSChandler Carruth 
8327adc3a2bSChandler Carruth bool CallAnalyzer::visitInsertValue(InsertValueInst &I) {
8337adc3a2bSChandler Carruth   // Constant folding for insert value is trivial.
8347adc3a2bSChandler Carruth   Constant *AggC = dyn_cast<Constant>(I.getAggregateOperand());
8357adc3a2bSChandler Carruth   if (!AggC)
8367adc3a2bSChandler Carruth     AggC = SimplifiedValues.lookup(I.getAggregateOperand());
8377adc3a2bSChandler Carruth   Constant *InsertedC = dyn_cast<Constant>(I.getInsertedValueOperand());
8387adc3a2bSChandler Carruth   if (!InsertedC)
8397adc3a2bSChandler Carruth     InsertedC = SimplifiedValues.lookup(I.getInsertedValueOperand());
8407adc3a2bSChandler Carruth   if (AggC && InsertedC) {
841567556aaSChad Rosier     SimplifiedValues[&I] =
842567556aaSChad Rosier         ConstantExpr::getInsertValue(AggC, InsertedC, I.getIndices());
8437adc3a2bSChandler Carruth     return true;
8447adc3a2bSChandler Carruth   }
8457adc3a2bSChandler Carruth 
8467adc3a2bSChandler Carruth   // SROA can look through these but give them a cost.
8477adc3a2bSChandler Carruth   return false;
8487adc3a2bSChandler Carruth }
8497adc3a2bSChandler Carruth 
8507adc3a2bSChandler Carruth /// \brief Try to simplify a call site.
8517adc3a2bSChandler Carruth ///
8527adc3a2bSChandler Carruth /// Takes a concrete function and callsite and tries to actually simplify it by
8537adc3a2bSChandler Carruth /// analyzing the arguments and call itself with instsimplify. Returns true if
8547adc3a2bSChandler Carruth /// it has simplified the callsite to some other entity (a constant), making it
8557adc3a2bSChandler Carruth /// free.
8567adc3a2bSChandler Carruth bool CallAnalyzer::simplifyCallSite(Function *F, CallSite CS) {
8577adc3a2bSChandler Carruth   // FIXME: Using the instsimplify logic directly for this is inefficient
8587adc3a2bSChandler Carruth   // because we have to continually rebuild the argument list even when no
8597adc3a2bSChandler Carruth   // simplifications can be performed. Until that is fixed with remapping
8607adc3a2bSChandler Carruth   // inside of instsimplify, directly constant fold calls here.
8617adc3a2bSChandler Carruth   if (!canConstantFoldCallTo(F))
8627adc3a2bSChandler Carruth     return false;
8637adc3a2bSChandler Carruth 
8647adc3a2bSChandler Carruth   // Try to re-map the arguments to constants.
8657adc3a2bSChandler Carruth   SmallVector<Constant *, 4> ConstantArgs;
8667adc3a2bSChandler Carruth   ConstantArgs.reserve(CS.arg_size());
867567556aaSChad Rosier   for (CallSite::arg_iterator I = CS.arg_begin(), E = CS.arg_end(); I != E;
868567556aaSChad Rosier        ++I) {
8697adc3a2bSChandler Carruth     Constant *C = dyn_cast<Constant>(*I);
8707adc3a2bSChandler Carruth     if (!C)
8717adc3a2bSChandler Carruth       C = dyn_cast_or_null<Constant>(SimplifiedValues.lookup(*I));
8727adc3a2bSChandler Carruth     if (!C)
8737adc3a2bSChandler Carruth       return false; // This argument doesn't map to a constant.
8747adc3a2bSChandler Carruth 
8757adc3a2bSChandler Carruth     ConstantArgs.push_back(C);
8767adc3a2bSChandler Carruth   }
8777adc3a2bSChandler Carruth   if (Constant *C = ConstantFoldCall(F, ConstantArgs)) {
8787adc3a2bSChandler Carruth     SimplifiedValues[CS.getInstruction()] = C;
8797adc3a2bSChandler Carruth     return true;
8807adc3a2bSChandler Carruth   }
8817adc3a2bSChandler Carruth 
8827adc3a2bSChandler Carruth   return false;
8837adc3a2bSChandler Carruth }
8847adc3a2bSChandler Carruth 
8857adc3a2bSChandler Carruth bool CallAnalyzer::visitCallSite(CallSite CS) {
8867adc3a2bSChandler Carruth   if (CS.hasFnAttr(Attribute::ReturnsTwice) &&
8877adc3a2bSChandler Carruth       !F.hasFnAttribute(Attribute::ReturnsTwice)) {
8887adc3a2bSChandler Carruth     // This aborts the entire analysis.
8897adc3a2bSChandler Carruth     ExposesReturnsTwice = true;
8907adc3a2bSChandler Carruth     return false;
8917adc3a2bSChandler Carruth   }
892567556aaSChad Rosier   if (CS.isCall() && cast<CallInst>(CS.getInstruction())->cannotDuplicate())
8937adc3a2bSChandler Carruth     ContainsNoDuplicateCall = true;
8947adc3a2bSChandler Carruth 
8957adc3a2bSChandler Carruth   if (Function *F = CS.getCalledFunction()) {
8967adc3a2bSChandler Carruth     // When we have a concrete function, first try to simplify it directly.
8977adc3a2bSChandler Carruth     if (simplifyCallSite(F, CS))
8987adc3a2bSChandler Carruth       return true;
8997adc3a2bSChandler Carruth 
9007adc3a2bSChandler Carruth     // Next check if it is an intrinsic we know about.
9017adc3a2bSChandler Carruth     // FIXME: Lift this into part of the InstVisitor.
9027adc3a2bSChandler Carruth     if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(CS.getInstruction())) {
9037adc3a2bSChandler Carruth       switch (II->getIntrinsicID()) {
9047adc3a2bSChandler Carruth       default:
9057adc3a2bSChandler Carruth         return Base::visitCallSite(CS);
9067adc3a2bSChandler Carruth 
9077dd8dbf4SPeter Collingbourne       case Intrinsic::load_relative:
9087dd8dbf4SPeter Collingbourne         // This is normally lowered to 4 LLVM instructions.
9097dd8dbf4SPeter Collingbourne         Cost += 3 * InlineConstants::InstrCost;
9107dd8dbf4SPeter Collingbourne         return false;
9117dd8dbf4SPeter Collingbourne 
9127adc3a2bSChandler Carruth       case Intrinsic::memset:
9137adc3a2bSChandler Carruth       case Intrinsic::memcpy:
9147adc3a2bSChandler Carruth       case Intrinsic::memmove:
9157adc3a2bSChandler Carruth         // SROA can usually chew through these intrinsics, but they aren't free.
9167adc3a2bSChandler Carruth         return false;
9177adc3a2bSChandler Carruth       case Intrinsic::localescape:
9187adc3a2bSChandler Carruth         HasFrameEscape = true;
9197adc3a2bSChandler Carruth         return false;
9207adc3a2bSChandler Carruth       }
9217adc3a2bSChandler Carruth     }
9227adc3a2bSChandler Carruth 
9237adc3a2bSChandler Carruth     if (F == CS.getInstruction()->getParent()->getParent()) {
9247adc3a2bSChandler Carruth       // This flag will fully abort the analysis, so don't bother with anything
9257adc3a2bSChandler Carruth       // else.
9267adc3a2bSChandler Carruth       IsRecursiveCall = true;
9277adc3a2bSChandler Carruth       return false;
9287adc3a2bSChandler Carruth     }
9297adc3a2bSChandler Carruth 
9307adc3a2bSChandler Carruth     if (TTI.isLoweredToCall(F)) {
9317adc3a2bSChandler Carruth       // We account for the average 1 instruction per call argument setup
9327adc3a2bSChandler Carruth       // here.
9337adc3a2bSChandler Carruth       Cost += CS.arg_size() * InlineConstants::InstrCost;
9347adc3a2bSChandler Carruth 
9357adc3a2bSChandler Carruth       // Everything other than inline ASM will also have a significant cost
9367adc3a2bSChandler Carruth       // merely from making the call.
9377adc3a2bSChandler Carruth       if (!isa<InlineAsm>(CS.getCalledValue()))
9387adc3a2bSChandler Carruth         Cost += InlineConstants::CallPenalty;
9397adc3a2bSChandler Carruth     }
9407adc3a2bSChandler Carruth 
9417adc3a2bSChandler Carruth     return Base::visitCallSite(CS);
9427adc3a2bSChandler Carruth   }
9437adc3a2bSChandler Carruth 
9447adc3a2bSChandler Carruth   // Otherwise we're in a very special case -- an indirect function call. See
9457adc3a2bSChandler Carruth   // if we can be particularly clever about this.
9467adc3a2bSChandler Carruth   Value *Callee = CS.getCalledValue();
9477adc3a2bSChandler Carruth 
9487adc3a2bSChandler Carruth   // First, pay the price of the argument setup. We account for the average
9497adc3a2bSChandler Carruth   // 1 instruction per call argument setup here.
9507adc3a2bSChandler Carruth   Cost += CS.arg_size() * InlineConstants::InstrCost;
9517adc3a2bSChandler Carruth 
9527adc3a2bSChandler Carruth   // Next, check if this happens to be an indirect function call to a known
9537adc3a2bSChandler Carruth   // function in this inline context. If not, we've done all we can.
9547adc3a2bSChandler Carruth   Function *F = dyn_cast_or_null<Function>(SimplifiedValues.lookup(Callee));
9557adc3a2bSChandler Carruth   if (!F)
9567adc3a2bSChandler Carruth     return Base::visitCallSite(CS);
9577adc3a2bSChandler Carruth 
9587adc3a2bSChandler Carruth   // If we have a constant that we are calling as a function, we can peer
9597adc3a2bSChandler Carruth   // through it and see the function target. This happens not infrequently
9607adc3a2bSChandler Carruth   // during devirtualization and so we want to give it a hefty bonus for
9617adc3a2bSChandler Carruth   // inlining, but cap that bonus in the event that inlining wouldn't pan
9627adc3a2bSChandler Carruth   // out. Pretend to inline the function, with a custom threshold.
963b1bd398cSEaswaran Raman   CallAnalyzer CA(TTI, ACT, *F, InlineConstants::IndirectCallThreshold, CS);
9647adc3a2bSChandler Carruth   if (CA.analyzeCall(CS)) {
9657adc3a2bSChandler Carruth     // We were able to inline the indirect call! Subtract the cost from the
9666d90d9f1SEaswaran Raman     // threshold to get the bonus we want to apply, but don't go below zero.
9676d90d9f1SEaswaran Raman     Cost -= std::max(0, CA.getThreshold() - CA.getCost());
9687adc3a2bSChandler Carruth   }
9697adc3a2bSChandler Carruth 
9707adc3a2bSChandler Carruth   return Base::visitCallSite(CS);
9717adc3a2bSChandler Carruth }
9727adc3a2bSChandler Carruth 
9737adc3a2bSChandler Carruth bool CallAnalyzer::visitReturnInst(ReturnInst &RI) {
9747adc3a2bSChandler Carruth   // At least one return instruction will be free after inlining.
9757adc3a2bSChandler Carruth   bool Free = !HasReturn;
9767adc3a2bSChandler Carruth   HasReturn = true;
9777adc3a2bSChandler Carruth   return Free;
9787adc3a2bSChandler Carruth }
9797adc3a2bSChandler Carruth 
9807adc3a2bSChandler Carruth bool CallAnalyzer::visitBranchInst(BranchInst &BI) {
9817adc3a2bSChandler Carruth   // We model unconditional branches as essentially free -- they really
9827adc3a2bSChandler Carruth   // shouldn't exist at all, but handling them makes the behavior of the
9837adc3a2bSChandler Carruth   // inliner more regular and predictable. Interestingly, conditional branches
9847adc3a2bSChandler Carruth   // which will fold away are also free.
9857adc3a2bSChandler Carruth   return BI.isUnconditional() || isa<ConstantInt>(BI.getCondition()) ||
9867adc3a2bSChandler Carruth          dyn_cast_or_null<ConstantInt>(
9877adc3a2bSChandler Carruth              SimplifiedValues.lookup(BI.getCondition()));
9887adc3a2bSChandler Carruth }
9897adc3a2bSChandler Carruth 
9907adc3a2bSChandler Carruth bool CallAnalyzer::visitSwitchInst(SwitchInst &SI) {
9917adc3a2bSChandler Carruth   // We model unconditional switches as free, see the comments on handling
9927adc3a2bSChandler Carruth   // branches.
9937adc3a2bSChandler Carruth   if (isa<ConstantInt>(SI.getCondition()))
9947adc3a2bSChandler Carruth     return true;
9957adc3a2bSChandler Carruth   if (Value *V = SimplifiedValues.lookup(SI.getCondition()))
9967adc3a2bSChandler Carruth     if (isa<ConstantInt>(V))
9977adc3a2bSChandler Carruth       return true;
9987adc3a2bSChandler Carruth 
9997adc3a2bSChandler Carruth   // Otherwise, we need to accumulate a cost proportional to the number of
10007adc3a2bSChandler Carruth   // distinct successor blocks. This fan-out in the CFG cannot be represented
10017adc3a2bSChandler Carruth   // for free even if we can represent the core switch as a jumptable that
10027adc3a2bSChandler Carruth   // takes a single instruction.
10037adc3a2bSChandler Carruth   //
10047adc3a2bSChandler Carruth   // NB: We convert large switches which are just used to initialize large phi
10057adc3a2bSChandler Carruth   // nodes to lookup tables instead in simplify-cfg, so this shouldn't prevent
10067adc3a2bSChandler Carruth   // inlining those. It will prevent inlining in cases where the optimization
10077adc3a2bSChandler Carruth   // does not (yet) fire.
10087adc3a2bSChandler Carruth   SmallPtrSet<BasicBlock *, 8> SuccessorBlocks;
10097adc3a2bSChandler Carruth   SuccessorBlocks.insert(SI.getDefaultDest());
10107adc3a2bSChandler Carruth   for (auto I = SI.case_begin(), E = SI.case_end(); I != E; ++I)
10117adc3a2bSChandler Carruth     SuccessorBlocks.insert(I.getCaseSuccessor());
10127adc3a2bSChandler Carruth   // Add cost corresponding to the number of distinct destinations. The first
10137adc3a2bSChandler Carruth   // we model as free because of fallthrough.
10147adc3a2bSChandler Carruth   Cost += (SuccessorBlocks.size() - 1) * InlineConstants::InstrCost;
10157adc3a2bSChandler Carruth   return false;
10167adc3a2bSChandler Carruth }
10177adc3a2bSChandler Carruth 
10187adc3a2bSChandler Carruth bool CallAnalyzer::visitIndirectBrInst(IndirectBrInst &IBI) {
10197adc3a2bSChandler Carruth   // We never want to inline functions that contain an indirectbr.  This is
10207adc3a2bSChandler Carruth   // incorrect because all the blockaddress's (in static global initializers
10217adc3a2bSChandler Carruth   // for example) would be referring to the original function, and this
10227adc3a2bSChandler Carruth   // indirect jump would jump from the inlined copy of the function into the
10237adc3a2bSChandler Carruth   // original function which is extremely undefined behavior.
10247adc3a2bSChandler Carruth   // FIXME: This logic isn't really right; we can safely inline functions with
10257adc3a2bSChandler Carruth   // indirectbr's as long as no other function or global references the
10267adc3a2bSChandler Carruth   // blockaddress of a block within the current function.
10277adc3a2bSChandler Carruth   HasIndirectBr = true;
10287adc3a2bSChandler Carruth   return false;
10297adc3a2bSChandler Carruth }
10307adc3a2bSChandler Carruth 
10317adc3a2bSChandler Carruth bool CallAnalyzer::visitResumeInst(ResumeInst &RI) {
10327adc3a2bSChandler Carruth   // FIXME: It's not clear that a single instruction is an accurate model for
10337adc3a2bSChandler Carruth   // the inline cost of a resume instruction.
10347adc3a2bSChandler Carruth   return false;
10357adc3a2bSChandler Carruth }
10367adc3a2bSChandler Carruth 
10377adc3a2bSChandler Carruth bool CallAnalyzer::visitCleanupReturnInst(CleanupReturnInst &CRI) {
10387adc3a2bSChandler Carruth   // FIXME: It's not clear that a single instruction is an accurate model for
10397adc3a2bSChandler Carruth   // the inline cost of a cleanupret instruction.
10407adc3a2bSChandler Carruth   return false;
10417adc3a2bSChandler Carruth }
10427adc3a2bSChandler Carruth 
10437adc3a2bSChandler Carruth bool CallAnalyzer::visitCatchReturnInst(CatchReturnInst &CRI) {
10447adc3a2bSChandler Carruth   // FIXME: It's not clear that a single instruction is an accurate model for
10458220bcc5SJoseph Tremoulet   // the inline cost of a catchret instruction.
10467adc3a2bSChandler Carruth   return false;
10477adc3a2bSChandler Carruth }
10487adc3a2bSChandler Carruth 
10497adc3a2bSChandler Carruth bool CallAnalyzer::visitUnreachableInst(UnreachableInst &I) {
10507adc3a2bSChandler Carruth   // FIXME: It might be reasonably to discount the cost of instructions leading
10517adc3a2bSChandler Carruth   // to unreachable as they have the lowest possible impact on both runtime and
10527adc3a2bSChandler Carruth   // code size.
10537adc3a2bSChandler Carruth   return true; // No actual code is needed for unreachable.
10547adc3a2bSChandler Carruth }
10557adc3a2bSChandler Carruth 
10567adc3a2bSChandler Carruth bool CallAnalyzer::visitInstruction(Instruction &I) {
10577adc3a2bSChandler Carruth   // Some instructions are free. All of the free intrinsics can also be
10587adc3a2bSChandler Carruth   // handled by SROA, etc.
10597adc3a2bSChandler Carruth   if (TargetTransformInfo::TCC_Free == TTI.getUserCost(&I))
10607adc3a2bSChandler Carruth     return true;
10617adc3a2bSChandler Carruth 
10627adc3a2bSChandler Carruth   // We found something we don't understand or can't handle. Mark any SROA-able
10637adc3a2bSChandler Carruth   // values in the operand list as no longer viable.
10647adc3a2bSChandler Carruth   for (User::op_iterator OI = I.op_begin(), OE = I.op_end(); OI != OE; ++OI)
10657adc3a2bSChandler Carruth     disableSROA(*OI);
10667adc3a2bSChandler Carruth 
10677adc3a2bSChandler Carruth   return false;
10687adc3a2bSChandler Carruth }
10697adc3a2bSChandler Carruth 
10707adc3a2bSChandler Carruth /// \brief Analyze a basic block for its contribution to the inline cost.
10717adc3a2bSChandler Carruth ///
10727adc3a2bSChandler Carruth /// This method walks the analyzer over every instruction in the given basic
10737adc3a2bSChandler Carruth /// block and accounts for their cost during inlining at this callsite. It
10747adc3a2bSChandler Carruth /// aborts early if the threshold has been exceeded or an impossible to inline
10757adc3a2bSChandler Carruth /// construct has been detected. It returns false if inlining is no longer
10767adc3a2bSChandler Carruth /// viable, and true if inlining remains viable.
10777adc3a2bSChandler Carruth bool CallAnalyzer::analyzeBlock(BasicBlock *BB,
10787adc3a2bSChandler Carruth                                 SmallPtrSetImpl<const Value *> &EphValues) {
10797adc3a2bSChandler Carruth   for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I) {
10807adc3a2bSChandler Carruth     // FIXME: Currently, the number of instructions in a function regardless of
10817adc3a2bSChandler Carruth     // our ability to simplify them during inline to constants or dead code,
10827adc3a2bSChandler Carruth     // are actually used by the vector bonus heuristic. As long as that's true,
10837adc3a2bSChandler Carruth     // we have to special case debug intrinsics here to prevent differences in
10847adc3a2bSChandler Carruth     // inlining due to debug symbols. Eventually, the number of unsimplified
10857adc3a2bSChandler Carruth     // instructions shouldn't factor into the cost computation, but until then,
10867adc3a2bSChandler Carruth     // hack around it here.
10877adc3a2bSChandler Carruth     if (isa<DbgInfoIntrinsic>(I))
10887adc3a2bSChandler Carruth       continue;
10897adc3a2bSChandler Carruth 
10907adc3a2bSChandler Carruth     // Skip ephemeral values.
10915a82c916SDuncan P. N. Exon Smith     if (EphValues.count(&*I))
10927adc3a2bSChandler Carruth       continue;
10937adc3a2bSChandler Carruth 
10947adc3a2bSChandler Carruth     ++NumInstructions;
10957adc3a2bSChandler Carruth     if (isa<ExtractElementInst>(I) || I->getType()->isVectorTy())
10967adc3a2bSChandler Carruth       ++NumVectorInstructions;
10977adc3a2bSChandler Carruth 
1098e9434e80SSanjay Patel     // If the instruction is floating point, and the target says this operation
1099e9434e80SSanjay Patel     // is expensive or the function has the "use-soft-float" attribute, this may
11007adc3a2bSChandler Carruth     // eventually become a library call. Treat the cost as such.
11017adc3a2bSChandler Carruth     if (I->getType()->isFloatingPointTy()) {
11027adc3a2bSChandler Carruth       bool hasSoftFloatAttr = false;
11037adc3a2bSChandler Carruth 
1104e9434e80SSanjay Patel       // If the function has the "use-soft-float" attribute, mark it as
1105e9434e80SSanjay Patel       // expensive.
11067adc3a2bSChandler Carruth       if (F.hasFnAttribute("use-soft-float")) {
11077adc3a2bSChandler Carruth         Attribute Attr = F.getFnAttribute("use-soft-float");
11087adc3a2bSChandler Carruth         StringRef Val = Attr.getValueAsString();
11097adc3a2bSChandler Carruth         if (Val == "true")
11107adc3a2bSChandler Carruth           hasSoftFloatAttr = true;
11117adc3a2bSChandler Carruth       }
11127adc3a2bSChandler Carruth 
11137adc3a2bSChandler Carruth       if (TTI.getFPOpCost(I->getType()) == TargetTransformInfo::TCC_Expensive ||
11147adc3a2bSChandler Carruth           hasSoftFloatAttr)
11157adc3a2bSChandler Carruth         Cost += InlineConstants::CallPenalty;
11167adc3a2bSChandler Carruth     }
11177adc3a2bSChandler Carruth 
11187adc3a2bSChandler Carruth     // If the instruction simplified to a constant, there is no cost to this
11197adc3a2bSChandler Carruth     // instruction. Visit the instructions using our InstVisitor to account for
11207adc3a2bSChandler Carruth     // all of the per-instruction logic. The visit tree returns true if we
11217adc3a2bSChandler Carruth     // consumed the instruction in any way, and false if the instruction's base
11227adc3a2bSChandler Carruth     // cost should count against inlining.
11235a82c916SDuncan P. N. Exon Smith     if (Base::visit(&*I))
11247adc3a2bSChandler Carruth       ++NumInstructionsSimplified;
11257adc3a2bSChandler Carruth     else
11267adc3a2bSChandler Carruth       Cost += InlineConstants::InstrCost;
11277adc3a2bSChandler Carruth 
11287adc3a2bSChandler Carruth     // If the visit this instruction detected an uninlinable pattern, abort.
11297adc3a2bSChandler Carruth     if (IsRecursiveCall || ExposesReturnsTwice || HasDynamicAlloca ||
11307adc3a2bSChandler Carruth         HasIndirectBr || HasFrameEscape)
11317adc3a2bSChandler Carruth       return false;
11327adc3a2bSChandler Carruth 
11337adc3a2bSChandler Carruth     // If the caller is a recursive function then we don't want to inline
11347adc3a2bSChandler Carruth     // functions which allocate a lot of stack space because it would increase
11357adc3a2bSChandler Carruth     // the caller stack usage dramatically.
11367adc3a2bSChandler Carruth     if (IsCallerRecursive &&
11377adc3a2bSChandler Carruth         AllocatedSize > InlineConstants::TotalAllocaSizeRecursiveCaller)
11387adc3a2bSChandler Carruth       return false;
11397adc3a2bSChandler Carruth 
11407adc3a2bSChandler Carruth     // Check if we've past the maximum possible threshold so we don't spin in
11417adc3a2bSChandler Carruth     // huge basic blocks that will never inline.
11427adc3a2bSChandler Carruth     if (Cost > Threshold)
11437adc3a2bSChandler Carruth       return false;
11447adc3a2bSChandler Carruth   }
11457adc3a2bSChandler Carruth 
11467adc3a2bSChandler Carruth   return true;
11477adc3a2bSChandler Carruth }
11487adc3a2bSChandler Carruth 
11497adc3a2bSChandler Carruth /// \brief Compute the base pointer and cumulative constant offsets for V.
11507adc3a2bSChandler Carruth ///
11517adc3a2bSChandler Carruth /// This strips all constant offsets off of V, leaving it the base pointer, and
11527adc3a2bSChandler Carruth /// accumulates the total constant offset applied in the returned constant. It
11537adc3a2bSChandler Carruth /// returns 0 if V is not a pointer, and returns the constant '0' if there are
11547adc3a2bSChandler Carruth /// no constant offsets applied.
11557adc3a2bSChandler Carruth ConstantInt *CallAnalyzer::stripAndComputeInBoundsConstantOffsets(Value *&V) {
11567adc3a2bSChandler Carruth   if (!V->getType()->isPointerTy())
11577adc3a2bSChandler Carruth     return nullptr;
11587adc3a2bSChandler Carruth 
11597adc3a2bSChandler Carruth   const DataLayout &DL = F.getParent()->getDataLayout();
11607adc3a2bSChandler Carruth   unsigned IntPtrWidth = DL.getPointerSizeInBits();
11617adc3a2bSChandler Carruth   APInt Offset = APInt::getNullValue(IntPtrWidth);
11627adc3a2bSChandler Carruth 
11637adc3a2bSChandler Carruth   // Even though we don't look through PHI nodes, we could be called on an
11647adc3a2bSChandler Carruth   // instruction in an unreachable block, which may be on a cycle.
11657adc3a2bSChandler Carruth   SmallPtrSet<Value *, 4> Visited;
11667adc3a2bSChandler Carruth   Visited.insert(V);
11677adc3a2bSChandler Carruth   do {
11687adc3a2bSChandler Carruth     if (GEPOperator *GEP = dyn_cast<GEPOperator>(V)) {
11697adc3a2bSChandler Carruth       if (!GEP->isInBounds() || !accumulateGEPOffset(*GEP, Offset))
11707adc3a2bSChandler Carruth         return nullptr;
11717adc3a2bSChandler Carruth       V = GEP->getPointerOperand();
11727adc3a2bSChandler Carruth     } else if (Operator::getOpcode(V) == Instruction::BitCast) {
11737adc3a2bSChandler Carruth       V = cast<Operator>(V)->getOperand(0);
11747adc3a2bSChandler Carruth     } else if (GlobalAlias *GA = dyn_cast<GlobalAlias>(V)) {
11755ce32728SSanjoy Das       if (GA->isInterposable())
11767adc3a2bSChandler Carruth         break;
11777adc3a2bSChandler Carruth       V = GA->getAliasee();
11787adc3a2bSChandler Carruth     } else {
11797adc3a2bSChandler Carruth       break;
11807adc3a2bSChandler Carruth     }
11817adc3a2bSChandler Carruth     assert(V->getType()->isPointerTy() && "Unexpected operand type!");
11827adc3a2bSChandler Carruth   } while (Visited.insert(V).second);
11837adc3a2bSChandler Carruth 
11847adc3a2bSChandler Carruth   Type *IntPtrTy = DL.getIntPtrType(V->getContext());
11857adc3a2bSChandler Carruth   return cast<ConstantInt>(ConstantInt::get(IntPtrTy, Offset));
11867adc3a2bSChandler Carruth }
11877adc3a2bSChandler Carruth 
11887adc3a2bSChandler Carruth /// \brief Analyze a call site for potential inlining.
11897adc3a2bSChandler Carruth ///
11907adc3a2bSChandler Carruth /// Returns true if inlining this call is viable, and false if it is not
11917adc3a2bSChandler Carruth /// viable. It computes the cost and adjusts the threshold based on numerous
11927adc3a2bSChandler Carruth /// factors and heuristics. If this method returns false but the computed cost
11937adc3a2bSChandler Carruth /// is below the computed threshold, then inlining was forcibly disabled by
11947adc3a2bSChandler Carruth /// some artifact of the routine.
11957adc3a2bSChandler Carruth bool CallAnalyzer::analyzeCall(CallSite CS) {
11967adc3a2bSChandler Carruth   ++NumCallsAnalyzed;
11977adc3a2bSChandler Carruth 
11987adc3a2bSChandler Carruth   // Perform some tweaks to the cost and threshold based on the direct
11997adc3a2bSChandler Carruth   // callsite information.
12007adc3a2bSChandler Carruth 
12017adc3a2bSChandler Carruth   // We want to more aggressively inline vector-dense kernels, so up the
12027adc3a2bSChandler Carruth   // threshold, and we'll lower it if the % of vector instructions gets too
12037adc3a2bSChandler Carruth   // low. Note that these bonuses are some what arbitrary and evolved over time
12047adc3a2bSChandler Carruth   // by accident as much as because they are principled bonuses.
12057adc3a2bSChandler Carruth   //
12067adc3a2bSChandler Carruth   // FIXME: It would be nice to remove all such bonuses. At least it would be
12077adc3a2bSChandler Carruth   // nice to base the bonus values on something more scientific.
12087adc3a2bSChandler Carruth   assert(NumInstructions == 0);
12097adc3a2bSChandler Carruth   assert(NumVectorInstructions == 0);
1210f4bb2f0dSEaswaran Raman 
1211f4bb2f0dSEaswaran Raman   // Update the threshold based on callsite properties
1212f4bb2f0dSEaswaran Raman   updateThreshold(CS, F);
1213f4bb2f0dSEaswaran Raman 
12147adc3a2bSChandler Carruth   FiftyPercentVectorBonus = 3 * Threshold / 2;
12157adc3a2bSChandler Carruth   TenPercentVectorBonus = 3 * Threshold / 4;
12167adc3a2bSChandler Carruth   const DataLayout &DL = F.getParent()->getDataLayout();
12177adc3a2bSChandler Carruth 
12187adc3a2bSChandler Carruth   // Track whether the post-inlining function would have more than one basic
12197adc3a2bSChandler Carruth   // block. A single basic block is often intended for inlining. Balloon the
12207adc3a2bSChandler Carruth   // threshold by 50% until we pass the single-BB phase.
12217adc3a2bSChandler Carruth   bool SingleBB = true;
12227adc3a2bSChandler Carruth   int SingleBBBonus = Threshold / 2;
12237adc3a2bSChandler Carruth 
12247adc3a2bSChandler Carruth   // Speculatively apply all possible bonuses to Threshold. If cost exceeds
12257adc3a2bSChandler Carruth   // this Threshold any time, and cost cannot decrease, we can stop processing
12267adc3a2bSChandler Carruth   // the rest of the function body.
12277adc3a2bSChandler Carruth   Threshold += (SingleBBBonus + FiftyPercentVectorBonus);
12287adc3a2bSChandler Carruth 
12297adc3a2bSChandler Carruth   // Give out bonuses per argument, as the instructions setting them up will
12307adc3a2bSChandler Carruth   // be gone after inlining.
12317adc3a2bSChandler Carruth   for (unsigned I = 0, E = CS.arg_size(); I != E; ++I) {
12327adc3a2bSChandler Carruth     if (CS.isByValArgument(I)) {
12337adc3a2bSChandler Carruth       // We approximate the number of loads and stores needed by dividing the
12347adc3a2bSChandler Carruth       // size of the byval type by the target's pointer size.
12357adc3a2bSChandler Carruth       PointerType *PTy = cast<PointerType>(CS.getArgument(I)->getType());
12367adc3a2bSChandler Carruth       unsigned TypeSize = DL.getTypeSizeInBits(PTy->getElementType());
12377adc3a2bSChandler Carruth       unsigned PointerSize = DL.getPointerSizeInBits();
12387adc3a2bSChandler Carruth       // Ceiling division.
12397adc3a2bSChandler Carruth       unsigned NumStores = (TypeSize + PointerSize - 1) / PointerSize;
12407adc3a2bSChandler Carruth 
12417adc3a2bSChandler Carruth       // If it generates more than 8 stores it is likely to be expanded as an
12427adc3a2bSChandler Carruth       // inline memcpy so we take that as an upper bound. Otherwise we assume
12437adc3a2bSChandler Carruth       // one load and one store per word copied.
12447adc3a2bSChandler Carruth       // FIXME: The maxStoresPerMemcpy setting from the target should be used
12457adc3a2bSChandler Carruth       // here instead of a magic number of 8, but it's not available via
12467adc3a2bSChandler Carruth       // DataLayout.
12477adc3a2bSChandler Carruth       NumStores = std::min(NumStores, 8U);
12487adc3a2bSChandler Carruth 
12497adc3a2bSChandler Carruth       Cost -= 2 * NumStores * InlineConstants::InstrCost;
12507adc3a2bSChandler Carruth     } else {
12517adc3a2bSChandler Carruth       // For non-byval arguments subtract off one instruction per call
12527adc3a2bSChandler Carruth       // argument.
12537adc3a2bSChandler Carruth       Cost -= InlineConstants::InstrCost;
12547adc3a2bSChandler Carruth     }
12557adc3a2bSChandler Carruth   }
12567adc3a2bSChandler Carruth 
12577adc3a2bSChandler Carruth   // If there is only one call of the function, and it has internal linkage,
12587adc3a2bSChandler Carruth   // the cost of inlining it drops dramatically.
1259567556aaSChad Rosier   bool OnlyOneCallAndLocalLinkage =
1260567556aaSChad Rosier       F.hasLocalLinkage() && F.hasOneUse() && &F == CS.getCalledFunction();
12617adc3a2bSChandler Carruth   if (OnlyOneCallAndLocalLinkage)
12627adc3a2bSChandler Carruth     Cost += InlineConstants::LastCallToStaticBonus;
12637adc3a2bSChandler Carruth 
12647adc3a2bSChandler Carruth   // If this function uses the coldcc calling convention, prefer not to inline
12657adc3a2bSChandler Carruth   // it.
12667adc3a2bSChandler Carruth   if (F.getCallingConv() == CallingConv::Cold)
12677adc3a2bSChandler Carruth     Cost += InlineConstants::ColdccPenalty;
12687adc3a2bSChandler Carruth 
12697adc3a2bSChandler Carruth   // Check if we're done. This can happen due to bonuses and penalties.
12707adc3a2bSChandler Carruth   if (Cost > Threshold)
12717adc3a2bSChandler Carruth     return false;
12727adc3a2bSChandler Carruth 
12737adc3a2bSChandler Carruth   if (F.empty())
12747adc3a2bSChandler Carruth     return true;
12757adc3a2bSChandler Carruth 
12767adc3a2bSChandler Carruth   Function *Caller = CS.getInstruction()->getParent()->getParent();
12777adc3a2bSChandler Carruth   // Check if the caller function is recursive itself.
12787adc3a2bSChandler Carruth   for (User *U : Caller->users()) {
12797adc3a2bSChandler Carruth     CallSite Site(U);
12807adc3a2bSChandler Carruth     if (!Site)
12817adc3a2bSChandler Carruth       continue;
12827adc3a2bSChandler Carruth     Instruction *I = Site.getInstruction();
12837adc3a2bSChandler Carruth     if (I->getParent()->getParent() == Caller) {
12847adc3a2bSChandler Carruth       IsCallerRecursive = true;
12857adc3a2bSChandler Carruth       break;
12867adc3a2bSChandler Carruth     }
12877adc3a2bSChandler Carruth   }
12887adc3a2bSChandler Carruth 
12897adc3a2bSChandler Carruth   // Populate our simplified values by mapping from function arguments to call
12907adc3a2bSChandler Carruth   // arguments with known important simplifications.
12917adc3a2bSChandler Carruth   CallSite::arg_iterator CAI = CS.arg_begin();
12927adc3a2bSChandler Carruth   for (Function::arg_iterator FAI = F.arg_begin(), FAE = F.arg_end();
12937adc3a2bSChandler Carruth        FAI != FAE; ++FAI, ++CAI) {
12947adc3a2bSChandler Carruth     assert(CAI != CS.arg_end());
12957adc3a2bSChandler Carruth     if (Constant *C = dyn_cast<Constant>(CAI))
12965a82c916SDuncan P. N. Exon Smith       SimplifiedValues[&*FAI] = C;
12977adc3a2bSChandler Carruth 
12987adc3a2bSChandler Carruth     Value *PtrArg = *CAI;
12997adc3a2bSChandler Carruth     if (ConstantInt *C = stripAndComputeInBoundsConstantOffsets(PtrArg)) {
13005a82c916SDuncan P. N. Exon Smith       ConstantOffsetPtrs[&*FAI] = std::make_pair(PtrArg, C->getValue());
13017adc3a2bSChandler Carruth 
13027adc3a2bSChandler Carruth       // We can SROA any pointer arguments derived from alloca instructions.
13037adc3a2bSChandler Carruth       if (isa<AllocaInst>(PtrArg)) {
13045a82c916SDuncan P. N. Exon Smith         SROAArgValues[&*FAI] = PtrArg;
13057adc3a2bSChandler Carruth         SROAArgCosts[PtrArg] = 0;
13067adc3a2bSChandler Carruth       }
13077adc3a2bSChandler Carruth     }
13087adc3a2bSChandler Carruth   }
13097adc3a2bSChandler Carruth   NumConstantArgs = SimplifiedValues.size();
13107adc3a2bSChandler Carruth   NumConstantOffsetPtrArgs = ConstantOffsetPtrs.size();
13117adc3a2bSChandler Carruth   NumAllocaArgs = SROAArgValues.size();
13127adc3a2bSChandler Carruth 
13137adc3a2bSChandler Carruth   // FIXME: If a caller has multiple calls to a callee, we end up recomputing
13147adc3a2bSChandler Carruth   // the ephemeral values multiple times (and they're completely determined by
13157adc3a2bSChandler Carruth   // the callee, so this is purely duplicate work).
13167adc3a2bSChandler Carruth   SmallPtrSet<const Value *, 32> EphValues;
1317567556aaSChad Rosier   CodeMetrics::collectEphemeralValues(&F, &ACT->getAssumptionCache(F),
1318567556aaSChad Rosier                                       EphValues);
13197adc3a2bSChandler Carruth 
13207adc3a2bSChandler Carruth   // The worklist of live basic blocks in the callee *after* inlining. We avoid
13217adc3a2bSChandler Carruth   // adding basic blocks of the callee which can be proven to be dead for this
13227adc3a2bSChandler Carruth   // particular call site in order to get more accurate cost estimates. This
13237adc3a2bSChandler Carruth   // requires a somewhat heavyweight iteration pattern: we need to walk the
13247adc3a2bSChandler Carruth   // basic blocks in a breadth-first order as we insert live successors. To
13257adc3a2bSChandler Carruth   // accomplish this, prioritizing for small iterations because we exit after
13267adc3a2bSChandler Carruth   // crossing our threshold, we use a small-size optimized SetVector.
13277adc3a2bSChandler Carruth   typedef SetVector<BasicBlock *, SmallVector<BasicBlock *, 16>,
1328567556aaSChad Rosier                     SmallPtrSet<BasicBlock *, 16>>
1329567556aaSChad Rosier       BBSetVector;
13307adc3a2bSChandler Carruth   BBSetVector BBWorklist;
13317adc3a2bSChandler Carruth   BBWorklist.insert(&F.getEntryBlock());
13327adc3a2bSChandler Carruth   // Note that we *must not* cache the size, this loop grows the worklist.
13337adc3a2bSChandler Carruth   for (unsigned Idx = 0; Idx != BBWorklist.size(); ++Idx) {
13347adc3a2bSChandler Carruth     // Bail out the moment we cross the threshold. This means we'll under-count
13357adc3a2bSChandler Carruth     // the cost, but only when undercounting doesn't matter.
13367adc3a2bSChandler Carruth     if (Cost > Threshold)
13377adc3a2bSChandler Carruth       break;
13387adc3a2bSChandler Carruth 
13397adc3a2bSChandler Carruth     BasicBlock *BB = BBWorklist[Idx];
13407adc3a2bSChandler Carruth     if (BB->empty())
13417adc3a2bSChandler Carruth       continue;
13427adc3a2bSChandler Carruth 
13437adc3a2bSChandler Carruth     // Disallow inlining a blockaddress. A blockaddress only has defined
13447adc3a2bSChandler Carruth     // behavior for an indirect branch in the same function, and we do not
13457adc3a2bSChandler Carruth     // currently support inlining indirect branches. But, the inliner may not
13467adc3a2bSChandler Carruth     // see an indirect branch that ends up being dead code at a particular call
13477adc3a2bSChandler Carruth     // site. If the blockaddress escapes the function, e.g., via a global
13487adc3a2bSChandler Carruth     // variable, inlining may lead to an invalid cross-function reference.
13497adc3a2bSChandler Carruth     if (BB->hasAddressTaken())
13507adc3a2bSChandler Carruth       return false;
13517adc3a2bSChandler Carruth 
13527adc3a2bSChandler Carruth     // Analyze the cost of this block. If we blow through the threshold, this
13537adc3a2bSChandler Carruth     // returns false, and we can bail on out.
1354d295b00aSEaswaran Raman     if (!analyzeBlock(BB, EphValues))
13557adc3a2bSChandler Carruth       return false;
13567adc3a2bSChandler Carruth 
13577adc3a2bSChandler Carruth     TerminatorInst *TI = BB->getTerminator();
13587adc3a2bSChandler Carruth 
13597adc3a2bSChandler Carruth     // Add in the live successors by first checking whether we have terminator
13607adc3a2bSChandler Carruth     // that may be simplified based on the values simplified by this call.
13617adc3a2bSChandler Carruth     if (BranchInst *BI = dyn_cast<BranchInst>(TI)) {
13627adc3a2bSChandler Carruth       if (BI->isConditional()) {
13637adc3a2bSChandler Carruth         Value *Cond = BI->getCondition();
1364567556aaSChad Rosier         if (ConstantInt *SimpleCond =
1365567556aaSChad Rosier                 dyn_cast_or_null<ConstantInt>(SimplifiedValues.lookup(Cond))) {
13667adc3a2bSChandler Carruth           BBWorklist.insert(BI->getSuccessor(SimpleCond->isZero() ? 1 : 0));
13677adc3a2bSChandler Carruth           continue;
13687adc3a2bSChandler Carruth         }
13697adc3a2bSChandler Carruth       }
13707adc3a2bSChandler Carruth     } else if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) {
13717adc3a2bSChandler Carruth       Value *Cond = SI->getCondition();
1372567556aaSChad Rosier       if (ConstantInt *SimpleCond =
1373567556aaSChad Rosier               dyn_cast_or_null<ConstantInt>(SimplifiedValues.lookup(Cond))) {
13747adc3a2bSChandler Carruth         BBWorklist.insert(SI->findCaseValue(SimpleCond).getCaseSuccessor());
13757adc3a2bSChandler Carruth         continue;
13767adc3a2bSChandler Carruth       }
13777adc3a2bSChandler Carruth     }
13787adc3a2bSChandler Carruth 
13797adc3a2bSChandler Carruth     // If we're unable to select a particular successor, just count all of
13807adc3a2bSChandler Carruth     // them.
13817adc3a2bSChandler Carruth     for (unsigned TIdx = 0, TSize = TI->getNumSuccessors(); TIdx != TSize;
13827adc3a2bSChandler Carruth          ++TIdx)
13837adc3a2bSChandler Carruth       BBWorklist.insert(TI->getSuccessor(TIdx));
13847adc3a2bSChandler Carruth 
13857adc3a2bSChandler Carruth     // If we had any successors at this point, than post-inlining is likely to
13867adc3a2bSChandler Carruth     // have them as well. Note that we assume any basic blocks which existed
13877adc3a2bSChandler Carruth     // due to branches or switches which folded above will also fold after
13887adc3a2bSChandler Carruth     // inlining.
13897adc3a2bSChandler Carruth     if (SingleBB && TI->getNumSuccessors() > 1) {
13907adc3a2bSChandler Carruth       // Take off the bonus we applied to the threshold.
13917adc3a2bSChandler Carruth       Threshold -= SingleBBBonus;
13927adc3a2bSChandler Carruth       SingleBB = false;
13937adc3a2bSChandler Carruth     }
13947adc3a2bSChandler Carruth   }
13957adc3a2bSChandler Carruth 
13967adc3a2bSChandler Carruth   // If this is a noduplicate call, we can still inline as long as
13977adc3a2bSChandler Carruth   // inlining this would cause the removal of the caller (so the instruction
13987adc3a2bSChandler Carruth   // is not actually duplicated, just moved).
13997adc3a2bSChandler Carruth   if (!OnlyOneCallAndLocalLinkage && ContainsNoDuplicateCall)
14007adc3a2bSChandler Carruth     return false;
14017adc3a2bSChandler Carruth 
14027adc3a2bSChandler Carruth   // We applied the maximum possible vector bonus at the beginning. Now,
14037adc3a2bSChandler Carruth   // subtract the excess bonus, if any, from the Threshold before
14047adc3a2bSChandler Carruth   // comparing against Cost.
14057adc3a2bSChandler Carruth   if (NumVectorInstructions <= NumInstructions / 10)
14067adc3a2bSChandler Carruth     Threshold -= FiftyPercentVectorBonus;
14077adc3a2bSChandler Carruth   else if (NumVectorInstructions <= NumInstructions / 2)
14087adc3a2bSChandler Carruth     Threshold -= (FiftyPercentVectorBonus - TenPercentVectorBonus);
14097adc3a2bSChandler Carruth 
141000ab73dcSHans Wennborg   return Cost < std::max(1, Threshold);
14117adc3a2bSChandler Carruth }
14127adc3a2bSChandler Carruth 
14137adc3a2bSChandler Carruth #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
14147adc3a2bSChandler Carruth /// \brief Dump stats about this call's analysis.
1415eb2a2546SYaron Keren LLVM_DUMP_METHOD void CallAnalyzer::dump() {
14167adc3a2bSChandler Carruth #define DEBUG_PRINT_STAT(x) dbgs() << "      " #x ": " << x << "\n"
14177adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumConstantArgs);
14187adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumConstantOffsetPtrArgs);
14197adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumAllocaArgs);
14207adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumConstantPtrCmps);
14217adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumConstantPtrDiffs);
14227adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumInstructionsSimplified);
14237adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumInstructions);
14247adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(SROACostSavings);
14257adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(SROACostSavingsLost);
14267adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(ContainsNoDuplicateCall);
14277adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(Cost);
14287adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(Threshold);
14297adc3a2bSChandler Carruth #undef DEBUG_PRINT_STAT
14307adc3a2bSChandler Carruth }
14317adc3a2bSChandler Carruth #endif
14327adc3a2bSChandler Carruth 
14335af7ace4SAkira Hatanaka /// \brief Test that two functions either have or have not the given attribute
14345af7ace4SAkira Hatanaka ///        at the same time.
14355af7ace4SAkira Hatanaka template <typename AttrKind>
14365af7ace4SAkira Hatanaka static bool attributeMatches(Function *F1, Function *F2, AttrKind Attr) {
14375af7ace4SAkira Hatanaka   return F1->getFnAttribute(Attr) == F2->getFnAttribute(Attr);
14385af7ace4SAkira Hatanaka }
14395af7ace4SAkira Hatanaka 
14407adc3a2bSChandler Carruth /// \brief Test that there are no attribute conflicts between Caller and Callee
14417adc3a2bSChandler Carruth ///        that prevent inlining.
14427adc3a2bSChandler Carruth static bool functionsHaveCompatibleAttributes(Function *Caller,
14437adc3a2bSChandler Carruth                                               Function *Callee,
14447adc3a2bSChandler Carruth                                               TargetTransformInfo &TTI) {
14457adc3a2bSChandler Carruth   return TTI.areInlineCompatible(Caller, Callee) &&
14461cb242ebSAkira Hatanaka          AttributeFuncs::areInlineCompatible(*Caller, *Callee);
14477adc3a2bSChandler Carruth }
14487adc3a2bSChandler Carruth 
1449f4bb2f0dSEaswaran Raman InlineCost llvm::getInlineCost(CallSite CS, int DefaultThreshold,
1450b9f7120eSEaswaran Raman                                TargetTransformInfo &CalleeTTI,
1451b1bd398cSEaswaran Raman                                AssumptionCacheTracker *ACT) {
1452f4bb2f0dSEaswaran Raman   return getInlineCost(CS, CS.getCalledFunction(), DefaultThreshold, CalleeTTI,
1453b1bd398cSEaswaran Raman                        ACT);
1454b9f7120eSEaswaran Raman }
1455b9f7120eSEaswaran Raman 
1456f4bb2f0dSEaswaran Raman int llvm::computeThresholdFromOptLevels(unsigned OptLevel,
1457f4bb2f0dSEaswaran Raman                                         unsigned SizeOptLevel) {
1458f4bb2f0dSEaswaran Raman   if (OptLevel > 2)
1459f4bb2f0dSEaswaran Raman     return OptAggressiveThreshold;
1460f4bb2f0dSEaswaran Raman   if (SizeOptLevel == 1) // -Os
1461f4bb2f0dSEaswaran Raman     return OptSizeThreshold;
1462f4bb2f0dSEaswaran Raman   if (SizeOptLevel == 2) // -Oz
1463f4bb2f0dSEaswaran Raman     return OptMinSizeThreshold;
1464f4bb2f0dSEaswaran Raman   return DefaultInlineThreshold;
1465f4bb2f0dSEaswaran Raman }
1466f4bb2f0dSEaswaran Raman 
1467f4bb2f0dSEaswaran Raman int llvm::getDefaultInlineThreshold() { return DefaultInlineThreshold; }
1468f4bb2f0dSEaswaran Raman 
1469f4bb2f0dSEaswaran Raman InlineCost llvm::getInlineCost(CallSite CS, Function *Callee,
1470f4bb2f0dSEaswaran Raman                                int DefaultThreshold,
1471b9f7120eSEaswaran Raman                                TargetTransformInfo &CalleeTTI,
1472b1bd398cSEaswaran Raman                                AssumptionCacheTracker *ACT) {
1473f4bb2f0dSEaswaran Raman 
14747adc3a2bSChandler Carruth   // Cannot inline indirect calls.
14757adc3a2bSChandler Carruth   if (!Callee)
14767adc3a2bSChandler Carruth     return llvm::InlineCost::getNever();
14777adc3a2bSChandler Carruth 
14787adc3a2bSChandler Carruth   // Calls to functions with always-inline attributes should be inlined
14797adc3a2bSChandler Carruth   // whenever possible.
14807adc3a2bSChandler Carruth   if (CS.hasFnAttr(Attribute::AlwaysInline)) {
14817adc3a2bSChandler Carruth     if (isInlineViable(*Callee))
14827adc3a2bSChandler Carruth       return llvm::InlineCost::getAlways();
14837adc3a2bSChandler Carruth     return llvm::InlineCost::getNever();
14847adc3a2bSChandler Carruth   }
14857adc3a2bSChandler Carruth 
14867adc3a2bSChandler Carruth   // Never inline functions with conflicting attributes (unless callee has
14877adc3a2bSChandler Carruth   // always-inline attribute).
1488b9f7120eSEaswaran Raman   if (!functionsHaveCompatibleAttributes(CS.getCaller(), Callee, CalleeTTI))
14897adc3a2bSChandler Carruth     return llvm::InlineCost::getNever();
14907adc3a2bSChandler Carruth 
14917adc3a2bSChandler Carruth   // Don't inline this call if the caller has the optnone attribute.
14927adc3a2bSChandler Carruth   if (CS.getCaller()->hasFnAttribute(Attribute::OptimizeNone))
14937adc3a2bSChandler Carruth     return llvm::InlineCost::getNever();
14947adc3a2bSChandler Carruth 
14955ce32728SSanjoy Das   // Don't inline functions which can be interposed at link-time.  Don't inline
14965ce32728SSanjoy Das   // functions marked noinline or call sites marked noinline.
14975ce32728SSanjoy Das   // Note: inlining non-exact non-interposable fucntions is fine, since we know
14985ce32728SSanjoy Das   // we have *a* correct implementation of the source level function.
1499567556aaSChad Rosier   if (Callee->isInterposable() || Callee->hasFnAttribute(Attribute::NoInline) ||
1500567556aaSChad Rosier       CS.isNoInline())
15017adc3a2bSChandler Carruth     return llvm::InlineCost::getNever();
15027adc3a2bSChandler Carruth 
15037adc3a2bSChandler Carruth   DEBUG(llvm::dbgs() << "      Analyzing call of " << Callee->getName()
15047adc3a2bSChandler Carruth                      << "...\n");
15057adc3a2bSChandler Carruth 
1506b1bd398cSEaswaran Raman   CallAnalyzer CA(CalleeTTI, ACT, *Callee, DefaultThreshold, CS);
15077adc3a2bSChandler Carruth   bool ShouldInline = CA.analyzeCall(CS);
15087adc3a2bSChandler Carruth 
15097adc3a2bSChandler Carruth   DEBUG(CA.dump());
15107adc3a2bSChandler Carruth 
15117adc3a2bSChandler Carruth   // Check if there was a reason to force inlining or no inlining.
15127adc3a2bSChandler Carruth   if (!ShouldInline && CA.getCost() < CA.getThreshold())
15137adc3a2bSChandler Carruth     return InlineCost::getNever();
15147adc3a2bSChandler Carruth   if (ShouldInline && CA.getCost() >= CA.getThreshold())
15157adc3a2bSChandler Carruth     return InlineCost::getAlways();
15167adc3a2bSChandler Carruth 
15177adc3a2bSChandler Carruth   return llvm::InlineCost::get(CA.getCost(), CA.getThreshold());
15187adc3a2bSChandler Carruth }
15197adc3a2bSChandler Carruth 
1520b9f7120eSEaswaran Raman bool llvm::isInlineViable(Function &F) {
15217adc3a2bSChandler Carruth   bool ReturnsTwice = F.hasFnAttribute(Attribute::ReturnsTwice);
15227adc3a2bSChandler Carruth   for (Function::iterator BI = F.begin(), BE = F.end(); BI != BE; ++BI) {
15237adc3a2bSChandler Carruth     // Disallow inlining of functions which contain indirect branches or
15247adc3a2bSChandler Carruth     // blockaddresses.
15257adc3a2bSChandler Carruth     if (isa<IndirectBrInst>(BI->getTerminator()) || BI->hasAddressTaken())
15267adc3a2bSChandler Carruth       return false;
15277adc3a2bSChandler Carruth 
15285a82c916SDuncan P. N. Exon Smith     for (auto &II : *BI) {
15295a82c916SDuncan P. N. Exon Smith       CallSite CS(&II);
15307adc3a2bSChandler Carruth       if (!CS)
15317adc3a2bSChandler Carruth         continue;
15327adc3a2bSChandler Carruth 
15337adc3a2bSChandler Carruth       // Disallow recursive calls.
15347adc3a2bSChandler Carruth       if (&F == CS.getCalledFunction())
15357adc3a2bSChandler Carruth         return false;
15367adc3a2bSChandler Carruth 
15377adc3a2bSChandler Carruth       // Disallow calls which expose returns-twice to a function not previously
15387adc3a2bSChandler Carruth       // attributed as such.
15397adc3a2bSChandler Carruth       if (!ReturnsTwice && CS.isCall() &&
15407adc3a2bSChandler Carruth           cast<CallInst>(CS.getInstruction())->canReturnTwice())
15417adc3a2bSChandler Carruth         return false;
15427adc3a2bSChandler Carruth 
15437adc3a2bSChandler Carruth       // Disallow inlining functions that call @llvm.localescape. Doing this
15447adc3a2bSChandler Carruth       // correctly would require major changes to the inliner.
15457adc3a2bSChandler Carruth       if (CS.getCalledFunction() &&
15467adc3a2bSChandler Carruth           CS.getCalledFunction()->getIntrinsicID() ==
15477adc3a2bSChandler Carruth               llvm::Intrinsic::localescape)
15487adc3a2bSChandler Carruth         return false;
15497adc3a2bSChandler Carruth     }
15507adc3a2bSChandler Carruth   }
15517adc3a2bSChandler Carruth 
15527adc3a2bSChandler Carruth   return true;
15537adc3a2bSChandler Carruth }
1554