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" 20aec2fa35SDaniel Jasper #include "llvm/Analysis/AssumptionCache.h" 2112585b01SEaswaran Raman #include "llvm/Analysis/BlockFrequencyInfo.h" 227adc3a2bSChandler Carruth #include "llvm/Analysis/CodeMetrics.h" 237adc3a2bSChandler Carruth #include "llvm/Analysis/ConstantFolding.h" 247adc3a2bSChandler Carruth #include "llvm/Analysis/InstructionSimplify.h" 2571069cf6SEaswaran Raman #include "llvm/Analysis/ProfileSummaryInfo.h" 267adc3a2bSChandler Carruth #include "llvm/Analysis/TargetTransformInfo.h" 277adc3a2bSChandler Carruth #include "llvm/IR/CallSite.h" 287adc3a2bSChandler Carruth #include "llvm/IR/CallingConv.h" 297adc3a2bSChandler Carruth #include "llvm/IR/DataLayout.h" 307adc3a2bSChandler Carruth #include "llvm/IR/GetElementPtrTypeIterator.h" 317adc3a2bSChandler Carruth #include "llvm/IR/GlobalAlias.h" 327adc3a2bSChandler Carruth #include "llvm/IR/InstVisitor.h" 337adc3a2bSChandler Carruth #include "llvm/IR/IntrinsicInst.h" 347adc3a2bSChandler Carruth #include "llvm/IR/Operator.h" 357adc3a2bSChandler Carruth #include "llvm/Support/Debug.h" 367adc3a2bSChandler Carruth #include "llvm/Support/raw_ostream.h" 377adc3a2bSChandler Carruth 387adc3a2bSChandler Carruth using namespace llvm; 397adc3a2bSChandler Carruth 407adc3a2bSChandler Carruth #define DEBUG_TYPE "inline-cost" 417adc3a2bSChandler Carruth 427adc3a2bSChandler Carruth STATISTIC(NumCallsAnalyzed, "Number of call sites analyzed"); 437adc3a2bSChandler Carruth 441c57cc2bSEaswaran Raman static cl::opt<int> InlineThreshold( 45f4bb2f0dSEaswaran Raman "inline-threshold", cl::Hidden, cl::init(225), cl::ZeroOrMore, 46f4bb2f0dSEaswaran Raman cl::desc("Control the amount of inlining to perform (default = 225)")); 47f4bb2f0dSEaswaran Raman 48f4bb2f0dSEaswaran Raman static cl::opt<int> HintThreshold( 49f4bb2f0dSEaswaran Raman "inlinehint-threshold", cl::Hidden, cl::init(325), 50f4bb2f0dSEaswaran Raman cl::desc("Threshold for inlining functions with inline hint")); 51f4bb2f0dSEaswaran Raman 5212585b01SEaswaran Raman static cl::opt<int> 5312585b01SEaswaran Raman ColdCallSiteThreshold("inline-cold-callsite-threshold", cl::Hidden, 5412585b01SEaswaran Raman cl::init(45), 5512585b01SEaswaran Raman cl::desc("Threshold for inlining cold callsites")); 5612585b01SEaswaran Raman 57f4bb2f0dSEaswaran Raman // We introduce this threshold to help performance of instrumentation based 58f4bb2f0dSEaswaran Raman // PGO before we actually hook up inliner with analysis passes such as BPI and 59f4bb2f0dSEaswaran Raman // BFI. 60f4bb2f0dSEaswaran Raman static cl::opt<int> ColdThreshold( 61c103ef89SEaswaran Raman "inlinecold-threshold", cl::Hidden, cl::init(45), 62f4bb2f0dSEaswaran Raman cl::desc("Threshold for inlining functions with cold attribute")); 63f4bb2f0dSEaswaran Raman 64de39cb93SDehao Chen static cl::opt<int> 65de39cb93SDehao Chen HotCallSiteThreshold("hot-callsite-threshold", cl::Hidden, cl::init(3000), 66de39cb93SDehao Chen cl::ZeroOrMore, 67de39cb93SDehao Chen cl::desc("Threshold for hot callsites ")); 68de39cb93SDehao Chen 69974d4eeaSEaswaran Raman static cl::opt<int> LocallyHotCallSiteThreshold( 70974d4eeaSEaswaran Raman "locally-hot-callsite-threshold", cl::Hidden, cl::init(525), cl::ZeroOrMore, 71974d4eeaSEaswaran Raman cl::desc("Threshold for locally hot callsites ")); 72974d4eeaSEaswaran Raman 73c5fa6358SEaswaran Raman static cl::opt<int> ColdCallSiteRelFreq( 74c5fa6358SEaswaran Raman "cold-callsite-rel-freq", cl::Hidden, cl::init(2), cl::ZeroOrMore, 75c5fa6358SEaswaran Raman cl::desc("Maxmimum block frequency, expressed as a percentage of caller's " 76c5fa6358SEaswaran Raman "entry frequency, for a callsite to be cold in the absence of " 77c5fa6358SEaswaran Raman "profile information.")); 78c5fa6358SEaswaran Raman 79974d4eeaSEaswaran Raman static cl::opt<int> HotCallSiteRelFreq( 80974d4eeaSEaswaran Raman "hot-callsite-rel-freq", cl::Hidden, cl::init(60), cl::ZeroOrMore, 81ff77cc75SEaswaran Raman cl::desc("Minimum block frequency, expressed as a multiple of caller's " 82974d4eeaSEaswaran Raman "entry frequency, for a callsite to be hot in the absence of " 83974d4eeaSEaswaran Raman "profile information.")); 84974d4eeaSEaswaran Raman 850812c5beSHaicheng Wu static cl::opt<bool> ComputeFullInlineCost( 860812c5beSHaicheng Wu "inline-cost-full", cl::Hidden, cl::init(false), 870812c5beSHaicheng Wu cl::desc("Compute the full inline cost of a call site even when the cost " 880812c5beSHaicheng Wu "exceeds the threshold.")); 890812c5beSHaicheng Wu 907adc3a2bSChandler Carruth namespace { 917adc3a2bSChandler Carruth 927adc3a2bSChandler Carruth class CallAnalyzer : public InstVisitor<CallAnalyzer, bool> { 937adc3a2bSChandler Carruth typedef InstVisitor<CallAnalyzer, bool> Base; 947adc3a2bSChandler Carruth friend class InstVisitor<CallAnalyzer, bool>; 957adc3a2bSChandler Carruth 967adc3a2bSChandler Carruth /// The TargetTransformInfo available for this compilation. 977adc3a2bSChandler Carruth const TargetTransformInfo &TTI; 987adc3a2bSChandler Carruth 99aec2fa35SDaniel Jasper /// Getter for the cache of @llvm.assume intrinsics. 100aec2fa35SDaniel Jasper std::function<AssumptionCache &(Function &)> &GetAssumptionCache; 101aec2fa35SDaniel Jasper 10212585b01SEaswaran Raman /// Getter for BlockFrequencyInfo 10312585b01SEaswaran Raman Optional<function_ref<BlockFrequencyInfo &(Function &)>> &GetBFI; 10412585b01SEaswaran Raman 10571069cf6SEaswaran Raman /// Profile summary information. 10671069cf6SEaswaran Raman ProfileSummaryInfo *PSI; 10771069cf6SEaswaran Raman 108f3d122cdSPiotr Padlewski /// The called function. 1097adc3a2bSChandler Carruth Function &F; 1107adc3a2bSChandler Carruth 11185be8ca8SEric Christopher // Cache the DataLayout since we use it a lot. 11285be8ca8SEric Christopher const DataLayout &DL; 11385be8ca8SEric Christopher 1140812c5beSHaicheng Wu /// The OptimizationRemarkEmitter available for this compilation. 1150812c5beSHaicheng Wu OptimizationRemarkEmitter *ORE; 1160812c5beSHaicheng Wu 117f3d122cdSPiotr Padlewski /// The candidate callsite being analyzed. Please do not use this to do 118f3d122cdSPiotr Padlewski /// analysis in the caller function; we want the inline cost query to be 119f3d122cdSPiotr Padlewski /// easily cacheable. Instead, use the cover function paramHasAttr. 1207adc3a2bSChandler Carruth CallSite CandidateCS; 1217adc3a2bSChandler Carruth 122f3d122cdSPiotr Padlewski /// Tunable parameters that control the analysis. 1231c57cc2bSEaswaran Raman const InlineParams &Params; 1241c57cc2bSEaswaran Raman 1257adc3a2bSChandler Carruth int Threshold; 1267adc3a2bSChandler Carruth int Cost; 1277adc3a2bSChandler Carruth 1287adc3a2bSChandler Carruth bool IsCallerRecursive; 1297adc3a2bSChandler Carruth bool IsRecursiveCall; 1307adc3a2bSChandler Carruth bool ExposesReturnsTwice; 1317adc3a2bSChandler Carruth bool HasDynamicAlloca; 1327adc3a2bSChandler Carruth bool ContainsNoDuplicateCall; 1337adc3a2bSChandler Carruth bool HasReturn; 1347adc3a2bSChandler Carruth bool HasIndirectBr; 1357adc3a2bSChandler Carruth bool HasFrameEscape; 1367adc3a2bSChandler Carruth 1377adc3a2bSChandler Carruth /// Number of bytes allocated statically by the callee. 1387adc3a2bSChandler Carruth uint64_t AllocatedSize; 1397adc3a2bSChandler Carruth unsigned NumInstructions, NumVectorInstructions; 14051b809bfSEaswaran Raman int VectorBonus, TenPercentVectorBonus; 14151b809bfSEaswaran Raman // Bonus to be applied when the callee has only one reachable basic block. 14251b809bfSEaswaran Raman int SingleBBBonus; 1437adc3a2bSChandler Carruth 144f3d122cdSPiotr Padlewski /// While we walk the potentially-inlined instructions, we build up and 145f3d122cdSPiotr Padlewski /// maintain a mapping of simplified values specific to this callsite. The 146f3d122cdSPiotr Padlewski /// idea is to propagate any special information we have about arguments to 147f3d122cdSPiotr Padlewski /// this call through the inlinable section of the function, and account for 148f3d122cdSPiotr Padlewski /// likely simplifications post-inlining. The most important aspect we track 149f3d122cdSPiotr Padlewski /// is CFG altering simplifications -- when we prove a basic block dead, that 150f3d122cdSPiotr Padlewski /// can cause dramatic shifts in the cost of inlining a function. 1517adc3a2bSChandler Carruth DenseMap<Value *, Constant *> SimplifiedValues; 1527adc3a2bSChandler Carruth 153f3d122cdSPiotr Padlewski /// Keep track of the values which map back (through function arguments) to 154f3d122cdSPiotr Padlewski /// allocas on the caller stack which could be simplified through SROA. 1557adc3a2bSChandler Carruth DenseMap<Value *, Value *> SROAArgValues; 1567adc3a2bSChandler Carruth 157f3d122cdSPiotr Padlewski /// The mapping of caller Alloca values to their accumulated cost savings. If 158f3d122cdSPiotr Padlewski /// we have to disable SROA for one of the allocas, this tells us how much 159f3d122cdSPiotr Padlewski /// cost must be added. 1607adc3a2bSChandler Carruth DenseMap<Value *, int> SROAArgCosts; 1617adc3a2bSChandler Carruth 162f3d122cdSPiotr Padlewski /// Keep track of values which map to a pointer base and constant offset. 1637adc3a2bSChandler Carruth DenseMap<Value *, std::pair<Value *, APInt>> ConstantOffsetPtrs; 1647adc3a2bSChandler Carruth 1657adc3a2bSChandler Carruth // Custom simplification helper routines. 1667adc3a2bSChandler Carruth bool isAllocaDerivedArg(Value *V); 1677adc3a2bSChandler Carruth bool lookupSROAArgAndCost(Value *V, Value *&Arg, 1687adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator &CostIt); 1697adc3a2bSChandler Carruth void disableSROA(DenseMap<Value *, int>::iterator CostIt); 1707adc3a2bSChandler Carruth void disableSROA(Value *V); 1717adc3a2bSChandler Carruth void accumulateSROACost(DenseMap<Value *, int>::iterator CostIt, 1727adc3a2bSChandler Carruth int InstructionCost); 173201b191bSHaicheng Wu bool isGEPFree(GetElementPtrInst &GEP); 1747adc3a2bSChandler Carruth bool accumulateGEPOffset(GEPOperator &GEP, APInt &Offset); 1757adc3a2bSChandler Carruth bool simplifyCallSite(Function *F, CallSite CS); 176617f6364SEaswaran Raman template <typename Callable> 177617f6364SEaswaran Raman bool simplifyInstruction(Instruction &I, Callable Evaluate); 1787adc3a2bSChandler Carruth ConstantInt *stripAndComputeInBoundsConstantOffsets(Value *&V); 1797adc3a2bSChandler Carruth 1807adc3a2bSChandler Carruth /// Return true if the given argument to the function being considered for 1817adc3a2bSChandler Carruth /// inlining has the given attribute set either at the call site or the 1827adc3a2bSChandler Carruth /// function declaration. Primarily used to inspect call site specific 1837adc3a2bSChandler Carruth /// attributes since these can be more precise than the ones on the callee 1847adc3a2bSChandler Carruth /// itself. 1857adc3a2bSChandler Carruth bool paramHasAttr(Argument *A, Attribute::AttrKind Attr); 1867adc3a2bSChandler Carruth 1877adc3a2bSChandler Carruth /// Return true if the given value is known non null within the callee if 1887adc3a2bSChandler Carruth /// inlined through this particular callsite. 1897adc3a2bSChandler Carruth bool isKnownNonNullInCallee(Value *V); 1907adc3a2bSChandler Carruth 191f4bb2f0dSEaswaran Raman /// Update Threshold based on callsite properties such as callee 192f4bb2f0dSEaswaran Raman /// attributes and callee hotness for PGO builds. The Callee is explicitly 193f4bb2f0dSEaswaran Raman /// passed to support analyzing indirect calls whose target is inferred by 194f4bb2f0dSEaswaran Raman /// analysis. 195f4bb2f0dSEaswaran Raman void updateThreshold(CallSite CS, Function &Callee); 196f4bb2f0dSEaswaran Raman 1979a3fc17aSEaswaran Raman /// Return true if size growth is allowed when inlining the callee at CS. 1989a3fc17aSEaswaran Raman bool allowSizeGrowth(CallSite CS); 1999a3fc17aSEaswaran Raman 200c5fa6358SEaswaran Raman /// Return true if \p CS is a cold callsite. 201c5fa6358SEaswaran Raman bool isColdCallSite(CallSite CS, BlockFrequencyInfo *CallerBFI); 202c5fa6358SEaswaran Raman 203974d4eeaSEaswaran Raman /// Return a higher threshold if \p CS is a hot callsite. 204974d4eeaSEaswaran Raman Optional<int> getHotCallSiteThreshold(CallSite CS, 205974d4eeaSEaswaran Raman BlockFrequencyInfo *CallerBFI); 206974d4eeaSEaswaran Raman 2077adc3a2bSChandler Carruth // Custom analysis routines. 2087adc3a2bSChandler Carruth bool analyzeBlock(BasicBlock *BB, SmallPtrSetImpl<const Value *> &EphValues); 2097adc3a2bSChandler Carruth 2107adc3a2bSChandler Carruth // Disable several entry points to the visitor so we don't accidentally use 2117adc3a2bSChandler Carruth // them by declaring but not defining them here. 212567556aaSChad Rosier void visit(Module *); 213567556aaSChad Rosier void visit(Module &); 214567556aaSChad Rosier void visit(Function *); 215567556aaSChad Rosier void visit(Function &); 216567556aaSChad Rosier void visit(BasicBlock *); 217567556aaSChad Rosier void visit(BasicBlock &); 2187adc3a2bSChandler Carruth 2197adc3a2bSChandler Carruth // Provide base case for our instruction visit. 2207adc3a2bSChandler Carruth bool visitInstruction(Instruction &I); 2217adc3a2bSChandler Carruth 2227adc3a2bSChandler Carruth // Our visit overrides. 2237adc3a2bSChandler Carruth bool visitAlloca(AllocaInst &I); 2247adc3a2bSChandler Carruth bool visitPHI(PHINode &I); 2257adc3a2bSChandler Carruth bool visitGetElementPtr(GetElementPtrInst &I); 2267adc3a2bSChandler Carruth bool visitBitCast(BitCastInst &I); 2277adc3a2bSChandler Carruth bool visitPtrToInt(PtrToIntInst &I); 2287adc3a2bSChandler Carruth bool visitIntToPtr(IntToPtrInst &I); 2297adc3a2bSChandler Carruth bool visitCastInst(CastInst &I); 2307adc3a2bSChandler Carruth bool visitUnaryInstruction(UnaryInstruction &I); 2317adc3a2bSChandler Carruth bool visitCmpInst(CmpInst &I); 2322e1c050eSChad Rosier bool visitAnd(BinaryOperator &I); 2332e1c050eSChad Rosier bool visitOr(BinaryOperator &I); 2347adc3a2bSChandler Carruth bool visitSub(BinaryOperator &I); 2357adc3a2bSChandler Carruth bool visitBinaryOperator(BinaryOperator &I); 2367adc3a2bSChandler Carruth bool visitLoad(LoadInst &I); 2377adc3a2bSChandler Carruth bool visitStore(StoreInst &I); 2387adc3a2bSChandler Carruth bool visitExtractValue(ExtractValueInst &I); 2397adc3a2bSChandler Carruth bool visitInsertValue(InsertValueInst &I); 2407adc3a2bSChandler Carruth bool visitCallSite(CallSite CS); 2417adc3a2bSChandler Carruth bool visitReturnInst(ReturnInst &RI); 2427adc3a2bSChandler Carruth bool visitBranchInst(BranchInst &BI); 2437adc3a2bSChandler Carruth bool visitSwitchInst(SwitchInst &SI); 2447adc3a2bSChandler Carruth bool visitIndirectBrInst(IndirectBrInst &IBI); 2457adc3a2bSChandler Carruth bool visitResumeInst(ResumeInst &RI); 2467adc3a2bSChandler Carruth bool visitCleanupReturnInst(CleanupReturnInst &RI); 2477adc3a2bSChandler Carruth bool visitCatchReturnInst(CatchReturnInst &RI); 2487adc3a2bSChandler Carruth bool visitUnreachableInst(UnreachableInst &I); 2497adc3a2bSChandler Carruth 2507adc3a2bSChandler Carruth public: 251ab6a6837SSean Silva CallAnalyzer(const TargetTransformInfo &TTI, 252aec2fa35SDaniel Jasper std::function<AssumptionCache &(Function &)> &GetAssumptionCache, 25312585b01SEaswaran Raman Optional<function_ref<BlockFrequencyInfo &(Function &)>> &GetBFI, 2540812c5beSHaicheng Wu ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE, 2550812c5beSHaicheng Wu Function &Callee, CallSite CSArg, const InlineParams &Params) 25612585b01SEaswaran Raman : TTI(TTI), GetAssumptionCache(GetAssumptionCache), GetBFI(GetBFI), 2570812c5beSHaicheng Wu PSI(PSI), F(Callee), DL(F.getParent()->getDataLayout()), ORE(ORE), 25885be8ca8SEric Christopher CandidateCS(CSArg), Params(Params), Threshold(Params.DefaultThreshold), 25985be8ca8SEric Christopher Cost(0), IsCallerRecursive(false), IsRecursiveCall(false), 26085be8ca8SEric Christopher ExposesReturnsTwice(false), HasDynamicAlloca(false), 26185be8ca8SEric Christopher ContainsNoDuplicateCall(false), HasReturn(false), HasIndirectBr(false), 26285be8ca8SEric Christopher HasFrameEscape(false), AllocatedSize(0), NumInstructions(0), 26351b809bfSEaswaran Raman NumVectorInstructions(0), VectorBonus(0), SingleBBBonus(0), 26451b809bfSEaswaran Raman NumConstantArgs(0), NumConstantOffsetPtrArgs(0), NumAllocaArgs(0), 26551b809bfSEaswaran Raman NumConstantPtrCmps(0), NumConstantPtrDiffs(0), 26651b809bfSEaswaran Raman NumInstructionsSimplified(0), SROACostSavings(0), 26751b809bfSEaswaran Raman SROACostSavingsLost(0) {} 2687adc3a2bSChandler Carruth 2697adc3a2bSChandler Carruth bool analyzeCall(CallSite CS); 2707adc3a2bSChandler Carruth 2717adc3a2bSChandler Carruth int getThreshold() { return Threshold; } 2727adc3a2bSChandler Carruth int getCost() { return Cost; } 2737adc3a2bSChandler Carruth 2747adc3a2bSChandler Carruth // Keep a bunch of stats about the cost savings found so we can print them 2757adc3a2bSChandler Carruth // out when debugging. 2767adc3a2bSChandler Carruth unsigned NumConstantArgs; 2777adc3a2bSChandler Carruth unsigned NumConstantOffsetPtrArgs; 2787adc3a2bSChandler Carruth unsigned NumAllocaArgs; 2797adc3a2bSChandler Carruth unsigned NumConstantPtrCmps; 2807adc3a2bSChandler Carruth unsigned NumConstantPtrDiffs; 2817adc3a2bSChandler Carruth unsigned NumInstructionsSimplified; 2827adc3a2bSChandler Carruth unsigned SROACostSavings; 2837adc3a2bSChandler Carruth unsigned SROACostSavingsLost; 2847adc3a2bSChandler Carruth 2857adc3a2bSChandler Carruth void dump(); 2867adc3a2bSChandler Carruth }; 2877adc3a2bSChandler Carruth 2887adc3a2bSChandler Carruth } // namespace 2897adc3a2bSChandler Carruth 2907adc3a2bSChandler Carruth /// \brief Test whether the given value is an Alloca-derived function argument. 2917adc3a2bSChandler Carruth bool CallAnalyzer::isAllocaDerivedArg(Value *V) { 2927adc3a2bSChandler Carruth return SROAArgValues.count(V); 2937adc3a2bSChandler Carruth } 2947adc3a2bSChandler Carruth 2957adc3a2bSChandler Carruth /// \brief Lookup the SROA-candidate argument and cost iterator which V maps to. 2967adc3a2bSChandler Carruth /// Returns false if V does not map to a SROA-candidate. 2977adc3a2bSChandler Carruth bool CallAnalyzer::lookupSROAArgAndCost( 2987adc3a2bSChandler Carruth Value *V, Value *&Arg, DenseMap<Value *, int>::iterator &CostIt) { 2997adc3a2bSChandler Carruth if (SROAArgValues.empty() || SROAArgCosts.empty()) 3007adc3a2bSChandler Carruth return false; 3017adc3a2bSChandler Carruth 3027adc3a2bSChandler Carruth DenseMap<Value *, Value *>::iterator ArgIt = SROAArgValues.find(V); 3037adc3a2bSChandler Carruth if (ArgIt == SROAArgValues.end()) 3047adc3a2bSChandler Carruth return false; 3057adc3a2bSChandler Carruth 3067adc3a2bSChandler Carruth Arg = ArgIt->second; 3077adc3a2bSChandler Carruth CostIt = SROAArgCosts.find(Arg); 3087adc3a2bSChandler Carruth return CostIt != SROAArgCosts.end(); 3097adc3a2bSChandler Carruth } 3107adc3a2bSChandler Carruth 3117adc3a2bSChandler Carruth /// \brief Disable SROA for the candidate marked by this cost iterator. 3127adc3a2bSChandler Carruth /// 3137adc3a2bSChandler Carruth /// This marks the candidate as no longer viable for SROA, and adds the cost 3147adc3a2bSChandler Carruth /// savings associated with it back into the inline cost measurement. 3157adc3a2bSChandler Carruth void CallAnalyzer::disableSROA(DenseMap<Value *, int>::iterator CostIt) { 3167adc3a2bSChandler Carruth // If we're no longer able to perform SROA we need to undo its cost savings 3177adc3a2bSChandler Carruth // and prevent subsequent analysis. 3187adc3a2bSChandler Carruth Cost += CostIt->second; 3197adc3a2bSChandler Carruth SROACostSavings -= CostIt->second; 3207adc3a2bSChandler Carruth SROACostSavingsLost += CostIt->second; 3217adc3a2bSChandler Carruth SROAArgCosts.erase(CostIt); 3227adc3a2bSChandler Carruth } 3237adc3a2bSChandler Carruth 3247adc3a2bSChandler Carruth /// \brief If 'V' maps to a SROA candidate, disable SROA for it. 3257adc3a2bSChandler Carruth void CallAnalyzer::disableSROA(Value *V) { 3267adc3a2bSChandler Carruth Value *SROAArg; 3277adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 3287adc3a2bSChandler Carruth if (lookupSROAArgAndCost(V, SROAArg, CostIt)) 3297adc3a2bSChandler Carruth disableSROA(CostIt); 3307adc3a2bSChandler Carruth } 3317adc3a2bSChandler Carruth 3327adc3a2bSChandler Carruth /// \brief Accumulate the given cost for a particular SROA candidate. 3337adc3a2bSChandler Carruth void CallAnalyzer::accumulateSROACost(DenseMap<Value *, int>::iterator CostIt, 3347adc3a2bSChandler Carruth int InstructionCost) { 3357adc3a2bSChandler Carruth CostIt->second += InstructionCost; 3367adc3a2bSChandler Carruth SROACostSavings += InstructionCost; 3377adc3a2bSChandler Carruth } 3387adc3a2bSChandler Carruth 3397adc3a2bSChandler Carruth /// \brief Accumulate a constant GEP offset into an APInt if possible. 3407adc3a2bSChandler Carruth /// 3417adc3a2bSChandler Carruth /// Returns false if unable to compute the offset for any reason. Respects any 3427adc3a2bSChandler Carruth /// simplified values known during the analysis of this callsite. 3437adc3a2bSChandler Carruth bool CallAnalyzer::accumulateGEPOffset(GEPOperator &GEP, APInt &Offset) { 3447adc3a2bSChandler Carruth unsigned IntPtrWidth = DL.getPointerSizeInBits(); 3457adc3a2bSChandler Carruth assert(IntPtrWidth == Offset.getBitWidth()); 3467adc3a2bSChandler Carruth 3477adc3a2bSChandler Carruth for (gep_type_iterator GTI = gep_type_begin(GEP), GTE = gep_type_end(GEP); 3487adc3a2bSChandler Carruth GTI != GTE; ++GTI) { 3497adc3a2bSChandler Carruth ConstantInt *OpC = dyn_cast<ConstantInt>(GTI.getOperand()); 3507adc3a2bSChandler Carruth if (!OpC) 3517adc3a2bSChandler Carruth if (Constant *SimpleOp = SimplifiedValues.lookup(GTI.getOperand())) 3527adc3a2bSChandler Carruth OpC = dyn_cast<ConstantInt>(SimpleOp); 3537adc3a2bSChandler Carruth if (!OpC) 3547adc3a2bSChandler Carruth return false; 355567556aaSChad Rosier if (OpC->isZero()) 356567556aaSChad Rosier continue; 3577adc3a2bSChandler Carruth 3587adc3a2bSChandler Carruth // Handle a struct index, which adds its field offset to the pointer. 359ab85225bSPeter Collingbourne if (StructType *STy = GTI.getStructTypeOrNull()) { 3607adc3a2bSChandler Carruth unsigned ElementIdx = OpC->getZExtValue(); 3617adc3a2bSChandler Carruth const StructLayout *SL = DL.getStructLayout(STy); 3627adc3a2bSChandler Carruth Offset += APInt(IntPtrWidth, SL->getElementOffset(ElementIdx)); 3637adc3a2bSChandler Carruth continue; 3647adc3a2bSChandler Carruth } 3657adc3a2bSChandler Carruth 3667adc3a2bSChandler Carruth APInt TypeSize(IntPtrWidth, DL.getTypeAllocSize(GTI.getIndexedType())); 3677adc3a2bSChandler Carruth Offset += OpC->getValue().sextOrTrunc(IntPtrWidth) * TypeSize; 3687adc3a2bSChandler Carruth } 3697adc3a2bSChandler Carruth return true; 3707adc3a2bSChandler Carruth } 3717adc3a2bSChandler Carruth 372201b191bSHaicheng Wu /// \brief Use TTI to check whether a GEP is free. 373201b191bSHaicheng Wu /// 374201b191bSHaicheng Wu /// Respects any simplified values known during the analysis of this callsite. 375201b191bSHaicheng Wu bool CallAnalyzer::isGEPFree(GetElementPtrInst &GEP) { 37661c1bd5aSEvgeny Astigeevich SmallVector<Value *, 4> Operands; 37761c1bd5aSEvgeny Astigeevich Operands.push_back(GEP.getOperand(0)); 378201b191bSHaicheng Wu for (User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end(); I != E; ++I) 379201b191bSHaicheng Wu if (Constant *SimpleOp = SimplifiedValues.lookup(*I)) 38061c1bd5aSEvgeny Astigeevich Operands.push_back(SimpleOp); 381201b191bSHaicheng Wu else 38261c1bd5aSEvgeny Astigeevich Operands.push_back(*I); 38361c1bd5aSEvgeny Astigeevich return TargetTransformInfo::TCC_Free == TTI.getUserCost(&GEP, Operands); 384201b191bSHaicheng Wu } 385201b191bSHaicheng Wu 3867adc3a2bSChandler Carruth bool CallAnalyzer::visitAlloca(AllocaInst &I) { 3877adc3a2bSChandler Carruth // Check whether inlining will turn a dynamic alloca into a static 3880f153424SSanjay Patel // alloca and handle that case. 3897adc3a2bSChandler Carruth if (I.isArrayAllocation()) { 3900f153424SSanjay Patel Constant *Size = SimplifiedValues.lookup(I.getArraySize()); 3910f153424SSanjay Patel if (auto *AllocSize = dyn_cast_or_null<ConstantInt>(Size)) { 3927adc3a2bSChandler Carruth Type *Ty = I.getAllocatedType(); 39322eb80a1SEaswaran Raman AllocatedSize = SaturatingMultiplyAdd( 39422eb80a1SEaswaran Raman AllocSize->getLimitedValue(), DL.getTypeAllocSize(Ty), AllocatedSize); 3957adc3a2bSChandler Carruth return Base::visitAlloca(I); 3967adc3a2bSChandler Carruth } 3977adc3a2bSChandler Carruth } 3987adc3a2bSChandler Carruth 3997adc3a2bSChandler Carruth // Accumulate the allocated size. 4007adc3a2bSChandler Carruth if (I.isStaticAlloca()) { 4017adc3a2bSChandler Carruth Type *Ty = I.getAllocatedType(); 40222eb80a1SEaswaran Raman AllocatedSize = SaturatingAdd(DL.getTypeAllocSize(Ty), AllocatedSize); 4037adc3a2bSChandler Carruth } 4047adc3a2bSChandler Carruth 4057adc3a2bSChandler Carruth // We will happily inline static alloca instructions. 4067adc3a2bSChandler Carruth if (I.isStaticAlloca()) 4077adc3a2bSChandler Carruth return Base::visitAlloca(I); 4087adc3a2bSChandler Carruth 4097adc3a2bSChandler Carruth // FIXME: This is overly conservative. Dynamic allocas are inefficient for 4107adc3a2bSChandler Carruth // a variety of reasons, and so we would like to not inline them into 4117adc3a2bSChandler Carruth // functions which don't currently have a dynamic alloca. This simply 4127adc3a2bSChandler Carruth // disables inlining altogether in the presence of a dynamic alloca. 4137adc3a2bSChandler Carruth HasDynamicAlloca = true; 4147adc3a2bSChandler Carruth return false; 4157adc3a2bSChandler Carruth } 4167adc3a2bSChandler Carruth 4177adc3a2bSChandler Carruth bool CallAnalyzer::visitPHI(PHINode &I) { 4187adc3a2bSChandler Carruth // FIXME: We should potentially be tracking values through phi nodes, 4197adc3a2bSChandler Carruth // especially when they collapse to a single value due to deleted CFG edges 4207adc3a2bSChandler Carruth // during inlining. 4217adc3a2bSChandler Carruth 4227adc3a2bSChandler Carruth // FIXME: We need to propagate SROA *disabling* through phi nodes, even 4237adc3a2bSChandler Carruth // though we don't want to propagate it's bonuses. The idea is to disable 4247adc3a2bSChandler Carruth // SROA if it *might* be used in an inappropriate manner. 4257adc3a2bSChandler Carruth 4267adc3a2bSChandler Carruth // Phi nodes are always zero-cost. 4277adc3a2bSChandler Carruth return true; 4287adc3a2bSChandler Carruth } 4297adc3a2bSChandler Carruth 4307adc3a2bSChandler Carruth bool CallAnalyzer::visitGetElementPtr(GetElementPtrInst &I) { 4317adc3a2bSChandler Carruth Value *SROAArg; 4327adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 433567556aaSChad Rosier bool SROACandidate = 434567556aaSChad Rosier lookupSROAArgAndCost(I.getPointerOperand(), SROAArg, CostIt); 4357adc3a2bSChandler Carruth 4367adc3a2bSChandler Carruth // Try to fold GEPs of constant-offset call site argument pointers. This 4377adc3a2bSChandler Carruth // requires target data and inbounds GEPs. 4387adc3a2bSChandler Carruth if (I.isInBounds()) { 4397adc3a2bSChandler Carruth // Check if we have a base + offset for the pointer. 4407adc3a2bSChandler Carruth Value *Ptr = I.getPointerOperand(); 4417adc3a2bSChandler Carruth std::pair<Value *, APInt> BaseAndOffset = ConstantOffsetPtrs.lookup(Ptr); 4427adc3a2bSChandler Carruth if (BaseAndOffset.first) { 4437adc3a2bSChandler Carruth // Check if the offset of this GEP is constant, and if so accumulate it 4447adc3a2bSChandler Carruth // into Offset. 4457adc3a2bSChandler Carruth if (!accumulateGEPOffset(cast<GEPOperator>(I), BaseAndOffset.second)) { 4467adc3a2bSChandler Carruth // Non-constant GEPs aren't folded, and disable SROA. 4477adc3a2bSChandler Carruth if (SROACandidate) 4487adc3a2bSChandler Carruth disableSROA(CostIt); 449201b191bSHaicheng Wu return isGEPFree(I); 4507adc3a2bSChandler Carruth } 4517adc3a2bSChandler Carruth 4527adc3a2bSChandler Carruth // Add the result as a new mapping to Base + Offset. 4537adc3a2bSChandler Carruth ConstantOffsetPtrs[&I] = BaseAndOffset; 4547adc3a2bSChandler Carruth 4557adc3a2bSChandler Carruth // Also handle SROA candidates here, we already know that the GEP is 4567adc3a2bSChandler Carruth // all-constant indexed. 4577adc3a2bSChandler Carruth if (SROACandidate) 4587adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 4597adc3a2bSChandler Carruth 4607adc3a2bSChandler Carruth return true; 4617adc3a2bSChandler Carruth } 4627adc3a2bSChandler Carruth } 4637adc3a2bSChandler Carruth 464a8b9cdc9SEaswaran Raman // Lambda to check whether a GEP's indices are all constant. 465a8b9cdc9SEaswaran Raman auto IsGEPOffsetConstant = [&](GetElementPtrInst &GEP) { 466a8b9cdc9SEaswaran Raman for (User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end(); I != E; ++I) 467a8b9cdc9SEaswaran Raman if (!isa<Constant>(*I) && !SimplifiedValues.lookup(*I)) 468a8b9cdc9SEaswaran Raman return false; 469a8b9cdc9SEaswaran Raman return true; 470a8b9cdc9SEaswaran Raman }; 471a8b9cdc9SEaswaran Raman 472a8b9cdc9SEaswaran Raman if (IsGEPOffsetConstant(I)) { 4737adc3a2bSChandler Carruth if (SROACandidate) 4747adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 4757adc3a2bSChandler Carruth 4767adc3a2bSChandler Carruth // Constant GEPs are modeled as free. 4777adc3a2bSChandler Carruth return true; 4787adc3a2bSChandler Carruth } 4797adc3a2bSChandler Carruth 4807adc3a2bSChandler Carruth // Variable GEPs will require math and will disable SROA. 4817adc3a2bSChandler Carruth if (SROACandidate) 4827adc3a2bSChandler Carruth disableSROA(CostIt); 483201b191bSHaicheng Wu return isGEPFree(I); 4847adc3a2bSChandler Carruth } 4857adc3a2bSChandler Carruth 486617f6364SEaswaran Raman /// Simplify \p I if its operands are constants and update SimplifiedValues. 487617f6364SEaswaran Raman /// \p Evaluate is a callable specific to instruction type that evaluates the 488617f6364SEaswaran Raman /// instruction when all the operands are constants. 489617f6364SEaswaran Raman template <typename Callable> 490617f6364SEaswaran Raman bool CallAnalyzer::simplifyInstruction(Instruction &I, Callable Evaluate) { 491617f6364SEaswaran Raman SmallVector<Constant *, 2> COps; 492617f6364SEaswaran Raman for (Value *Op : I.operands()) { 493617f6364SEaswaran Raman Constant *COp = dyn_cast<Constant>(Op); 4947adc3a2bSChandler Carruth if (!COp) 495617f6364SEaswaran Raman COp = SimplifiedValues.lookup(Op); 496617f6364SEaswaran Raman if (!COp) 497617f6364SEaswaran Raman return false; 498617f6364SEaswaran Raman COps.push_back(COp); 499617f6364SEaswaran Raman } 500617f6364SEaswaran Raman auto *C = Evaluate(COps); 501617f6364SEaswaran Raman if (!C) 502617f6364SEaswaran Raman return false; 5037adc3a2bSChandler Carruth SimplifiedValues[&I] = C; 5047adc3a2bSChandler Carruth return true; 5057adc3a2bSChandler Carruth } 5067adc3a2bSChandler Carruth 507617f6364SEaswaran Raman bool CallAnalyzer::visitBitCast(BitCastInst &I) { 508617f6364SEaswaran Raman // Propagate constants through bitcasts. 509617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 510617f6364SEaswaran Raman return ConstantExpr::getBitCast(COps[0], I.getType()); 511617f6364SEaswaran Raman })) 512617f6364SEaswaran Raman return true; 513617f6364SEaswaran Raman 5147adc3a2bSChandler Carruth // Track base/offsets through casts 515567556aaSChad Rosier std::pair<Value *, APInt> BaseAndOffset = 516567556aaSChad Rosier ConstantOffsetPtrs.lookup(I.getOperand(0)); 5177adc3a2bSChandler Carruth // Casts don't change the offset, just wrap it up. 5187adc3a2bSChandler Carruth if (BaseAndOffset.first) 5197adc3a2bSChandler Carruth ConstantOffsetPtrs[&I] = BaseAndOffset; 5207adc3a2bSChandler Carruth 5217adc3a2bSChandler Carruth // Also look for SROA candidates here. 5227adc3a2bSChandler Carruth Value *SROAArg; 5237adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 5247adc3a2bSChandler Carruth if (lookupSROAArgAndCost(I.getOperand(0), SROAArg, CostIt)) 5257adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 5267adc3a2bSChandler Carruth 5277adc3a2bSChandler Carruth // Bitcasts are always zero cost. 5287adc3a2bSChandler Carruth return true; 5297adc3a2bSChandler Carruth } 5307adc3a2bSChandler Carruth 5317adc3a2bSChandler Carruth bool CallAnalyzer::visitPtrToInt(PtrToIntInst &I) { 5327adc3a2bSChandler Carruth // Propagate constants through ptrtoint. 533617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 534617f6364SEaswaran Raman return ConstantExpr::getPtrToInt(COps[0], I.getType()); 535617f6364SEaswaran Raman })) 5367adc3a2bSChandler Carruth return true; 5377adc3a2bSChandler Carruth 5387adc3a2bSChandler Carruth // Track base/offset pairs when converted to a plain integer provided the 5397adc3a2bSChandler Carruth // integer is large enough to represent the pointer. 5407adc3a2bSChandler Carruth unsigned IntegerSize = I.getType()->getScalarSizeInBits(); 5417adc3a2bSChandler Carruth if (IntegerSize >= DL.getPointerSizeInBits()) { 542567556aaSChad Rosier std::pair<Value *, APInt> BaseAndOffset = 543567556aaSChad Rosier ConstantOffsetPtrs.lookup(I.getOperand(0)); 5447adc3a2bSChandler Carruth if (BaseAndOffset.first) 5457adc3a2bSChandler Carruth ConstantOffsetPtrs[&I] = BaseAndOffset; 5467adc3a2bSChandler Carruth } 5477adc3a2bSChandler Carruth 5487adc3a2bSChandler Carruth // This is really weird. Technically, ptrtoint will disable SROA. However, 5497adc3a2bSChandler Carruth // unless that ptrtoint is *used* somewhere in the live basic blocks after 5507adc3a2bSChandler Carruth // inlining, it will be nuked, and SROA should proceed. All of the uses which 5517adc3a2bSChandler Carruth // would block SROA would also block SROA if applied directly to a pointer, 5527adc3a2bSChandler Carruth // and so we can just add the integer in here. The only places where SROA is 5537adc3a2bSChandler Carruth // preserved either cannot fire on an integer, or won't in-and-of themselves 5547adc3a2bSChandler Carruth // disable SROA (ext) w/o some later use that we would see and disable. 5557adc3a2bSChandler Carruth Value *SROAArg; 5567adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 5577adc3a2bSChandler Carruth if (lookupSROAArgAndCost(I.getOperand(0), SROAArg, CostIt)) 5587adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 5597adc3a2bSChandler Carruth 5607adc3a2bSChandler Carruth return TargetTransformInfo::TCC_Free == TTI.getUserCost(&I); 5617adc3a2bSChandler Carruth } 5627adc3a2bSChandler Carruth 5637adc3a2bSChandler Carruth bool CallAnalyzer::visitIntToPtr(IntToPtrInst &I) { 5647adc3a2bSChandler Carruth // Propagate constants through ptrtoint. 565617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 566617f6364SEaswaran Raman return ConstantExpr::getIntToPtr(COps[0], I.getType()); 567617f6364SEaswaran Raman })) 5687adc3a2bSChandler Carruth return true; 5697adc3a2bSChandler Carruth 5707adc3a2bSChandler Carruth // Track base/offset pairs when round-tripped through a pointer without 5717adc3a2bSChandler Carruth // modifications provided the integer is not too large. 5727adc3a2bSChandler Carruth Value *Op = I.getOperand(0); 5737adc3a2bSChandler Carruth unsigned IntegerSize = Op->getType()->getScalarSizeInBits(); 5747adc3a2bSChandler Carruth if (IntegerSize <= DL.getPointerSizeInBits()) { 5757adc3a2bSChandler Carruth std::pair<Value *, APInt> BaseAndOffset = ConstantOffsetPtrs.lookup(Op); 5767adc3a2bSChandler Carruth if (BaseAndOffset.first) 5777adc3a2bSChandler Carruth ConstantOffsetPtrs[&I] = BaseAndOffset; 5787adc3a2bSChandler Carruth } 5797adc3a2bSChandler Carruth 5807adc3a2bSChandler Carruth // "Propagate" SROA here in the same manner as we do for ptrtoint above. 5817adc3a2bSChandler Carruth Value *SROAArg; 5827adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 5837adc3a2bSChandler Carruth if (lookupSROAArgAndCost(Op, SROAArg, CostIt)) 5847adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 5857adc3a2bSChandler Carruth 5867adc3a2bSChandler Carruth return TargetTransformInfo::TCC_Free == TTI.getUserCost(&I); 5877adc3a2bSChandler Carruth } 5887adc3a2bSChandler Carruth 5897adc3a2bSChandler Carruth bool CallAnalyzer::visitCastInst(CastInst &I) { 5907adc3a2bSChandler Carruth // Propagate constants through ptrtoint. 591617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 592617f6364SEaswaran Raman return ConstantExpr::getCast(I.getOpcode(), COps[0], I.getType()); 593617f6364SEaswaran Raman })) 5947adc3a2bSChandler Carruth return true; 5957adc3a2bSChandler Carruth 5967adc3a2bSChandler Carruth // Disable SROA in the face of arbitrary casts we don't whitelist elsewhere. 5977adc3a2bSChandler Carruth disableSROA(I.getOperand(0)); 5987adc3a2bSChandler Carruth 5997adc3a2bSChandler Carruth return TargetTransformInfo::TCC_Free == TTI.getUserCost(&I); 6007adc3a2bSChandler Carruth } 6017adc3a2bSChandler Carruth 6027adc3a2bSChandler Carruth bool CallAnalyzer::visitUnaryInstruction(UnaryInstruction &I) { 6037adc3a2bSChandler Carruth Value *Operand = I.getOperand(0); 604617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 605617f6364SEaswaran Raman return ConstantFoldInstOperands(&I, COps[0], DL); 606617f6364SEaswaran Raman })) 6077adc3a2bSChandler Carruth return true; 6087adc3a2bSChandler Carruth 6097adc3a2bSChandler Carruth // Disable any SROA on the argument to arbitrary unary operators. 6107adc3a2bSChandler Carruth disableSROA(Operand); 6117adc3a2bSChandler Carruth 6127adc3a2bSChandler Carruth return false; 6137adc3a2bSChandler Carruth } 6147adc3a2bSChandler Carruth 6157adc3a2bSChandler Carruth bool CallAnalyzer::paramHasAttr(Argument *A, Attribute::AttrKind Attr) { 616fb502d2fSReid Kleckner return CandidateCS.paramHasAttr(A->getArgNo(), Attr); 6177adc3a2bSChandler Carruth } 6187adc3a2bSChandler Carruth 6197adc3a2bSChandler Carruth bool CallAnalyzer::isKnownNonNullInCallee(Value *V) { 6207adc3a2bSChandler Carruth // Does the *call site* have the NonNull attribute set on an argument? We 6217adc3a2bSChandler Carruth // use the attribute on the call site to memoize any analysis done in the 6227adc3a2bSChandler Carruth // caller. This will also trip if the callee function has a non-null 6237adc3a2bSChandler Carruth // parameter attribute, but that's a less interesting case because hopefully 6247adc3a2bSChandler Carruth // the callee would already have been simplified based on that. 6257adc3a2bSChandler Carruth if (Argument *A = dyn_cast<Argument>(V)) 6267adc3a2bSChandler Carruth if (paramHasAttr(A, Attribute::NonNull)) 6277adc3a2bSChandler Carruth return true; 6287adc3a2bSChandler Carruth 6297adc3a2bSChandler Carruth // Is this an alloca in the caller? This is distinct from the attribute case 6307adc3a2bSChandler Carruth // above because attributes aren't updated within the inliner itself and we 6317adc3a2bSChandler Carruth // always want to catch the alloca derived case. 6327adc3a2bSChandler Carruth if (isAllocaDerivedArg(V)) 6337adc3a2bSChandler Carruth // We can actually predict the result of comparisons between an 6347adc3a2bSChandler Carruth // alloca-derived value and null. Note that this fires regardless of 6357adc3a2bSChandler Carruth // SROA firing. 6367adc3a2bSChandler Carruth return true; 6377adc3a2bSChandler Carruth 6387adc3a2bSChandler Carruth return false; 6397adc3a2bSChandler Carruth } 6407adc3a2bSChandler Carruth 6419a3fc17aSEaswaran Raman bool CallAnalyzer::allowSizeGrowth(CallSite CS) { 6429a3fc17aSEaswaran Raman // If the normal destination of the invoke or the parent block of the call 6439a3fc17aSEaswaran Raman // site is unreachable-terminated, there is little point in inlining this 6449a3fc17aSEaswaran Raman // unless there is literally zero cost. 6459a3fc17aSEaswaran Raman // FIXME: Note that it is possible that an unreachable-terminated block has a 6469a3fc17aSEaswaran Raman // hot entry. For example, in below scenario inlining hot_call_X() may be 6479a3fc17aSEaswaran Raman // beneficial : 6489a3fc17aSEaswaran Raman // main() { 6499a3fc17aSEaswaran Raman // hot_call_1(); 6509a3fc17aSEaswaran Raman // ... 6519a3fc17aSEaswaran Raman // hot_call_N() 6529a3fc17aSEaswaran Raman // exit(0); 6539a3fc17aSEaswaran Raman // } 6549a3fc17aSEaswaran Raman // For now, we are not handling this corner case here as it is rare in real 6559a3fc17aSEaswaran Raman // code. In future, we should elaborate this based on BPI and BFI in more 6569a3fc17aSEaswaran Raman // general threshold adjusting heuristics in updateThreshold(). 6579a3fc17aSEaswaran Raman Instruction *Instr = CS.getInstruction(); 6589a3fc17aSEaswaran Raman if (InvokeInst *II = dyn_cast<InvokeInst>(Instr)) { 6599a3fc17aSEaswaran Raman if (isa<UnreachableInst>(II->getNormalDest()->getTerminator())) 6609a3fc17aSEaswaran Raman return false; 6619a3fc17aSEaswaran Raman } else if (isa<UnreachableInst>(Instr->getParent()->getTerminator())) 6629a3fc17aSEaswaran Raman return false; 6639a3fc17aSEaswaran Raman 6649a3fc17aSEaswaran Raman return true; 6659a3fc17aSEaswaran Raman } 6669a3fc17aSEaswaran Raman 667c5fa6358SEaswaran Raman bool CallAnalyzer::isColdCallSite(CallSite CS, BlockFrequencyInfo *CallerBFI) { 668c5fa6358SEaswaran Raman // If global profile summary is available, then callsite's coldness is 669c5fa6358SEaswaran Raman // determined based on that. 670bba762a1SChandler Carruth if (PSI && PSI->hasProfileSummary()) 671c5fa6358SEaswaran Raman return PSI->isColdCallSite(CS, CallerBFI); 672bba762a1SChandler Carruth 673bba762a1SChandler Carruth // Otherwise we need BFI to be available. 674c5fa6358SEaswaran Raman if (!CallerBFI) 675c5fa6358SEaswaran Raman return false; 676c5fa6358SEaswaran Raman 677bba762a1SChandler Carruth // Determine if the callsite is cold relative to caller's entry. We could 678bba762a1SChandler Carruth // potentially cache the computation of scaled entry frequency, but the added 679bba762a1SChandler Carruth // complexity is not worth it unless this scaling shows up high in the 680bba762a1SChandler Carruth // profiles. 681c5fa6358SEaswaran Raman const BranchProbability ColdProb(ColdCallSiteRelFreq, 100); 682c5fa6358SEaswaran Raman auto CallSiteBB = CS.getInstruction()->getParent(); 683c5fa6358SEaswaran Raman auto CallSiteFreq = CallerBFI->getBlockFreq(CallSiteBB); 684c5fa6358SEaswaran Raman auto CallerEntryFreq = 685c5fa6358SEaswaran Raman CallerBFI->getBlockFreq(&(CS.getCaller()->getEntryBlock())); 686c5fa6358SEaswaran Raman return CallSiteFreq < CallerEntryFreq * ColdProb; 687c5fa6358SEaswaran Raman } 688c5fa6358SEaswaran Raman 689974d4eeaSEaswaran Raman Optional<int> 690974d4eeaSEaswaran Raman CallAnalyzer::getHotCallSiteThreshold(CallSite CS, 691974d4eeaSEaswaran Raman BlockFrequencyInfo *CallerBFI) { 692bba762a1SChandler Carruth 693974d4eeaSEaswaran Raman // If global profile summary is available, then callsite's hotness is 694974d4eeaSEaswaran Raman // determined based on that. 695bba762a1SChandler Carruth if (PSI && PSI->hasProfileSummary() && PSI->isHotCallSite(CS, CallerBFI)) 696bba762a1SChandler Carruth return Params.HotCallSiteThreshold; 697974d4eeaSEaswaran Raman 698bba762a1SChandler Carruth // Otherwise we need BFI to be available and to have a locally hot callsite 699bba762a1SChandler Carruth // threshold. 700bba762a1SChandler Carruth if (!CallerBFI || !Params.LocallyHotCallSiteThreshold) 701974d4eeaSEaswaran Raman return None; 702974d4eeaSEaswaran Raman 703bba762a1SChandler Carruth // Determine if the callsite is hot relative to caller's entry. We could 704bba762a1SChandler Carruth // potentially cache the computation of scaled entry frequency, but the added 705bba762a1SChandler Carruth // complexity is not worth it unless this scaling shows up high in the 706bba762a1SChandler Carruth // profiles. 707974d4eeaSEaswaran Raman auto CallSiteBB = CS.getInstruction()->getParent(); 708974d4eeaSEaswaran Raman auto CallSiteFreq = CallerBFI->getBlockFreq(CallSiteBB).getFrequency(); 709974d4eeaSEaswaran Raman auto CallerEntryFreq = CallerBFI->getEntryFreq(); 710974d4eeaSEaswaran Raman if (CallSiteFreq >= CallerEntryFreq * HotCallSiteRelFreq) 711bba762a1SChandler Carruth return Params.LocallyHotCallSiteThreshold; 712bba762a1SChandler Carruth 713bba762a1SChandler Carruth // Otherwise treat it normally. 714974d4eeaSEaswaran Raman return None; 715974d4eeaSEaswaran Raman } 716974d4eeaSEaswaran Raman 717f4bb2f0dSEaswaran Raman void CallAnalyzer::updateThreshold(CallSite CS, Function &Callee) { 7189a3fc17aSEaswaran Raman // If no size growth is allowed for this inlining, set Threshold to 0. 7199a3fc17aSEaswaran Raman if (!allowSizeGrowth(CS)) { 7209a3fc17aSEaswaran Raman Threshold = 0; 7219a3fc17aSEaswaran Raman return; 7229a3fc17aSEaswaran Raman } 7239a3fc17aSEaswaran Raman 724bb578ef0SEaswaran Raman Function *Caller = CS.getCaller(); 7251c57cc2bSEaswaran Raman 7261c57cc2bSEaswaran Raman // return min(A, B) if B is valid. 7271c57cc2bSEaswaran Raman auto MinIfValid = [](int A, Optional<int> B) { 7281c57cc2bSEaswaran Raman return B ? std::min(A, B.getValue()) : A; 7291c57cc2bSEaswaran Raman }; 7301c57cc2bSEaswaran Raman 7310d58fcacSEaswaran Raman // return max(A, B) if B is valid. 7320d58fcacSEaswaran Raman auto MaxIfValid = [](int A, Optional<int> B) { 7330d58fcacSEaswaran Raman return B ? std::max(A, B.getValue()) : A; 7340d58fcacSEaswaran Raman }; 7350d58fcacSEaswaran Raman 73651b809bfSEaswaran Raman // Various bonus percentages. These are multiplied by Threshold to get the 73751b809bfSEaswaran Raman // bonus values. 73851b809bfSEaswaran Raman // SingleBBBonus: This bonus is applied if the callee has a single reachable 73951b809bfSEaswaran Raman // basic block at the given callsite context. This is speculatively applied 74051b809bfSEaswaran Raman // and withdrawn if more than one basic block is seen. 74151b809bfSEaswaran Raman // 74251b809bfSEaswaran Raman // Vector bonuses: We want to more aggressively inline vector-dense kernels 74351b809bfSEaswaran Raman // and apply this bonus based on the percentage of vector instructions. A 74451b809bfSEaswaran Raman // bonus is applied if the vector instructions exceed 50% and half that amount 74551b809bfSEaswaran Raman // is applied if it exceeds 10%. Note that these bonuses are some what 74651b809bfSEaswaran Raman // arbitrary and evolved over time by accident as much as because they are 74751b809bfSEaswaran Raman // principled bonuses. 74851b809bfSEaswaran Raman // FIXME: It would be nice to base the bonus values on something more 74951b809bfSEaswaran Raman // scientific. 75051b809bfSEaswaran Raman // 75151b809bfSEaswaran Raman // LstCallToStaticBonus: This large bonus is applied to ensure the inlining 75251b809bfSEaswaran Raman // of the last call to a static function as inlining such functions is 75351b809bfSEaswaran Raman // guaranteed to reduce code size. 75451b809bfSEaswaran Raman // 75551b809bfSEaswaran Raman // These bonus percentages may be set to 0 based on properties of the caller 75651b809bfSEaswaran Raman // and the callsite. 75751b809bfSEaswaran Raman int SingleBBBonusPercent = 50; 75851b809bfSEaswaran Raman int VectorBonusPercent = 150; 75951b809bfSEaswaran Raman int LastCallToStaticBonus = InlineConstants::LastCallToStaticBonus; 76051b809bfSEaswaran Raman 76151b809bfSEaswaran Raman // Lambda to set all the above bonus and bonus percentages to 0. 76251b809bfSEaswaran Raman auto DisallowAllBonuses = [&]() { 76351b809bfSEaswaran Raman SingleBBBonusPercent = 0; 76451b809bfSEaswaran Raman VectorBonusPercent = 0; 76551b809bfSEaswaran Raman LastCallToStaticBonus = 0; 76651b809bfSEaswaran Raman }; 76751b809bfSEaswaran Raman 7681c57cc2bSEaswaran Raman // Use the OptMinSizeThreshold or OptSizeThreshold knob if they are available 7691c57cc2bSEaswaran Raman // and reduce the threshold if the caller has the necessary attribute. 77051b809bfSEaswaran Raman if (Caller->optForMinSize()) { 7711c57cc2bSEaswaran Raman Threshold = MinIfValid(Threshold, Params.OptMinSizeThreshold); 77251b809bfSEaswaran Raman // For minsize, we want to disable the single BB bonus and the vector 77351b809bfSEaswaran Raman // bonuses, but not the last-call-to-static bonus. Inlining the last call to 77451b809bfSEaswaran Raman // a static function will, at the minimum, eliminate the parameter setup and 77551b809bfSEaswaran Raman // call/return instructions. 77651b809bfSEaswaran Raman SingleBBBonusPercent = 0; 77751b809bfSEaswaran Raman VectorBonusPercent = 0; 77851b809bfSEaswaran Raman } else if (Caller->optForSize()) 7791c57cc2bSEaswaran Raman Threshold = MinIfValid(Threshold, Params.OptSizeThreshold); 780f4bb2f0dSEaswaran Raman 781e08b139dSEaswaran Raman // Adjust the threshold based on inlinehint attribute and profile based 782e08b139dSEaswaran Raman // hotness information if the caller does not have MinSize attribute. 783e08b139dSEaswaran Raman if (!Caller->optForMinSize()) { 784e08b139dSEaswaran Raman if (Callee.hasFnAttribute(Attribute::InlineHint)) 7850d58fcacSEaswaran Raman Threshold = MaxIfValid(Threshold, Params.HintThreshold); 786bba762a1SChandler Carruth 7873cd1479cSEaswaran Raman // FIXME: After switching to the new passmanager, simplify the logic below 788bba762a1SChandler Carruth // by checking only the callsite hotness/coldness as we will reliably 789bba762a1SChandler Carruth // have local profile information. 7903cd1479cSEaswaran Raman // 791bba762a1SChandler Carruth // Callsite hotness and coldness can be determined if sample profile is 792bba762a1SChandler Carruth // used (which adds hotness metadata to calls) or if caller's 793bba762a1SChandler Carruth // BlockFrequencyInfo is available. 794bba762a1SChandler Carruth BlockFrequencyInfo *CallerBFI = GetBFI ? &((*GetBFI)(*Caller)) : nullptr; 795974d4eeaSEaswaran Raman auto HotCallSiteThreshold = getHotCallSiteThreshold(CS, CallerBFI); 796974d4eeaSEaswaran Raman if (!Caller->optForSize() && HotCallSiteThreshold) { 79712585b01SEaswaran Raman DEBUG(dbgs() << "Hot callsite.\n"); 798974d4eeaSEaswaran Raman // FIXME: This should update the threshold only if it exceeds the 799974d4eeaSEaswaran Raman // current threshold, but AutoFDO + ThinLTO currently relies on this 800974d4eeaSEaswaran Raman // behavior to prevent inlining of hot callsites during ThinLTO 801974d4eeaSEaswaran Raman // compile phase. 802974d4eeaSEaswaran Raman Threshold = HotCallSiteThreshold.getValue(); 803c5fa6358SEaswaran Raman } else if (isColdCallSite(CS, CallerBFI)) { 8043cd1479cSEaswaran Raman DEBUG(dbgs() << "Cold callsite.\n"); 80551b809bfSEaswaran Raman // Do not apply bonuses for a cold callsite including the 80651b809bfSEaswaran Raman // LastCallToStatic bonus. While this bonus might result in code size 80751b809bfSEaswaran Raman // reduction, it can cause the size of a non-cold caller to increase 80851b809bfSEaswaran Raman // preventing it from being inlined. 80951b809bfSEaswaran Raman DisallowAllBonuses(); 8103cd1479cSEaswaran Raman Threshold = MinIfValid(Threshold, Params.ColdCallSiteThreshold); 811bba762a1SChandler Carruth } else if (PSI) { 812bba762a1SChandler Carruth // Use callee's global profile information only if we have no way of 813bba762a1SChandler Carruth // determining this via callsite information. 8143cd1479cSEaswaran Raman if (PSI->isFunctionEntryHot(&Callee)) { 81512585b01SEaswaran Raman DEBUG(dbgs() << "Hot callee.\n"); 816e08b139dSEaswaran Raman // If callsite hotness can not be determined, we may still know 817e08b139dSEaswaran Raman // that the callee is hot and treat it as a weaker hint for threshold 818e08b139dSEaswaran Raman // increase. 819e08b139dSEaswaran Raman Threshold = MaxIfValid(Threshold, Params.HintThreshold); 820e08b139dSEaswaran Raman } else if (PSI->isFunctionEntryCold(&Callee)) { 82112585b01SEaswaran Raman DEBUG(dbgs() << "Cold callee.\n"); 82251b809bfSEaswaran Raman // Do not apply bonuses for a cold callee including the 82351b809bfSEaswaran Raman // LastCallToStatic bonus. While this bonus might result in code size 82451b809bfSEaswaran Raman // reduction, it can cause the size of a non-cold caller to increase 82551b809bfSEaswaran Raman // preventing it from being inlined. 82651b809bfSEaswaran Raman DisallowAllBonuses(); 8271c57cc2bSEaswaran Raman Threshold = MinIfValid(Threshold, Params.ColdThreshold); 828e08b139dSEaswaran Raman } 829e08b139dSEaswaran Raman } 830e08b139dSEaswaran Raman } 8318650a4daSJustin Lebar 8328650a4daSJustin Lebar // Finally, take the target-specific inlining threshold multiplier into 8338650a4daSJustin Lebar // account. 8348650a4daSJustin Lebar Threshold *= TTI.getInliningThresholdMultiplier(); 83551b809bfSEaswaran Raman 83651b809bfSEaswaran Raman SingleBBBonus = Threshold * SingleBBBonusPercent / 100; 83751b809bfSEaswaran Raman VectorBonus = Threshold * VectorBonusPercent / 100; 83851b809bfSEaswaran Raman 83951b809bfSEaswaran Raman bool OnlyOneCallAndLocalLinkage = 84051b809bfSEaswaran Raman F.hasLocalLinkage() && F.hasOneUse() && &F == CS.getCalledFunction(); 84151b809bfSEaswaran Raman // If there is only one call of the function, and it has internal linkage, 84251b809bfSEaswaran Raman // the cost of inlining it drops dramatically. It may seem odd to update 84351b809bfSEaswaran Raman // Cost in updateThreshold, but the bonus depends on the logic in this method. 84451b809bfSEaswaran Raman if (OnlyOneCallAndLocalLinkage) 84551b809bfSEaswaran Raman Cost -= LastCallToStaticBonus; 846f4bb2f0dSEaswaran Raman } 847f4bb2f0dSEaswaran Raman 8487adc3a2bSChandler Carruth bool CallAnalyzer::visitCmpInst(CmpInst &I) { 8497adc3a2bSChandler Carruth Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); 8507adc3a2bSChandler Carruth // First try to handle simplified comparisons. 851617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 852617f6364SEaswaran Raman return ConstantExpr::getCompare(I.getPredicate(), COps[0], COps[1]); 853617f6364SEaswaran Raman })) 8547adc3a2bSChandler Carruth return true; 8557adc3a2bSChandler Carruth 8567adc3a2bSChandler Carruth if (I.getOpcode() == Instruction::FCmp) 8577adc3a2bSChandler Carruth return false; 8587adc3a2bSChandler Carruth 8597adc3a2bSChandler Carruth // Otherwise look for a comparison between constant offset pointers with 8607adc3a2bSChandler Carruth // a common base. 8617adc3a2bSChandler Carruth Value *LHSBase, *RHSBase; 8627adc3a2bSChandler Carruth APInt LHSOffset, RHSOffset; 8637adc3a2bSChandler Carruth std::tie(LHSBase, LHSOffset) = ConstantOffsetPtrs.lookup(LHS); 8647adc3a2bSChandler Carruth if (LHSBase) { 8657adc3a2bSChandler Carruth std::tie(RHSBase, RHSOffset) = ConstantOffsetPtrs.lookup(RHS); 8667adc3a2bSChandler Carruth if (RHSBase && LHSBase == RHSBase) { 8677adc3a2bSChandler Carruth // We have common bases, fold the icmp to a constant based on the 8687adc3a2bSChandler Carruth // offsets. 8697adc3a2bSChandler Carruth Constant *CLHS = ConstantInt::get(LHS->getContext(), LHSOffset); 8707adc3a2bSChandler Carruth Constant *CRHS = ConstantInt::get(RHS->getContext(), RHSOffset); 8717adc3a2bSChandler Carruth if (Constant *C = ConstantExpr::getICmp(I.getPredicate(), CLHS, CRHS)) { 8727adc3a2bSChandler Carruth SimplifiedValues[&I] = C; 8737adc3a2bSChandler Carruth ++NumConstantPtrCmps; 8747adc3a2bSChandler Carruth return true; 8757adc3a2bSChandler Carruth } 8767adc3a2bSChandler Carruth } 8777adc3a2bSChandler Carruth } 8787adc3a2bSChandler Carruth 8797adc3a2bSChandler Carruth // If the comparison is an equality comparison with null, we can simplify it 8807adc3a2bSChandler Carruth // if we know the value (argument) can't be null 8817adc3a2bSChandler Carruth if (I.isEquality() && isa<ConstantPointerNull>(I.getOperand(1)) && 8827adc3a2bSChandler Carruth isKnownNonNullInCallee(I.getOperand(0))) { 8837adc3a2bSChandler Carruth bool IsNotEqual = I.getPredicate() == CmpInst::ICMP_NE; 8847adc3a2bSChandler Carruth SimplifiedValues[&I] = IsNotEqual ? ConstantInt::getTrue(I.getType()) 8857adc3a2bSChandler Carruth : ConstantInt::getFalse(I.getType()); 8867adc3a2bSChandler Carruth return true; 8877adc3a2bSChandler Carruth } 8887adc3a2bSChandler Carruth // Finally check for SROA candidates in comparisons. 8897adc3a2bSChandler Carruth Value *SROAArg; 8907adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 8917adc3a2bSChandler Carruth if (lookupSROAArgAndCost(I.getOperand(0), SROAArg, CostIt)) { 8927adc3a2bSChandler Carruth if (isa<ConstantPointerNull>(I.getOperand(1))) { 8937adc3a2bSChandler Carruth accumulateSROACost(CostIt, InlineConstants::InstrCost); 8947adc3a2bSChandler Carruth return true; 8957adc3a2bSChandler Carruth } 8967adc3a2bSChandler Carruth 8977adc3a2bSChandler Carruth disableSROA(CostIt); 8987adc3a2bSChandler Carruth } 8997adc3a2bSChandler Carruth 9007adc3a2bSChandler Carruth return false; 9017adc3a2bSChandler Carruth } 9027adc3a2bSChandler Carruth 9032e1c050eSChad Rosier bool CallAnalyzer::visitOr(BinaryOperator &I) { 9042e1c050eSChad Rosier // This is necessary because the generic simplify instruction only works if 9052e1c050eSChad Rosier // both operands are constants. 9062e1c050eSChad Rosier for (unsigned i = 0; i < 2; ++i) { 9072e1c050eSChad Rosier if (ConstantInt *C = dyn_cast_or_null<ConstantInt>( 9082e1c050eSChad Rosier SimplifiedValues.lookup(I.getOperand(i)))) 9092e1c050eSChad Rosier if (C->isAllOnesValue()) { 9102e1c050eSChad Rosier SimplifiedValues[&I] = C; 9112e1c050eSChad Rosier return true; 9122e1c050eSChad Rosier } 9132e1c050eSChad Rosier } 9142e1c050eSChad Rosier return Base::visitOr(I); 9152e1c050eSChad Rosier } 9162e1c050eSChad Rosier 9172e1c050eSChad Rosier bool CallAnalyzer::visitAnd(BinaryOperator &I) { 9182e1c050eSChad Rosier // This is necessary because the generic simplify instruction only works if 9192e1c050eSChad Rosier // both operands are constants. 9202e1c050eSChad Rosier for (unsigned i = 0; i < 2; ++i) { 9212e1c050eSChad Rosier if (ConstantInt *C = dyn_cast_or_null<ConstantInt>( 9222e1c050eSChad Rosier SimplifiedValues.lookup(I.getOperand(i)))) 9232e1c050eSChad Rosier if (C->isZero()) { 9242e1c050eSChad Rosier SimplifiedValues[&I] = C; 9252e1c050eSChad Rosier return true; 9262e1c050eSChad Rosier } 9272e1c050eSChad Rosier } 9282e1c050eSChad Rosier return Base::visitAnd(I); 9292e1c050eSChad Rosier } 9302e1c050eSChad Rosier 9317adc3a2bSChandler Carruth bool CallAnalyzer::visitSub(BinaryOperator &I) { 9327adc3a2bSChandler Carruth // Try to handle a special case: we can fold computing the difference of two 9337adc3a2bSChandler Carruth // constant-related pointers. 9347adc3a2bSChandler Carruth Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); 9357adc3a2bSChandler Carruth Value *LHSBase, *RHSBase; 9367adc3a2bSChandler Carruth APInt LHSOffset, RHSOffset; 9377adc3a2bSChandler Carruth std::tie(LHSBase, LHSOffset) = ConstantOffsetPtrs.lookup(LHS); 9387adc3a2bSChandler Carruth if (LHSBase) { 9397adc3a2bSChandler Carruth std::tie(RHSBase, RHSOffset) = ConstantOffsetPtrs.lookup(RHS); 9407adc3a2bSChandler Carruth if (RHSBase && LHSBase == RHSBase) { 9417adc3a2bSChandler Carruth // We have common bases, fold the subtract to a constant based on the 9427adc3a2bSChandler Carruth // offsets. 9437adc3a2bSChandler Carruth Constant *CLHS = ConstantInt::get(LHS->getContext(), LHSOffset); 9447adc3a2bSChandler Carruth Constant *CRHS = ConstantInt::get(RHS->getContext(), RHSOffset); 9457adc3a2bSChandler Carruth if (Constant *C = ConstantExpr::getSub(CLHS, CRHS)) { 9467adc3a2bSChandler Carruth SimplifiedValues[&I] = C; 9477adc3a2bSChandler Carruth ++NumConstantPtrDiffs; 9487adc3a2bSChandler Carruth return true; 9497adc3a2bSChandler Carruth } 9507adc3a2bSChandler Carruth } 9517adc3a2bSChandler Carruth } 9527adc3a2bSChandler Carruth 9537adc3a2bSChandler Carruth // Otherwise, fall back to the generic logic for simplifying and handling 9547adc3a2bSChandler Carruth // instructions. 9557adc3a2bSChandler Carruth return Base::visitSub(I); 9567adc3a2bSChandler Carruth } 9577adc3a2bSChandler Carruth 9587adc3a2bSChandler Carruth bool CallAnalyzer::visitBinaryOperator(BinaryOperator &I) { 9597adc3a2bSChandler Carruth Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); 960617f6364SEaswaran Raman auto Evaluate = [&](SmallVectorImpl<Constant *> &COps) { 9617adc3a2bSChandler Carruth Value *SimpleV = nullptr; 9627adc3a2bSChandler Carruth if (auto FI = dyn_cast<FPMathOperator>(&I)) 963617f6364SEaswaran Raman SimpleV = SimplifyFPBinOp(I.getOpcode(), COps[0], COps[1], 964617f6364SEaswaran Raman FI->getFastMathFlags(), DL); 9657adc3a2bSChandler Carruth else 966617f6364SEaswaran Raman SimpleV = SimplifyBinOp(I.getOpcode(), COps[0], COps[1], DL); 967617f6364SEaswaran Raman return dyn_cast_or_null<Constant>(SimpleV); 968617f6364SEaswaran Raman }; 9697adc3a2bSChandler Carruth 970617f6364SEaswaran Raman if (simplifyInstruction(I, Evaluate)) 9717adc3a2bSChandler Carruth return true; 9727adc3a2bSChandler Carruth 9737adc3a2bSChandler Carruth // Disable any SROA on arguments to arbitrary, unsimplified binary operators. 9747adc3a2bSChandler Carruth disableSROA(LHS); 9757adc3a2bSChandler Carruth disableSROA(RHS); 9767adc3a2bSChandler Carruth 9777adc3a2bSChandler Carruth return false; 9787adc3a2bSChandler Carruth } 9797adc3a2bSChandler Carruth 9807adc3a2bSChandler Carruth bool CallAnalyzer::visitLoad(LoadInst &I) { 9817adc3a2bSChandler Carruth Value *SROAArg; 9827adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 9837adc3a2bSChandler Carruth if (lookupSROAArgAndCost(I.getPointerOperand(), SROAArg, CostIt)) { 9847adc3a2bSChandler Carruth if (I.isSimple()) { 9857adc3a2bSChandler Carruth accumulateSROACost(CostIt, InlineConstants::InstrCost); 9867adc3a2bSChandler Carruth return true; 9877adc3a2bSChandler Carruth } 9887adc3a2bSChandler Carruth 9897adc3a2bSChandler Carruth disableSROA(CostIt); 9907adc3a2bSChandler Carruth } 9917adc3a2bSChandler Carruth 9927adc3a2bSChandler Carruth return false; 9937adc3a2bSChandler Carruth } 9947adc3a2bSChandler Carruth 9957adc3a2bSChandler Carruth bool CallAnalyzer::visitStore(StoreInst &I) { 9967adc3a2bSChandler Carruth Value *SROAArg; 9977adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 9987adc3a2bSChandler Carruth if (lookupSROAArgAndCost(I.getPointerOperand(), SROAArg, CostIt)) { 9997adc3a2bSChandler Carruth if (I.isSimple()) { 10007adc3a2bSChandler Carruth accumulateSROACost(CostIt, InlineConstants::InstrCost); 10017adc3a2bSChandler Carruth return true; 10027adc3a2bSChandler Carruth } 10037adc3a2bSChandler Carruth 10047adc3a2bSChandler Carruth disableSROA(CostIt); 10057adc3a2bSChandler Carruth } 10067adc3a2bSChandler Carruth 10077adc3a2bSChandler Carruth return false; 10087adc3a2bSChandler Carruth } 10097adc3a2bSChandler Carruth 10107adc3a2bSChandler Carruth bool CallAnalyzer::visitExtractValue(ExtractValueInst &I) { 10117adc3a2bSChandler Carruth // Constant folding for extract value is trivial. 1012617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 1013617f6364SEaswaran Raman return ConstantExpr::getExtractValue(COps[0], I.getIndices()); 1014617f6364SEaswaran Raman })) 10157adc3a2bSChandler Carruth return true; 10167adc3a2bSChandler Carruth 10177adc3a2bSChandler Carruth // SROA can look through these but give them a cost. 10187adc3a2bSChandler Carruth return false; 10197adc3a2bSChandler Carruth } 10207adc3a2bSChandler Carruth 10217adc3a2bSChandler Carruth bool CallAnalyzer::visitInsertValue(InsertValueInst &I) { 10227adc3a2bSChandler Carruth // Constant folding for insert value is trivial. 1023617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 1024617f6364SEaswaran Raman return ConstantExpr::getInsertValue(/*AggregateOperand*/ COps[0], 1025617f6364SEaswaran Raman /*InsertedValueOperand*/ COps[1], 1026617f6364SEaswaran Raman I.getIndices()); 1027617f6364SEaswaran Raman })) 10287adc3a2bSChandler Carruth return true; 10297adc3a2bSChandler Carruth 10307adc3a2bSChandler Carruth // SROA can look through these but give them a cost. 10317adc3a2bSChandler Carruth return false; 10327adc3a2bSChandler Carruth } 10337adc3a2bSChandler Carruth 10347adc3a2bSChandler Carruth /// \brief Try to simplify a call site. 10357adc3a2bSChandler Carruth /// 10367adc3a2bSChandler Carruth /// Takes a concrete function and callsite and tries to actually simplify it by 10377adc3a2bSChandler Carruth /// analyzing the arguments and call itself with instsimplify. Returns true if 10387adc3a2bSChandler Carruth /// it has simplified the callsite to some other entity (a constant), making it 10397adc3a2bSChandler Carruth /// free. 10407adc3a2bSChandler Carruth bool CallAnalyzer::simplifyCallSite(Function *F, CallSite CS) { 10417adc3a2bSChandler Carruth // FIXME: Using the instsimplify logic directly for this is inefficient 10427adc3a2bSChandler Carruth // because we have to continually rebuild the argument list even when no 10437adc3a2bSChandler Carruth // simplifications can be performed. Until that is fixed with remapping 10447adc3a2bSChandler Carruth // inside of instsimplify, directly constant fold calls here. 1045647025f9SAndrew Kaylor if (!canConstantFoldCallTo(CS, F)) 10467adc3a2bSChandler Carruth return false; 10477adc3a2bSChandler Carruth 10487adc3a2bSChandler Carruth // Try to re-map the arguments to constants. 10497adc3a2bSChandler Carruth SmallVector<Constant *, 4> ConstantArgs; 10507adc3a2bSChandler Carruth ConstantArgs.reserve(CS.arg_size()); 1051567556aaSChad Rosier for (CallSite::arg_iterator I = CS.arg_begin(), E = CS.arg_end(); I != E; 1052567556aaSChad Rosier ++I) { 10537adc3a2bSChandler Carruth Constant *C = dyn_cast<Constant>(*I); 10547adc3a2bSChandler Carruth if (!C) 10557adc3a2bSChandler Carruth C = dyn_cast_or_null<Constant>(SimplifiedValues.lookup(*I)); 10567adc3a2bSChandler Carruth if (!C) 10577adc3a2bSChandler Carruth return false; // This argument doesn't map to a constant. 10587adc3a2bSChandler Carruth 10597adc3a2bSChandler Carruth ConstantArgs.push_back(C); 10607adc3a2bSChandler Carruth } 1061647025f9SAndrew Kaylor if (Constant *C = ConstantFoldCall(CS, F, ConstantArgs)) { 10627adc3a2bSChandler Carruth SimplifiedValues[CS.getInstruction()] = C; 10637adc3a2bSChandler Carruth return true; 10647adc3a2bSChandler Carruth } 10657adc3a2bSChandler Carruth 10667adc3a2bSChandler Carruth return false; 10677adc3a2bSChandler Carruth } 10687adc3a2bSChandler Carruth 10697adc3a2bSChandler Carruth bool CallAnalyzer::visitCallSite(CallSite CS) { 10707adc3a2bSChandler Carruth if (CS.hasFnAttr(Attribute::ReturnsTwice) && 10717adc3a2bSChandler Carruth !F.hasFnAttribute(Attribute::ReturnsTwice)) { 10727adc3a2bSChandler Carruth // This aborts the entire analysis. 10737adc3a2bSChandler Carruth ExposesReturnsTwice = true; 10747adc3a2bSChandler Carruth return false; 10757adc3a2bSChandler Carruth } 1076567556aaSChad Rosier if (CS.isCall() && cast<CallInst>(CS.getInstruction())->cannotDuplicate()) 10777adc3a2bSChandler Carruth ContainsNoDuplicateCall = true; 10787adc3a2bSChandler Carruth 10797adc3a2bSChandler Carruth if (Function *F = CS.getCalledFunction()) { 10807adc3a2bSChandler Carruth // When we have a concrete function, first try to simplify it directly. 10817adc3a2bSChandler Carruth if (simplifyCallSite(F, CS)) 10827adc3a2bSChandler Carruth return true; 10837adc3a2bSChandler Carruth 10847adc3a2bSChandler Carruth // Next check if it is an intrinsic we know about. 10857adc3a2bSChandler Carruth // FIXME: Lift this into part of the InstVisitor. 10867adc3a2bSChandler Carruth if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(CS.getInstruction())) { 10877adc3a2bSChandler Carruth switch (II->getIntrinsicID()) { 10887adc3a2bSChandler Carruth default: 10897adc3a2bSChandler Carruth return Base::visitCallSite(CS); 10907adc3a2bSChandler Carruth 10917dd8dbf4SPeter Collingbourne case Intrinsic::load_relative: 10927dd8dbf4SPeter Collingbourne // This is normally lowered to 4 LLVM instructions. 10937dd8dbf4SPeter Collingbourne Cost += 3 * InlineConstants::InstrCost; 10947dd8dbf4SPeter Collingbourne return false; 10957dd8dbf4SPeter Collingbourne 10967adc3a2bSChandler Carruth case Intrinsic::memset: 10977adc3a2bSChandler Carruth case Intrinsic::memcpy: 10987adc3a2bSChandler Carruth case Intrinsic::memmove: 10997adc3a2bSChandler Carruth // SROA can usually chew through these intrinsics, but they aren't free. 11007adc3a2bSChandler Carruth return false; 11017adc3a2bSChandler Carruth case Intrinsic::localescape: 11027adc3a2bSChandler Carruth HasFrameEscape = true; 11037adc3a2bSChandler Carruth return false; 11047adc3a2bSChandler Carruth } 11057adc3a2bSChandler Carruth } 11067adc3a2bSChandler Carruth 11077adc3a2bSChandler Carruth if (F == CS.getInstruction()->getParent()->getParent()) { 11087adc3a2bSChandler Carruth // This flag will fully abort the analysis, so don't bother with anything 11097adc3a2bSChandler Carruth // else. 11107adc3a2bSChandler Carruth IsRecursiveCall = true; 11117adc3a2bSChandler Carruth return false; 11127adc3a2bSChandler Carruth } 11137adc3a2bSChandler Carruth 11147adc3a2bSChandler Carruth if (TTI.isLoweredToCall(F)) { 11157adc3a2bSChandler Carruth // We account for the average 1 instruction per call argument setup 11167adc3a2bSChandler Carruth // here. 11177adc3a2bSChandler Carruth Cost += CS.arg_size() * InlineConstants::InstrCost; 11187adc3a2bSChandler Carruth 11197adc3a2bSChandler Carruth // Everything other than inline ASM will also have a significant cost 11207adc3a2bSChandler Carruth // merely from making the call. 11217adc3a2bSChandler Carruth if (!isa<InlineAsm>(CS.getCalledValue())) 11227adc3a2bSChandler Carruth Cost += InlineConstants::CallPenalty; 11237adc3a2bSChandler Carruth } 11247adc3a2bSChandler Carruth 11257adc3a2bSChandler Carruth return Base::visitCallSite(CS); 11267adc3a2bSChandler Carruth } 11277adc3a2bSChandler Carruth 11287adc3a2bSChandler Carruth // Otherwise we're in a very special case -- an indirect function call. See 11297adc3a2bSChandler Carruth // if we can be particularly clever about this. 11307adc3a2bSChandler Carruth Value *Callee = CS.getCalledValue(); 11317adc3a2bSChandler Carruth 11327adc3a2bSChandler Carruth // First, pay the price of the argument setup. We account for the average 11337adc3a2bSChandler Carruth // 1 instruction per call argument setup here. 11347adc3a2bSChandler Carruth Cost += CS.arg_size() * InlineConstants::InstrCost; 11357adc3a2bSChandler Carruth 11367adc3a2bSChandler Carruth // Next, check if this happens to be an indirect function call to a known 11377adc3a2bSChandler Carruth // function in this inline context. If not, we've done all we can. 11387adc3a2bSChandler Carruth Function *F = dyn_cast_or_null<Function>(SimplifiedValues.lookup(Callee)); 11397adc3a2bSChandler Carruth if (!F) 11407adc3a2bSChandler Carruth return Base::visitCallSite(CS); 11417adc3a2bSChandler Carruth 11427adc3a2bSChandler Carruth // If we have a constant that we are calling as a function, we can peer 11437adc3a2bSChandler Carruth // through it and see the function target. This happens not infrequently 11447adc3a2bSChandler Carruth // during devirtualization and so we want to give it a hefty bonus for 11457adc3a2bSChandler Carruth // inlining, but cap that bonus in the event that inlining wouldn't pan 11467adc3a2bSChandler Carruth // out. Pretend to inline the function, with a custom threshold. 11471c57cc2bSEaswaran Raman auto IndirectCallParams = Params; 11481c57cc2bSEaswaran Raman IndirectCallParams.DefaultThreshold = InlineConstants::IndirectCallThreshold; 11490812c5beSHaicheng Wu CallAnalyzer CA(TTI, GetAssumptionCache, GetBFI, PSI, ORE, *F, CS, 115012585b01SEaswaran Raman IndirectCallParams); 11517adc3a2bSChandler Carruth if (CA.analyzeCall(CS)) { 11527adc3a2bSChandler Carruth // We were able to inline the indirect call! Subtract the cost from the 11536d90d9f1SEaswaran Raman // threshold to get the bonus we want to apply, but don't go below zero. 11546d90d9f1SEaswaran Raman Cost -= std::max(0, CA.getThreshold() - CA.getCost()); 11557adc3a2bSChandler Carruth } 11567adc3a2bSChandler Carruth 11577adc3a2bSChandler Carruth return Base::visitCallSite(CS); 11587adc3a2bSChandler Carruth } 11597adc3a2bSChandler Carruth 11607adc3a2bSChandler Carruth bool CallAnalyzer::visitReturnInst(ReturnInst &RI) { 11617adc3a2bSChandler Carruth // At least one return instruction will be free after inlining. 11627adc3a2bSChandler Carruth bool Free = !HasReturn; 11637adc3a2bSChandler Carruth HasReturn = true; 11647adc3a2bSChandler Carruth return Free; 11657adc3a2bSChandler Carruth } 11667adc3a2bSChandler Carruth 11677adc3a2bSChandler Carruth bool CallAnalyzer::visitBranchInst(BranchInst &BI) { 11687adc3a2bSChandler Carruth // We model unconditional branches as essentially free -- they really 11697adc3a2bSChandler Carruth // shouldn't exist at all, but handling them makes the behavior of the 11707adc3a2bSChandler Carruth // inliner more regular and predictable. Interestingly, conditional branches 11717adc3a2bSChandler Carruth // which will fold away are also free. 11727adc3a2bSChandler Carruth return BI.isUnconditional() || isa<ConstantInt>(BI.getCondition()) || 11737adc3a2bSChandler Carruth dyn_cast_or_null<ConstantInt>( 11747adc3a2bSChandler Carruth SimplifiedValues.lookup(BI.getCondition())); 11757adc3a2bSChandler Carruth } 11767adc3a2bSChandler Carruth 11777adc3a2bSChandler Carruth bool CallAnalyzer::visitSwitchInst(SwitchInst &SI) { 11787adc3a2bSChandler Carruth // We model unconditional switches as free, see the comments on handling 11797adc3a2bSChandler Carruth // branches. 11807adc3a2bSChandler Carruth if (isa<ConstantInt>(SI.getCondition())) 11817adc3a2bSChandler Carruth return true; 11827adc3a2bSChandler Carruth if (Value *V = SimplifiedValues.lookup(SI.getCondition())) 11837adc3a2bSChandler Carruth if (isa<ConstantInt>(V)) 11847adc3a2bSChandler Carruth return true; 11857adc3a2bSChandler Carruth 11867ad02eeeSEric Christopher // Assume the most general case where the switch is lowered into 1187919f9e8dSJun Bum Lim // either a jump table, bit test, or a balanced binary tree consisting of 1188919f9e8dSJun Bum Lim // case clusters without merging adjacent clusters with the same 1189919f9e8dSJun Bum Lim // destination. We do not consider the switches that are lowered with a mix 1190919f9e8dSJun Bum Lim // of jump table/bit test/binary search tree. The cost of the switch is 1191919f9e8dSJun Bum Lim // proportional to the size of the tree or the size of jump table range. 11922960d41eSJun Bum Lim // 11932960d41eSJun Bum Lim // NB: We convert large switches which are just used to initialize large phi 11942960d41eSJun Bum Lim // nodes to lookup tables instead in simplify-cfg, so this shouldn't prevent 11952960d41eSJun Bum Lim // inlining those. It will prevent inlining in cases where the optimization 11962960d41eSJun Bum Lim // does not (yet) fire. 1197919f9e8dSJun Bum Lim 1198506cfb7aSJun Bum Lim // Maximum valid cost increased in this function. 1199506cfb7aSJun Bum Lim int CostUpperBound = INT_MAX - InlineConstants::InstrCost - 1; 1200506cfb7aSJun Bum Lim 1201919f9e8dSJun Bum Lim // Exit early for a large switch, assuming one case needs at least one 1202919f9e8dSJun Bum Lim // instruction. 1203919f9e8dSJun Bum Lim // FIXME: This is not true for a bit test, but ignore such case for now to 1204919f9e8dSJun Bum Lim // save compile-time. 1205919f9e8dSJun Bum Lim int64_t CostLowerBound = 1206506cfb7aSJun Bum Lim std::min((int64_t)CostUpperBound, 1207919f9e8dSJun Bum Lim (int64_t)SI.getNumCases() * InlineConstants::InstrCost + Cost); 1208919f9e8dSJun Bum Lim 12090812c5beSHaicheng Wu if (CostLowerBound > Threshold && !ComputeFullInlineCost) { 1210919f9e8dSJun Bum Lim Cost = CostLowerBound; 1211919f9e8dSJun Bum Lim return false; 1212919f9e8dSJun Bum Lim } 1213919f9e8dSJun Bum Lim 1214919f9e8dSJun Bum Lim unsigned JumpTableSize = 0; 1215919f9e8dSJun Bum Lim unsigned NumCaseCluster = 1216919f9e8dSJun Bum Lim TTI.getEstimatedNumberOfCaseClusters(SI, JumpTableSize); 1217919f9e8dSJun Bum Lim 1218919f9e8dSJun Bum Lim // If suitable for a jump table, consider the cost for the table size and 1219919f9e8dSJun Bum Lim // branch to destination. 1220919f9e8dSJun Bum Lim if (JumpTableSize) { 1221919f9e8dSJun Bum Lim int64_t JTCost = (int64_t)JumpTableSize * InlineConstants::InstrCost + 1222919f9e8dSJun Bum Lim 4 * InlineConstants::InstrCost; 1223506cfb7aSJun Bum Lim 1224506cfb7aSJun Bum Lim Cost = std::min((int64_t)CostUpperBound, JTCost + Cost); 1225919f9e8dSJun Bum Lim return false; 1226919f9e8dSJun Bum Lim } 1227919f9e8dSJun Bum Lim 1228919f9e8dSJun Bum Lim // Considering forming a binary search, we should find the number of nodes 1229919f9e8dSJun Bum Lim // which is same as the number of comparisons when lowered. For a given 1230919f9e8dSJun Bum Lim // number of clusters, n, we can define a recursive function, f(n), to find 1231919f9e8dSJun Bum Lim // the number of nodes in the tree. The recursion is : 1232919f9e8dSJun Bum Lim // f(n) = 1 + f(n/2) + f (n - n/2), when n > 3, 1233919f9e8dSJun Bum Lim // and f(n) = n, when n <= 3. 1234919f9e8dSJun Bum Lim // This will lead a binary tree where the leaf should be either f(2) or f(3) 1235919f9e8dSJun Bum Lim // when n > 3. So, the number of comparisons from leaves should be n, while 1236919f9e8dSJun Bum Lim // the number of non-leaf should be : 1237919f9e8dSJun Bum Lim // 2^(log2(n) - 1) - 1 1238919f9e8dSJun Bum Lim // = 2^log2(n) * 2^-1 - 1 1239919f9e8dSJun Bum Lim // = n / 2 - 1. 1240919f9e8dSJun Bum Lim // Considering comparisons from leaf and non-leaf nodes, we can estimate the 1241919f9e8dSJun Bum Lim // number of comparisons in a simple closed form : 1242919f9e8dSJun Bum Lim // n + n / 2 - 1 = n * 3 / 2 - 1 1243919f9e8dSJun Bum Lim if (NumCaseCluster <= 3) { 1244919f9e8dSJun Bum Lim // Suppose a comparison includes one compare and one conditional branch. 1245919f9e8dSJun Bum Lim Cost += NumCaseCluster * 2 * InlineConstants::InstrCost; 1246919f9e8dSJun Bum Lim return false; 1247919f9e8dSJun Bum Lim } 1248506cfb7aSJun Bum Lim 1249506cfb7aSJun Bum Lim int64_t ExpectedNumberOfCompare = 3 * (int64_t)NumCaseCluster / 2 - 1; 1250506cfb7aSJun Bum Lim int64_t SwitchCost = 1251919f9e8dSJun Bum Lim ExpectedNumberOfCompare * 2 * InlineConstants::InstrCost; 1252506cfb7aSJun Bum Lim 1253506cfb7aSJun Bum Lim Cost = std::min((int64_t)CostUpperBound, SwitchCost + Cost); 1254919f9e8dSJun Bum Lim return false; 1255919f9e8dSJun Bum Lim } 1256919f9e8dSJun Bum Lim 12577adc3a2bSChandler Carruth bool CallAnalyzer::visitIndirectBrInst(IndirectBrInst &IBI) { 12587adc3a2bSChandler Carruth // We never want to inline functions that contain an indirectbr. This is 12597adc3a2bSChandler Carruth // incorrect because all the blockaddress's (in static global initializers 12607adc3a2bSChandler Carruth // for example) would be referring to the original function, and this 12617adc3a2bSChandler Carruth // indirect jump would jump from the inlined copy of the function into the 12627adc3a2bSChandler Carruth // original function which is extremely undefined behavior. 12637adc3a2bSChandler Carruth // FIXME: This logic isn't really right; we can safely inline functions with 12647adc3a2bSChandler Carruth // indirectbr's as long as no other function or global references the 12657adc3a2bSChandler Carruth // blockaddress of a block within the current function. 12667adc3a2bSChandler Carruth HasIndirectBr = true; 12677adc3a2bSChandler Carruth return false; 12687adc3a2bSChandler Carruth } 12697adc3a2bSChandler Carruth 12707adc3a2bSChandler Carruth bool CallAnalyzer::visitResumeInst(ResumeInst &RI) { 12717adc3a2bSChandler Carruth // FIXME: It's not clear that a single instruction is an accurate model for 12727adc3a2bSChandler Carruth // the inline cost of a resume instruction. 12737adc3a2bSChandler Carruth return false; 12747adc3a2bSChandler Carruth } 12757adc3a2bSChandler Carruth 12767adc3a2bSChandler Carruth bool CallAnalyzer::visitCleanupReturnInst(CleanupReturnInst &CRI) { 12777adc3a2bSChandler Carruth // FIXME: It's not clear that a single instruction is an accurate model for 12787adc3a2bSChandler Carruth // the inline cost of a cleanupret instruction. 12797adc3a2bSChandler Carruth return false; 12807adc3a2bSChandler Carruth } 12817adc3a2bSChandler Carruth 12827adc3a2bSChandler Carruth bool CallAnalyzer::visitCatchReturnInst(CatchReturnInst &CRI) { 12837adc3a2bSChandler Carruth // FIXME: It's not clear that a single instruction is an accurate model for 12848220bcc5SJoseph Tremoulet // the inline cost of a catchret instruction. 12857adc3a2bSChandler Carruth return false; 12867adc3a2bSChandler Carruth } 12877adc3a2bSChandler Carruth 12887adc3a2bSChandler Carruth bool CallAnalyzer::visitUnreachableInst(UnreachableInst &I) { 12897adc3a2bSChandler Carruth // FIXME: It might be reasonably to discount the cost of instructions leading 12907adc3a2bSChandler Carruth // to unreachable as they have the lowest possible impact on both runtime and 12917adc3a2bSChandler Carruth // code size. 12927adc3a2bSChandler Carruth return true; // No actual code is needed for unreachable. 12937adc3a2bSChandler Carruth } 12947adc3a2bSChandler Carruth 12957adc3a2bSChandler Carruth bool CallAnalyzer::visitInstruction(Instruction &I) { 12967adc3a2bSChandler Carruth // Some instructions are free. All of the free intrinsics can also be 12977adc3a2bSChandler Carruth // handled by SROA, etc. 12987adc3a2bSChandler Carruth if (TargetTransformInfo::TCC_Free == TTI.getUserCost(&I)) 12997adc3a2bSChandler Carruth return true; 13007adc3a2bSChandler Carruth 13017adc3a2bSChandler Carruth // We found something we don't understand or can't handle. Mark any SROA-able 13027adc3a2bSChandler Carruth // values in the operand list as no longer viable. 13037adc3a2bSChandler Carruth for (User::op_iterator OI = I.op_begin(), OE = I.op_end(); OI != OE; ++OI) 13047adc3a2bSChandler Carruth disableSROA(*OI); 13057adc3a2bSChandler Carruth 13067adc3a2bSChandler Carruth return false; 13077adc3a2bSChandler Carruth } 13087adc3a2bSChandler Carruth 13097adc3a2bSChandler Carruth /// \brief Analyze a basic block for its contribution to the inline cost. 13107adc3a2bSChandler Carruth /// 13117adc3a2bSChandler Carruth /// This method walks the analyzer over every instruction in the given basic 13127adc3a2bSChandler Carruth /// block and accounts for their cost during inlining at this callsite. It 13137adc3a2bSChandler Carruth /// aborts early if the threshold has been exceeded or an impossible to inline 13147adc3a2bSChandler Carruth /// construct has been detected. It returns false if inlining is no longer 13157adc3a2bSChandler Carruth /// viable, and true if inlining remains viable. 13167adc3a2bSChandler Carruth bool CallAnalyzer::analyzeBlock(BasicBlock *BB, 13177adc3a2bSChandler Carruth SmallPtrSetImpl<const Value *> &EphValues) { 13187adc3a2bSChandler Carruth for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I) { 13197adc3a2bSChandler Carruth // FIXME: Currently, the number of instructions in a function regardless of 13207adc3a2bSChandler Carruth // our ability to simplify them during inline to constants or dead code, 13217adc3a2bSChandler Carruth // are actually used by the vector bonus heuristic. As long as that's true, 13227adc3a2bSChandler Carruth // we have to special case debug intrinsics here to prevent differences in 13237adc3a2bSChandler Carruth // inlining due to debug symbols. Eventually, the number of unsimplified 13247adc3a2bSChandler Carruth // instructions shouldn't factor into the cost computation, but until then, 13257adc3a2bSChandler Carruth // hack around it here. 13267adc3a2bSChandler Carruth if (isa<DbgInfoIntrinsic>(I)) 13277adc3a2bSChandler Carruth continue; 13287adc3a2bSChandler Carruth 13297adc3a2bSChandler Carruth // Skip ephemeral values. 13305a82c916SDuncan P. N. Exon Smith if (EphValues.count(&*I)) 13317adc3a2bSChandler Carruth continue; 13327adc3a2bSChandler Carruth 13337adc3a2bSChandler Carruth ++NumInstructions; 13347adc3a2bSChandler Carruth if (isa<ExtractElementInst>(I) || I->getType()->isVectorTy()) 13357adc3a2bSChandler Carruth ++NumVectorInstructions; 13367adc3a2bSChandler Carruth 1337e9434e80SSanjay Patel // If the instruction is floating point, and the target says this operation 1338e9434e80SSanjay Patel // is expensive or the function has the "use-soft-float" attribute, this may 13397adc3a2bSChandler Carruth // eventually become a library call. Treat the cost as such. 13407adc3a2bSChandler Carruth if (I->getType()->isFloatingPointTy()) { 1341e9434e80SSanjay Patel // If the function has the "use-soft-float" attribute, mark it as 1342e9434e80SSanjay Patel // expensive. 13437adc3a2bSChandler Carruth if (TTI.getFPOpCost(I->getType()) == TargetTransformInfo::TCC_Expensive || 1344908ed7f2SEric Christopher (F.getFnAttribute("use-soft-float").getValueAsString() == "true")) 13457adc3a2bSChandler Carruth Cost += InlineConstants::CallPenalty; 13467adc3a2bSChandler Carruth } 13477adc3a2bSChandler Carruth 13487adc3a2bSChandler Carruth // If the instruction simplified to a constant, there is no cost to this 13497adc3a2bSChandler Carruth // instruction. Visit the instructions using our InstVisitor to account for 13507adc3a2bSChandler Carruth // all of the per-instruction logic. The visit tree returns true if we 13517adc3a2bSChandler Carruth // consumed the instruction in any way, and false if the instruction's base 13527adc3a2bSChandler Carruth // cost should count against inlining. 13535a82c916SDuncan P. N. Exon Smith if (Base::visit(&*I)) 13547adc3a2bSChandler Carruth ++NumInstructionsSimplified; 13557adc3a2bSChandler Carruth else 13567adc3a2bSChandler Carruth Cost += InlineConstants::InstrCost; 13577adc3a2bSChandler Carruth 13580812c5beSHaicheng Wu using namespace ore; 13597adc3a2bSChandler Carruth // If the visit this instruction detected an uninlinable pattern, abort. 13607adc3a2bSChandler Carruth if (IsRecursiveCall || ExposesReturnsTwice || HasDynamicAlloca || 13610812c5beSHaicheng Wu HasIndirectBr || HasFrameEscape) { 13620812c5beSHaicheng Wu if (ORE) 13630812c5beSHaicheng Wu ORE->emit(OptimizationRemarkMissed(DEBUG_TYPE, "NeverInline", 13640812c5beSHaicheng Wu CandidateCS.getInstruction()) 13650812c5beSHaicheng Wu << NV("Callee", &F) 13660812c5beSHaicheng Wu << " has uninlinable pattern and cost is not fully computed"); 13677adc3a2bSChandler Carruth return false; 13680812c5beSHaicheng Wu } 13697adc3a2bSChandler Carruth 13707adc3a2bSChandler Carruth // If the caller is a recursive function then we don't want to inline 13717adc3a2bSChandler Carruth // functions which allocate a lot of stack space because it would increase 13727adc3a2bSChandler Carruth // the caller stack usage dramatically. 13737adc3a2bSChandler Carruth if (IsCallerRecursive && 13740812c5beSHaicheng Wu AllocatedSize > InlineConstants::TotalAllocaSizeRecursiveCaller) { 13750812c5beSHaicheng Wu if (ORE) 13760812c5beSHaicheng Wu ORE->emit( 13770812c5beSHaicheng Wu OptimizationRemarkMissed(DEBUG_TYPE, "NeverInline", 13780812c5beSHaicheng Wu CandidateCS.getInstruction()) 13790812c5beSHaicheng Wu << NV("Callee", &F) 13800812c5beSHaicheng Wu << " is recursive and allocates too much stack space. Cost is " 13810812c5beSHaicheng Wu "not fully computed"); 13827adc3a2bSChandler Carruth return false; 13830812c5beSHaicheng Wu } 13847adc3a2bSChandler Carruth 13857adc3a2bSChandler Carruth // Check if we've past the maximum possible threshold so we don't spin in 13867adc3a2bSChandler Carruth // huge basic blocks that will never inline. 1387*61995364SHaicheng Wu if (Cost >= Threshold && !ComputeFullInlineCost) 13887adc3a2bSChandler Carruth return false; 13897adc3a2bSChandler Carruth } 13907adc3a2bSChandler Carruth 13917adc3a2bSChandler Carruth return true; 13927adc3a2bSChandler Carruth } 13937adc3a2bSChandler Carruth 13947adc3a2bSChandler Carruth /// \brief Compute the base pointer and cumulative constant offsets for V. 13957adc3a2bSChandler Carruth /// 13967adc3a2bSChandler Carruth /// This strips all constant offsets off of V, leaving it the base pointer, and 13977adc3a2bSChandler Carruth /// accumulates the total constant offset applied in the returned constant. It 13987adc3a2bSChandler Carruth /// returns 0 if V is not a pointer, and returns the constant '0' if there are 13997adc3a2bSChandler Carruth /// no constant offsets applied. 14007adc3a2bSChandler Carruth ConstantInt *CallAnalyzer::stripAndComputeInBoundsConstantOffsets(Value *&V) { 14017adc3a2bSChandler Carruth if (!V->getType()->isPointerTy()) 14027adc3a2bSChandler Carruth return nullptr; 14037adc3a2bSChandler Carruth 14047adc3a2bSChandler Carruth unsigned IntPtrWidth = DL.getPointerSizeInBits(); 14057adc3a2bSChandler Carruth APInt Offset = APInt::getNullValue(IntPtrWidth); 14067adc3a2bSChandler Carruth 14077adc3a2bSChandler Carruth // Even though we don't look through PHI nodes, we could be called on an 14087adc3a2bSChandler Carruth // instruction in an unreachable block, which may be on a cycle. 14097adc3a2bSChandler Carruth SmallPtrSet<Value *, 4> Visited; 14107adc3a2bSChandler Carruth Visited.insert(V); 14117adc3a2bSChandler Carruth do { 14127adc3a2bSChandler Carruth if (GEPOperator *GEP = dyn_cast<GEPOperator>(V)) { 14137adc3a2bSChandler Carruth if (!GEP->isInBounds() || !accumulateGEPOffset(*GEP, Offset)) 14147adc3a2bSChandler Carruth return nullptr; 14157adc3a2bSChandler Carruth V = GEP->getPointerOperand(); 14167adc3a2bSChandler Carruth } else if (Operator::getOpcode(V) == Instruction::BitCast) { 14177adc3a2bSChandler Carruth V = cast<Operator>(V)->getOperand(0); 14187adc3a2bSChandler Carruth } else if (GlobalAlias *GA = dyn_cast<GlobalAlias>(V)) { 14195ce32728SSanjoy Das if (GA->isInterposable()) 14207adc3a2bSChandler Carruth break; 14217adc3a2bSChandler Carruth V = GA->getAliasee(); 14227adc3a2bSChandler Carruth } else { 14237adc3a2bSChandler Carruth break; 14247adc3a2bSChandler Carruth } 14257adc3a2bSChandler Carruth assert(V->getType()->isPointerTy() && "Unexpected operand type!"); 14267adc3a2bSChandler Carruth } while (Visited.insert(V).second); 14277adc3a2bSChandler Carruth 14287adc3a2bSChandler Carruth Type *IntPtrTy = DL.getIntPtrType(V->getContext()); 14297adc3a2bSChandler Carruth return cast<ConstantInt>(ConstantInt::get(IntPtrTy, Offset)); 14307adc3a2bSChandler Carruth } 14317adc3a2bSChandler Carruth 14327adc3a2bSChandler Carruth /// \brief Analyze a call site for potential inlining. 14337adc3a2bSChandler Carruth /// 14347adc3a2bSChandler Carruth /// Returns true if inlining this call is viable, and false if it is not 14357adc3a2bSChandler Carruth /// viable. It computes the cost and adjusts the threshold based on numerous 14367adc3a2bSChandler Carruth /// factors and heuristics. If this method returns false but the computed cost 14377adc3a2bSChandler Carruth /// is below the computed threshold, then inlining was forcibly disabled by 14387adc3a2bSChandler Carruth /// some artifact of the routine. 14397adc3a2bSChandler Carruth bool CallAnalyzer::analyzeCall(CallSite CS) { 14407adc3a2bSChandler Carruth ++NumCallsAnalyzed; 14417adc3a2bSChandler Carruth 14427adc3a2bSChandler Carruth // Perform some tweaks to the cost and threshold based on the direct 14437adc3a2bSChandler Carruth // callsite information. 14447adc3a2bSChandler Carruth 14457adc3a2bSChandler Carruth // We want to more aggressively inline vector-dense kernels, so up the 14467adc3a2bSChandler Carruth // threshold, and we'll lower it if the % of vector instructions gets too 14477adc3a2bSChandler Carruth // low. Note that these bonuses are some what arbitrary and evolved over time 14487adc3a2bSChandler Carruth // by accident as much as because they are principled bonuses. 14497adc3a2bSChandler Carruth // 14507adc3a2bSChandler Carruth // FIXME: It would be nice to remove all such bonuses. At least it would be 14517adc3a2bSChandler Carruth // nice to base the bonus values on something more scientific. 14527adc3a2bSChandler Carruth assert(NumInstructions == 0); 14537adc3a2bSChandler Carruth assert(NumVectorInstructions == 0); 1454f4bb2f0dSEaswaran Raman 1455f4bb2f0dSEaswaran Raman // Update the threshold based on callsite properties 1456f4bb2f0dSEaswaran Raman updateThreshold(CS, F); 1457f4bb2f0dSEaswaran Raman 14587adc3a2bSChandler Carruth // Speculatively apply all possible bonuses to Threshold. If cost exceeds 14597adc3a2bSChandler Carruth // this Threshold any time, and cost cannot decrease, we can stop processing 14607adc3a2bSChandler Carruth // the rest of the function body. 146151b809bfSEaswaran Raman Threshold += (SingleBBBonus + VectorBonus); 14627adc3a2bSChandler Carruth 1463351d9b01SXinliang David Li // Give out bonuses for the callsite, as the instructions setting them up 1464351d9b01SXinliang David Li // will be gone after inlining. 1465351d9b01SXinliang David Li Cost -= getCallsiteCost(CS, DL); 14667adc3a2bSChandler Carruth 14677adc3a2bSChandler Carruth // If this function uses the coldcc calling convention, prefer not to inline 14687adc3a2bSChandler Carruth // it. 14697adc3a2bSChandler Carruth if (F.getCallingConv() == CallingConv::Cold) 14707adc3a2bSChandler Carruth Cost += InlineConstants::ColdccPenalty; 14717adc3a2bSChandler Carruth 14727adc3a2bSChandler Carruth // Check if we're done. This can happen due to bonuses and penalties. 1473*61995364SHaicheng Wu if (Cost >= Threshold && !ComputeFullInlineCost) 14747adc3a2bSChandler Carruth return false; 14757adc3a2bSChandler Carruth 14767adc3a2bSChandler Carruth if (F.empty()) 14777adc3a2bSChandler Carruth return true; 14787adc3a2bSChandler Carruth 14797adc3a2bSChandler Carruth Function *Caller = CS.getInstruction()->getParent()->getParent(); 14807adc3a2bSChandler Carruth // Check if the caller function is recursive itself. 14817adc3a2bSChandler Carruth for (User *U : Caller->users()) { 14827adc3a2bSChandler Carruth CallSite Site(U); 14837adc3a2bSChandler Carruth if (!Site) 14847adc3a2bSChandler Carruth continue; 14857adc3a2bSChandler Carruth Instruction *I = Site.getInstruction(); 14867adc3a2bSChandler Carruth if (I->getParent()->getParent() == Caller) { 14877adc3a2bSChandler Carruth IsCallerRecursive = true; 14887adc3a2bSChandler Carruth break; 14897adc3a2bSChandler Carruth } 14907adc3a2bSChandler Carruth } 14917adc3a2bSChandler Carruth 14927adc3a2bSChandler Carruth // Populate our simplified values by mapping from function arguments to call 14937adc3a2bSChandler Carruth // arguments with known important simplifications. 14947adc3a2bSChandler Carruth CallSite::arg_iterator CAI = CS.arg_begin(); 14957adc3a2bSChandler Carruth for (Function::arg_iterator FAI = F.arg_begin(), FAE = F.arg_end(); 14967adc3a2bSChandler Carruth FAI != FAE; ++FAI, ++CAI) { 14977adc3a2bSChandler Carruth assert(CAI != CS.arg_end()); 14987adc3a2bSChandler Carruth if (Constant *C = dyn_cast<Constant>(CAI)) 14995a82c916SDuncan P. N. Exon Smith SimplifiedValues[&*FAI] = C; 15007adc3a2bSChandler Carruth 15017adc3a2bSChandler Carruth Value *PtrArg = *CAI; 15027adc3a2bSChandler Carruth if (ConstantInt *C = stripAndComputeInBoundsConstantOffsets(PtrArg)) { 15035a82c916SDuncan P. N. Exon Smith ConstantOffsetPtrs[&*FAI] = std::make_pair(PtrArg, C->getValue()); 15047adc3a2bSChandler Carruth 15057adc3a2bSChandler Carruth // We can SROA any pointer arguments derived from alloca instructions. 15067adc3a2bSChandler Carruth if (isa<AllocaInst>(PtrArg)) { 15075a82c916SDuncan P. N. Exon Smith SROAArgValues[&*FAI] = PtrArg; 15087adc3a2bSChandler Carruth SROAArgCosts[PtrArg] = 0; 15097adc3a2bSChandler Carruth } 15107adc3a2bSChandler Carruth } 15117adc3a2bSChandler Carruth } 15127adc3a2bSChandler Carruth NumConstantArgs = SimplifiedValues.size(); 15137adc3a2bSChandler Carruth NumConstantOffsetPtrArgs = ConstantOffsetPtrs.size(); 15147adc3a2bSChandler Carruth NumAllocaArgs = SROAArgValues.size(); 15157adc3a2bSChandler Carruth 15167adc3a2bSChandler Carruth // FIXME: If a caller has multiple calls to a callee, we end up recomputing 15177adc3a2bSChandler Carruth // the ephemeral values multiple times (and they're completely determined by 15187adc3a2bSChandler Carruth // the callee, so this is purely duplicate work). 15197adc3a2bSChandler Carruth SmallPtrSet<const Value *, 32> EphValues; 1520aec2fa35SDaniel Jasper CodeMetrics::collectEphemeralValues(&F, &GetAssumptionCache(F), EphValues); 15217adc3a2bSChandler Carruth 15227adc3a2bSChandler Carruth // The worklist of live basic blocks in the callee *after* inlining. We avoid 15237adc3a2bSChandler Carruth // adding basic blocks of the callee which can be proven to be dead for this 15247adc3a2bSChandler Carruth // particular call site in order to get more accurate cost estimates. This 15257adc3a2bSChandler Carruth // requires a somewhat heavyweight iteration pattern: we need to walk the 15267adc3a2bSChandler Carruth // basic blocks in a breadth-first order as we insert live successors. To 15277adc3a2bSChandler Carruth // accomplish this, prioritizing for small iterations because we exit after 15287adc3a2bSChandler Carruth // crossing our threshold, we use a small-size optimized SetVector. 15297adc3a2bSChandler Carruth typedef SetVector<BasicBlock *, SmallVector<BasicBlock *, 16>, 1530567556aaSChad Rosier SmallPtrSet<BasicBlock *, 16>> 1531567556aaSChad Rosier BBSetVector; 15327adc3a2bSChandler Carruth BBSetVector BBWorklist; 15337adc3a2bSChandler Carruth BBWorklist.insert(&F.getEntryBlock()); 153451b809bfSEaswaran Raman bool SingleBB = true; 15357adc3a2bSChandler Carruth // Note that we *must not* cache the size, this loop grows the worklist. 15367adc3a2bSChandler Carruth for (unsigned Idx = 0; Idx != BBWorklist.size(); ++Idx) { 15377adc3a2bSChandler Carruth // Bail out the moment we cross the threshold. This means we'll under-count 15387adc3a2bSChandler Carruth // the cost, but only when undercounting doesn't matter. 1539*61995364SHaicheng Wu if (Cost >= Threshold && !ComputeFullInlineCost) 15407adc3a2bSChandler Carruth break; 15417adc3a2bSChandler Carruth 15427adc3a2bSChandler Carruth BasicBlock *BB = BBWorklist[Idx]; 15437adc3a2bSChandler Carruth if (BB->empty()) 15447adc3a2bSChandler Carruth continue; 15457adc3a2bSChandler Carruth 15467adc3a2bSChandler Carruth // Disallow inlining a blockaddress. A blockaddress only has defined 15477adc3a2bSChandler Carruth // behavior for an indirect branch in the same function, and we do not 15487adc3a2bSChandler Carruth // currently support inlining indirect branches. But, the inliner may not 15497adc3a2bSChandler Carruth // see an indirect branch that ends up being dead code at a particular call 15507adc3a2bSChandler Carruth // site. If the blockaddress escapes the function, e.g., via a global 15517adc3a2bSChandler Carruth // variable, inlining may lead to an invalid cross-function reference. 15527adc3a2bSChandler Carruth if (BB->hasAddressTaken()) 15537adc3a2bSChandler Carruth return false; 15547adc3a2bSChandler Carruth 15557adc3a2bSChandler Carruth // Analyze the cost of this block. If we blow through the threshold, this 15567adc3a2bSChandler Carruth // returns false, and we can bail on out. 1557d295b00aSEaswaran Raman if (!analyzeBlock(BB, EphValues)) 15587adc3a2bSChandler Carruth return false; 15597adc3a2bSChandler Carruth 15607adc3a2bSChandler Carruth TerminatorInst *TI = BB->getTerminator(); 15617adc3a2bSChandler Carruth 15627adc3a2bSChandler Carruth // Add in the live successors by first checking whether we have terminator 15637adc3a2bSChandler Carruth // that may be simplified based on the values simplified by this call. 15647adc3a2bSChandler Carruth if (BranchInst *BI = dyn_cast<BranchInst>(TI)) { 15657adc3a2bSChandler Carruth if (BI->isConditional()) { 15667adc3a2bSChandler Carruth Value *Cond = BI->getCondition(); 1567567556aaSChad Rosier if (ConstantInt *SimpleCond = 1568567556aaSChad Rosier dyn_cast_or_null<ConstantInt>(SimplifiedValues.lookup(Cond))) { 15697adc3a2bSChandler Carruth BBWorklist.insert(BI->getSuccessor(SimpleCond->isZero() ? 1 : 0)); 15707adc3a2bSChandler Carruth continue; 15717adc3a2bSChandler Carruth } 15727adc3a2bSChandler Carruth } 15737adc3a2bSChandler Carruth } else if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) { 15747adc3a2bSChandler Carruth Value *Cond = SI->getCondition(); 1575567556aaSChad Rosier if (ConstantInt *SimpleCond = 1576567556aaSChad Rosier dyn_cast_or_null<ConstantInt>(SimplifiedValues.lookup(Cond))) { 1577927d8e61SChandler Carruth BBWorklist.insert(SI->findCaseValue(SimpleCond)->getCaseSuccessor()); 15787adc3a2bSChandler Carruth continue; 15797adc3a2bSChandler Carruth } 15807adc3a2bSChandler Carruth } 15817adc3a2bSChandler Carruth 15827adc3a2bSChandler Carruth // If we're unable to select a particular successor, just count all of 15837adc3a2bSChandler Carruth // them. 15847adc3a2bSChandler Carruth for (unsigned TIdx = 0, TSize = TI->getNumSuccessors(); TIdx != TSize; 15857adc3a2bSChandler Carruth ++TIdx) 15867adc3a2bSChandler Carruth BBWorklist.insert(TI->getSuccessor(TIdx)); 15877adc3a2bSChandler Carruth 15887adc3a2bSChandler Carruth // If we had any successors at this point, than post-inlining is likely to 15897adc3a2bSChandler Carruth // have them as well. Note that we assume any basic blocks which existed 15907adc3a2bSChandler Carruth // due to branches or switches which folded above will also fold after 15917adc3a2bSChandler Carruth // inlining. 15927adc3a2bSChandler Carruth if (SingleBB && TI->getNumSuccessors() > 1) { 15937adc3a2bSChandler Carruth // Take off the bonus we applied to the threshold. 15947adc3a2bSChandler Carruth Threshold -= SingleBBBonus; 15957adc3a2bSChandler Carruth SingleBB = false; 15967adc3a2bSChandler Carruth } 15977adc3a2bSChandler Carruth } 15987adc3a2bSChandler Carruth 159951b809bfSEaswaran Raman bool OnlyOneCallAndLocalLinkage = 160051b809bfSEaswaran Raman F.hasLocalLinkage() && F.hasOneUse() && &F == CS.getCalledFunction(); 16017adc3a2bSChandler Carruth // If this is a noduplicate call, we can still inline as long as 16027adc3a2bSChandler Carruth // inlining this would cause the removal of the caller (so the instruction 16037adc3a2bSChandler Carruth // is not actually duplicated, just moved). 16047adc3a2bSChandler Carruth if (!OnlyOneCallAndLocalLinkage && ContainsNoDuplicateCall) 16057adc3a2bSChandler Carruth return false; 16067adc3a2bSChandler Carruth 16077adc3a2bSChandler Carruth // We applied the maximum possible vector bonus at the beginning. Now, 16087adc3a2bSChandler Carruth // subtract the excess bonus, if any, from the Threshold before 16097adc3a2bSChandler Carruth // comparing against Cost. 16107adc3a2bSChandler Carruth if (NumVectorInstructions <= NumInstructions / 10) 161151b809bfSEaswaran Raman Threshold -= VectorBonus; 16127adc3a2bSChandler Carruth else if (NumVectorInstructions <= NumInstructions / 2) 161351b809bfSEaswaran Raman Threshold -= VectorBonus/2; 16147adc3a2bSChandler Carruth 161500ab73dcSHans Wennborg return Cost < std::max(1, Threshold); 16167adc3a2bSChandler Carruth } 16177adc3a2bSChandler Carruth 16187adc3a2bSChandler Carruth #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 16197adc3a2bSChandler Carruth /// \brief Dump stats about this call's analysis. 1620eb2a2546SYaron Keren LLVM_DUMP_METHOD void CallAnalyzer::dump() { 16217adc3a2bSChandler Carruth #define DEBUG_PRINT_STAT(x) dbgs() << " " #x ": " << x << "\n" 16227adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantArgs); 16237adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantOffsetPtrArgs); 16247adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumAllocaArgs); 16257adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantPtrCmps); 16267adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantPtrDiffs); 16277adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumInstructionsSimplified); 16287adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumInstructions); 16297adc3a2bSChandler Carruth DEBUG_PRINT_STAT(SROACostSavings); 16307adc3a2bSChandler Carruth DEBUG_PRINT_STAT(SROACostSavingsLost); 16317adc3a2bSChandler Carruth DEBUG_PRINT_STAT(ContainsNoDuplicateCall); 16327adc3a2bSChandler Carruth DEBUG_PRINT_STAT(Cost); 16337adc3a2bSChandler Carruth DEBUG_PRINT_STAT(Threshold); 16347adc3a2bSChandler Carruth #undef DEBUG_PRINT_STAT 16357adc3a2bSChandler Carruth } 16367adc3a2bSChandler Carruth #endif 16377adc3a2bSChandler Carruth 16387adc3a2bSChandler Carruth /// \brief Test that there are no attribute conflicts between Caller and Callee 16397adc3a2bSChandler Carruth /// that prevent inlining. 16407adc3a2bSChandler Carruth static bool functionsHaveCompatibleAttributes(Function *Caller, 16417adc3a2bSChandler Carruth Function *Callee, 16427adc3a2bSChandler Carruth TargetTransformInfo &TTI) { 16437adc3a2bSChandler Carruth return TTI.areInlineCompatible(Caller, Callee) && 16441cb242ebSAkira Hatanaka AttributeFuncs::areInlineCompatible(*Caller, *Callee); 16457adc3a2bSChandler Carruth } 16467adc3a2bSChandler Carruth 1647351d9b01SXinliang David Li int llvm::getCallsiteCost(CallSite CS, const DataLayout &DL) { 1648351d9b01SXinliang David Li int Cost = 0; 1649351d9b01SXinliang David Li for (unsigned I = 0, E = CS.arg_size(); I != E; ++I) { 1650351d9b01SXinliang David Li if (CS.isByValArgument(I)) { 1651351d9b01SXinliang David Li // We approximate the number of loads and stores needed by dividing the 1652351d9b01SXinliang David Li // size of the byval type by the target's pointer size. 1653351d9b01SXinliang David Li PointerType *PTy = cast<PointerType>(CS.getArgument(I)->getType()); 1654351d9b01SXinliang David Li unsigned TypeSize = DL.getTypeSizeInBits(PTy->getElementType()); 1655351d9b01SXinliang David Li unsigned PointerSize = DL.getPointerSizeInBits(); 1656351d9b01SXinliang David Li // Ceiling division. 1657351d9b01SXinliang David Li unsigned NumStores = (TypeSize + PointerSize - 1) / PointerSize; 1658351d9b01SXinliang David Li 1659351d9b01SXinliang David Li // If it generates more than 8 stores it is likely to be expanded as an 1660351d9b01SXinliang David Li // inline memcpy so we take that as an upper bound. Otherwise we assume 1661351d9b01SXinliang David Li // one load and one store per word copied. 1662351d9b01SXinliang David Li // FIXME: The maxStoresPerMemcpy setting from the target should be used 1663351d9b01SXinliang David Li // here instead of a magic number of 8, but it's not available via 1664351d9b01SXinliang David Li // DataLayout. 1665351d9b01SXinliang David Li NumStores = std::min(NumStores, 8U); 1666351d9b01SXinliang David Li 1667351d9b01SXinliang David Li Cost += 2 * NumStores * InlineConstants::InstrCost; 1668351d9b01SXinliang David Li } else { 1669351d9b01SXinliang David Li // For non-byval arguments subtract off one instruction per call 1670351d9b01SXinliang David Li // argument. 1671351d9b01SXinliang David Li Cost += InlineConstants::InstrCost; 1672351d9b01SXinliang David Li } 1673351d9b01SXinliang David Li } 1674351d9b01SXinliang David Li // The call instruction also disappears after inlining. 1675351d9b01SXinliang David Li Cost += InlineConstants::InstrCost + InlineConstants::CallPenalty; 1676351d9b01SXinliang David Li return Cost; 1677351d9b01SXinliang David Li } 1678351d9b01SXinliang David Li 1679ab6a6837SSean Silva InlineCost llvm::getInlineCost( 16801c57cc2bSEaswaran Raman CallSite CS, const InlineParams &Params, TargetTransformInfo &CalleeTTI, 1681aec2fa35SDaniel Jasper std::function<AssumptionCache &(Function &)> &GetAssumptionCache, 168212585b01SEaswaran Raman Optional<function_ref<BlockFrequencyInfo &(Function &)>> GetBFI, 16830812c5beSHaicheng Wu ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE) { 1684aec2fa35SDaniel Jasper return getInlineCost(CS, CS.getCalledFunction(), Params, CalleeTTI, 16850812c5beSHaicheng Wu GetAssumptionCache, GetBFI, PSI, ORE); 1686b9f7120eSEaswaran Raman } 1687b9f7120eSEaswaran Raman 1688ab6a6837SSean Silva InlineCost llvm::getInlineCost( 16891c57cc2bSEaswaran Raman CallSite CS, Function *Callee, const InlineParams &Params, 1690aec2fa35SDaniel Jasper TargetTransformInfo &CalleeTTI, 1691aec2fa35SDaniel Jasper std::function<AssumptionCache &(Function &)> &GetAssumptionCache, 169212585b01SEaswaran Raman Optional<function_ref<BlockFrequencyInfo &(Function &)>> GetBFI, 16930812c5beSHaicheng Wu ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE) { 1694f4bb2f0dSEaswaran Raman 16957adc3a2bSChandler Carruth // Cannot inline indirect calls. 16967adc3a2bSChandler Carruth if (!Callee) 16977adc3a2bSChandler Carruth return llvm::InlineCost::getNever(); 16987adc3a2bSChandler Carruth 16997adc3a2bSChandler Carruth // Calls to functions with always-inline attributes should be inlined 17007adc3a2bSChandler Carruth // whenever possible. 17017adc3a2bSChandler Carruth if (CS.hasFnAttr(Attribute::AlwaysInline)) { 17027adc3a2bSChandler Carruth if (isInlineViable(*Callee)) 17037adc3a2bSChandler Carruth return llvm::InlineCost::getAlways(); 17047adc3a2bSChandler Carruth return llvm::InlineCost::getNever(); 17057adc3a2bSChandler Carruth } 17067adc3a2bSChandler Carruth 17077adc3a2bSChandler Carruth // Never inline functions with conflicting attributes (unless callee has 17087adc3a2bSChandler Carruth // always-inline attribute). 17095ce28f4fSChad Rosier Function *Caller = CS.getCaller(); 17105ce28f4fSChad Rosier if (!functionsHaveCompatibleAttributes(Caller, Callee, CalleeTTI)) 17117adc3a2bSChandler Carruth return llvm::InlineCost::getNever(); 17127adc3a2bSChandler Carruth 17137adc3a2bSChandler Carruth // Don't inline this call if the caller has the optnone attribute. 17145ce28f4fSChad Rosier if (Caller->hasFnAttribute(Attribute::OptimizeNone)) 17157adc3a2bSChandler Carruth return llvm::InlineCost::getNever(); 17167adc3a2bSChandler Carruth 17175ce32728SSanjoy Das // Don't inline functions which can be interposed at link-time. Don't inline 17185ce32728SSanjoy Das // functions marked noinline or call sites marked noinline. 1719107b187dSCraig Topper // Note: inlining non-exact non-interposable functions is fine, since we know 17205ce32728SSanjoy Das // we have *a* correct implementation of the source level function. 1721567556aaSChad Rosier if (Callee->isInterposable() || Callee->hasFnAttribute(Attribute::NoInline) || 1722567556aaSChad Rosier CS.isNoInline()) 17237adc3a2bSChandler Carruth return llvm::InlineCost::getNever(); 17247adc3a2bSChandler Carruth 17250812c5beSHaicheng Wu if (ORE) 17260812c5beSHaicheng Wu ComputeFullInlineCost = true; 17270812c5beSHaicheng Wu 17287adc3a2bSChandler Carruth DEBUG(llvm::dbgs() << " Analyzing call of " << Callee->getName() 17294eb18742SChad Rosier << "... (caller:" << Caller->getName() << ")\n"); 17307adc3a2bSChandler Carruth 17310812c5beSHaicheng Wu CallAnalyzer CA(CalleeTTI, GetAssumptionCache, GetBFI, PSI, ORE, *Callee, CS, 173212585b01SEaswaran Raman Params); 17337adc3a2bSChandler Carruth bool ShouldInline = CA.analyzeCall(CS); 17347adc3a2bSChandler Carruth 17357adc3a2bSChandler Carruth DEBUG(CA.dump()); 17367adc3a2bSChandler Carruth 17377adc3a2bSChandler Carruth // Check if there was a reason to force inlining or no inlining. 17387adc3a2bSChandler Carruth if (!ShouldInline && CA.getCost() < CA.getThreshold()) 17397adc3a2bSChandler Carruth return InlineCost::getNever(); 17407adc3a2bSChandler Carruth if (ShouldInline && CA.getCost() >= CA.getThreshold()) 17417adc3a2bSChandler Carruth return InlineCost::getAlways(); 17427adc3a2bSChandler Carruth 17437adc3a2bSChandler Carruth return llvm::InlineCost::get(CA.getCost(), CA.getThreshold()); 17447adc3a2bSChandler Carruth } 17457adc3a2bSChandler Carruth 1746b9f7120eSEaswaran Raman bool llvm::isInlineViable(Function &F) { 17477adc3a2bSChandler Carruth bool ReturnsTwice = F.hasFnAttribute(Attribute::ReturnsTwice); 17487adc3a2bSChandler Carruth for (Function::iterator BI = F.begin(), BE = F.end(); BI != BE; ++BI) { 17497adc3a2bSChandler Carruth // Disallow inlining of functions which contain indirect branches or 17507adc3a2bSChandler Carruth // blockaddresses. 17517adc3a2bSChandler Carruth if (isa<IndirectBrInst>(BI->getTerminator()) || BI->hasAddressTaken()) 17527adc3a2bSChandler Carruth return false; 17537adc3a2bSChandler Carruth 17545a82c916SDuncan P. N. Exon Smith for (auto &II : *BI) { 17555a82c916SDuncan P. N. Exon Smith CallSite CS(&II); 17567adc3a2bSChandler Carruth if (!CS) 17577adc3a2bSChandler Carruth continue; 17587adc3a2bSChandler Carruth 17597adc3a2bSChandler Carruth // Disallow recursive calls. 17607adc3a2bSChandler Carruth if (&F == CS.getCalledFunction()) 17617adc3a2bSChandler Carruth return false; 17627adc3a2bSChandler Carruth 17637adc3a2bSChandler Carruth // Disallow calls which expose returns-twice to a function not previously 17647adc3a2bSChandler Carruth // attributed as such. 17657adc3a2bSChandler Carruth if (!ReturnsTwice && CS.isCall() && 17667adc3a2bSChandler Carruth cast<CallInst>(CS.getInstruction())->canReturnTwice()) 17677adc3a2bSChandler Carruth return false; 17687adc3a2bSChandler Carruth 17697adc3a2bSChandler Carruth // Disallow inlining functions that call @llvm.localescape. Doing this 17707adc3a2bSChandler Carruth // correctly would require major changes to the inliner. 17717adc3a2bSChandler Carruth if (CS.getCalledFunction() && 17727adc3a2bSChandler Carruth CS.getCalledFunction()->getIntrinsicID() == 17737adc3a2bSChandler Carruth llvm::Intrinsic::localescape) 17747adc3a2bSChandler Carruth return false; 17757adc3a2bSChandler Carruth } 17767adc3a2bSChandler Carruth } 17777adc3a2bSChandler Carruth 17787adc3a2bSChandler Carruth return true; 17797adc3a2bSChandler Carruth } 17801c57cc2bSEaswaran Raman 17811c57cc2bSEaswaran Raman // APIs to create InlineParams based on command line flags and/or other 17821c57cc2bSEaswaran Raman // parameters. 17831c57cc2bSEaswaran Raman 17841c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams(int Threshold) { 17851c57cc2bSEaswaran Raman InlineParams Params; 17861c57cc2bSEaswaran Raman 17871c57cc2bSEaswaran Raman // This field is the threshold to use for a callee by default. This is 17881c57cc2bSEaswaran Raman // derived from one or more of: 17891c57cc2bSEaswaran Raman // * optimization or size-optimization levels, 17901c57cc2bSEaswaran Raman // * a value passed to createFunctionInliningPass function, or 17911c57cc2bSEaswaran Raman // * the -inline-threshold flag. 17921c57cc2bSEaswaran Raman // If the -inline-threshold flag is explicitly specified, that is used 17931c57cc2bSEaswaran Raman // irrespective of anything else. 17941c57cc2bSEaswaran Raman if (InlineThreshold.getNumOccurrences() > 0) 17951c57cc2bSEaswaran Raman Params.DefaultThreshold = InlineThreshold; 17961c57cc2bSEaswaran Raman else 17971c57cc2bSEaswaran Raman Params.DefaultThreshold = Threshold; 17981c57cc2bSEaswaran Raman 17991c57cc2bSEaswaran Raman // Set the HintThreshold knob from the -inlinehint-threshold. 18001c57cc2bSEaswaran Raman Params.HintThreshold = HintThreshold; 18011c57cc2bSEaswaran Raman 18021c57cc2bSEaswaran Raman // Set the HotCallSiteThreshold knob from the -hot-callsite-threshold. 18031c57cc2bSEaswaran Raman Params.HotCallSiteThreshold = HotCallSiteThreshold; 18041c57cc2bSEaswaran Raman 1805974d4eeaSEaswaran Raman // If the -locally-hot-callsite-threshold is explicitly specified, use it to 1806974d4eeaSEaswaran Raman // populate LocallyHotCallSiteThreshold. Later, we populate 1807974d4eeaSEaswaran Raman // Params.LocallyHotCallSiteThreshold from -locally-hot-callsite-threshold if 1808974d4eeaSEaswaran Raman // we know that optimization level is O3 (in the getInlineParams variant that 1809974d4eeaSEaswaran Raman // takes the opt and size levels). 1810974d4eeaSEaswaran Raman // FIXME: Remove this check (and make the assignment unconditional) after 1811974d4eeaSEaswaran Raman // addressing size regression issues at O2. 1812974d4eeaSEaswaran Raman if (LocallyHotCallSiteThreshold.getNumOccurrences() > 0) 1813974d4eeaSEaswaran Raman Params.LocallyHotCallSiteThreshold = LocallyHotCallSiteThreshold; 1814974d4eeaSEaswaran Raman 181512585b01SEaswaran Raman // Set the ColdCallSiteThreshold knob from the -inline-cold-callsite-threshold. 181612585b01SEaswaran Raman Params.ColdCallSiteThreshold = ColdCallSiteThreshold; 181712585b01SEaswaran Raman 18181c57cc2bSEaswaran Raman // Set the OptMinSizeThreshold and OptSizeThreshold params only if the 18191c57cc2bSEaswaran Raman // -inlinehint-threshold commandline option is not explicitly given. If that 18201c57cc2bSEaswaran Raman // option is present, then its value applies even for callees with size and 18211c57cc2bSEaswaran Raman // minsize attributes. 18221c57cc2bSEaswaran Raman // If the -inline-threshold is not specified, set the ColdThreshold from the 18231c57cc2bSEaswaran Raman // -inlinecold-threshold even if it is not explicitly passed. If 18241c57cc2bSEaswaran Raman // -inline-threshold is specified, then -inlinecold-threshold needs to be 18251c57cc2bSEaswaran Raman // explicitly specified to set the ColdThreshold knob 18261c57cc2bSEaswaran Raman if (InlineThreshold.getNumOccurrences() == 0) { 18271c57cc2bSEaswaran Raman Params.OptMinSizeThreshold = InlineConstants::OptMinSizeThreshold; 18281c57cc2bSEaswaran Raman Params.OptSizeThreshold = InlineConstants::OptSizeThreshold; 18291c57cc2bSEaswaran Raman Params.ColdThreshold = ColdThreshold; 18301c57cc2bSEaswaran Raman } else if (ColdThreshold.getNumOccurrences() > 0) { 18311c57cc2bSEaswaran Raman Params.ColdThreshold = ColdThreshold; 18321c57cc2bSEaswaran Raman } 18331c57cc2bSEaswaran Raman return Params; 18341c57cc2bSEaswaran Raman } 18351c57cc2bSEaswaran Raman 18361c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams() { 18371c57cc2bSEaswaran Raman return getInlineParams(InlineThreshold); 18381c57cc2bSEaswaran Raman } 18391c57cc2bSEaswaran Raman 18401c57cc2bSEaswaran Raman // Compute the default threshold for inlining based on the opt level and the 18411c57cc2bSEaswaran Raman // size opt level. 18421c57cc2bSEaswaran Raman static int computeThresholdFromOptLevels(unsigned OptLevel, 18431c57cc2bSEaswaran Raman unsigned SizeOptLevel) { 18441c57cc2bSEaswaran Raman if (OptLevel > 2) 18451c57cc2bSEaswaran Raman return InlineConstants::OptAggressiveThreshold; 18461c57cc2bSEaswaran Raman if (SizeOptLevel == 1) // -Os 18471c57cc2bSEaswaran Raman return InlineConstants::OptSizeThreshold; 18481c57cc2bSEaswaran Raman if (SizeOptLevel == 2) // -Oz 18491c57cc2bSEaswaran Raman return InlineConstants::OptMinSizeThreshold; 18501c57cc2bSEaswaran Raman return InlineThreshold; 18511c57cc2bSEaswaran Raman } 18521c57cc2bSEaswaran Raman 18531c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams(unsigned OptLevel, unsigned SizeOptLevel) { 1854974d4eeaSEaswaran Raman auto Params = 1855974d4eeaSEaswaran Raman getInlineParams(computeThresholdFromOptLevels(OptLevel, SizeOptLevel)); 1856974d4eeaSEaswaran Raman // At O3, use the value of -locally-hot-callsite-threshold option to populate 1857974d4eeaSEaswaran Raman // Params.LocallyHotCallSiteThreshold. Below O3, this flag has effect only 1858974d4eeaSEaswaran Raman // when it is specified explicitly. 1859974d4eeaSEaswaran Raman if (OptLevel > 2) 1860974d4eeaSEaswaran Raman Params.LocallyHotCallSiteThreshold = LocallyHotCallSiteThreshold; 1861974d4eeaSEaswaran Raman return Params; 18621c57cc2bSEaswaran Raman } 1863