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. 165*567556aaSChad Rosier void visit(Module *); 166*567556aaSChad Rosier void visit(Module &); 167*567556aaSChad Rosier void visit(Function *); 168*567556aaSChad Rosier void visit(Function &); 169*567556aaSChad Rosier void visit(BasicBlock *); 170*567556aaSChad 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; 313*567556aaSChad Rosier if (OpC->isZero()) 314*567556aaSChad 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 3327adc3a2bSChandler Carruth // alloca, and handle that case. 3337adc3a2bSChandler Carruth if (I.isArrayAllocation()) { 3347adc3a2bSChandler Carruth if (Constant *Size = SimplifiedValues.lookup(I.getArraySize())) { 3357adc3a2bSChandler Carruth ConstantInt *AllocSize = dyn_cast<ConstantInt>(Size); 3367adc3a2bSChandler Carruth assert(AllocSize && "Allocation size not a constant int?"); 3377adc3a2bSChandler Carruth Type *Ty = I.getAllocatedType(); 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; 378*567556aaSChad Rosier bool SROACandidate = 379*567556aaSChad 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 435*567556aaSChad Rosier std::pair<Value *, APInt> BaseAndOffset = 436*567556aaSChad 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()) { 467*567556aaSChad Rosier std::pair<Value *, APInt> BaseAndOffset = 468*567556aaSChad 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)) 679*567556aaSChad Rosier if (Constant *C = 680*567556aaSChad 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) { 841*567556aaSChad Rosier SimplifiedValues[&I] = 842*567556aaSChad 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()); 867*567556aaSChad Rosier for (CallSite::arg_iterator I = CS.arg_begin(), E = CS.arg_end(); I != E; 868*567556aaSChad 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 } 892*567556aaSChad 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. 1259*567556aaSChad Rosier bool OnlyOneCallAndLocalLinkage = 1260*567556aaSChad 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; 1317*567556aaSChad Rosier CodeMetrics::collectEphemeralValues(&F, &ACT->getAssumptionCache(F), 1318*567556aaSChad 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>, 1328*567556aaSChad Rosier SmallPtrSet<BasicBlock *, 16>> 1329*567556aaSChad 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(); 1364*567556aaSChad Rosier if (ConstantInt *SimpleCond = 1365*567556aaSChad 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(); 1372*567556aaSChad Rosier if (ConstantInt *SimpleCond = 1373*567556aaSChad 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. 1499*567556aaSChad Rosier if (Callee->isInterposable() || Callee->hasFnAttribute(Attribute::NoInline) || 1500*567556aaSChad 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