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" 243739e14aSHaicheng Wu #include "llvm/Analysis/CFG.h" 257adc3a2bSChandler Carruth #include "llvm/Analysis/InstructionSimplify.h" 26ba9f245bSDavid Green #include "llvm/Analysis/LoopInfo.h" 2771069cf6SEaswaran Raman #include "llvm/Analysis/ProfileSummaryInfo.h" 287adc3a2bSChandler Carruth #include "llvm/Analysis/TargetTransformInfo.h" 29a4461515SHaicheng Wu #include "llvm/Analysis/ValueTracking.h" 30432a3883SNico Weber #include "llvm/Config/llvm-config.h" 317adc3a2bSChandler Carruth #include "llvm/IR/CallSite.h" 327adc3a2bSChandler Carruth #include "llvm/IR/CallingConv.h" 337adc3a2bSChandler Carruth #include "llvm/IR/DataLayout.h" 34ba9f245bSDavid Green #include "llvm/IR/Dominators.h" 357adc3a2bSChandler Carruth #include "llvm/IR/GetElementPtrTypeIterator.h" 367adc3a2bSChandler Carruth #include "llvm/IR/GlobalAlias.h" 377adc3a2bSChandler Carruth #include "llvm/IR/InstVisitor.h" 387adc3a2bSChandler Carruth #include "llvm/IR/IntrinsicInst.h" 397adc3a2bSChandler Carruth #include "llvm/IR/Operator.h" 407adc3a2bSChandler Carruth #include "llvm/Support/Debug.h" 417adc3a2bSChandler Carruth #include "llvm/Support/raw_ostream.h" 427adc3a2bSChandler Carruth 437adc3a2bSChandler Carruth using namespace llvm; 447adc3a2bSChandler Carruth 457adc3a2bSChandler Carruth #define DEBUG_TYPE "inline-cost" 467adc3a2bSChandler Carruth 477adc3a2bSChandler Carruth STATISTIC(NumCallsAnalyzed, "Number of call sites analyzed"); 487adc3a2bSChandler Carruth 491c57cc2bSEaswaran Raman static cl::opt<int> InlineThreshold( 50f4bb2f0dSEaswaran Raman "inline-threshold", cl::Hidden, cl::init(225), cl::ZeroOrMore, 51f4bb2f0dSEaswaran Raman cl::desc("Control the amount of inlining to perform (default = 225)")); 52f4bb2f0dSEaswaran Raman 53f4bb2f0dSEaswaran Raman static cl::opt<int> HintThreshold( 54f4bb2f0dSEaswaran Raman "inlinehint-threshold", cl::Hidden, cl::init(325), 55f4bb2f0dSEaswaran Raman cl::desc("Threshold for inlining functions with inline hint")); 56f4bb2f0dSEaswaran Raman 5712585b01SEaswaran Raman static cl::opt<int> 5812585b01SEaswaran Raman ColdCallSiteThreshold("inline-cold-callsite-threshold", cl::Hidden, 5912585b01SEaswaran Raman cl::init(45), 6012585b01SEaswaran Raman cl::desc("Threshold for inlining cold callsites")); 6112585b01SEaswaran Raman 62f4bb2f0dSEaswaran Raman // We introduce this threshold to help performance of instrumentation based 63f4bb2f0dSEaswaran Raman // PGO before we actually hook up inliner with analysis passes such as BPI and 64f4bb2f0dSEaswaran Raman // BFI. 65f4bb2f0dSEaswaran Raman static cl::opt<int> ColdThreshold( 66c103ef89SEaswaran Raman "inlinecold-threshold", cl::Hidden, cl::init(45), 67f4bb2f0dSEaswaran Raman cl::desc("Threshold for inlining functions with cold attribute")); 68f4bb2f0dSEaswaran Raman 69de39cb93SDehao Chen static cl::opt<int> 70de39cb93SDehao Chen HotCallSiteThreshold("hot-callsite-threshold", cl::Hidden, cl::init(3000), 71de39cb93SDehao Chen cl::ZeroOrMore, 72de39cb93SDehao Chen cl::desc("Threshold for hot callsites ")); 73de39cb93SDehao Chen 74974d4eeaSEaswaran Raman static cl::opt<int> LocallyHotCallSiteThreshold( 75974d4eeaSEaswaran Raman "locally-hot-callsite-threshold", cl::Hidden, cl::init(525), cl::ZeroOrMore, 76974d4eeaSEaswaran Raman cl::desc("Threshold for locally hot callsites ")); 77974d4eeaSEaswaran Raman 78c5fa6358SEaswaran Raman static cl::opt<int> ColdCallSiteRelFreq( 79c5fa6358SEaswaran Raman "cold-callsite-rel-freq", cl::Hidden, cl::init(2), cl::ZeroOrMore, 80c5fa6358SEaswaran Raman cl::desc("Maxmimum block frequency, expressed as a percentage of caller's " 81c5fa6358SEaswaran Raman "entry frequency, for a callsite to be cold in the absence of " 82c5fa6358SEaswaran Raman "profile information.")); 83c5fa6358SEaswaran Raman 84974d4eeaSEaswaran Raman static cl::opt<int> HotCallSiteRelFreq( 85974d4eeaSEaswaran Raman "hot-callsite-rel-freq", cl::Hidden, cl::init(60), cl::ZeroOrMore, 86ff77cc75SEaswaran Raman cl::desc("Minimum block frequency, expressed as a multiple of caller's " 87974d4eeaSEaswaran Raman "entry frequency, for a callsite to be hot in the absence of " 88974d4eeaSEaswaran Raman "profile information.")); 89974d4eeaSEaswaran Raman 904924bb00SEaswaran Raman static cl::opt<bool> OptComputeFullInlineCost( 910812c5beSHaicheng Wu "inline-cost-full", cl::Hidden, cl::init(false), 920812c5beSHaicheng Wu cl::desc("Compute the full inline cost of a call site even when the cost " 930812c5beSHaicheng Wu "exceeds the threshold.")); 940812c5beSHaicheng Wu 957adc3a2bSChandler Carruth namespace { 967adc3a2bSChandler Carruth 977adc3a2bSChandler Carruth class CallAnalyzer : public InstVisitor<CallAnalyzer, bool> { 987adc3a2bSChandler Carruth typedef InstVisitor<CallAnalyzer, bool> Base; 997adc3a2bSChandler Carruth friend class InstVisitor<CallAnalyzer, bool>; 1007adc3a2bSChandler Carruth 1017adc3a2bSChandler Carruth /// The TargetTransformInfo available for this compilation. 1027adc3a2bSChandler Carruth const TargetTransformInfo &TTI; 1037adc3a2bSChandler Carruth 104aec2fa35SDaniel Jasper /// Getter for the cache of @llvm.assume intrinsics. 105aec2fa35SDaniel Jasper std::function<AssumptionCache &(Function &)> &GetAssumptionCache; 106aec2fa35SDaniel Jasper 10712585b01SEaswaran Raman /// Getter for BlockFrequencyInfo 10812585b01SEaswaran Raman Optional<function_ref<BlockFrequencyInfo &(Function &)>> &GetBFI; 10912585b01SEaswaran Raman 11071069cf6SEaswaran Raman /// Profile summary information. 11171069cf6SEaswaran Raman ProfileSummaryInfo *PSI; 11271069cf6SEaswaran Raman 113f3d122cdSPiotr Padlewski /// The called function. 1147adc3a2bSChandler Carruth Function &F; 1157adc3a2bSChandler Carruth 11685be8ca8SEric Christopher // Cache the DataLayout since we use it a lot. 11785be8ca8SEric Christopher const DataLayout &DL; 11885be8ca8SEric Christopher 1190812c5beSHaicheng Wu /// The OptimizationRemarkEmitter available for this compilation. 1200812c5beSHaicheng Wu OptimizationRemarkEmitter *ORE; 1210812c5beSHaicheng Wu 122f3d122cdSPiotr Padlewski /// The candidate callsite being analyzed. Please do not use this to do 123f3d122cdSPiotr Padlewski /// analysis in the caller function; we want the inline cost query to be 124f3d122cdSPiotr Padlewski /// easily cacheable. Instead, use the cover function paramHasAttr. 1257adc3a2bSChandler Carruth CallSite CandidateCS; 1267adc3a2bSChandler Carruth 127f3d122cdSPiotr Padlewski /// Tunable parameters that control the analysis. 1281c57cc2bSEaswaran Raman const InlineParams &Params; 1291c57cc2bSEaswaran Raman 1307adc3a2bSChandler Carruth int Threshold; 1317adc3a2bSChandler Carruth int Cost; 1324924bb00SEaswaran Raman bool ComputeFullInlineCost; 1337adc3a2bSChandler Carruth 1347adc3a2bSChandler Carruth bool IsCallerRecursive; 1357adc3a2bSChandler Carruth bool IsRecursiveCall; 1367adc3a2bSChandler Carruth bool ExposesReturnsTwice; 1377adc3a2bSChandler Carruth bool HasDynamicAlloca; 1387adc3a2bSChandler Carruth bool ContainsNoDuplicateCall; 1397adc3a2bSChandler Carruth bool HasReturn; 1407adc3a2bSChandler Carruth bool HasIndirectBr; 1414296ea72SVitaly Buka bool HasUninlineableIntrinsic; 14277beee41SSameer AbuAsal bool InitsVargArgs; 1437adc3a2bSChandler Carruth 1447adc3a2bSChandler Carruth /// Number of bytes allocated statically by the callee. 1457adc3a2bSChandler Carruth uint64_t AllocatedSize; 1467adc3a2bSChandler Carruth unsigned NumInstructions, NumVectorInstructions; 14751b809bfSEaswaran Raman int VectorBonus, TenPercentVectorBonus; 14851b809bfSEaswaran Raman // Bonus to be applied when the callee has only one reachable basic block. 14951b809bfSEaswaran Raman int SingleBBBonus; 1507adc3a2bSChandler Carruth 151f3d122cdSPiotr Padlewski /// While we walk the potentially-inlined instructions, we build up and 152f3d122cdSPiotr Padlewski /// maintain a mapping of simplified values specific to this callsite. The 153f3d122cdSPiotr Padlewski /// idea is to propagate any special information we have about arguments to 154f3d122cdSPiotr Padlewski /// this call through the inlinable section of the function, and account for 155f3d122cdSPiotr Padlewski /// likely simplifications post-inlining. The most important aspect we track 156f3d122cdSPiotr Padlewski /// is CFG altering simplifications -- when we prove a basic block dead, that 157f3d122cdSPiotr Padlewski /// can cause dramatic shifts in the cost of inlining a function. 1587adc3a2bSChandler Carruth DenseMap<Value *, Constant *> SimplifiedValues; 1597adc3a2bSChandler Carruth 160f3d122cdSPiotr Padlewski /// Keep track of the values which map back (through function arguments) to 161f3d122cdSPiotr Padlewski /// allocas on the caller stack which could be simplified through SROA. 1627adc3a2bSChandler Carruth DenseMap<Value *, Value *> SROAArgValues; 1637adc3a2bSChandler Carruth 164f3d122cdSPiotr Padlewski /// The mapping of caller Alloca values to their accumulated cost savings. If 165f3d122cdSPiotr Padlewski /// we have to disable SROA for one of the allocas, this tells us how much 166f3d122cdSPiotr Padlewski /// cost must be added. 1677adc3a2bSChandler Carruth DenseMap<Value *, int> SROAArgCosts; 1687adc3a2bSChandler Carruth 169f3d122cdSPiotr Padlewski /// Keep track of values which map to a pointer base and constant offset. 1707adc3a2bSChandler Carruth DenseMap<Value *, std::pair<Value *, APInt>> ConstantOffsetPtrs; 1717adc3a2bSChandler Carruth 1723739e14aSHaicheng Wu /// Keep track of dead blocks due to the constant arguments. 1733739e14aSHaicheng Wu SetVector<BasicBlock *> DeadBlocks; 1743739e14aSHaicheng Wu 1753739e14aSHaicheng Wu /// The mapping of the blocks to their known unique successors due to the 1763739e14aSHaicheng Wu /// constant arguments. 1773739e14aSHaicheng Wu DenseMap<BasicBlock *, BasicBlock *> KnownSuccessors; 1783739e14aSHaicheng Wu 179a4461515SHaicheng Wu /// Model the elimination of repeated loads that is expected to happen 180a4461515SHaicheng Wu /// whenever we simplify away the stores that would otherwise cause them to be 181a4461515SHaicheng Wu /// loads. 182a4461515SHaicheng Wu bool EnableLoadElimination; 183a4461515SHaicheng Wu SmallPtrSet<Value *, 16> LoadAddrSet; 184a4461515SHaicheng Wu int LoadEliminationCost; 185a4461515SHaicheng Wu 1867adc3a2bSChandler Carruth // Custom simplification helper routines. 1877adc3a2bSChandler Carruth bool isAllocaDerivedArg(Value *V); 1887adc3a2bSChandler Carruth bool lookupSROAArgAndCost(Value *V, Value *&Arg, 1897adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator &CostIt); 1907adc3a2bSChandler Carruth void disableSROA(DenseMap<Value *, int>::iterator CostIt); 1917adc3a2bSChandler Carruth void disableSROA(Value *V); 1923739e14aSHaicheng Wu void findDeadBlocks(BasicBlock *CurrBB, BasicBlock *NextBB); 1937adc3a2bSChandler Carruth void accumulateSROACost(DenseMap<Value *, int>::iterator CostIt, 1947adc3a2bSChandler Carruth int InstructionCost); 195a4461515SHaicheng Wu void disableLoadElimination(); 196201b191bSHaicheng Wu bool isGEPFree(GetElementPtrInst &GEP); 197d3558b5dSEvgeny Astigeevich bool canFoldInboundsGEP(GetElementPtrInst &I); 1987adc3a2bSChandler Carruth bool accumulateGEPOffset(GEPOperator &GEP, APInt &Offset); 1997adc3a2bSChandler Carruth bool simplifyCallSite(Function *F, CallSite CS); 200617f6364SEaswaran Raman template <typename Callable> 201617f6364SEaswaran Raman bool simplifyInstruction(Instruction &I, Callable Evaluate); 2027adc3a2bSChandler Carruth ConstantInt *stripAndComputeInBoundsConstantOffsets(Value *&V); 2037adc3a2bSChandler Carruth 2047adc3a2bSChandler Carruth /// Return true if the given argument to the function being considered for 2057adc3a2bSChandler Carruth /// inlining has the given attribute set either at the call site or the 2067adc3a2bSChandler Carruth /// function declaration. Primarily used to inspect call site specific 2077adc3a2bSChandler Carruth /// attributes since these can be more precise than the ones on the callee 2087adc3a2bSChandler Carruth /// itself. 2097adc3a2bSChandler Carruth bool paramHasAttr(Argument *A, Attribute::AttrKind Attr); 2107adc3a2bSChandler Carruth 2117adc3a2bSChandler Carruth /// Return true if the given value is known non null within the callee if 2127adc3a2bSChandler Carruth /// inlined through this particular callsite. 2137adc3a2bSChandler Carruth bool isKnownNonNullInCallee(Value *V); 2147adc3a2bSChandler Carruth 215f4bb2f0dSEaswaran Raman /// Update Threshold based on callsite properties such as callee 216f4bb2f0dSEaswaran Raman /// attributes and callee hotness for PGO builds. The Callee is explicitly 217f4bb2f0dSEaswaran Raman /// passed to support analyzing indirect calls whose target is inferred by 218f4bb2f0dSEaswaran Raman /// analysis. 219f4bb2f0dSEaswaran Raman void updateThreshold(CallSite CS, Function &Callee); 220f4bb2f0dSEaswaran Raman 2219a3fc17aSEaswaran Raman /// Return true if size growth is allowed when inlining the callee at CS. 2229a3fc17aSEaswaran Raman bool allowSizeGrowth(CallSite CS); 2239a3fc17aSEaswaran Raman 224c5fa6358SEaswaran Raman /// Return true if \p CS is a cold callsite. 225c5fa6358SEaswaran Raman bool isColdCallSite(CallSite CS, BlockFrequencyInfo *CallerBFI); 226c5fa6358SEaswaran Raman 227974d4eeaSEaswaran Raman /// Return a higher threshold if \p CS is a hot callsite. 228974d4eeaSEaswaran Raman Optional<int> getHotCallSiteThreshold(CallSite CS, 229974d4eeaSEaswaran Raman BlockFrequencyInfo *CallerBFI); 230974d4eeaSEaswaran Raman 2317adc3a2bSChandler Carruth // Custom analysis routines. 232c0aa4b75SDavid Bolvansky InlineResult analyzeBlock(BasicBlock *BB, 233c0aa4b75SDavid Bolvansky SmallPtrSetImpl<const Value *> &EphValues); 2347adc3a2bSChandler Carruth 2357adc3a2bSChandler Carruth // Disable several entry points to the visitor so we don't accidentally use 2367adc3a2bSChandler Carruth // them by declaring but not defining them here. 237567556aaSChad Rosier void visit(Module *); 238567556aaSChad Rosier void visit(Module &); 239567556aaSChad Rosier void visit(Function *); 240567556aaSChad Rosier void visit(Function &); 241567556aaSChad Rosier void visit(BasicBlock *); 242567556aaSChad Rosier void visit(BasicBlock &); 2437adc3a2bSChandler Carruth 2447adc3a2bSChandler Carruth // Provide base case for our instruction visit. 2457adc3a2bSChandler Carruth bool visitInstruction(Instruction &I); 2467adc3a2bSChandler Carruth 2477adc3a2bSChandler Carruth // Our visit overrides. 2487adc3a2bSChandler Carruth bool visitAlloca(AllocaInst &I); 2497adc3a2bSChandler Carruth bool visitPHI(PHINode &I); 2507adc3a2bSChandler Carruth bool visitGetElementPtr(GetElementPtrInst &I); 2517adc3a2bSChandler Carruth bool visitBitCast(BitCastInst &I); 2527adc3a2bSChandler Carruth bool visitPtrToInt(PtrToIntInst &I); 2537adc3a2bSChandler Carruth bool visitIntToPtr(IntToPtrInst &I); 2547adc3a2bSChandler Carruth bool visitCastInst(CastInst &I); 2557adc3a2bSChandler Carruth bool visitUnaryInstruction(UnaryInstruction &I); 2567adc3a2bSChandler Carruth bool visitCmpInst(CmpInst &I); 2577adc3a2bSChandler Carruth bool visitSub(BinaryOperator &I); 2587adc3a2bSChandler Carruth bool visitBinaryOperator(BinaryOperator &I); 2597adc3a2bSChandler Carruth bool visitLoad(LoadInst &I); 2607adc3a2bSChandler Carruth bool visitStore(StoreInst &I); 2617adc3a2bSChandler Carruth bool visitExtractValue(ExtractValueInst &I); 2627adc3a2bSChandler Carruth bool visitInsertValue(InsertValueInst &I); 2637adc3a2bSChandler Carruth bool visitCallSite(CallSite CS); 2647adc3a2bSChandler Carruth bool visitReturnInst(ReturnInst &RI); 2657adc3a2bSChandler Carruth bool visitBranchInst(BranchInst &BI); 2663ec848bcSHaicheng Wu bool visitSelectInst(SelectInst &SI); 2677adc3a2bSChandler Carruth bool visitSwitchInst(SwitchInst &SI); 2687adc3a2bSChandler Carruth bool visitIndirectBrInst(IndirectBrInst &IBI); 2697adc3a2bSChandler Carruth bool visitResumeInst(ResumeInst &RI); 2707adc3a2bSChandler Carruth bool visitCleanupReturnInst(CleanupReturnInst &RI); 2717adc3a2bSChandler Carruth bool visitCatchReturnInst(CatchReturnInst &RI); 2727adc3a2bSChandler Carruth bool visitUnreachableInst(UnreachableInst &I); 2737adc3a2bSChandler Carruth 2747adc3a2bSChandler Carruth public: 275ab6a6837SSean Silva CallAnalyzer(const TargetTransformInfo &TTI, 276aec2fa35SDaniel Jasper std::function<AssumptionCache &(Function &)> &GetAssumptionCache, 27712585b01SEaswaran Raman Optional<function_ref<BlockFrequencyInfo &(Function &)>> &GetBFI, 2780812c5beSHaicheng Wu ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE, 2790812c5beSHaicheng Wu Function &Callee, CallSite CSArg, const InlineParams &Params) 28012585b01SEaswaran Raman : TTI(TTI), GetAssumptionCache(GetAssumptionCache), GetBFI(GetBFI), 2810812c5beSHaicheng Wu PSI(PSI), F(Callee), DL(F.getParent()->getDataLayout()), ORE(ORE), 28285be8ca8SEric Christopher CandidateCS(CSArg), Params(Params), Threshold(Params.DefaultThreshold), 2834924bb00SEaswaran Raman Cost(0), ComputeFullInlineCost(OptComputeFullInlineCost || 2844924bb00SEaswaran Raman Params.ComputeFullInlineCost || ORE), 2854924bb00SEaswaran Raman IsCallerRecursive(false), IsRecursiveCall(false), 28685be8ca8SEric Christopher ExposesReturnsTwice(false), HasDynamicAlloca(false), 28785be8ca8SEric Christopher ContainsNoDuplicateCall(false), HasReturn(false), HasIndirectBr(false), 28877beee41SSameer AbuAsal HasUninlineableIntrinsic(false), InitsVargArgs(false), AllocatedSize(0), 2894296ea72SVitaly Buka NumInstructions(0), NumVectorInstructions(0), VectorBonus(0), 2904296ea72SVitaly Buka SingleBBBonus(0), EnableLoadElimination(true), LoadEliminationCost(0), 2914296ea72SVitaly Buka NumConstantArgs(0), NumConstantOffsetPtrArgs(0), NumAllocaArgs(0), 2924296ea72SVitaly Buka NumConstantPtrCmps(0), NumConstantPtrDiffs(0), 2934296ea72SVitaly Buka NumInstructionsSimplified(0), SROACostSavings(0), 2944296ea72SVitaly Buka SROACostSavingsLost(0) {} 2957adc3a2bSChandler Carruth 296c0aa4b75SDavid Bolvansky InlineResult analyzeCall(CallSite CS); 2977adc3a2bSChandler Carruth 2987adc3a2bSChandler Carruth int getThreshold() { return Threshold; } 2997adc3a2bSChandler Carruth int getCost() { return Cost; } 3007adc3a2bSChandler Carruth 3017adc3a2bSChandler Carruth // Keep a bunch of stats about the cost savings found so we can print them 3027adc3a2bSChandler Carruth // out when debugging. 3037adc3a2bSChandler Carruth unsigned NumConstantArgs; 3047adc3a2bSChandler Carruth unsigned NumConstantOffsetPtrArgs; 3057adc3a2bSChandler Carruth unsigned NumAllocaArgs; 3067adc3a2bSChandler Carruth unsigned NumConstantPtrCmps; 3077adc3a2bSChandler Carruth unsigned NumConstantPtrDiffs; 3087adc3a2bSChandler Carruth unsigned NumInstructionsSimplified; 3097adc3a2bSChandler Carruth unsigned SROACostSavings; 3107adc3a2bSChandler Carruth unsigned SROACostSavingsLost; 3117adc3a2bSChandler Carruth 3127adc3a2bSChandler Carruth void dump(); 3137adc3a2bSChandler Carruth }; 3147adc3a2bSChandler Carruth 3157adc3a2bSChandler Carruth } // namespace 3167adc3a2bSChandler Carruth 3175f8f34e4SAdrian Prantl /// Test whether the given value is an Alloca-derived function argument. 3187adc3a2bSChandler Carruth bool CallAnalyzer::isAllocaDerivedArg(Value *V) { 3197adc3a2bSChandler Carruth return SROAArgValues.count(V); 3207adc3a2bSChandler Carruth } 3217adc3a2bSChandler Carruth 3225f8f34e4SAdrian Prantl /// Lookup the SROA-candidate argument and cost iterator which V maps to. 3237adc3a2bSChandler Carruth /// Returns false if V does not map to a SROA-candidate. 3247adc3a2bSChandler Carruth bool CallAnalyzer::lookupSROAArgAndCost( 3257adc3a2bSChandler Carruth Value *V, Value *&Arg, DenseMap<Value *, int>::iterator &CostIt) { 3267adc3a2bSChandler Carruth if (SROAArgValues.empty() || SROAArgCosts.empty()) 3277adc3a2bSChandler Carruth return false; 3287adc3a2bSChandler Carruth 3297adc3a2bSChandler Carruth DenseMap<Value *, Value *>::iterator ArgIt = SROAArgValues.find(V); 3307adc3a2bSChandler Carruth if (ArgIt == SROAArgValues.end()) 3317adc3a2bSChandler Carruth return false; 3327adc3a2bSChandler Carruth 3337adc3a2bSChandler Carruth Arg = ArgIt->second; 3347adc3a2bSChandler Carruth CostIt = SROAArgCosts.find(Arg); 3357adc3a2bSChandler Carruth return CostIt != SROAArgCosts.end(); 3367adc3a2bSChandler Carruth } 3377adc3a2bSChandler Carruth 3385f8f34e4SAdrian Prantl /// Disable SROA for the candidate marked by this cost iterator. 3397adc3a2bSChandler Carruth /// 3407adc3a2bSChandler Carruth /// This marks the candidate as no longer viable for SROA, and adds the cost 3417adc3a2bSChandler Carruth /// savings associated with it back into the inline cost measurement. 3427adc3a2bSChandler Carruth void CallAnalyzer::disableSROA(DenseMap<Value *, int>::iterator CostIt) { 3437adc3a2bSChandler Carruth // If we're no longer able to perform SROA we need to undo its cost savings 3447adc3a2bSChandler Carruth // and prevent subsequent analysis. 3457adc3a2bSChandler Carruth Cost += CostIt->second; 3467adc3a2bSChandler Carruth SROACostSavings -= CostIt->second; 3477adc3a2bSChandler Carruth SROACostSavingsLost += CostIt->second; 3487adc3a2bSChandler Carruth SROAArgCosts.erase(CostIt); 349a4461515SHaicheng Wu disableLoadElimination(); 3507adc3a2bSChandler Carruth } 3517adc3a2bSChandler Carruth 3525f8f34e4SAdrian Prantl /// If 'V' maps to a SROA candidate, disable SROA for it. 3537adc3a2bSChandler Carruth void CallAnalyzer::disableSROA(Value *V) { 3547adc3a2bSChandler Carruth Value *SROAArg; 3557adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 3567adc3a2bSChandler Carruth if (lookupSROAArgAndCost(V, SROAArg, CostIt)) 3577adc3a2bSChandler Carruth disableSROA(CostIt); 3587adc3a2bSChandler Carruth } 3597adc3a2bSChandler Carruth 3605f8f34e4SAdrian Prantl /// Accumulate the given cost for a particular SROA candidate. 3617adc3a2bSChandler Carruth void CallAnalyzer::accumulateSROACost(DenseMap<Value *, int>::iterator CostIt, 3627adc3a2bSChandler Carruth int InstructionCost) { 3637adc3a2bSChandler Carruth CostIt->second += InstructionCost; 3647adc3a2bSChandler Carruth SROACostSavings += InstructionCost; 3657adc3a2bSChandler Carruth } 3667adc3a2bSChandler Carruth 367a4461515SHaicheng Wu void CallAnalyzer::disableLoadElimination() { 368a4461515SHaicheng Wu if (EnableLoadElimination) { 369a4461515SHaicheng Wu Cost += LoadEliminationCost; 370b3689cabSHaicheng Wu LoadEliminationCost = 0; 371a4461515SHaicheng Wu EnableLoadElimination = false; 372a4461515SHaicheng Wu } 373a4461515SHaicheng Wu } 374a4461515SHaicheng Wu 3755f8f34e4SAdrian Prantl /// Accumulate a constant GEP offset into an APInt if possible. 3767adc3a2bSChandler Carruth /// 3777adc3a2bSChandler Carruth /// Returns false if unable to compute the offset for any reason. Respects any 3787adc3a2bSChandler Carruth /// simplified values known during the analysis of this callsite. 3797adc3a2bSChandler Carruth bool CallAnalyzer::accumulateGEPOffset(GEPOperator &GEP, APInt &Offset) { 380945b7e5aSElena Demikhovsky unsigned IntPtrWidth = DL.getIndexTypeSizeInBits(GEP.getType()); 3817adc3a2bSChandler Carruth assert(IntPtrWidth == Offset.getBitWidth()); 3827adc3a2bSChandler Carruth 3837adc3a2bSChandler Carruth for (gep_type_iterator GTI = gep_type_begin(GEP), GTE = gep_type_end(GEP); 3847adc3a2bSChandler Carruth GTI != GTE; ++GTI) { 3857adc3a2bSChandler Carruth ConstantInt *OpC = dyn_cast<ConstantInt>(GTI.getOperand()); 3867adc3a2bSChandler Carruth if (!OpC) 3877adc3a2bSChandler Carruth if (Constant *SimpleOp = SimplifiedValues.lookup(GTI.getOperand())) 3887adc3a2bSChandler Carruth OpC = dyn_cast<ConstantInt>(SimpleOp); 3897adc3a2bSChandler Carruth if (!OpC) 3907adc3a2bSChandler Carruth return false; 391567556aaSChad Rosier if (OpC->isZero()) 392567556aaSChad Rosier continue; 3937adc3a2bSChandler Carruth 3947adc3a2bSChandler Carruth // Handle a struct index, which adds its field offset to the pointer. 395ab85225bSPeter Collingbourne if (StructType *STy = GTI.getStructTypeOrNull()) { 3967adc3a2bSChandler Carruth unsigned ElementIdx = OpC->getZExtValue(); 3977adc3a2bSChandler Carruth const StructLayout *SL = DL.getStructLayout(STy); 3987adc3a2bSChandler Carruth Offset += APInt(IntPtrWidth, SL->getElementOffset(ElementIdx)); 3997adc3a2bSChandler Carruth continue; 4007adc3a2bSChandler Carruth } 4017adc3a2bSChandler Carruth 4027adc3a2bSChandler Carruth APInt TypeSize(IntPtrWidth, DL.getTypeAllocSize(GTI.getIndexedType())); 4037adc3a2bSChandler Carruth Offset += OpC->getValue().sextOrTrunc(IntPtrWidth) * TypeSize; 4047adc3a2bSChandler Carruth } 4057adc3a2bSChandler Carruth return true; 4067adc3a2bSChandler Carruth } 4077adc3a2bSChandler Carruth 4085f8f34e4SAdrian Prantl /// Use TTI to check whether a GEP is free. 409201b191bSHaicheng Wu /// 410201b191bSHaicheng Wu /// Respects any simplified values known during the analysis of this callsite. 411201b191bSHaicheng Wu bool CallAnalyzer::isGEPFree(GetElementPtrInst &GEP) { 41261c1bd5aSEvgeny Astigeevich SmallVector<Value *, 4> Operands; 41361c1bd5aSEvgeny Astigeevich Operands.push_back(GEP.getOperand(0)); 414201b191bSHaicheng Wu for (User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end(); I != E; ++I) 415201b191bSHaicheng Wu if (Constant *SimpleOp = SimplifiedValues.lookup(*I)) 41661c1bd5aSEvgeny Astigeevich Operands.push_back(SimpleOp); 417201b191bSHaicheng Wu else 41861c1bd5aSEvgeny Astigeevich Operands.push_back(*I); 41961c1bd5aSEvgeny Astigeevich return TargetTransformInfo::TCC_Free == TTI.getUserCost(&GEP, Operands); 420201b191bSHaicheng Wu } 421201b191bSHaicheng Wu 4227adc3a2bSChandler Carruth bool CallAnalyzer::visitAlloca(AllocaInst &I) { 4237adc3a2bSChandler Carruth // Check whether inlining will turn a dynamic alloca into a static 4240f153424SSanjay Patel // alloca and handle that case. 4257adc3a2bSChandler Carruth if (I.isArrayAllocation()) { 4260f153424SSanjay Patel Constant *Size = SimplifiedValues.lookup(I.getArraySize()); 4270f153424SSanjay Patel if (auto *AllocSize = dyn_cast_or_null<ConstantInt>(Size)) { 4287adc3a2bSChandler Carruth Type *Ty = I.getAllocatedType(); 42922eb80a1SEaswaran Raman AllocatedSize = SaturatingMultiplyAdd( 43022eb80a1SEaswaran Raman AllocSize->getLimitedValue(), DL.getTypeAllocSize(Ty), AllocatedSize); 4317adc3a2bSChandler Carruth return Base::visitAlloca(I); 4327adc3a2bSChandler Carruth } 4337adc3a2bSChandler Carruth } 4347adc3a2bSChandler Carruth 4357adc3a2bSChandler Carruth // Accumulate the allocated size. 4367adc3a2bSChandler Carruth if (I.isStaticAlloca()) { 4377adc3a2bSChandler Carruth Type *Ty = I.getAllocatedType(); 43822eb80a1SEaswaran Raman AllocatedSize = SaturatingAdd(DL.getTypeAllocSize(Ty), AllocatedSize); 4397adc3a2bSChandler Carruth } 4407adc3a2bSChandler Carruth 4417adc3a2bSChandler Carruth // We will happily inline static alloca instructions. 4427adc3a2bSChandler Carruth if (I.isStaticAlloca()) 4437adc3a2bSChandler Carruth return Base::visitAlloca(I); 4447adc3a2bSChandler Carruth 4457adc3a2bSChandler Carruth // FIXME: This is overly conservative. Dynamic allocas are inefficient for 4467adc3a2bSChandler Carruth // a variety of reasons, and so we would like to not inline them into 4477adc3a2bSChandler Carruth // functions which don't currently have a dynamic alloca. This simply 4487adc3a2bSChandler Carruth // disables inlining altogether in the presence of a dynamic alloca. 4497adc3a2bSChandler Carruth HasDynamicAlloca = true; 4507adc3a2bSChandler Carruth return false; 4517adc3a2bSChandler Carruth } 4527adc3a2bSChandler Carruth 4537adc3a2bSChandler Carruth bool CallAnalyzer::visitPHI(PHINode &I) { 4547adc3a2bSChandler Carruth // FIXME: We need to propagate SROA *disabling* through phi nodes, even 4557adc3a2bSChandler Carruth // though we don't want to propagate it's bonuses. The idea is to disable 4567adc3a2bSChandler Carruth // SROA if it *might* be used in an inappropriate manner. 4577adc3a2bSChandler Carruth 4587adc3a2bSChandler Carruth // Phi nodes are always zero-cost. 45977f32994SBjorn Pettersson // FIXME: Pointer sizes may differ between different address spaces, so do we 46077f32994SBjorn Pettersson // need to use correct address space in the call to getPointerSizeInBits here? 46177f32994SBjorn Pettersson // Or could we skip the getPointerSizeInBits call completely? As far as I can 46277f32994SBjorn Pettersson // see the ZeroOffset is used as a dummy value, so we can probably use any 46377f32994SBjorn Pettersson // bit width for the ZeroOffset? 46477f32994SBjorn Pettersson APInt ZeroOffset = APInt::getNullValue(DL.getPointerSizeInBits(0)); 4653739e14aSHaicheng Wu bool CheckSROA = I.getType()->isPointerTy(); 4663739e14aSHaicheng Wu 4673739e14aSHaicheng Wu // Track the constant or pointer with constant offset we've seen so far. 4683739e14aSHaicheng Wu Constant *FirstC = nullptr; 4693739e14aSHaicheng Wu std::pair<Value *, APInt> FirstBaseAndOffset = {nullptr, ZeroOffset}; 4703739e14aSHaicheng Wu Value *FirstV = nullptr; 4713739e14aSHaicheng Wu 4723739e14aSHaicheng Wu for (unsigned i = 0, e = I.getNumIncomingValues(); i != e; ++i) { 4733739e14aSHaicheng Wu BasicBlock *Pred = I.getIncomingBlock(i); 4743739e14aSHaicheng Wu // If the incoming block is dead, skip the incoming block. 4753739e14aSHaicheng Wu if (DeadBlocks.count(Pred)) 4763739e14aSHaicheng Wu continue; 4773739e14aSHaicheng Wu // If the parent block of phi is not the known successor of the incoming 4783739e14aSHaicheng Wu // block, skip the incoming block. 4793739e14aSHaicheng Wu BasicBlock *KnownSuccessor = KnownSuccessors[Pred]; 4803739e14aSHaicheng Wu if (KnownSuccessor && KnownSuccessor != I.getParent()) 4813739e14aSHaicheng Wu continue; 4823739e14aSHaicheng Wu 4833739e14aSHaicheng Wu Value *V = I.getIncomingValue(i); 4843739e14aSHaicheng Wu // If the incoming value is this phi itself, skip the incoming value. 4853739e14aSHaicheng Wu if (&I == V) 4863739e14aSHaicheng Wu continue; 4873739e14aSHaicheng Wu 4883739e14aSHaicheng Wu Constant *C = dyn_cast<Constant>(V); 4893739e14aSHaicheng Wu if (!C) 4903739e14aSHaicheng Wu C = SimplifiedValues.lookup(V); 4913739e14aSHaicheng Wu 4923739e14aSHaicheng Wu std::pair<Value *, APInt> BaseAndOffset = {nullptr, ZeroOffset}; 4933739e14aSHaicheng Wu if (!C && CheckSROA) 4943739e14aSHaicheng Wu BaseAndOffset = ConstantOffsetPtrs.lookup(V); 4953739e14aSHaicheng Wu 4963739e14aSHaicheng Wu if (!C && !BaseAndOffset.first) 4973739e14aSHaicheng Wu // The incoming value is neither a constant nor a pointer with constant 4983739e14aSHaicheng Wu // offset, exit early. 4993739e14aSHaicheng Wu return true; 5003739e14aSHaicheng Wu 5013739e14aSHaicheng Wu if (FirstC) { 5023739e14aSHaicheng Wu if (FirstC == C) 5033739e14aSHaicheng Wu // If we've seen a constant incoming value before and it is the same 5043739e14aSHaicheng Wu // constant we see this time, continue checking the next incoming value. 5053739e14aSHaicheng Wu continue; 5063739e14aSHaicheng Wu // Otherwise early exit because we either see a different constant or saw 5073739e14aSHaicheng Wu // a constant before but we have a pointer with constant offset this time. 5083739e14aSHaicheng Wu return true; 5093739e14aSHaicheng Wu } 5103739e14aSHaicheng Wu 5113739e14aSHaicheng Wu if (FirstV) { 5123739e14aSHaicheng Wu // The same logic as above, but check pointer with constant offset here. 5133739e14aSHaicheng Wu if (FirstBaseAndOffset == BaseAndOffset) 5143739e14aSHaicheng Wu continue; 5153739e14aSHaicheng Wu return true; 5163739e14aSHaicheng Wu } 5173739e14aSHaicheng Wu 5183739e14aSHaicheng Wu if (C) { 5193739e14aSHaicheng Wu // This is the 1st time we've seen a constant, record it. 5203739e14aSHaicheng Wu FirstC = C; 5213739e14aSHaicheng Wu continue; 5223739e14aSHaicheng Wu } 5233739e14aSHaicheng Wu 5243739e14aSHaicheng Wu // The remaining case is that this is the 1st time we've seen a pointer with 5253739e14aSHaicheng Wu // constant offset, record it. 5263739e14aSHaicheng Wu FirstV = V; 5273739e14aSHaicheng Wu FirstBaseAndOffset = BaseAndOffset; 5283739e14aSHaicheng Wu } 5293739e14aSHaicheng Wu 5303739e14aSHaicheng Wu // Check if we can map phi to a constant. 5313739e14aSHaicheng Wu if (FirstC) { 5323739e14aSHaicheng Wu SimplifiedValues[&I] = FirstC; 5333739e14aSHaicheng Wu return true; 5343739e14aSHaicheng Wu } 5353739e14aSHaicheng Wu 5363739e14aSHaicheng Wu // Check if we can map phi to a pointer with constant offset. 5373739e14aSHaicheng Wu if (FirstBaseAndOffset.first) { 5383739e14aSHaicheng Wu ConstantOffsetPtrs[&I] = FirstBaseAndOffset; 5393739e14aSHaicheng Wu 5403739e14aSHaicheng Wu Value *SROAArg; 5413739e14aSHaicheng Wu DenseMap<Value *, int>::iterator CostIt; 5423739e14aSHaicheng Wu if (lookupSROAArgAndCost(FirstV, SROAArg, CostIt)) 5433739e14aSHaicheng Wu SROAArgValues[&I] = SROAArg; 5443739e14aSHaicheng Wu } 5453739e14aSHaicheng Wu 5467adc3a2bSChandler Carruth return true; 5477adc3a2bSChandler Carruth } 5487adc3a2bSChandler Carruth 5495f8f34e4SAdrian Prantl /// Check we can fold GEPs of constant-offset call site argument pointers. 550d3558b5dSEvgeny Astigeevich /// This requires target data and inbounds GEPs. 551d3558b5dSEvgeny Astigeevich /// 552d3558b5dSEvgeny Astigeevich /// \return true if the specified GEP can be folded. 553d3558b5dSEvgeny Astigeevich bool CallAnalyzer::canFoldInboundsGEP(GetElementPtrInst &I) { 554d3558b5dSEvgeny Astigeevich // Check if we have a base + offset for the pointer. 555d3558b5dSEvgeny Astigeevich std::pair<Value *, APInt> BaseAndOffset = 556d3558b5dSEvgeny Astigeevich ConstantOffsetPtrs.lookup(I.getPointerOperand()); 557d3558b5dSEvgeny Astigeevich if (!BaseAndOffset.first) 558d3558b5dSEvgeny Astigeevich return false; 559d3558b5dSEvgeny Astigeevich 560d3558b5dSEvgeny Astigeevich // Check if the offset of this GEP is constant, and if so accumulate it 561d3558b5dSEvgeny Astigeevich // into Offset. 562d3558b5dSEvgeny Astigeevich if (!accumulateGEPOffset(cast<GEPOperator>(I), BaseAndOffset.second)) 563d3558b5dSEvgeny Astigeevich return false; 564d3558b5dSEvgeny Astigeevich 565d3558b5dSEvgeny Astigeevich // Add the result as a new mapping to Base + Offset. 566d3558b5dSEvgeny Astigeevich ConstantOffsetPtrs[&I] = BaseAndOffset; 567d3558b5dSEvgeny Astigeevich 568d3558b5dSEvgeny Astigeevich return true; 569d3558b5dSEvgeny Astigeevich } 570d3558b5dSEvgeny Astigeevich 5717adc3a2bSChandler Carruth bool CallAnalyzer::visitGetElementPtr(GetElementPtrInst &I) { 5727adc3a2bSChandler Carruth Value *SROAArg; 5737adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 574567556aaSChad Rosier bool SROACandidate = 575567556aaSChad Rosier lookupSROAArgAndCost(I.getPointerOperand(), SROAArg, CostIt); 5767adc3a2bSChandler Carruth 577a8b9cdc9SEaswaran Raman // Lambda to check whether a GEP's indices are all constant. 578a8b9cdc9SEaswaran Raman auto IsGEPOffsetConstant = [&](GetElementPtrInst &GEP) { 579a8b9cdc9SEaswaran Raman for (User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end(); I != E; ++I) 580a8b9cdc9SEaswaran Raman if (!isa<Constant>(*I) && !SimplifiedValues.lookup(*I)) 581a8b9cdc9SEaswaran Raman return false; 582a8b9cdc9SEaswaran Raman return true; 583a8b9cdc9SEaswaran Raman }; 584a8b9cdc9SEaswaran Raman 585d3558b5dSEvgeny Astigeevich if ((I.isInBounds() && canFoldInboundsGEP(I)) || IsGEPOffsetConstant(I)) { 5867adc3a2bSChandler Carruth if (SROACandidate) 5877adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 5887adc3a2bSChandler Carruth 5897adc3a2bSChandler Carruth // Constant GEPs are modeled as free. 5907adc3a2bSChandler Carruth return true; 5917adc3a2bSChandler Carruth } 5927adc3a2bSChandler Carruth 5937adc3a2bSChandler Carruth // Variable GEPs will require math and will disable SROA. 5947adc3a2bSChandler Carruth if (SROACandidate) 5957adc3a2bSChandler Carruth disableSROA(CostIt); 596201b191bSHaicheng Wu return isGEPFree(I); 5977adc3a2bSChandler Carruth } 5987adc3a2bSChandler Carruth 599617f6364SEaswaran Raman /// Simplify \p I if its operands are constants and update SimplifiedValues. 600617f6364SEaswaran Raman /// \p Evaluate is a callable specific to instruction type that evaluates the 601617f6364SEaswaran Raman /// instruction when all the operands are constants. 602617f6364SEaswaran Raman template <typename Callable> 603617f6364SEaswaran Raman bool CallAnalyzer::simplifyInstruction(Instruction &I, Callable Evaluate) { 604617f6364SEaswaran Raman SmallVector<Constant *, 2> COps; 605617f6364SEaswaran Raman for (Value *Op : I.operands()) { 606617f6364SEaswaran Raman Constant *COp = dyn_cast<Constant>(Op); 6077adc3a2bSChandler Carruth if (!COp) 608617f6364SEaswaran Raman COp = SimplifiedValues.lookup(Op); 609617f6364SEaswaran Raman if (!COp) 610617f6364SEaswaran Raman return false; 611617f6364SEaswaran Raman COps.push_back(COp); 612617f6364SEaswaran Raman } 613617f6364SEaswaran Raman auto *C = Evaluate(COps); 614617f6364SEaswaran Raman if (!C) 615617f6364SEaswaran Raman return false; 6167adc3a2bSChandler Carruth SimplifiedValues[&I] = C; 6177adc3a2bSChandler Carruth return true; 6187adc3a2bSChandler Carruth } 6197adc3a2bSChandler Carruth 620617f6364SEaswaran Raman bool CallAnalyzer::visitBitCast(BitCastInst &I) { 621617f6364SEaswaran Raman // Propagate constants through bitcasts. 622617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 623617f6364SEaswaran Raman return ConstantExpr::getBitCast(COps[0], I.getType()); 624617f6364SEaswaran Raman })) 625617f6364SEaswaran Raman return true; 626617f6364SEaswaran Raman 6277adc3a2bSChandler Carruth // Track base/offsets through casts 628567556aaSChad Rosier std::pair<Value *, APInt> BaseAndOffset = 629567556aaSChad Rosier ConstantOffsetPtrs.lookup(I.getOperand(0)); 6307adc3a2bSChandler Carruth // Casts don't change the offset, just wrap it up. 6317adc3a2bSChandler Carruth if (BaseAndOffset.first) 6327adc3a2bSChandler Carruth ConstantOffsetPtrs[&I] = BaseAndOffset; 6337adc3a2bSChandler Carruth 6347adc3a2bSChandler Carruth // Also look for SROA candidates here. 6357adc3a2bSChandler Carruth Value *SROAArg; 6367adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 6377adc3a2bSChandler Carruth if (lookupSROAArgAndCost(I.getOperand(0), SROAArg, CostIt)) 6387adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 6397adc3a2bSChandler Carruth 6407adc3a2bSChandler Carruth // Bitcasts are always zero cost. 6417adc3a2bSChandler Carruth return true; 6427adc3a2bSChandler Carruth } 6437adc3a2bSChandler Carruth 6447adc3a2bSChandler Carruth bool CallAnalyzer::visitPtrToInt(PtrToIntInst &I) { 6457adc3a2bSChandler Carruth // Propagate constants through ptrtoint. 646617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 647617f6364SEaswaran Raman return ConstantExpr::getPtrToInt(COps[0], I.getType()); 648617f6364SEaswaran Raman })) 6497adc3a2bSChandler Carruth return true; 6507adc3a2bSChandler Carruth 6517adc3a2bSChandler Carruth // Track base/offset pairs when converted to a plain integer provided the 6527adc3a2bSChandler Carruth // integer is large enough to represent the pointer. 6537adc3a2bSChandler Carruth unsigned IntegerSize = I.getType()->getScalarSizeInBits(); 65477f32994SBjorn Pettersson unsigned AS = I.getOperand(0)->getType()->getPointerAddressSpace(); 65577f32994SBjorn Pettersson if (IntegerSize >= DL.getPointerSizeInBits(AS)) { 656567556aaSChad Rosier std::pair<Value *, APInt> BaseAndOffset = 657567556aaSChad Rosier ConstantOffsetPtrs.lookup(I.getOperand(0)); 6587adc3a2bSChandler Carruth if (BaseAndOffset.first) 6597adc3a2bSChandler Carruth ConstantOffsetPtrs[&I] = BaseAndOffset; 6607adc3a2bSChandler Carruth } 6617adc3a2bSChandler Carruth 6627adc3a2bSChandler Carruth // This is really weird. Technically, ptrtoint will disable SROA. However, 6637adc3a2bSChandler Carruth // unless that ptrtoint is *used* somewhere in the live basic blocks after 6647adc3a2bSChandler Carruth // inlining, it will be nuked, and SROA should proceed. All of the uses which 6657adc3a2bSChandler Carruth // would block SROA would also block SROA if applied directly to a pointer, 6667adc3a2bSChandler Carruth // and so we can just add the integer in here. The only places where SROA is 6677adc3a2bSChandler Carruth // preserved either cannot fire on an integer, or won't in-and-of themselves 6687adc3a2bSChandler Carruth // disable SROA (ext) w/o some later use that we would see and disable. 6697adc3a2bSChandler Carruth Value *SROAArg; 6707adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 6717adc3a2bSChandler Carruth if (lookupSROAArgAndCost(I.getOperand(0), SROAArg, CostIt)) 6727adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 6737adc3a2bSChandler Carruth 6747adc3a2bSChandler Carruth return TargetTransformInfo::TCC_Free == TTI.getUserCost(&I); 6757adc3a2bSChandler Carruth } 6767adc3a2bSChandler Carruth 6777adc3a2bSChandler Carruth bool CallAnalyzer::visitIntToPtr(IntToPtrInst &I) { 6787adc3a2bSChandler Carruth // Propagate constants through ptrtoint. 679617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 680617f6364SEaswaran Raman return ConstantExpr::getIntToPtr(COps[0], I.getType()); 681617f6364SEaswaran Raman })) 6827adc3a2bSChandler Carruth return true; 6837adc3a2bSChandler Carruth 6847adc3a2bSChandler Carruth // Track base/offset pairs when round-tripped through a pointer without 6857adc3a2bSChandler Carruth // modifications provided the integer is not too large. 6867adc3a2bSChandler Carruth Value *Op = I.getOperand(0); 6877adc3a2bSChandler Carruth unsigned IntegerSize = Op->getType()->getScalarSizeInBits(); 68877f32994SBjorn Pettersson if (IntegerSize <= DL.getPointerTypeSizeInBits(I.getType())) { 6897adc3a2bSChandler Carruth std::pair<Value *, APInt> BaseAndOffset = ConstantOffsetPtrs.lookup(Op); 6907adc3a2bSChandler Carruth if (BaseAndOffset.first) 6917adc3a2bSChandler Carruth ConstantOffsetPtrs[&I] = BaseAndOffset; 6927adc3a2bSChandler Carruth } 6937adc3a2bSChandler Carruth 6947adc3a2bSChandler Carruth // "Propagate" SROA here in the same manner as we do for ptrtoint above. 6957adc3a2bSChandler Carruth Value *SROAArg; 6967adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 6977adc3a2bSChandler Carruth if (lookupSROAArgAndCost(Op, SROAArg, CostIt)) 6987adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 6997adc3a2bSChandler Carruth 7007adc3a2bSChandler Carruth return TargetTransformInfo::TCC_Free == TTI.getUserCost(&I); 7017adc3a2bSChandler Carruth } 7027adc3a2bSChandler Carruth 7037adc3a2bSChandler Carruth bool CallAnalyzer::visitCastInst(CastInst &I) { 7047adc3a2bSChandler Carruth // Propagate constants through ptrtoint. 705617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 706617f6364SEaswaran Raman return ConstantExpr::getCast(I.getOpcode(), COps[0], I.getType()); 707617f6364SEaswaran Raman })) 7087adc3a2bSChandler Carruth return true; 7097adc3a2bSChandler Carruth 7107adc3a2bSChandler Carruth // Disable SROA in the face of arbitrary casts we don't whitelist elsewhere. 7117adc3a2bSChandler Carruth disableSROA(I.getOperand(0)); 7127adc3a2bSChandler Carruth 71339ed9a60SEli Friedman // If this is a floating-point cast, and the target says this operation 71439ed9a60SEli Friedman // is expensive, this may eventually become a library call. Treat the cost 71539ed9a60SEli Friedman // as such. 71639ed9a60SEli Friedman switch (I.getOpcode()) { 71739ed9a60SEli Friedman case Instruction::FPTrunc: 71839ed9a60SEli Friedman case Instruction::FPExt: 71939ed9a60SEli Friedman case Instruction::UIToFP: 72039ed9a60SEli Friedman case Instruction::SIToFP: 72139ed9a60SEli Friedman case Instruction::FPToUI: 72239ed9a60SEli Friedman case Instruction::FPToSI: 72339ed9a60SEli Friedman if (TTI.getFPOpCost(I.getType()) == TargetTransformInfo::TCC_Expensive) 72439ed9a60SEli Friedman Cost += InlineConstants::CallPenalty; 7254dc0b1acSReid Kleckner break; 72639ed9a60SEli Friedman default: 72739ed9a60SEli Friedman break; 72839ed9a60SEli Friedman } 72939ed9a60SEli Friedman 7307adc3a2bSChandler Carruth return TargetTransformInfo::TCC_Free == TTI.getUserCost(&I); 7317adc3a2bSChandler Carruth } 7327adc3a2bSChandler Carruth 7337adc3a2bSChandler Carruth bool CallAnalyzer::visitUnaryInstruction(UnaryInstruction &I) { 7347adc3a2bSChandler Carruth Value *Operand = I.getOperand(0); 735617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 736617f6364SEaswaran Raman return ConstantFoldInstOperands(&I, COps[0], DL); 737617f6364SEaswaran Raman })) 7387adc3a2bSChandler Carruth return true; 7397adc3a2bSChandler Carruth 7407adc3a2bSChandler Carruth // Disable any SROA on the argument to arbitrary unary operators. 7417adc3a2bSChandler Carruth disableSROA(Operand); 7427adc3a2bSChandler Carruth 7437adc3a2bSChandler Carruth return false; 7447adc3a2bSChandler Carruth } 7457adc3a2bSChandler Carruth 7467adc3a2bSChandler Carruth bool CallAnalyzer::paramHasAttr(Argument *A, Attribute::AttrKind Attr) { 747fb502d2fSReid Kleckner return CandidateCS.paramHasAttr(A->getArgNo(), Attr); 7487adc3a2bSChandler Carruth } 7497adc3a2bSChandler Carruth 7507adc3a2bSChandler Carruth bool CallAnalyzer::isKnownNonNullInCallee(Value *V) { 7517adc3a2bSChandler Carruth // Does the *call site* have the NonNull attribute set on an argument? We 7527adc3a2bSChandler Carruth // use the attribute on the call site to memoize any analysis done in the 7537adc3a2bSChandler Carruth // caller. This will also trip if the callee function has a non-null 7547adc3a2bSChandler Carruth // parameter attribute, but that's a less interesting case because hopefully 7557adc3a2bSChandler Carruth // the callee would already have been simplified based on that. 7567adc3a2bSChandler Carruth if (Argument *A = dyn_cast<Argument>(V)) 7577adc3a2bSChandler Carruth if (paramHasAttr(A, Attribute::NonNull)) 7587adc3a2bSChandler Carruth return true; 7597adc3a2bSChandler Carruth 7607adc3a2bSChandler Carruth // Is this an alloca in the caller? This is distinct from the attribute case 7617adc3a2bSChandler Carruth // above because attributes aren't updated within the inliner itself and we 7627adc3a2bSChandler Carruth // always want to catch the alloca derived case. 7637adc3a2bSChandler Carruth if (isAllocaDerivedArg(V)) 7647adc3a2bSChandler Carruth // We can actually predict the result of comparisons between an 7657adc3a2bSChandler Carruth // alloca-derived value and null. Note that this fires regardless of 7667adc3a2bSChandler Carruth // SROA firing. 7677adc3a2bSChandler Carruth return true; 7687adc3a2bSChandler Carruth 7697adc3a2bSChandler Carruth return false; 7707adc3a2bSChandler Carruth } 7717adc3a2bSChandler Carruth 7729a3fc17aSEaswaran Raman bool CallAnalyzer::allowSizeGrowth(CallSite CS) { 7739a3fc17aSEaswaran Raman // If the normal destination of the invoke or the parent block of the call 7749a3fc17aSEaswaran Raman // site is unreachable-terminated, there is little point in inlining this 7759a3fc17aSEaswaran Raman // unless there is literally zero cost. 7769a3fc17aSEaswaran Raman // FIXME: Note that it is possible that an unreachable-terminated block has a 7779a3fc17aSEaswaran Raman // hot entry. For example, in below scenario inlining hot_call_X() may be 7789a3fc17aSEaswaran Raman // beneficial : 7799a3fc17aSEaswaran Raman // main() { 7809a3fc17aSEaswaran Raman // hot_call_1(); 7819a3fc17aSEaswaran Raman // ... 7829a3fc17aSEaswaran Raman // hot_call_N() 7839a3fc17aSEaswaran Raman // exit(0); 7849a3fc17aSEaswaran Raman // } 7859a3fc17aSEaswaran Raman // For now, we are not handling this corner case here as it is rare in real 7869a3fc17aSEaswaran Raman // code. In future, we should elaborate this based on BPI and BFI in more 7879a3fc17aSEaswaran Raman // general threshold adjusting heuristics in updateThreshold(). 7889a3fc17aSEaswaran Raman Instruction *Instr = CS.getInstruction(); 7899a3fc17aSEaswaran Raman if (InvokeInst *II = dyn_cast<InvokeInst>(Instr)) { 7909a3fc17aSEaswaran Raman if (isa<UnreachableInst>(II->getNormalDest()->getTerminator())) 7919a3fc17aSEaswaran Raman return false; 7929a3fc17aSEaswaran Raman } else if (isa<UnreachableInst>(Instr->getParent()->getTerminator())) 7939a3fc17aSEaswaran Raman return false; 7949a3fc17aSEaswaran Raman 7959a3fc17aSEaswaran Raman return true; 7969a3fc17aSEaswaran Raman } 7979a3fc17aSEaswaran Raman 798c5fa6358SEaswaran Raman bool CallAnalyzer::isColdCallSite(CallSite CS, BlockFrequencyInfo *CallerBFI) { 799c5fa6358SEaswaran Raman // If global profile summary is available, then callsite's coldness is 800c5fa6358SEaswaran Raman // determined based on that. 801bba762a1SChandler Carruth if (PSI && PSI->hasProfileSummary()) 802c5fa6358SEaswaran Raman return PSI->isColdCallSite(CS, CallerBFI); 803bba762a1SChandler Carruth 804bba762a1SChandler Carruth // Otherwise we need BFI to be available. 805c5fa6358SEaswaran Raman if (!CallerBFI) 806c5fa6358SEaswaran Raman return false; 807c5fa6358SEaswaran Raman 808bba762a1SChandler Carruth // Determine if the callsite is cold relative to caller's entry. We could 809bba762a1SChandler Carruth // potentially cache the computation of scaled entry frequency, but the added 810bba762a1SChandler Carruth // complexity is not worth it unless this scaling shows up high in the 811bba762a1SChandler Carruth // profiles. 812c5fa6358SEaswaran Raman const BranchProbability ColdProb(ColdCallSiteRelFreq, 100); 813c5fa6358SEaswaran Raman auto CallSiteBB = CS.getInstruction()->getParent(); 814c5fa6358SEaswaran Raman auto CallSiteFreq = CallerBFI->getBlockFreq(CallSiteBB); 815c5fa6358SEaswaran Raman auto CallerEntryFreq = 816c5fa6358SEaswaran Raman CallerBFI->getBlockFreq(&(CS.getCaller()->getEntryBlock())); 817c5fa6358SEaswaran Raman return CallSiteFreq < CallerEntryFreq * ColdProb; 818c5fa6358SEaswaran Raman } 819c5fa6358SEaswaran Raman 820974d4eeaSEaswaran Raman Optional<int> 821974d4eeaSEaswaran Raman CallAnalyzer::getHotCallSiteThreshold(CallSite CS, 822974d4eeaSEaswaran Raman BlockFrequencyInfo *CallerBFI) { 823bba762a1SChandler Carruth 824974d4eeaSEaswaran Raman // If global profile summary is available, then callsite's hotness is 825974d4eeaSEaswaran Raman // determined based on that. 826bba762a1SChandler Carruth if (PSI && PSI->hasProfileSummary() && PSI->isHotCallSite(CS, CallerBFI)) 827bba762a1SChandler Carruth return Params.HotCallSiteThreshold; 828974d4eeaSEaswaran Raman 829bba762a1SChandler Carruth // Otherwise we need BFI to be available and to have a locally hot callsite 830bba762a1SChandler Carruth // threshold. 831bba762a1SChandler Carruth if (!CallerBFI || !Params.LocallyHotCallSiteThreshold) 832974d4eeaSEaswaran Raman return None; 833974d4eeaSEaswaran Raman 834bba762a1SChandler Carruth // Determine if the callsite is hot relative to caller's entry. We could 835bba762a1SChandler Carruth // potentially cache the computation of scaled entry frequency, but the added 836bba762a1SChandler Carruth // complexity is not worth it unless this scaling shows up high in the 837bba762a1SChandler Carruth // profiles. 838974d4eeaSEaswaran Raman auto CallSiteBB = CS.getInstruction()->getParent(); 839974d4eeaSEaswaran Raman auto CallSiteFreq = CallerBFI->getBlockFreq(CallSiteBB).getFrequency(); 840974d4eeaSEaswaran Raman auto CallerEntryFreq = CallerBFI->getEntryFreq(); 841974d4eeaSEaswaran Raman if (CallSiteFreq >= CallerEntryFreq * HotCallSiteRelFreq) 842bba762a1SChandler Carruth return Params.LocallyHotCallSiteThreshold; 843bba762a1SChandler Carruth 844bba762a1SChandler Carruth // Otherwise treat it normally. 845974d4eeaSEaswaran Raman return None; 846974d4eeaSEaswaran Raman } 847974d4eeaSEaswaran Raman 848f4bb2f0dSEaswaran Raman void CallAnalyzer::updateThreshold(CallSite CS, Function &Callee) { 8499a3fc17aSEaswaran Raman // If no size growth is allowed for this inlining, set Threshold to 0. 8509a3fc17aSEaswaran Raman if (!allowSizeGrowth(CS)) { 8519a3fc17aSEaswaran Raman Threshold = 0; 8529a3fc17aSEaswaran Raman return; 8539a3fc17aSEaswaran Raman } 8549a3fc17aSEaswaran Raman 855bb578ef0SEaswaran Raman Function *Caller = CS.getCaller(); 8561c57cc2bSEaswaran Raman 8571c57cc2bSEaswaran Raman // return min(A, B) if B is valid. 8581c57cc2bSEaswaran Raman auto MinIfValid = [](int A, Optional<int> B) { 8591c57cc2bSEaswaran Raman return B ? std::min(A, B.getValue()) : A; 8601c57cc2bSEaswaran Raman }; 8611c57cc2bSEaswaran Raman 8620d58fcacSEaswaran Raman // return max(A, B) if B is valid. 8630d58fcacSEaswaran Raman auto MaxIfValid = [](int A, Optional<int> B) { 8640d58fcacSEaswaran Raman return B ? std::max(A, B.getValue()) : A; 8650d58fcacSEaswaran Raman }; 8660d58fcacSEaswaran Raman 86751b809bfSEaswaran Raman // Various bonus percentages. These are multiplied by Threshold to get the 86851b809bfSEaswaran Raman // bonus values. 86951b809bfSEaswaran Raman // SingleBBBonus: This bonus is applied if the callee has a single reachable 87051b809bfSEaswaran Raman // basic block at the given callsite context. This is speculatively applied 87151b809bfSEaswaran Raman // and withdrawn if more than one basic block is seen. 87251b809bfSEaswaran Raman // 87351b809bfSEaswaran Raman // Vector bonuses: We want to more aggressively inline vector-dense kernels 87451b809bfSEaswaran Raman // and apply this bonus based on the percentage of vector instructions. A 87551b809bfSEaswaran Raman // bonus is applied if the vector instructions exceed 50% and half that amount 87651b809bfSEaswaran Raman // is applied if it exceeds 10%. Note that these bonuses are some what 87751b809bfSEaswaran Raman // arbitrary and evolved over time by accident as much as because they are 87851b809bfSEaswaran Raman // principled bonuses. 87951b809bfSEaswaran Raman // FIXME: It would be nice to base the bonus values on something more 88051b809bfSEaswaran Raman // scientific. 88151b809bfSEaswaran Raman // 88251b809bfSEaswaran Raman // LstCallToStaticBonus: This large bonus is applied to ensure the inlining 88351b809bfSEaswaran Raman // of the last call to a static function as inlining such functions is 88451b809bfSEaswaran Raman // guaranteed to reduce code size. 88551b809bfSEaswaran Raman // 88651b809bfSEaswaran Raman // These bonus percentages may be set to 0 based on properties of the caller 88751b809bfSEaswaran Raman // and the callsite. 88851b809bfSEaswaran Raman int SingleBBBonusPercent = 50; 88951b809bfSEaswaran Raman int VectorBonusPercent = 150; 89051b809bfSEaswaran Raman int LastCallToStaticBonus = InlineConstants::LastCallToStaticBonus; 89151b809bfSEaswaran Raman 89251b809bfSEaswaran Raman // Lambda to set all the above bonus and bonus percentages to 0. 89351b809bfSEaswaran Raman auto DisallowAllBonuses = [&]() { 89451b809bfSEaswaran Raman SingleBBBonusPercent = 0; 89551b809bfSEaswaran Raman VectorBonusPercent = 0; 89651b809bfSEaswaran Raman LastCallToStaticBonus = 0; 89751b809bfSEaswaran Raman }; 89851b809bfSEaswaran Raman 8991c57cc2bSEaswaran Raman // Use the OptMinSizeThreshold or OptSizeThreshold knob if they are available 9001c57cc2bSEaswaran Raman // and reduce the threshold if the caller has the necessary attribute. 90151b809bfSEaswaran Raman if (Caller->optForMinSize()) { 9021c57cc2bSEaswaran Raman Threshold = MinIfValid(Threshold, Params.OptMinSizeThreshold); 90351b809bfSEaswaran Raman // For minsize, we want to disable the single BB bonus and the vector 90451b809bfSEaswaran Raman // bonuses, but not the last-call-to-static bonus. Inlining the last call to 90551b809bfSEaswaran Raman // a static function will, at the minimum, eliminate the parameter setup and 90651b809bfSEaswaran Raman // call/return instructions. 90751b809bfSEaswaran Raman SingleBBBonusPercent = 0; 90851b809bfSEaswaran Raman VectorBonusPercent = 0; 90951b809bfSEaswaran Raman } else if (Caller->optForSize()) 9101c57cc2bSEaswaran Raman Threshold = MinIfValid(Threshold, Params.OptSizeThreshold); 911f4bb2f0dSEaswaran Raman 912e08b139dSEaswaran Raman // Adjust the threshold based on inlinehint attribute and profile based 913e08b139dSEaswaran Raman // hotness information if the caller does not have MinSize attribute. 914e08b139dSEaswaran Raman if (!Caller->optForMinSize()) { 915e08b139dSEaswaran Raman if (Callee.hasFnAttribute(Attribute::InlineHint)) 9160d58fcacSEaswaran Raman Threshold = MaxIfValid(Threshold, Params.HintThreshold); 917bba762a1SChandler Carruth 9183cd1479cSEaswaran Raman // FIXME: After switching to the new passmanager, simplify the logic below 919bba762a1SChandler Carruth // by checking only the callsite hotness/coldness as we will reliably 920bba762a1SChandler Carruth // have local profile information. 9213cd1479cSEaswaran Raman // 922bba762a1SChandler Carruth // Callsite hotness and coldness can be determined if sample profile is 923bba762a1SChandler Carruth // used (which adds hotness metadata to calls) or if caller's 924bba762a1SChandler Carruth // BlockFrequencyInfo is available. 925bba762a1SChandler Carruth BlockFrequencyInfo *CallerBFI = GetBFI ? &((*GetBFI)(*Caller)) : nullptr; 926974d4eeaSEaswaran Raman auto HotCallSiteThreshold = getHotCallSiteThreshold(CS, CallerBFI); 927974d4eeaSEaswaran Raman if (!Caller->optForSize() && HotCallSiteThreshold) { 928d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "Hot callsite.\n"); 929974d4eeaSEaswaran Raman // FIXME: This should update the threshold only if it exceeds the 930974d4eeaSEaswaran Raman // current threshold, but AutoFDO + ThinLTO currently relies on this 931974d4eeaSEaswaran Raman // behavior to prevent inlining of hot callsites during ThinLTO 932974d4eeaSEaswaran Raman // compile phase. 933974d4eeaSEaswaran Raman Threshold = HotCallSiteThreshold.getValue(); 934c5fa6358SEaswaran Raman } else if (isColdCallSite(CS, CallerBFI)) { 935d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "Cold callsite.\n"); 93651b809bfSEaswaran Raman // Do not apply bonuses for a cold callsite including the 93751b809bfSEaswaran Raman // LastCallToStatic bonus. While this bonus might result in code size 93851b809bfSEaswaran Raman // reduction, it can cause the size of a non-cold caller to increase 93951b809bfSEaswaran Raman // preventing it from being inlined. 94051b809bfSEaswaran Raman DisallowAllBonuses(); 9413cd1479cSEaswaran Raman Threshold = MinIfValid(Threshold, Params.ColdCallSiteThreshold); 942bba762a1SChandler Carruth } else if (PSI) { 943bba762a1SChandler Carruth // Use callee's global profile information only if we have no way of 944bba762a1SChandler Carruth // determining this via callsite information. 9453cd1479cSEaswaran Raman if (PSI->isFunctionEntryHot(&Callee)) { 946d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "Hot callee.\n"); 947e08b139dSEaswaran Raman // If callsite hotness can not be determined, we may still know 948e08b139dSEaswaran Raman // that the callee is hot and treat it as a weaker hint for threshold 949e08b139dSEaswaran Raman // increase. 950e08b139dSEaswaran Raman Threshold = MaxIfValid(Threshold, Params.HintThreshold); 951e08b139dSEaswaran Raman } else if (PSI->isFunctionEntryCold(&Callee)) { 952d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "Cold callee.\n"); 95351b809bfSEaswaran Raman // Do not apply bonuses for a cold callee including the 95451b809bfSEaswaran Raman // LastCallToStatic bonus. While this bonus might result in code size 95551b809bfSEaswaran Raman // reduction, it can cause the size of a non-cold caller to increase 95651b809bfSEaswaran Raman // preventing it from being inlined. 95751b809bfSEaswaran Raman DisallowAllBonuses(); 9581c57cc2bSEaswaran Raman Threshold = MinIfValid(Threshold, Params.ColdThreshold); 959e08b139dSEaswaran Raman } 960e08b139dSEaswaran Raman } 961e08b139dSEaswaran Raman } 9628650a4daSJustin Lebar 9638650a4daSJustin Lebar // Finally, take the target-specific inlining threshold multiplier into 9648650a4daSJustin Lebar // account. 9658650a4daSJustin Lebar Threshold *= TTI.getInliningThresholdMultiplier(); 96651b809bfSEaswaran Raman 96751b809bfSEaswaran Raman SingleBBBonus = Threshold * SingleBBBonusPercent / 100; 96851b809bfSEaswaran Raman VectorBonus = Threshold * VectorBonusPercent / 100; 96951b809bfSEaswaran Raman 97051b809bfSEaswaran Raman bool OnlyOneCallAndLocalLinkage = 97151b809bfSEaswaran Raman F.hasLocalLinkage() && F.hasOneUse() && &F == CS.getCalledFunction(); 97251b809bfSEaswaran Raman // If there is only one call of the function, and it has internal linkage, 97351b809bfSEaswaran Raman // the cost of inlining it drops dramatically. It may seem odd to update 97451b809bfSEaswaran Raman // Cost in updateThreshold, but the bonus depends on the logic in this method. 97551b809bfSEaswaran Raman if (OnlyOneCallAndLocalLinkage) 97651b809bfSEaswaran Raman Cost -= LastCallToStaticBonus; 977f4bb2f0dSEaswaran Raman } 978f4bb2f0dSEaswaran Raman 9797adc3a2bSChandler Carruth bool CallAnalyzer::visitCmpInst(CmpInst &I) { 9807adc3a2bSChandler Carruth Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); 9817adc3a2bSChandler Carruth // First try to handle simplified comparisons. 982617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 983617f6364SEaswaran Raman return ConstantExpr::getCompare(I.getPredicate(), COps[0], COps[1]); 984617f6364SEaswaran Raman })) 9857adc3a2bSChandler Carruth return true; 9867adc3a2bSChandler Carruth 9877adc3a2bSChandler Carruth if (I.getOpcode() == Instruction::FCmp) 9887adc3a2bSChandler Carruth return false; 9897adc3a2bSChandler Carruth 9907adc3a2bSChandler Carruth // Otherwise look for a comparison between constant offset pointers with 9917adc3a2bSChandler Carruth // a common base. 9927adc3a2bSChandler Carruth Value *LHSBase, *RHSBase; 9937adc3a2bSChandler Carruth APInt LHSOffset, RHSOffset; 9947adc3a2bSChandler Carruth std::tie(LHSBase, LHSOffset) = ConstantOffsetPtrs.lookup(LHS); 9957adc3a2bSChandler Carruth if (LHSBase) { 9967adc3a2bSChandler Carruth std::tie(RHSBase, RHSOffset) = ConstantOffsetPtrs.lookup(RHS); 9977adc3a2bSChandler Carruth if (RHSBase && LHSBase == RHSBase) { 9987adc3a2bSChandler Carruth // We have common bases, fold the icmp to a constant based on the 9997adc3a2bSChandler Carruth // offsets. 10007adc3a2bSChandler Carruth Constant *CLHS = ConstantInt::get(LHS->getContext(), LHSOffset); 10017adc3a2bSChandler Carruth Constant *CRHS = ConstantInt::get(RHS->getContext(), RHSOffset); 10027adc3a2bSChandler Carruth if (Constant *C = ConstantExpr::getICmp(I.getPredicate(), CLHS, CRHS)) { 10037adc3a2bSChandler Carruth SimplifiedValues[&I] = C; 10047adc3a2bSChandler Carruth ++NumConstantPtrCmps; 10057adc3a2bSChandler Carruth return true; 10067adc3a2bSChandler Carruth } 10077adc3a2bSChandler Carruth } 10087adc3a2bSChandler Carruth } 10097adc3a2bSChandler Carruth 10107adc3a2bSChandler Carruth // If the comparison is an equality comparison with null, we can simplify it 10117adc3a2bSChandler Carruth // if we know the value (argument) can't be null 10127adc3a2bSChandler Carruth if (I.isEquality() && isa<ConstantPointerNull>(I.getOperand(1)) && 10137adc3a2bSChandler Carruth isKnownNonNullInCallee(I.getOperand(0))) { 10147adc3a2bSChandler Carruth bool IsNotEqual = I.getPredicate() == CmpInst::ICMP_NE; 10157adc3a2bSChandler Carruth SimplifiedValues[&I] = IsNotEqual ? ConstantInt::getTrue(I.getType()) 10167adc3a2bSChandler Carruth : ConstantInt::getFalse(I.getType()); 10177adc3a2bSChandler Carruth return true; 10187adc3a2bSChandler Carruth } 10197adc3a2bSChandler Carruth // Finally check for SROA candidates in comparisons. 10207adc3a2bSChandler Carruth Value *SROAArg; 10217adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 10227adc3a2bSChandler Carruth if (lookupSROAArgAndCost(I.getOperand(0), SROAArg, CostIt)) { 10237adc3a2bSChandler Carruth if (isa<ConstantPointerNull>(I.getOperand(1))) { 10247adc3a2bSChandler Carruth accumulateSROACost(CostIt, InlineConstants::InstrCost); 10257adc3a2bSChandler Carruth return true; 10267adc3a2bSChandler Carruth } 10277adc3a2bSChandler Carruth 10287adc3a2bSChandler Carruth disableSROA(CostIt); 10297adc3a2bSChandler Carruth } 10307adc3a2bSChandler Carruth 10317adc3a2bSChandler Carruth return false; 10327adc3a2bSChandler Carruth } 10337adc3a2bSChandler Carruth 10347adc3a2bSChandler Carruth bool CallAnalyzer::visitSub(BinaryOperator &I) { 10357adc3a2bSChandler Carruth // Try to handle a special case: we can fold computing the difference of two 10367adc3a2bSChandler Carruth // constant-related pointers. 10377adc3a2bSChandler Carruth Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); 10387adc3a2bSChandler Carruth Value *LHSBase, *RHSBase; 10397adc3a2bSChandler Carruth APInt LHSOffset, RHSOffset; 10407adc3a2bSChandler Carruth std::tie(LHSBase, LHSOffset) = ConstantOffsetPtrs.lookup(LHS); 10417adc3a2bSChandler Carruth if (LHSBase) { 10427adc3a2bSChandler Carruth std::tie(RHSBase, RHSOffset) = ConstantOffsetPtrs.lookup(RHS); 10437adc3a2bSChandler Carruth if (RHSBase && LHSBase == RHSBase) { 10447adc3a2bSChandler Carruth // We have common bases, fold the subtract to a constant based on the 10457adc3a2bSChandler Carruth // offsets. 10467adc3a2bSChandler Carruth Constant *CLHS = ConstantInt::get(LHS->getContext(), LHSOffset); 10477adc3a2bSChandler Carruth Constant *CRHS = ConstantInt::get(RHS->getContext(), RHSOffset); 10487adc3a2bSChandler Carruth if (Constant *C = ConstantExpr::getSub(CLHS, CRHS)) { 10497adc3a2bSChandler Carruth SimplifiedValues[&I] = C; 10507adc3a2bSChandler Carruth ++NumConstantPtrDiffs; 10517adc3a2bSChandler Carruth return true; 10527adc3a2bSChandler Carruth } 10537adc3a2bSChandler Carruth } 10547adc3a2bSChandler Carruth } 10557adc3a2bSChandler Carruth 10567adc3a2bSChandler Carruth // Otherwise, fall back to the generic logic for simplifying and handling 10577adc3a2bSChandler Carruth // instructions. 10587adc3a2bSChandler Carruth return Base::visitSub(I); 10597adc3a2bSChandler Carruth } 10607adc3a2bSChandler Carruth 10617adc3a2bSChandler Carruth bool CallAnalyzer::visitBinaryOperator(BinaryOperator &I) { 10627adc3a2bSChandler Carruth Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); 10636d14dfe8SHaicheng Wu Constant *CLHS = dyn_cast<Constant>(LHS); 10646d14dfe8SHaicheng Wu if (!CLHS) 10656d14dfe8SHaicheng Wu CLHS = SimplifiedValues.lookup(LHS); 10666d14dfe8SHaicheng Wu Constant *CRHS = dyn_cast<Constant>(RHS); 10676d14dfe8SHaicheng Wu if (!CRHS) 10686d14dfe8SHaicheng Wu CRHS = SimplifiedValues.lookup(RHS); 10696d14dfe8SHaicheng Wu 10707adc3a2bSChandler Carruth Value *SimpleV = nullptr; 10717adc3a2bSChandler Carruth if (auto FI = dyn_cast<FPMathOperator>(&I)) 10726d14dfe8SHaicheng Wu SimpleV = SimplifyFPBinOp(I.getOpcode(), CLHS ? CLHS : LHS, 10736d14dfe8SHaicheng Wu CRHS ? CRHS : RHS, FI->getFastMathFlags(), DL); 10747adc3a2bSChandler Carruth else 10756d14dfe8SHaicheng Wu SimpleV = 10766d14dfe8SHaicheng Wu SimplifyBinOp(I.getOpcode(), CLHS ? CLHS : LHS, CRHS ? CRHS : RHS, DL); 10777adc3a2bSChandler Carruth 10786d14dfe8SHaicheng Wu if (Constant *C = dyn_cast_or_null<Constant>(SimpleV)) 10796d14dfe8SHaicheng Wu SimplifiedValues[&I] = C; 10806d14dfe8SHaicheng Wu 10816d14dfe8SHaicheng Wu if (SimpleV) 10827adc3a2bSChandler Carruth return true; 10837adc3a2bSChandler Carruth 10847adc3a2bSChandler Carruth // Disable any SROA on arguments to arbitrary, unsimplified binary operators. 10857adc3a2bSChandler Carruth disableSROA(LHS); 10867adc3a2bSChandler Carruth disableSROA(RHS); 10877adc3a2bSChandler Carruth 108839ed9a60SEli Friedman // If the instruction is floating point, and the target says this operation 108939ed9a60SEli Friedman // is expensive, this may eventually become a library call. Treat the cost 109039ed9a60SEli Friedman // as such. 109139ed9a60SEli Friedman if (I.getType()->isFloatingPointTy() && 109239ed9a60SEli Friedman TTI.getFPOpCost(I.getType()) == TargetTransformInfo::TCC_Expensive) 109339ed9a60SEli Friedman Cost += InlineConstants::CallPenalty; 109439ed9a60SEli Friedman 10957adc3a2bSChandler Carruth return false; 10967adc3a2bSChandler Carruth } 10977adc3a2bSChandler Carruth 10987adc3a2bSChandler Carruth bool CallAnalyzer::visitLoad(LoadInst &I) { 10997adc3a2bSChandler Carruth Value *SROAArg; 11007adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 11017adc3a2bSChandler Carruth if (lookupSROAArgAndCost(I.getPointerOperand(), SROAArg, CostIt)) { 11027adc3a2bSChandler Carruth if (I.isSimple()) { 11037adc3a2bSChandler Carruth accumulateSROACost(CostIt, InlineConstants::InstrCost); 11047adc3a2bSChandler Carruth return true; 11057adc3a2bSChandler Carruth } 11067adc3a2bSChandler Carruth 11077adc3a2bSChandler Carruth disableSROA(CostIt); 11087adc3a2bSChandler Carruth } 11097adc3a2bSChandler Carruth 1110a4461515SHaicheng Wu // If the data is already loaded from this address and hasn't been clobbered 1111a4461515SHaicheng Wu // by any stores or calls, this load is likely to be redundant and can be 1112a4461515SHaicheng Wu // eliminated. 1113a4461515SHaicheng Wu if (EnableLoadElimination && 1114b3689cabSHaicheng Wu !LoadAddrSet.insert(I.getPointerOperand()).second && I.isUnordered()) { 1115a4461515SHaicheng Wu LoadEliminationCost += InlineConstants::InstrCost; 1116a4461515SHaicheng Wu return true; 1117a4461515SHaicheng Wu } 1118a4461515SHaicheng Wu 11197adc3a2bSChandler Carruth return false; 11207adc3a2bSChandler Carruth } 11217adc3a2bSChandler Carruth 11227adc3a2bSChandler Carruth bool CallAnalyzer::visitStore(StoreInst &I) { 11237adc3a2bSChandler Carruth Value *SROAArg; 11247adc3a2bSChandler Carruth DenseMap<Value *, int>::iterator CostIt; 11257adc3a2bSChandler Carruth if (lookupSROAArgAndCost(I.getPointerOperand(), SROAArg, CostIt)) { 11267adc3a2bSChandler Carruth if (I.isSimple()) { 11277adc3a2bSChandler Carruth accumulateSROACost(CostIt, InlineConstants::InstrCost); 11287adc3a2bSChandler Carruth return true; 11297adc3a2bSChandler Carruth } 11307adc3a2bSChandler Carruth 11317adc3a2bSChandler Carruth disableSROA(CostIt); 11327adc3a2bSChandler Carruth } 11337adc3a2bSChandler Carruth 1134a4461515SHaicheng Wu // The store can potentially clobber loads and prevent repeated loads from 1135a4461515SHaicheng Wu // being eliminated. 1136a4461515SHaicheng Wu // FIXME: 1137a4461515SHaicheng Wu // 1. We can probably keep an initial set of eliminatable loads substracted 1138a4461515SHaicheng Wu // from the cost even when we finally see a store. We just need to disable 1139a4461515SHaicheng Wu // *further* accumulation of elimination savings. 1140a4461515SHaicheng Wu // 2. We should probably at some point thread MemorySSA for the callee into 1141a4461515SHaicheng Wu // this and then use that to actually compute *really* precise savings. 1142a4461515SHaicheng Wu disableLoadElimination(); 11437adc3a2bSChandler Carruth return false; 11447adc3a2bSChandler Carruth } 11457adc3a2bSChandler Carruth 11467adc3a2bSChandler Carruth bool CallAnalyzer::visitExtractValue(ExtractValueInst &I) { 11477adc3a2bSChandler Carruth // Constant folding for extract value is trivial. 1148617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 1149617f6364SEaswaran Raman return ConstantExpr::getExtractValue(COps[0], I.getIndices()); 1150617f6364SEaswaran Raman })) 11517adc3a2bSChandler Carruth return true; 11527adc3a2bSChandler Carruth 11537adc3a2bSChandler Carruth // SROA can look through these but give them a cost. 11547adc3a2bSChandler Carruth return false; 11557adc3a2bSChandler Carruth } 11567adc3a2bSChandler Carruth 11577adc3a2bSChandler Carruth bool CallAnalyzer::visitInsertValue(InsertValueInst &I) { 11587adc3a2bSChandler Carruth // Constant folding for insert value is trivial. 1159617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 1160617f6364SEaswaran Raman return ConstantExpr::getInsertValue(/*AggregateOperand*/ COps[0], 1161617f6364SEaswaran Raman /*InsertedValueOperand*/ COps[1], 1162617f6364SEaswaran Raman I.getIndices()); 1163617f6364SEaswaran Raman })) 11647adc3a2bSChandler Carruth return true; 11657adc3a2bSChandler Carruth 11667adc3a2bSChandler Carruth // SROA can look through these but give them a cost. 11677adc3a2bSChandler Carruth return false; 11687adc3a2bSChandler Carruth } 11697adc3a2bSChandler Carruth 11705f8f34e4SAdrian Prantl /// Try to simplify a call site. 11717adc3a2bSChandler Carruth /// 11727adc3a2bSChandler Carruth /// Takes a concrete function and callsite and tries to actually simplify it by 11737adc3a2bSChandler Carruth /// analyzing the arguments and call itself with instsimplify. Returns true if 11747adc3a2bSChandler Carruth /// it has simplified the callsite to some other entity (a constant), making it 11757adc3a2bSChandler Carruth /// free. 11767adc3a2bSChandler Carruth bool CallAnalyzer::simplifyCallSite(Function *F, CallSite CS) { 11777adc3a2bSChandler Carruth // FIXME: Using the instsimplify logic directly for this is inefficient 11787adc3a2bSChandler Carruth // because we have to continually rebuild the argument list even when no 11797adc3a2bSChandler Carruth // simplifications can be performed. Until that is fixed with remapping 11807adc3a2bSChandler Carruth // inside of instsimplify, directly constant fold calls here. 1181647025f9SAndrew Kaylor if (!canConstantFoldCallTo(CS, F)) 11827adc3a2bSChandler Carruth return false; 11837adc3a2bSChandler Carruth 11847adc3a2bSChandler Carruth // Try to re-map the arguments to constants. 11857adc3a2bSChandler Carruth SmallVector<Constant *, 4> ConstantArgs; 11867adc3a2bSChandler Carruth ConstantArgs.reserve(CS.arg_size()); 1187567556aaSChad Rosier for (CallSite::arg_iterator I = CS.arg_begin(), E = CS.arg_end(); I != E; 1188567556aaSChad Rosier ++I) { 11897adc3a2bSChandler Carruth Constant *C = dyn_cast<Constant>(*I); 11907adc3a2bSChandler Carruth if (!C) 11917adc3a2bSChandler Carruth C = dyn_cast_or_null<Constant>(SimplifiedValues.lookup(*I)); 11927adc3a2bSChandler Carruth if (!C) 11937adc3a2bSChandler Carruth return false; // This argument doesn't map to a constant. 11947adc3a2bSChandler Carruth 11957adc3a2bSChandler Carruth ConstantArgs.push_back(C); 11967adc3a2bSChandler Carruth } 1197647025f9SAndrew Kaylor if (Constant *C = ConstantFoldCall(CS, F, ConstantArgs)) { 11987adc3a2bSChandler Carruth SimplifiedValues[CS.getInstruction()] = C; 11997adc3a2bSChandler Carruth return true; 12007adc3a2bSChandler Carruth } 12017adc3a2bSChandler Carruth 12027adc3a2bSChandler Carruth return false; 12037adc3a2bSChandler Carruth } 12047adc3a2bSChandler Carruth 12057adc3a2bSChandler Carruth bool CallAnalyzer::visitCallSite(CallSite CS) { 12067adc3a2bSChandler Carruth if (CS.hasFnAttr(Attribute::ReturnsTwice) && 12077adc3a2bSChandler Carruth !F.hasFnAttribute(Attribute::ReturnsTwice)) { 12087adc3a2bSChandler Carruth // This aborts the entire analysis. 12097adc3a2bSChandler Carruth ExposesReturnsTwice = true; 12107adc3a2bSChandler Carruth return false; 12117adc3a2bSChandler Carruth } 1212567556aaSChad Rosier if (CS.isCall() && cast<CallInst>(CS.getInstruction())->cannotDuplicate()) 12137adc3a2bSChandler Carruth ContainsNoDuplicateCall = true; 12147adc3a2bSChandler Carruth 12157adc3a2bSChandler Carruth if (Function *F = CS.getCalledFunction()) { 12167adc3a2bSChandler Carruth // When we have a concrete function, first try to simplify it directly. 12177adc3a2bSChandler Carruth if (simplifyCallSite(F, CS)) 12187adc3a2bSChandler Carruth return true; 12197adc3a2bSChandler Carruth 12207adc3a2bSChandler Carruth // Next check if it is an intrinsic we know about. 12217adc3a2bSChandler Carruth // FIXME: Lift this into part of the InstVisitor. 12227adc3a2bSChandler Carruth if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(CS.getInstruction())) { 12237adc3a2bSChandler Carruth switch (II->getIntrinsicID()) { 12247adc3a2bSChandler Carruth default: 1225a4461515SHaicheng Wu if (!CS.onlyReadsMemory() && !isAssumeLikeIntrinsic(II)) 1226a4461515SHaicheng Wu disableLoadElimination(); 12277adc3a2bSChandler Carruth return Base::visitCallSite(CS); 12287adc3a2bSChandler Carruth 12297dd8dbf4SPeter Collingbourne case Intrinsic::load_relative: 12307dd8dbf4SPeter Collingbourne // This is normally lowered to 4 LLVM instructions. 12317dd8dbf4SPeter Collingbourne Cost += 3 * InlineConstants::InstrCost; 12327dd8dbf4SPeter Collingbourne return false; 12337dd8dbf4SPeter Collingbourne 12347adc3a2bSChandler Carruth case Intrinsic::memset: 12357adc3a2bSChandler Carruth case Intrinsic::memcpy: 12367adc3a2bSChandler Carruth case Intrinsic::memmove: 1237a4461515SHaicheng Wu disableLoadElimination(); 12387adc3a2bSChandler Carruth // SROA can usually chew through these intrinsics, but they aren't free. 12397adc3a2bSChandler Carruth return false; 12404296ea72SVitaly Buka case Intrinsic::icall_branch_funnel: 12417adc3a2bSChandler Carruth case Intrinsic::localescape: 12424296ea72SVitaly Buka HasUninlineableIntrinsic = true; 12437adc3a2bSChandler Carruth return false; 124480788d80SFlorian Hahn case Intrinsic::vastart: 124577beee41SSameer AbuAsal InitsVargArgs = true; 124680788d80SFlorian Hahn return false; 12477adc3a2bSChandler Carruth } 12487adc3a2bSChandler Carruth } 12497adc3a2bSChandler Carruth 12509d939c8fSDavide Italiano if (F == CS.getInstruction()->getFunction()) { 12517adc3a2bSChandler Carruth // This flag will fully abort the analysis, so don't bother with anything 12527adc3a2bSChandler Carruth // else. 12537adc3a2bSChandler Carruth IsRecursiveCall = true; 12547adc3a2bSChandler Carruth return false; 12557adc3a2bSChandler Carruth } 12567adc3a2bSChandler Carruth 12577adc3a2bSChandler Carruth if (TTI.isLoweredToCall(F)) { 12587adc3a2bSChandler Carruth // We account for the average 1 instruction per call argument setup 12597adc3a2bSChandler Carruth // here. 12607adc3a2bSChandler Carruth Cost += CS.arg_size() * InlineConstants::InstrCost; 12617adc3a2bSChandler Carruth 12627adc3a2bSChandler Carruth // Everything other than inline ASM will also have a significant cost 12637adc3a2bSChandler Carruth // merely from making the call. 12647adc3a2bSChandler Carruth if (!isa<InlineAsm>(CS.getCalledValue())) 12657adc3a2bSChandler Carruth Cost += InlineConstants::CallPenalty; 12667adc3a2bSChandler Carruth } 12677adc3a2bSChandler Carruth 1268a4461515SHaicheng Wu if (!CS.onlyReadsMemory()) 1269a4461515SHaicheng Wu disableLoadElimination(); 12707adc3a2bSChandler Carruth return Base::visitCallSite(CS); 12717adc3a2bSChandler Carruth } 12727adc3a2bSChandler Carruth 12737adc3a2bSChandler Carruth // Otherwise we're in a very special case -- an indirect function call. See 12747adc3a2bSChandler Carruth // if we can be particularly clever about this. 12757adc3a2bSChandler Carruth Value *Callee = CS.getCalledValue(); 12767adc3a2bSChandler Carruth 12777adc3a2bSChandler Carruth // First, pay the price of the argument setup. We account for the average 12787adc3a2bSChandler Carruth // 1 instruction per call argument setup here. 12797adc3a2bSChandler Carruth Cost += CS.arg_size() * InlineConstants::InstrCost; 12807adc3a2bSChandler Carruth 12817adc3a2bSChandler Carruth // Next, check if this happens to be an indirect function call to a known 12827adc3a2bSChandler Carruth // function in this inline context. If not, we've done all we can. 12837adc3a2bSChandler Carruth Function *F = dyn_cast_or_null<Function>(SimplifiedValues.lookup(Callee)); 1284a4461515SHaicheng Wu if (!F) { 1285a4461515SHaicheng Wu if (!CS.onlyReadsMemory()) 1286a4461515SHaicheng Wu disableLoadElimination(); 12877adc3a2bSChandler Carruth return Base::visitCallSite(CS); 1288a4461515SHaicheng Wu } 12897adc3a2bSChandler Carruth 12907adc3a2bSChandler Carruth // If we have a constant that we are calling as a function, we can peer 12917adc3a2bSChandler Carruth // through it and see the function target. This happens not infrequently 12927adc3a2bSChandler Carruth // during devirtualization and so we want to give it a hefty bonus for 12937adc3a2bSChandler Carruth // inlining, but cap that bonus in the event that inlining wouldn't pan 12947adc3a2bSChandler Carruth // out. Pretend to inline the function, with a custom threshold. 12951c57cc2bSEaswaran Raman auto IndirectCallParams = Params; 12961c57cc2bSEaswaran Raman IndirectCallParams.DefaultThreshold = InlineConstants::IndirectCallThreshold; 12970812c5beSHaicheng Wu CallAnalyzer CA(TTI, GetAssumptionCache, GetBFI, PSI, ORE, *F, CS, 129812585b01SEaswaran Raman IndirectCallParams); 12997adc3a2bSChandler Carruth if (CA.analyzeCall(CS)) { 13007adc3a2bSChandler Carruth // We were able to inline the indirect call! Subtract the cost from the 13016d90d9f1SEaswaran Raman // threshold to get the bonus we want to apply, but don't go below zero. 13026d90d9f1SEaswaran Raman Cost -= std::max(0, CA.getThreshold() - CA.getCost()); 13037adc3a2bSChandler Carruth } 13047adc3a2bSChandler Carruth 1305a4461515SHaicheng Wu if (!F->onlyReadsMemory()) 1306a4461515SHaicheng Wu disableLoadElimination(); 13077adc3a2bSChandler Carruth return Base::visitCallSite(CS); 13087adc3a2bSChandler Carruth } 13097adc3a2bSChandler Carruth 13107adc3a2bSChandler Carruth bool CallAnalyzer::visitReturnInst(ReturnInst &RI) { 13117adc3a2bSChandler Carruth // At least one return instruction will be free after inlining. 13127adc3a2bSChandler Carruth bool Free = !HasReturn; 13137adc3a2bSChandler Carruth HasReturn = true; 13147adc3a2bSChandler Carruth return Free; 13157adc3a2bSChandler Carruth } 13167adc3a2bSChandler Carruth 13177adc3a2bSChandler Carruth bool CallAnalyzer::visitBranchInst(BranchInst &BI) { 13187adc3a2bSChandler Carruth // We model unconditional branches as essentially free -- they really 13197adc3a2bSChandler Carruth // shouldn't exist at all, but handling them makes the behavior of the 13207adc3a2bSChandler Carruth // inliner more regular and predictable. Interestingly, conditional branches 13217adc3a2bSChandler Carruth // which will fold away are also free. 13227adc3a2bSChandler Carruth return BI.isUnconditional() || isa<ConstantInt>(BI.getCondition()) || 13237adc3a2bSChandler Carruth dyn_cast_or_null<ConstantInt>( 13247adc3a2bSChandler Carruth SimplifiedValues.lookup(BI.getCondition())); 13257adc3a2bSChandler Carruth } 13267adc3a2bSChandler Carruth 13273ec848bcSHaicheng Wu bool CallAnalyzer::visitSelectInst(SelectInst &SI) { 13283ec848bcSHaicheng Wu bool CheckSROA = SI.getType()->isPointerTy(); 13293ec848bcSHaicheng Wu Value *TrueVal = SI.getTrueValue(); 13303ec848bcSHaicheng Wu Value *FalseVal = SI.getFalseValue(); 13313ec848bcSHaicheng Wu 13323ec848bcSHaicheng Wu Constant *TrueC = dyn_cast<Constant>(TrueVal); 13333ec848bcSHaicheng Wu if (!TrueC) 13343ec848bcSHaicheng Wu TrueC = SimplifiedValues.lookup(TrueVal); 13353ec848bcSHaicheng Wu Constant *FalseC = dyn_cast<Constant>(FalseVal); 13363ec848bcSHaicheng Wu if (!FalseC) 13373ec848bcSHaicheng Wu FalseC = SimplifiedValues.lookup(FalseVal); 13383ec848bcSHaicheng Wu Constant *CondC = 13393ec848bcSHaicheng Wu dyn_cast_or_null<Constant>(SimplifiedValues.lookup(SI.getCondition())); 13403ec848bcSHaicheng Wu 13413ec848bcSHaicheng Wu if (!CondC) { 13423ec848bcSHaicheng Wu // Select C, X, X => X 13433ec848bcSHaicheng Wu if (TrueC == FalseC && TrueC) { 13443ec848bcSHaicheng Wu SimplifiedValues[&SI] = TrueC; 13453ec848bcSHaicheng Wu return true; 13463ec848bcSHaicheng Wu } 13473ec848bcSHaicheng Wu 13483ec848bcSHaicheng Wu if (!CheckSROA) 13493ec848bcSHaicheng Wu return Base::visitSelectInst(SI); 13503ec848bcSHaicheng Wu 13513ec848bcSHaicheng Wu std::pair<Value *, APInt> TrueBaseAndOffset = 13523ec848bcSHaicheng Wu ConstantOffsetPtrs.lookup(TrueVal); 13533ec848bcSHaicheng Wu std::pair<Value *, APInt> FalseBaseAndOffset = 13543ec848bcSHaicheng Wu ConstantOffsetPtrs.lookup(FalseVal); 13553ec848bcSHaicheng Wu if (TrueBaseAndOffset == FalseBaseAndOffset && TrueBaseAndOffset.first) { 13563ec848bcSHaicheng Wu ConstantOffsetPtrs[&SI] = TrueBaseAndOffset; 13573ec848bcSHaicheng Wu 13583ec848bcSHaicheng Wu Value *SROAArg; 13593ec848bcSHaicheng Wu DenseMap<Value *, int>::iterator CostIt; 13603ec848bcSHaicheng Wu if (lookupSROAArgAndCost(TrueVal, SROAArg, CostIt)) 13613ec848bcSHaicheng Wu SROAArgValues[&SI] = SROAArg; 13623ec848bcSHaicheng Wu return true; 13633ec848bcSHaicheng Wu } 13643ec848bcSHaicheng Wu 13653ec848bcSHaicheng Wu return Base::visitSelectInst(SI); 13663ec848bcSHaicheng Wu } 13673ec848bcSHaicheng Wu 13683ec848bcSHaicheng Wu // Select condition is a constant. 13693ec848bcSHaicheng Wu Value *SelectedV = CondC->isAllOnesValue() 13703ec848bcSHaicheng Wu ? TrueVal 13713ec848bcSHaicheng Wu : (CondC->isNullValue()) ? FalseVal : nullptr; 13723ec848bcSHaicheng Wu if (!SelectedV) { 13733ec848bcSHaicheng Wu // Condition is a vector constant that is not all 1s or all 0s. If all 13743ec848bcSHaicheng Wu // operands are constants, ConstantExpr::getSelect() can handle the cases 13753ec848bcSHaicheng Wu // such as select vectors. 13763ec848bcSHaicheng Wu if (TrueC && FalseC) { 13773ec848bcSHaicheng Wu if (auto *C = ConstantExpr::getSelect(CondC, TrueC, FalseC)) { 13783ec848bcSHaicheng Wu SimplifiedValues[&SI] = C; 13793ec848bcSHaicheng Wu return true; 13803ec848bcSHaicheng Wu } 13813ec848bcSHaicheng Wu } 13823ec848bcSHaicheng Wu return Base::visitSelectInst(SI); 13833ec848bcSHaicheng Wu } 13843ec848bcSHaicheng Wu 13853ec848bcSHaicheng Wu // Condition is either all 1s or all 0s. SI can be simplified. 13863ec848bcSHaicheng Wu if (Constant *SelectedC = dyn_cast<Constant>(SelectedV)) { 13873ec848bcSHaicheng Wu SimplifiedValues[&SI] = SelectedC; 13883ec848bcSHaicheng Wu return true; 13893ec848bcSHaicheng Wu } 13903ec848bcSHaicheng Wu 13913ec848bcSHaicheng Wu if (!CheckSROA) 13923ec848bcSHaicheng Wu return true; 13933ec848bcSHaicheng Wu 13943ec848bcSHaicheng Wu std::pair<Value *, APInt> BaseAndOffset = 13953ec848bcSHaicheng Wu ConstantOffsetPtrs.lookup(SelectedV); 13963ec848bcSHaicheng Wu if (BaseAndOffset.first) { 13973ec848bcSHaicheng Wu ConstantOffsetPtrs[&SI] = BaseAndOffset; 13983ec848bcSHaicheng Wu 13993ec848bcSHaicheng Wu Value *SROAArg; 14003ec848bcSHaicheng Wu DenseMap<Value *, int>::iterator CostIt; 14013ec848bcSHaicheng Wu if (lookupSROAArgAndCost(SelectedV, SROAArg, CostIt)) 14023ec848bcSHaicheng Wu SROAArgValues[&SI] = SROAArg; 14033ec848bcSHaicheng Wu } 14043ec848bcSHaicheng Wu 14053ec848bcSHaicheng Wu return true; 14063ec848bcSHaicheng Wu } 14073ec848bcSHaicheng Wu 14087adc3a2bSChandler Carruth bool CallAnalyzer::visitSwitchInst(SwitchInst &SI) { 14097adc3a2bSChandler Carruth // We model unconditional switches as free, see the comments on handling 14107adc3a2bSChandler Carruth // branches. 14117adc3a2bSChandler Carruth if (isa<ConstantInt>(SI.getCondition())) 14127adc3a2bSChandler Carruth return true; 14137adc3a2bSChandler Carruth if (Value *V = SimplifiedValues.lookup(SI.getCondition())) 14147adc3a2bSChandler Carruth if (isa<ConstantInt>(V)) 14157adc3a2bSChandler Carruth return true; 14167adc3a2bSChandler Carruth 14177ad02eeeSEric Christopher // Assume the most general case where the switch is lowered into 1418919f9e8dSJun Bum Lim // either a jump table, bit test, or a balanced binary tree consisting of 1419919f9e8dSJun Bum Lim // case clusters without merging adjacent clusters with the same 1420919f9e8dSJun Bum Lim // destination. We do not consider the switches that are lowered with a mix 1421919f9e8dSJun Bum Lim // of jump table/bit test/binary search tree. The cost of the switch is 1422919f9e8dSJun Bum Lim // proportional to the size of the tree or the size of jump table range. 14232960d41eSJun Bum Lim // 14242960d41eSJun Bum Lim // NB: We convert large switches which are just used to initialize large phi 14252960d41eSJun Bum Lim // nodes to lookup tables instead in simplify-cfg, so this shouldn't prevent 14262960d41eSJun Bum Lim // inlining those. It will prevent inlining in cases where the optimization 14272960d41eSJun Bum Lim // does not (yet) fire. 1428919f9e8dSJun Bum Lim 1429506cfb7aSJun Bum Lim // Maximum valid cost increased in this function. 1430506cfb7aSJun Bum Lim int CostUpperBound = INT_MAX - InlineConstants::InstrCost - 1; 1431506cfb7aSJun Bum Lim 1432919f9e8dSJun Bum Lim // Exit early for a large switch, assuming one case needs at least one 1433919f9e8dSJun Bum Lim // instruction. 1434919f9e8dSJun Bum Lim // FIXME: This is not true for a bit test, but ignore such case for now to 1435919f9e8dSJun Bum Lim // save compile-time. 1436919f9e8dSJun Bum Lim int64_t CostLowerBound = 1437506cfb7aSJun Bum Lim std::min((int64_t)CostUpperBound, 1438919f9e8dSJun Bum Lim (int64_t)SI.getNumCases() * InlineConstants::InstrCost + Cost); 1439919f9e8dSJun Bum Lim 14400812c5beSHaicheng Wu if (CostLowerBound > Threshold && !ComputeFullInlineCost) { 1441919f9e8dSJun Bum Lim Cost = CostLowerBound; 1442919f9e8dSJun Bum Lim return false; 1443919f9e8dSJun Bum Lim } 1444919f9e8dSJun Bum Lim 1445919f9e8dSJun Bum Lim unsigned JumpTableSize = 0; 1446919f9e8dSJun Bum Lim unsigned NumCaseCluster = 1447919f9e8dSJun Bum Lim TTI.getEstimatedNumberOfCaseClusters(SI, JumpTableSize); 1448919f9e8dSJun Bum Lim 1449919f9e8dSJun Bum Lim // If suitable for a jump table, consider the cost for the table size and 1450919f9e8dSJun Bum Lim // branch to destination. 1451919f9e8dSJun Bum Lim if (JumpTableSize) { 1452919f9e8dSJun Bum Lim int64_t JTCost = (int64_t)JumpTableSize * InlineConstants::InstrCost + 1453919f9e8dSJun Bum Lim 4 * InlineConstants::InstrCost; 1454506cfb7aSJun Bum Lim 1455506cfb7aSJun Bum Lim Cost = std::min((int64_t)CostUpperBound, JTCost + Cost); 1456919f9e8dSJun Bum Lim return false; 1457919f9e8dSJun Bum Lim } 1458919f9e8dSJun Bum Lim 1459919f9e8dSJun Bum Lim // Considering forming a binary search, we should find the number of nodes 1460919f9e8dSJun Bum Lim // which is same as the number of comparisons when lowered. For a given 1461919f9e8dSJun Bum Lim // number of clusters, n, we can define a recursive function, f(n), to find 1462919f9e8dSJun Bum Lim // the number of nodes in the tree. The recursion is : 1463919f9e8dSJun Bum Lim // f(n) = 1 + f(n/2) + f (n - n/2), when n > 3, 1464919f9e8dSJun Bum Lim // and f(n) = n, when n <= 3. 1465919f9e8dSJun Bum Lim // This will lead a binary tree where the leaf should be either f(2) or f(3) 1466919f9e8dSJun Bum Lim // when n > 3. So, the number of comparisons from leaves should be n, while 1467919f9e8dSJun Bum Lim // the number of non-leaf should be : 1468919f9e8dSJun Bum Lim // 2^(log2(n) - 1) - 1 1469919f9e8dSJun Bum Lim // = 2^log2(n) * 2^-1 - 1 1470919f9e8dSJun Bum Lim // = n / 2 - 1. 1471919f9e8dSJun Bum Lim // Considering comparisons from leaf and non-leaf nodes, we can estimate the 1472919f9e8dSJun Bum Lim // number of comparisons in a simple closed form : 1473919f9e8dSJun Bum Lim // n + n / 2 - 1 = n * 3 / 2 - 1 1474919f9e8dSJun Bum Lim if (NumCaseCluster <= 3) { 1475919f9e8dSJun Bum Lim // Suppose a comparison includes one compare and one conditional branch. 1476919f9e8dSJun Bum Lim Cost += NumCaseCluster * 2 * InlineConstants::InstrCost; 1477919f9e8dSJun Bum Lim return false; 1478919f9e8dSJun Bum Lim } 1479506cfb7aSJun Bum Lim 1480506cfb7aSJun Bum Lim int64_t ExpectedNumberOfCompare = 3 * (int64_t)NumCaseCluster / 2 - 1; 1481506cfb7aSJun Bum Lim int64_t SwitchCost = 1482919f9e8dSJun Bum Lim ExpectedNumberOfCompare * 2 * InlineConstants::InstrCost; 1483506cfb7aSJun Bum Lim 1484506cfb7aSJun Bum Lim Cost = std::min((int64_t)CostUpperBound, SwitchCost + Cost); 1485919f9e8dSJun Bum Lim return false; 1486919f9e8dSJun Bum Lim } 1487919f9e8dSJun Bum Lim 14887adc3a2bSChandler Carruth bool CallAnalyzer::visitIndirectBrInst(IndirectBrInst &IBI) { 14897adc3a2bSChandler Carruth // We never want to inline functions that contain an indirectbr. This is 14907adc3a2bSChandler Carruth // incorrect because all the blockaddress's (in static global initializers 14917adc3a2bSChandler Carruth // for example) would be referring to the original function, and this 14927adc3a2bSChandler Carruth // indirect jump would jump from the inlined copy of the function into the 14937adc3a2bSChandler Carruth // original function which is extremely undefined behavior. 14947adc3a2bSChandler Carruth // FIXME: This logic isn't really right; we can safely inline functions with 14957adc3a2bSChandler Carruth // indirectbr's as long as no other function or global references the 14967adc3a2bSChandler Carruth // blockaddress of a block within the current function. 14977adc3a2bSChandler Carruth HasIndirectBr = true; 14987adc3a2bSChandler Carruth return false; 14997adc3a2bSChandler Carruth } 15007adc3a2bSChandler Carruth 15017adc3a2bSChandler Carruth bool CallAnalyzer::visitResumeInst(ResumeInst &RI) { 15027adc3a2bSChandler Carruth // FIXME: It's not clear that a single instruction is an accurate model for 15037adc3a2bSChandler Carruth // the inline cost of a resume instruction. 15047adc3a2bSChandler Carruth return false; 15057adc3a2bSChandler Carruth } 15067adc3a2bSChandler Carruth 15077adc3a2bSChandler Carruth bool CallAnalyzer::visitCleanupReturnInst(CleanupReturnInst &CRI) { 15087adc3a2bSChandler Carruth // FIXME: It's not clear that a single instruction is an accurate model for 15097adc3a2bSChandler Carruth // the inline cost of a cleanupret instruction. 15107adc3a2bSChandler Carruth return false; 15117adc3a2bSChandler Carruth } 15127adc3a2bSChandler Carruth 15137adc3a2bSChandler Carruth bool CallAnalyzer::visitCatchReturnInst(CatchReturnInst &CRI) { 15147adc3a2bSChandler Carruth // FIXME: It's not clear that a single instruction is an accurate model for 15158220bcc5SJoseph Tremoulet // the inline cost of a catchret instruction. 15167adc3a2bSChandler Carruth return false; 15177adc3a2bSChandler Carruth } 15187adc3a2bSChandler Carruth 15197adc3a2bSChandler Carruth bool CallAnalyzer::visitUnreachableInst(UnreachableInst &I) { 15207adc3a2bSChandler Carruth // FIXME: It might be reasonably to discount the cost of instructions leading 15217adc3a2bSChandler Carruth // to unreachable as they have the lowest possible impact on both runtime and 15227adc3a2bSChandler Carruth // code size. 15237adc3a2bSChandler Carruth return true; // No actual code is needed for unreachable. 15247adc3a2bSChandler Carruth } 15257adc3a2bSChandler Carruth 15267adc3a2bSChandler Carruth bool CallAnalyzer::visitInstruction(Instruction &I) { 15277adc3a2bSChandler Carruth // Some instructions are free. All of the free intrinsics can also be 15287adc3a2bSChandler Carruth // handled by SROA, etc. 15297adc3a2bSChandler Carruth if (TargetTransformInfo::TCC_Free == TTI.getUserCost(&I)) 15307adc3a2bSChandler Carruth return true; 15317adc3a2bSChandler Carruth 15327adc3a2bSChandler Carruth // We found something we don't understand or can't handle. Mark any SROA-able 15337adc3a2bSChandler Carruth // values in the operand list as no longer viable. 15347adc3a2bSChandler Carruth for (User::op_iterator OI = I.op_begin(), OE = I.op_end(); OI != OE; ++OI) 15357adc3a2bSChandler Carruth disableSROA(*OI); 15367adc3a2bSChandler Carruth 15377adc3a2bSChandler Carruth return false; 15387adc3a2bSChandler Carruth } 15397adc3a2bSChandler Carruth 15405f8f34e4SAdrian Prantl /// Analyze a basic block for its contribution to the inline cost. 15417adc3a2bSChandler Carruth /// 15427adc3a2bSChandler Carruth /// This method walks the analyzer over every instruction in the given basic 15437adc3a2bSChandler Carruth /// block and accounts for their cost during inlining at this callsite. It 15447adc3a2bSChandler Carruth /// aborts early if the threshold has been exceeded or an impossible to inline 15457adc3a2bSChandler Carruth /// construct has been detected. It returns false if inlining is no longer 15467adc3a2bSChandler Carruth /// viable, and true if inlining remains viable. 1547c0aa4b75SDavid Bolvansky InlineResult 1548c0aa4b75SDavid Bolvansky CallAnalyzer::analyzeBlock(BasicBlock *BB, 15497adc3a2bSChandler Carruth SmallPtrSetImpl<const Value *> &EphValues) { 15507adc3a2bSChandler Carruth for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I) { 15517adc3a2bSChandler Carruth // FIXME: Currently, the number of instructions in a function regardless of 15527adc3a2bSChandler Carruth // our ability to simplify them during inline to constants or dead code, 15537adc3a2bSChandler Carruth // are actually used by the vector bonus heuristic. As long as that's true, 15547adc3a2bSChandler Carruth // we have to special case debug intrinsics here to prevent differences in 15557adc3a2bSChandler Carruth // inlining due to debug symbols. Eventually, the number of unsimplified 15567adc3a2bSChandler Carruth // instructions shouldn't factor into the cost computation, but until then, 15577adc3a2bSChandler Carruth // hack around it here. 15587adc3a2bSChandler Carruth if (isa<DbgInfoIntrinsic>(I)) 15597adc3a2bSChandler Carruth continue; 15607adc3a2bSChandler Carruth 15617adc3a2bSChandler Carruth // Skip ephemeral values. 15625a82c916SDuncan P. N. Exon Smith if (EphValues.count(&*I)) 15637adc3a2bSChandler Carruth continue; 15647adc3a2bSChandler Carruth 15657adc3a2bSChandler Carruth ++NumInstructions; 15667adc3a2bSChandler Carruth if (isa<ExtractElementInst>(I) || I->getType()->isVectorTy()) 15677adc3a2bSChandler Carruth ++NumVectorInstructions; 15687adc3a2bSChandler Carruth 15697adc3a2bSChandler Carruth // If the instruction simplified to a constant, there is no cost to this 15707adc3a2bSChandler Carruth // instruction. Visit the instructions using our InstVisitor to account for 15717adc3a2bSChandler Carruth // all of the per-instruction logic. The visit tree returns true if we 15727adc3a2bSChandler Carruth // consumed the instruction in any way, and false if the instruction's base 15737adc3a2bSChandler Carruth // cost should count against inlining. 15745a82c916SDuncan P. N. Exon Smith if (Base::visit(&*I)) 15757adc3a2bSChandler Carruth ++NumInstructionsSimplified; 15767adc3a2bSChandler Carruth else 15777adc3a2bSChandler Carruth Cost += InlineConstants::InstrCost; 15787adc3a2bSChandler Carruth 15790812c5beSHaicheng Wu using namespace ore; 15807adc3a2bSChandler Carruth // If the visit this instruction detected an uninlinable pattern, abort. 1581c0aa4b75SDavid Bolvansky InlineResult IR; 1582c0aa4b75SDavid Bolvansky if (IsRecursiveCall) 1583c0aa4b75SDavid Bolvansky IR = "recursive"; 1584c0aa4b75SDavid Bolvansky else if (ExposesReturnsTwice) 1585c0aa4b75SDavid Bolvansky IR = "exposes returns twice"; 1586c0aa4b75SDavid Bolvansky else if (HasDynamicAlloca) 1587c0aa4b75SDavid Bolvansky IR = "dynamic alloca"; 1588c0aa4b75SDavid Bolvansky else if (HasIndirectBr) 1589c0aa4b75SDavid Bolvansky IR = "indirect branch"; 1590c0aa4b75SDavid Bolvansky else if (HasUninlineableIntrinsic) 1591c0aa4b75SDavid Bolvansky IR = "uninlinable intrinsic"; 159277beee41SSameer AbuAsal else if (InitsVargArgs) 1593c0aa4b75SDavid Bolvansky IR = "varargs"; 1594c0aa4b75SDavid Bolvansky if (!IR) { 15950812c5beSHaicheng Wu if (ORE) 15969590658fSVivek Pandya ORE->emit([&]() { 15979590658fSVivek Pandya return OptimizationRemarkMissed(DEBUG_TYPE, "NeverInline", 15980812c5beSHaicheng Wu CandidateCS.getInstruction()) 1599c0aa4b75SDavid Bolvansky << NV("Callee", &F) << " has uninlinable pattern (" 1600c0aa4b75SDavid Bolvansky << NV("InlineResult", IR.message) 1601c0aa4b75SDavid Bolvansky << ") and cost is not fully computed"; 16029590658fSVivek Pandya }); 1603c0aa4b75SDavid Bolvansky return IR; 16040812c5beSHaicheng Wu } 16057adc3a2bSChandler Carruth 16067adc3a2bSChandler Carruth // If the caller is a recursive function then we don't want to inline 16077adc3a2bSChandler Carruth // functions which allocate a lot of stack space because it would increase 16087adc3a2bSChandler Carruth // the caller stack usage dramatically. 16097adc3a2bSChandler Carruth if (IsCallerRecursive && 16100812c5beSHaicheng Wu AllocatedSize > InlineConstants::TotalAllocaSizeRecursiveCaller) { 1611c0aa4b75SDavid Bolvansky InlineResult IR = "recursive and allocates too much stack space"; 16120812c5beSHaicheng Wu if (ORE) 16139590658fSVivek Pandya ORE->emit([&]() { 16149590658fSVivek Pandya return OptimizationRemarkMissed(DEBUG_TYPE, "NeverInline", 16150812c5beSHaicheng Wu CandidateCS.getInstruction()) 1616c0aa4b75SDavid Bolvansky << NV("Callee", &F) << " is " << NV("InlineResult", IR.message) 1617c0aa4b75SDavid Bolvansky << ". Cost is not fully computed"; 16189590658fSVivek Pandya }); 1619c0aa4b75SDavid Bolvansky return IR; 16200812c5beSHaicheng Wu } 16217adc3a2bSChandler Carruth 16227adc3a2bSChandler Carruth // Check if we've past the maximum possible threshold so we don't spin in 16237adc3a2bSChandler Carruth // huge basic blocks that will never inline. 162461995364SHaicheng Wu if (Cost >= Threshold && !ComputeFullInlineCost) 16257adc3a2bSChandler Carruth return false; 16267adc3a2bSChandler Carruth } 16277adc3a2bSChandler Carruth 16287adc3a2bSChandler Carruth return true; 16297adc3a2bSChandler Carruth } 16307adc3a2bSChandler Carruth 16315f8f34e4SAdrian Prantl /// Compute the base pointer and cumulative constant offsets for V. 16327adc3a2bSChandler Carruth /// 16337adc3a2bSChandler Carruth /// This strips all constant offsets off of V, leaving it the base pointer, and 16347adc3a2bSChandler Carruth /// accumulates the total constant offset applied in the returned constant. It 16357adc3a2bSChandler Carruth /// returns 0 if V is not a pointer, and returns the constant '0' if there are 16367adc3a2bSChandler Carruth /// no constant offsets applied. 16377adc3a2bSChandler Carruth ConstantInt *CallAnalyzer::stripAndComputeInBoundsConstantOffsets(Value *&V) { 16387adc3a2bSChandler Carruth if (!V->getType()->isPointerTy()) 16397adc3a2bSChandler Carruth return nullptr; 16407adc3a2bSChandler Carruth 164177f32994SBjorn Pettersson unsigned AS = V->getType()->getPointerAddressSpace(); 1642945b7e5aSElena Demikhovsky unsigned IntPtrWidth = DL.getIndexSizeInBits(AS); 16437adc3a2bSChandler Carruth APInt Offset = APInt::getNullValue(IntPtrWidth); 16447adc3a2bSChandler Carruth 16457adc3a2bSChandler Carruth // Even though we don't look through PHI nodes, we could be called on an 16467adc3a2bSChandler Carruth // instruction in an unreachable block, which may be on a cycle. 16477adc3a2bSChandler Carruth SmallPtrSet<Value *, 4> Visited; 16487adc3a2bSChandler Carruth Visited.insert(V); 16497adc3a2bSChandler Carruth do { 16507adc3a2bSChandler Carruth if (GEPOperator *GEP = dyn_cast<GEPOperator>(V)) { 16517adc3a2bSChandler Carruth if (!GEP->isInBounds() || !accumulateGEPOffset(*GEP, Offset)) 16527adc3a2bSChandler Carruth return nullptr; 16537adc3a2bSChandler Carruth V = GEP->getPointerOperand(); 16547adc3a2bSChandler Carruth } else if (Operator::getOpcode(V) == Instruction::BitCast) { 16557adc3a2bSChandler Carruth V = cast<Operator>(V)->getOperand(0); 16567adc3a2bSChandler Carruth } else if (GlobalAlias *GA = dyn_cast<GlobalAlias>(V)) { 16575ce32728SSanjoy Das if (GA->isInterposable()) 16587adc3a2bSChandler Carruth break; 16597adc3a2bSChandler Carruth V = GA->getAliasee(); 16607adc3a2bSChandler Carruth } else { 16617adc3a2bSChandler Carruth break; 16627adc3a2bSChandler Carruth } 16637adc3a2bSChandler Carruth assert(V->getType()->isPointerTy() && "Unexpected operand type!"); 16647adc3a2bSChandler Carruth } while (Visited.insert(V).second); 16657adc3a2bSChandler Carruth 166677f32994SBjorn Pettersson Type *IntPtrTy = DL.getIntPtrType(V->getContext(), AS); 16677adc3a2bSChandler Carruth return cast<ConstantInt>(ConstantInt::get(IntPtrTy, Offset)); 16687adc3a2bSChandler Carruth } 16697adc3a2bSChandler Carruth 16705f8f34e4SAdrian Prantl /// Find dead blocks due to deleted CFG edges during inlining. 16713739e14aSHaicheng Wu /// 16723739e14aSHaicheng Wu /// If we know the successor of the current block, \p CurrBB, has to be \p 16733739e14aSHaicheng Wu /// NextBB, the other successors of \p CurrBB are dead if these successors have 16743739e14aSHaicheng Wu /// no live incoming CFG edges. If one block is found to be dead, we can 16753739e14aSHaicheng Wu /// continue growing the dead block list by checking the successors of the dead 16763739e14aSHaicheng Wu /// blocks to see if all their incoming edges are dead or not. 16773739e14aSHaicheng Wu void CallAnalyzer::findDeadBlocks(BasicBlock *CurrBB, BasicBlock *NextBB) { 16783739e14aSHaicheng Wu auto IsEdgeDead = [&](BasicBlock *Pred, BasicBlock *Succ) { 16793739e14aSHaicheng Wu // A CFG edge is dead if the predecessor is dead or the predessor has a 16803739e14aSHaicheng Wu // known successor which is not the one under exam. 16813739e14aSHaicheng Wu return (DeadBlocks.count(Pred) || 16823739e14aSHaicheng Wu (KnownSuccessors[Pred] && KnownSuccessors[Pred] != Succ)); 16833739e14aSHaicheng Wu }; 16843739e14aSHaicheng Wu 16853739e14aSHaicheng Wu auto IsNewlyDead = [&](BasicBlock *BB) { 16863739e14aSHaicheng Wu // If all the edges to a block are dead, the block is also dead. 16873739e14aSHaicheng Wu return (!DeadBlocks.count(BB) && 16883739e14aSHaicheng Wu llvm::all_of(predecessors(BB), 16893739e14aSHaicheng Wu [&](BasicBlock *P) { return IsEdgeDead(P, BB); })); 16903739e14aSHaicheng Wu }; 16913739e14aSHaicheng Wu 16923739e14aSHaicheng Wu for (BasicBlock *Succ : successors(CurrBB)) { 16933739e14aSHaicheng Wu if (Succ == NextBB || !IsNewlyDead(Succ)) 16943739e14aSHaicheng Wu continue; 16953739e14aSHaicheng Wu SmallVector<BasicBlock *, 4> NewDead; 16963739e14aSHaicheng Wu NewDead.push_back(Succ); 16973739e14aSHaicheng Wu while (!NewDead.empty()) { 16983739e14aSHaicheng Wu BasicBlock *Dead = NewDead.pop_back_val(); 16993739e14aSHaicheng Wu if (DeadBlocks.insert(Dead)) 17003739e14aSHaicheng Wu // Continue growing the dead block lists. 17013739e14aSHaicheng Wu for (BasicBlock *S : successors(Dead)) 17023739e14aSHaicheng Wu if (IsNewlyDead(S)) 17033739e14aSHaicheng Wu NewDead.push_back(S); 17043739e14aSHaicheng Wu } 17053739e14aSHaicheng Wu } 17063739e14aSHaicheng Wu } 17073739e14aSHaicheng Wu 17085f8f34e4SAdrian Prantl /// Analyze a call site for potential inlining. 17097adc3a2bSChandler Carruth /// 17107adc3a2bSChandler Carruth /// Returns true if inlining this call is viable, and false if it is not 17117adc3a2bSChandler Carruth /// viable. It computes the cost and adjusts the threshold based on numerous 17127adc3a2bSChandler Carruth /// factors and heuristics. If this method returns false but the computed cost 17137adc3a2bSChandler Carruth /// is below the computed threshold, then inlining was forcibly disabled by 17147adc3a2bSChandler Carruth /// some artifact of the routine. 1715c0aa4b75SDavid Bolvansky InlineResult CallAnalyzer::analyzeCall(CallSite CS) { 17167adc3a2bSChandler Carruth ++NumCallsAnalyzed; 17177adc3a2bSChandler Carruth 17187adc3a2bSChandler Carruth // Perform some tweaks to the cost and threshold based on the direct 17197adc3a2bSChandler Carruth // callsite information. 17207adc3a2bSChandler Carruth 17217adc3a2bSChandler Carruth // We want to more aggressively inline vector-dense kernels, so up the 17227adc3a2bSChandler Carruth // threshold, and we'll lower it if the % of vector instructions gets too 17237adc3a2bSChandler Carruth // low. Note that these bonuses are some what arbitrary and evolved over time 17247adc3a2bSChandler Carruth // by accident as much as because they are principled bonuses. 17257adc3a2bSChandler Carruth // 17267adc3a2bSChandler Carruth // FIXME: It would be nice to remove all such bonuses. At least it would be 17277adc3a2bSChandler Carruth // nice to base the bonus values on something more scientific. 17287adc3a2bSChandler Carruth assert(NumInstructions == 0); 17297adc3a2bSChandler Carruth assert(NumVectorInstructions == 0); 1730f4bb2f0dSEaswaran Raman 1731f4bb2f0dSEaswaran Raman // Update the threshold based on callsite properties 1732f4bb2f0dSEaswaran Raman updateThreshold(CS, F); 1733f4bb2f0dSEaswaran Raman 1734*ed279752SEaswaran Raman // While Threshold depends on commandline options that can take negative 1735*ed279752SEaswaran Raman // values, we want to enforce the invariant that the computed threshold and 1736*ed279752SEaswaran Raman // bonuses are non-negative. 1737*ed279752SEaswaran Raman assert(Threshold >= 0); 1738*ed279752SEaswaran Raman assert(SingleBBBonus >= 0); 1739*ed279752SEaswaran Raman assert(VectorBonus >= 0); 1740*ed279752SEaswaran Raman 17417adc3a2bSChandler Carruth // Speculatively apply all possible bonuses to Threshold. If cost exceeds 17427adc3a2bSChandler Carruth // this Threshold any time, and cost cannot decrease, we can stop processing 17437adc3a2bSChandler Carruth // the rest of the function body. 174451b809bfSEaswaran Raman Threshold += (SingleBBBonus + VectorBonus); 17457adc3a2bSChandler Carruth 1746351d9b01SXinliang David Li // Give out bonuses for the callsite, as the instructions setting them up 1747351d9b01SXinliang David Li // will be gone after inlining. 1748351d9b01SXinliang David Li Cost -= getCallsiteCost(CS, DL); 17497adc3a2bSChandler Carruth 17507adc3a2bSChandler Carruth // If this function uses the coldcc calling convention, prefer not to inline 17517adc3a2bSChandler Carruth // it. 17527adc3a2bSChandler Carruth if (F.getCallingConv() == CallingConv::Cold) 17537adc3a2bSChandler Carruth Cost += InlineConstants::ColdccPenalty; 17547adc3a2bSChandler Carruth 17557adc3a2bSChandler Carruth // Check if we're done. This can happen due to bonuses and penalties. 175661995364SHaicheng Wu if (Cost >= Threshold && !ComputeFullInlineCost) 1757c0aa4b75SDavid Bolvansky return "high cost"; 17587adc3a2bSChandler Carruth 17597adc3a2bSChandler Carruth if (F.empty()) 17607adc3a2bSChandler Carruth return true; 17617adc3a2bSChandler Carruth 17629d939c8fSDavide Italiano Function *Caller = CS.getInstruction()->getFunction(); 17637adc3a2bSChandler Carruth // Check if the caller function is recursive itself. 17647adc3a2bSChandler Carruth for (User *U : Caller->users()) { 17657adc3a2bSChandler Carruth CallSite Site(U); 17667adc3a2bSChandler Carruth if (!Site) 17677adc3a2bSChandler Carruth continue; 17687adc3a2bSChandler Carruth Instruction *I = Site.getInstruction(); 17699d939c8fSDavide Italiano if (I->getFunction() == Caller) { 17707adc3a2bSChandler Carruth IsCallerRecursive = true; 17717adc3a2bSChandler Carruth break; 17727adc3a2bSChandler Carruth } 17737adc3a2bSChandler Carruth } 17747adc3a2bSChandler Carruth 17757adc3a2bSChandler Carruth // Populate our simplified values by mapping from function arguments to call 17767adc3a2bSChandler Carruth // arguments with known important simplifications. 17777adc3a2bSChandler Carruth CallSite::arg_iterator CAI = CS.arg_begin(); 17787adc3a2bSChandler Carruth for (Function::arg_iterator FAI = F.arg_begin(), FAE = F.arg_end(); 17797adc3a2bSChandler Carruth FAI != FAE; ++FAI, ++CAI) { 17807adc3a2bSChandler Carruth assert(CAI != CS.arg_end()); 17817adc3a2bSChandler Carruth if (Constant *C = dyn_cast<Constant>(CAI)) 17825a82c916SDuncan P. N. Exon Smith SimplifiedValues[&*FAI] = C; 17837adc3a2bSChandler Carruth 17847adc3a2bSChandler Carruth Value *PtrArg = *CAI; 17857adc3a2bSChandler Carruth if (ConstantInt *C = stripAndComputeInBoundsConstantOffsets(PtrArg)) { 17865a82c916SDuncan P. N. Exon Smith ConstantOffsetPtrs[&*FAI] = std::make_pair(PtrArg, C->getValue()); 17877adc3a2bSChandler Carruth 17887adc3a2bSChandler Carruth // We can SROA any pointer arguments derived from alloca instructions. 17897adc3a2bSChandler Carruth if (isa<AllocaInst>(PtrArg)) { 17905a82c916SDuncan P. N. Exon Smith SROAArgValues[&*FAI] = PtrArg; 17917adc3a2bSChandler Carruth SROAArgCosts[PtrArg] = 0; 17927adc3a2bSChandler Carruth } 17937adc3a2bSChandler Carruth } 17947adc3a2bSChandler Carruth } 17957adc3a2bSChandler Carruth NumConstantArgs = SimplifiedValues.size(); 17967adc3a2bSChandler Carruth NumConstantOffsetPtrArgs = ConstantOffsetPtrs.size(); 17977adc3a2bSChandler Carruth NumAllocaArgs = SROAArgValues.size(); 17987adc3a2bSChandler Carruth 17997adc3a2bSChandler Carruth // FIXME: If a caller has multiple calls to a callee, we end up recomputing 18007adc3a2bSChandler Carruth // the ephemeral values multiple times (and they're completely determined by 18017adc3a2bSChandler Carruth // the callee, so this is purely duplicate work). 18027adc3a2bSChandler Carruth SmallPtrSet<const Value *, 32> EphValues; 1803aec2fa35SDaniel Jasper CodeMetrics::collectEphemeralValues(&F, &GetAssumptionCache(F), EphValues); 18047adc3a2bSChandler Carruth 18057adc3a2bSChandler Carruth // The worklist of live basic blocks in the callee *after* inlining. We avoid 18067adc3a2bSChandler Carruth // adding basic blocks of the callee which can be proven to be dead for this 18077adc3a2bSChandler Carruth // particular call site in order to get more accurate cost estimates. This 18087adc3a2bSChandler Carruth // requires a somewhat heavyweight iteration pattern: we need to walk the 18097adc3a2bSChandler Carruth // basic blocks in a breadth-first order as we insert live successors. To 18107adc3a2bSChandler Carruth // accomplish this, prioritizing for small iterations because we exit after 18117adc3a2bSChandler Carruth // crossing our threshold, we use a small-size optimized SetVector. 18127adc3a2bSChandler Carruth typedef SetVector<BasicBlock *, SmallVector<BasicBlock *, 16>, 1813567556aaSChad Rosier SmallPtrSet<BasicBlock *, 16>> 1814567556aaSChad Rosier BBSetVector; 18157adc3a2bSChandler Carruth BBSetVector BBWorklist; 18167adc3a2bSChandler Carruth BBWorklist.insert(&F.getEntryBlock()); 181751b809bfSEaswaran Raman bool SingleBB = true; 18187adc3a2bSChandler Carruth // Note that we *must not* cache the size, this loop grows the worklist. 18197adc3a2bSChandler Carruth for (unsigned Idx = 0; Idx != BBWorklist.size(); ++Idx) { 18207adc3a2bSChandler Carruth // Bail out the moment we cross the threshold. This means we'll under-count 18217adc3a2bSChandler Carruth // the cost, but only when undercounting doesn't matter. 182261995364SHaicheng Wu if (Cost >= Threshold && !ComputeFullInlineCost) 18237adc3a2bSChandler Carruth break; 18247adc3a2bSChandler Carruth 18257adc3a2bSChandler Carruth BasicBlock *BB = BBWorklist[Idx]; 18267adc3a2bSChandler Carruth if (BB->empty()) 18277adc3a2bSChandler Carruth continue; 18287adc3a2bSChandler Carruth 18297adc3a2bSChandler Carruth // Disallow inlining a blockaddress. A blockaddress only has defined 18307adc3a2bSChandler Carruth // behavior for an indirect branch in the same function, and we do not 18317adc3a2bSChandler Carruth // currently support inlining indirect branches. But, the inliner may not 18327adc3a2bSChandler Carruth // see an indirect branch that ends up being dead code at a particular call 18337adc3a2bSChandler Carruth // site. If the blockaddress escapes the function, e.g., via a global 18347adc3a2bSChandler Carruth // variable, inlining may lead to an invalid cross-function reference. 18357adc3a2bSChandler Carruth if (BB->hasAddressTaken()) 1836c0aa4b75SDavid Bolvansky return "blockaddress"; 18377adc3a2bSChandler Carruth 18387adc3a2bSChandler Carruth // Analyze the cost of this block. If we blow through the threshold, this 18397adc3a2bSChandler Carruth // returns false, and we can bail on out. 1840c0aa4b75SDavid Bolvansky InlineResult IR = analyzeBlock(BB, EphValues); 1841c0aa4b75SDavid Bolvansky if (!IR) 1842c0aa4b75SDavid Bolvansky return IR; 18437adc3a2bSChandler Carruth 1844edb12a83SChandler Carruth Instruction *TI = BB->getTerminator(); 18457adc3a2bSChandler Carruth 18467adc3a2bSChandler Carruth // Add in the live successors by first checking whether we have terminator 18477adc3a2bSChandler Carruth // that may be simplified based on the values simplified by this call. 18487adc3a2bSChandler Carruth if (BranchInst *BI = dyn_cast<BranchInst>(TI)) { 18497adc3a2bSChandler Carruth if (BI->isConditional()) { 18507adc3a2bSChandler Carruth Value *Cond = BI->getCondition(); 1851567556aaSChad Rosier if (ConstantInt *SimpleCond = 1852567556aaSChad Rosier dyn_cast_or_null<ConstantInt>(SimplifiedValues.lookup(Cond))) { 18533739e14aSHaicheng Wu BasicBlock *NextBB = BI->getSuccessor(SimpleCond->isZero() ? 1 : 0); 18543739e14aSHaicheng Wu BBWorklist.insert(NextBB); 18553739e14aSHaicheng Wu KnownSuccessors[BB] = NextBB; 18563739e14aSHaicheng Wu findDeadBlocks(BB, NextBB); 18577adc3a2bSChandler Carruth continue; 18587adc3a2bSChandler Carruth } 18597adc3a2bSChandler Carruth } 18607adc3a2bSChandler Carruth } else if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) { 18617adc3a2bSChandler Carruth Value *Cond = SI->getCondition(); 1862567556aaSChad Rosier if (ConstantInt *SimpleCond = 1863567556aaSChad Rosier dyn_cast_or_null<ConstantInt>(SimplifiedValues.lookup(Cond))) { 18643739e14aSHaicheng Wu BasicBlock *NextBB = SI->findCaseValue(SimpleCond)->getCaseSuccessor(); 18653739e14aSHaicheng Wu BBWorklist.insert(NextBB); 18663739e14aSHaicheng Wu KnownSuccessors[BB] = NextBB; 18673739e14aSHaicheng Wu findDeadBlocks(BB, NextBB); 18687adc3a2bSChandler Carruth continue; 18697adc3a2bSChandler Carruth } 18707adc3a2bSChandler Carruth } 18717adc3a2bSChandler Carruth 18727adc3a2bSChandler Carruth // If we're unable to select a particular successor, just count all of 18737adc3a2bSChandler Carruth // them. 18747adc3a2bSChandler Carruth for (unsigned TIdx = 0, TSize = TI->getNumSuccessors(); TIdx != TSize; 18757adc3a2bSChandler Carruth ++TIdx) 18767adc3a2bSChandler Carruth BBWorklist.insert(TI->getSuccessor(TIdx)); 18777adc3a2bSChandler Carruth 18787adc3a2bSChandler Carruth // If we had any successors at this point, than post-inlining is likely to 18797adc3a2bSChandler Carruth // have them as well. Note that we assume any basic blocks which existed 18807adc3a2bSChandler Carruth // due to branches or switches which folded above will also fold after 18817adc3a2bSChandler Carruth // inlining. 18827adc3a2bSChandler Carruth if (SingleBB && TI->getNumSuccessors() > 1) { 18837adc3a2bSChandler Carruth // Take off the bonus we applied to the threshold. 18847adc3a2bSChandler Carruth Threshold -= SingleBBBonus; 18857adc3a2bSChandler Carruth SingleBB = false; 18867adc3a2bSChandler Carruth } 18877adc3a2bSChandler Carruth } 18887adc3a2bSChandler Carruth 188951b809bfSEaswaran Raman bool OnlyOneCallAndLocalLinkage = 189051b809bfSEaswaran Raman F.hasLocalLinkage() && F.hasOneUse() && &F == CS.getCalledFunction(); 18917adc3a2bSChandler Carruth // If this is a noduplicate call, we can still inline as long as 18927adc3a2bSChandler Carruth // inlining this would cause the removal of the caller (so the instruction 18937adc3a2bSChandler Carruth // is not actually duplicated, just moved). 18947adc3a2bSChandler Carruth if (!OnlyOneCallAndLocalLinkage && ContainsNoDuplicateCall) 1895c0aa4b75SDavid Bolvansky return "noduplicate"; 18967adc3a2bSChandler Carruth 1897ba9f245bSDavid Green // Loops generally act a lot like calls in that they act like barriers to 1898ba9f245bSDavid Green // movement, require a certain amount of setup, etc. So when optimising for 1899ba9f245bSDavid Green // size, we penalise any call sites that perform loops. We do this after all 1900ba9f245bSDavid Green // other costs here, so will likely only be dealing with relatively small 1901ba9f245bSDavid Green // functions (and hence DT and LI will hopefully be cheap). 1902ba9f245bSDavid Green if (Caller->optForMinSize()) { 1903ba9f245bSDavid Green DominatorTree DT(F); 1904ba9f245bSDavid Green LoopInfo LI(DT); 1905ba9f245bSDavid Green int NumLoops = 0; 1906ba9f245bSDavid Green for (Loop *L : LI) { 1907ba9f245bSDavid Green // Ignore loops that will not be executed 1908ba9f245bSDavid Green if (DeadBlocks.count(L->getHeader())) 1909ba9f245bSDavid Green continue; 1910ba9f245bSDavid Green NumLoops++; 1911ba9f245bSDavid Green } 1912ba9f245bSDavid Green Cost += NumLoops * InlineConstants::CallPenalty; 1913ba9f245bSDavid Green } 1914ba9f245bSDavid Green 19157adc3a2bSChandler Carruth // We applied the maximum possible vector bonus at the beginning. Now, 19167adc3a2bSChandler Carruth // subtract the excess bonus, if any, from the Threshold before 19177adc3a2bSChandler Carruth // comparing against Cost. 19187adc3a2bSChandler Carruth if (NumVectorInstructions <= NumInstructions / 10) 191951b809bfSEaswaran Raman Threshold -= VectorBonus; 19207adc3a2bSChandler Carruth else if (NumVectorInstructions <= NumInstructions / 2) 192151b809bfSEaswaran Raman Threshold -= VectorBonus/2; 19227adc3a2bSChandler Carruth 192300ab73dcSHans Wennborg return Cost < std::max(1, Threshold); 19247adc3a2bSChandler Carruth } 19257adc3a2bSChandler Carruth 1926615eb470SAaron Ballman #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 19275f8f34e4SAdrian Prantl /// Dump stats about this call's analysis. 1928eb2a2546SYaron Keren LLVM_DUMP_METHOD void CallAnalyzer::dump() { 19297adc3a2bSChandler Carruth #define DEBUG_PRINT_STAT(x) dbgs() << " " #x ": " << x << "\n" 19307adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantArgs); 19317adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantOffsetPtrArgs); 19327adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumAllocaArgs); 19337adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantPtrCmps); 19347adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantPtrDiffs); 19357adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumInstructionsSimplified); 19367adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumInstructions); 19377adc3a2bSChandler Carruth DEBUG_PRINT_STAT(SROACostSavings); 19387adc3a2bSChandler Carruth DEBUG_PRINT_STAT(SROACostSavingsLost); 1939a4461515SHaicheng Wu DEBUG_PRINT_STAT(LoadEliminationCost); 19407adc3a2bSChandler Carruth DEBUG_PRINT_STAT(ContainsNoDuplicateCall); 19417adc3a2bSChandler Carruth DEBUG_PRINT_STAT(Cost); 19427adc3a2bSChandler Carruth DEBUG_PRINT_STAT(Threshold); 19437adc3a2bSChandler Carruth #undef DEBUG_PRINT_STAT 19447adc3a2bSChandler Carruth } 19457adc3a2bSChandler Carruth #endif 19467adc3a2bSChandler Carruth 19475f8f34e4SAdrian Prantl /// Test that there are no attribute conflicts between Caller and Callee 19487adc3a2bSChandler Carruth /// that prevent inlining. 19497adc3a2bSChandler Carruth static bool functionsHaveCompatibleAttributes(Function *Caller, 19507adc3a2bSChandler Carruth Function *Callee, 19517adc3a2bSChandler Carruth TargetTransformInfo &TTI) { 19527adc3a2bSChandler Carruth return TTI.areInlineCompatible(Caller, Callee) && 19531cb242ebSAkira Hatanaka AttributeFuncs::areInlineCompatible(*Caller, *Callee); 19547adc3a2bSChandler Carruth } 19557adc3a2bSChandler Carruth 1956351d9b01SXinliang David Li int llvm::getCallsiteCost(CallSite CS, const DataLayout &DL) { 1957351d9b01SXinliang David Li int Cost = 0; 1958351d9b01SXinliang David Li for (unsigned I = 0, E = CS.arg_size(); I != E; ++I) { 1959351d9b01SXinliang David Li if (CS.isByValArgument(I)) { 1960351d9b01SXinliang David Li // We approximate the number of loads and stores needed by dividing the 1961351d9b01SXinliang David Li // size of the byval type by the target's pointer size. 1962351d9b01SXinliang David Li PointerType *PTy = cast<PointerType>(CS.getArgument(I)->getType()); 1963351d9b01SXinliang David Li unsigned TypeSize = DL.getTypeSizeInBits(PTy->getElementType()); 196477f32994SBjorn Pettersson unsigned AS = PTy->getAddressSpace(); 196577f32994SBjorn Pettersson unsigned PointerSize = DL.getPointerSizeInBits(AS); 1966351d9b01SXinliang David Li // Ceiling division. 1967351d9b01SXinliang David Li unsigned NumStores = (TypeSize + PointerSize - 1) / PointerSize; 1968351d9b01SXinliang David Li 1969351d9b01SXinliang David Li // If it generates more than 8 stores it is likely to be expanded as an 1970351d9b01SXinliang David Li // inline memcpy so we take that as an upper bound. Otherwise we assume 1971351d9b01SXinliang David Li // one load and one store per word copied. 1972351d9b01SXinliang David Li // FIXME: The maxStoresPerMemcpy setting from the target should be used 1973351d9b01SXinliang David Li // here instead of a magic number of 8, but it's not available via 1974351d9b01SXinliang David Li // DataLayout. 1975351d9b01SXinliang David Li NumStores = std::min(NumStores, 8U); 1976351d9b01SXinliang David Li 1977351d9b01SXinliang David Li Cost += 2 * NumStores * InlineConstants::InstrCost; 1978351d9b01SXinliang David Li } else { 1979351d9b01SXinliang David Li // For non-byval arguments subtract off one instruction per call 1980351d9b01SXinliang David Li // argument. 1981351d9b01SXinliang David Li Cost += InlineConstants::InstrCost; 1982351d9b01SXinliang David Li } 1983351d9b01SXinliang David Li } 1984351d9b01SXinliang David Li // The call instruction also disappears after inlining. 1985351d9b01SXinliang David Li Cost += InlineConstants::InstrCost + InlineConstants::CallPenalty; 1986351d9b01SXinliang David Li return Cost; 1987351d9b01SXinliang David Li } 1988351d9b01SXinliang David Li 1989ab6a6837SSean Silva InlineCost llvm::getInlineCost( 19901c57cc2bSEaswaran Raman CallSite CS, const InlineParams &Params, TargetTransformInfo &CalleeTTI, 1991aec2fa35SDaniel Jasper std::function<AssumptionCache &(Function &)> &GetAssumptionCache, 199212585b01SEaswaran Raman Optional<function_ref<BlockFrequencyInfo &(Function &)>> GetBFI, 19930812c5beSHaicheng Wu ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE) { 1994aec2fa35SDaniel Jasper return getInlineCost(CS, CS.getCalledFunction(), Params, CalleeTTI, 19950812c5beSHaicheng Wu GetAssumptionCache, GetBFI, PSI, ORE); 1996b9f7120eSEaswaran Raman } 1997b9f7120eSEaswaran Raman 1998ab6a6837SSean Silva InlineCost llvm::getInlineCost( 19991c57cc2bSEaswaran Raman CallSite CS, Function *Callee, const InlineParams &Params, 2000aec2fa35SDaniel Jasper TargetTransformInfo &CalleeTTI, 2001aec2fa35SDaniel Jasper std::function<AssumptionCache &(Function &)> &GetAssumptionCache, 200212585b01SEaswaran Raman Optional<function_ref<BlockFrequencyInfo &(Function &)>> GetBFI, 20030812c5beSHaicheng Wu ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE) { 2004f4bb2f0dSEaswaran Raman 20057adc3a2bSChandler Carruth // Cannot inline indirect calls. 20067adc3a2bSChandler Carruth if (!Callee) 2007c0aa4b75SDavid Bolvansky return llvm::InlineCost::getNever("indirect call"); 20087adc3a2bSChandler Carruth 20093851496eSBjorn Pettersson // Never inline calls with byval arguments that does not have the alloca 20103851496eSBjorn Pettersson // address space. Since byval arguments can be replaced with a copy to an 20113851496eSBjorn Pettersson // alloca, the inlined code would need to be adjusted to handle that the 20123851496eSBjorn Pettersson // argument is in the alloca address space (so it is a little bit complicated 20133851496eSBjorn Pettersson // to solve). 20143851496eSBjorn Pettersson unsigned AllocaAS = Callee->getParent()->getDataLayout().getAllocaAddrSpace(); 20153851496eSBjorn Pettersson for (unsigned I = 0, E = CS.arg_size(); I != E; ++I) 20163851496eSBjorn Pettersson if (CS.isByValArgument(I)) { 20173851496eSBjorn Pettersson PointerType *PTy = cast<PointerType>(CS.getArgument(I)->getType()); 20183851496eSBjorn Pettersson if (PTy->getAddressSpace() != AllocaAS) 2019c0aa4b75SDavid Bolvansky return llvm::InlineCost::getNever("byval arguments without alloca" 2020c0aa4b75SDavid Bolvansky " address space"); 20213851496eSBjorn Pettersson } 20223851496eSBjorn Pettersson 20237adc3a2bSChandler Carruth // Calls to functions with always-inline attributes should be inlined 20247adc3a2bSChandler Carruth // whenever possible. 20257adc3a2bSChandler Carruth if (CS.hasFnAttr(Attribute::AlwaysInline)) { 20267adc3a2bSChandler Carruth if (isInlineViable(*Callee)) 2027c0aa4b75SDavid Bolvansky return llvm::InlineCost::getAlways("always inline attribute"); 2028c0aa4b75SDavid Bolvansky return llvm::InlineCost::getNever("inapplicable always inline attribute"); 20297adc3a2bSChandler Carruth } 20307adc3a2bSChandler Carruth 20317adc3a2bSChandler Carruth // Never inline functions with conflicting attributes (unless callee has 20327adc3a2bSChandler Carruth // always-inline attribute). 20335ce28f4fSChad Rosier Function *Caller = CS.getCaller(); 20345ce28f4fSChad Rosier if (!functionsHaveCompatibleAttributes(Caller, Callee, CalleeTTI)) 2035c0aa4b75SDavid Bolvansky return llvm::InlineCost::getNever("conflicting attributes"); 20367adc3a2bSChandler Carruth 20377adc3a2bSChandler Carruth // Don't inline this call if the caller has the optnone attribute. 20385ce28f4fSChad Rosier if (Caller->hasFnAttribute(Attribute::OptimizeNone)) 2039c0aa4b75SDavid Bolvansky return llvm::InlineCost::getNever("optnone attribute"); 20407adc3a2bSChandler Carruth 204177eeac3dSManoj Gupta // Don't inline a function that treats null pointer as valid into a caller 204277eeac3dSManoj Gupta // that does not have this attribute. 204377eeac3dSManoj Gupta if (!Caller->nullPointerIsDefined() && Callee->nullPointerIsDefined()) 2044c0aa4b75SDavid Bolvansky return llvm::InlineCost::getNever("nullptr definitions incompatible"); 204577eeac3dSManoj Gupta 2046c0aa4b75SDavid Bolvansky // Don't inline functions which can be interposed at link-time. 2047c0aa4b75SDavid Bolvansky if (Callee->isInterposable()) 2048c0aa4b75SDavid Bolvansky return llvm::InlineCost::getNever("interposable"); 2049c0aa4b75SDavid Bolvansky 2050c0aa4b75SDavid Bolvansky // Don't inline functions marked noinline. 2051c0aa4b75SDavid Bolvansky if (Callee->hasFnAttribute(Attribute::NoInline)) 2052c0aa4b75SDavid Bolvansky return llvm::InlineCost::getNever("noinline function attribute"); 2053c0aa4b75SDavid Bolvansky 2054c0aa4b75SDavid Bolvansky // Don't inline call sites marked noinline. 2055c0aa4b75SDavid Bolvansky if (CS.isNoInline()) 2056c0aa4b75SDavid Bolvansky return llvm::InlineCost::getNever("noinline call site attribute"); 20577adc3a2bSChandler Carruth 2058d34e60caSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Analyzing call of " << Callee->getName() 20594eb18742SChad Rosier << "... (caller:" << Caller->getName() << ")\n"); 20607adc3a2bSChandler Carruth 20610812c5beSHaicheng Wu CallAnalyzer CA(CalleeTTI, GetAssumptionCache, GetBFI, PSI, ORE, *Callee, CS, 206212585b01SEaswaran Raman Params); 2063c0aa4b75SDavid Bolvansky InlineResult ShouldInline = CA.analyzeCall(CS); 20647adc3a2bSChandler Carruth 2065d34e60caSNicola Zaghen LLVM_DEBUG(CA.dump()); 20667adc3a2bSChandler Carruth 20677adc3a2bSChandler Carruth // Check if there was a reason to force inlining or no inlining. 20687adc3a2bSChandler Carruth if (!ShouldInline && CA.getCost() < CA.getThreshold()) 2069c0aa4b75SDavid Bolvansky return InlineCost::getNever(ShouldInline.message); 20707adc3a2bSChandler Carruth if (ShouldInline && CA.getCost() >= CA.getThreshold()) 2071c0aa4b75SDavid Bolvansky return InlineCost::getAlways("empty function"); 20727adc3a2bSChandler Carruth 20737adc3a2bSChandler Carruth return llvm::InlineCost::get(CA.getCost(), CA.getThreshold()); 20747adc3a2bSChandler Carruth } 20757adc3a2bSChandler Carruth 2076b9f7120eSEaswaran Raman bool llvm::isInlineViable(Function &F) { 20777adc3a2bSChandler Carruth bool ReturnsTwice = F.hasFnAttribute(Attribute::ReturnsTwice); 20787adc3a2bSChandler Carruth for (Function::iterator BI = F.begin(), BE = F.end(); BI != BE; ++BI) { 20797adc3a2bSChandler Carruth // Disallow inlining of functions which contain indirect branches or 20807adc3a2bSChandler Carruth // blockaddresses. 20817adc3a2bSChandler Carruth if (isa<IndirectBrInst>(BI->getTerminator()) || BI->hasAddressTaken()) 20827adc3a2bSChandler Carruth return false; 20837adc3a2bSChandler Carruth 20845a82c916SDuncan P. N. Exon Smith for (auto &II : *BI) { 20855a82c916SDuncan P. N. Exon Smith CallSite CS(&II); 20867adc3a2bSChandler Carruth if (!CS) 20877adc3a2bSChandler Carruth continue; 20887adc3a2bSChandler Carruth 20897adc3a2bSChandler Carruth // Disallow recursive calls. 20907adc3a2bSChandler Carruth if (&F == CS.getCalledFunction()) 20917adc3a2bSChandler Carruth return false; 20927adc3a2bSChandler Carruth 20937adc3a2bSChandler Carruth // Disallow calls which expose returns-twice to a function not previously 20947adc3a2bSChandler Carruth // attributed as such. 20957adc3a2bSChandler Carruth if (!ReturnsTwice && CS.isCall() && 20967adc3a2bSChandler Carruth cast<CallInst>(CS.getInstruction())->canReturnTwice()) 20977adc3a2bSChandler Carruth return false; 20987adc3a2bSChandler Carruth 20991636651eSFlorian Hahn if (CS.getCalledFunction()) 21001636651eSFlorian Hahn switch (CS.getCalledFunction()->getIntrinsicID()) { 21011636651eSFlorian Hahn default: 21021636651eSFlorian Hahn break; 21034296ea72SVitaly Buka // Disallow inlining of @llvm.icall.branch.funnel because current 21044296ea72SVitaly Buka // backend can't separate call targets from call arguments. 21054296ea72SVitaly Buka case llvm::Intrinsic::icall_branch_funnel: 21067adc3a2bSChandler Carruth // Disallow inlining functions that call @llvm.localescape. Doing this 21077adc3a2bSChandler Carruth // correctly would require major changes to the inliner. 21081636651eSFlorian Hahn case llvm::Intrinsic::localescape: 210977beee41SSameer AbuAsal // Disallow inlining of functions that initialize VarArgs with va_start. 21101636651eSFlorian Hahn case llvm::Intrinsic::vastart: 21117adc3a2bSChandler Carruth return false; 21127adc3a2bSChandler Carruth } 21137adc3a2bSChandler Carruth } 21141636651eSFlorian Hahn } 21157adc3a2bSChandler Carruth 21167adc3a2bSChandler Carruth return true; 21177adc3a2bSChandler Carruth } 21181c57cc2bSEaswaran Raman 21191c57cc2bSEaswaran Raman // APIs to create InlineParams based on command line flags and/or other 21201c57cc2bSEaswaran Raman // parameters. 21211c57cc2bSEaswaran Raman 21221c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams(int Threshold) { 21231c57cc2bSEaswaran Raman InlineParams Params; 21241c57cc2bSEaswaran Raman 21251c57cc2bSEaswaran Raman // This field is the threshold to use for a callee by default. This is 21261c57cc2bSEaswaran Raman // derived from one or more of: 21271c57cc2bSEaswaran Raman // * optimization or size-optimization levels, 21281c57cc2bSEaswaran Raman // * a value passed to createFunctionInliningPass function, or 21291c57cc2bSEaswaran Raman // * the -inline-threshold flag. 21301c57cc2bSEaswaran Raman // If the -inline-threshold flag is explicitly specified, that is used 21311c57cc2bSEaswaran Raman // irrespective of anything else. 21321c57cc2bSEaswaran Raman if (InlineThreshold.getNumOccurrences() > 0) 21331c57cc2bSEaswaran Raman Params.DefaultThreshold = InlineThreshold; 21341c57cc2bSEaswaran Raman else 21351c57cc2bSEaswaran Raman Params.DefaultThreshold = Threshold; 21361c57cc2bSEaswaran Raman 21371c57cc2bSEaswaran Raman // Set the HintThreshold knob from the -inlinehint-threshold. 21381c57cc2bSEaswaran Raman Params.HintThreshold = HintThreshold; 21391c57cc2bSEaswaran Raman 21401c57cc2bSEaswaran Raman // Set the HotCallSiteThreshold knob from the -hot-callsite-threshold. 21411c57cc2bSEaswaran Raman Params.HotCallSiteThreshold = HotCallSiteThreshold; 21421c57cc2bSEaswaran Raman 2143974d4eeaSEaswaran Raman // If the -locally-hot-callsite-threshold is explicitly specified, use it to 2144974d4eeaSEaswaran Raman // populate LocallyHotCallSiteThreshold. Later, we populate 2145974d4eeaSEaswaran Raman // Params.LocallyHotCallSiteThreshold from -locally-hot-callsite-threshold if 2146974d4eeaSEaswaran Raman // we know that optimization level is O3 (in the getInlineParams variant that 2147974d4eeaSEaswaran Raman // takes the opt and size levels). 2148974d4eeaSEaswaran Raman // FIXME: Remove this check (and make the assignment unconditional) after 2149974d4eeaSEaswaran Raman // addressing size regression issues at O2. 2150974d4eeaSEaswaran Raman if (LocallyHotCallSiteThreshold.getNumOccurrences() > 0) 2151974d4eeaSEaswaran Raman Params.LocallyHotCallSiteThreshold = LocallyHotCallSiteThreshold; 2152974d4eeaSEaswaran Raman 215312585b01SEaswaran Raman // Set the ColdCallSiteThreshold knob from the -inline-cold-callsite-threshold. 215412585b01SEaswaran Raman Params.ColdCallSiteThreshold = ColdCallSiteThreshold; 215512585b01SEaswaran Raman 21561c57cc2bSEaswaran Raman // Set the OptMinSizeThreshold and OptSizeThreshold params only if the 21571c57cc2bSEaswaran Raman // -inlinehint-threshold commandline option is not explicitly given. If that 21581c57cc2bSEaswaran Raman // option is present, then its value applies even for callees with size and 21591c57cc2bSEaswaran Raman // minsize attributes. 21601c57cc2bSEaswaran Raman // If the -inline-threshold is not specified, set the ColdThreshold from the 21611c57cc2bSEaswaran Raman // -inlinecold-threshold even if it is not explicitly passed. If 21621c57cc2bSEaswaran Raman // -inline-threshold is specified, then -inlinecold-threshold needs to be 21631c57cc2bSEaswaran Raman // explicitly specified to set the ColdThreshold knob 21641c57cc2bSEaswaran Raman if (InlineThreshold.getNumOccurrences() == 0) { 21651c57cc2bSEaswaran Raman Params.OptMinSizeThreshold = InlineConstants::OptMinSizeThreshold; 21661c57cc2bSEaswaran Raman Params.OptSizeThreshold = InlineConstants::OptSizeThreshold; 21671c57cc2bSEaswaran Raman Params.ColdThreshold = ColdThreshold; 21681c57cc2bSEaswaran Raman } else if (ColdThreshold.getNumOccurrences() > 0) { 21691c57cc2bSEaswaran Raman Params.ColdThreshold = ColdThreshold; 21701c57cc2bSEaswaran Raman } 21711c57cc2bSEaswaran Raman return Params; 21721c57cc2bSEaswaran Raman } 21731c57cc2bSEaswaran Raman 21741c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams() { 21751c57cc2bSEaswaran Raman return getInlineParams(InlineThreshold); 21761c57cc2bSEaswaran Raman } 21771c57cc2bSEaswaran Raman 21781c57cc2bSEaswaran Raman // Compute the default threshold for inlining based on the opt level and the 21791c57cc2bSEaswaran Raman // size opt level. 21801c57cc2bSEaswaran Raman static int computeThresholdFromOptLevels(unsigned OptLevel, 21811c57cc2bSEaswaran Raman unsigned SizeOptLevel) { 21821c57cc2bSEaswaran Raman if (OptLevel > 2) 21831c57cc2bSEaswaran Raman return InlineConstants::OptAggressiveThreshold; 21841c57cc2bSEaswaran Raman if (SizeOptLevel == 1) // -Os 21851c57cc2bSEaswaran Raman return InlineConstants::OptSizeThreshold; 21861c57cc2bSEaswaran Raman if (SizeOptLevel == 2) // -Oz 21871c57cc2bSEaswaran Raman return InlineConstants::OptMinSizeThreshold; 21881c57cc2bSEaswaran Raman return InlineThreshold; 21891c57cc2bSEaswaran Raman } 21901c57cc2bSEaswaran Raman 21911c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams(unsigned OptLevel, unsigned SizeOptLevel) { 2192974d4eeaSEaswaran Raman auto Params = 2193974d4eeaSEaswaran Raman getInlineParams(computeThresholdFromOptLevels(OptLevel, SizeOptLevel)); 2194974d4eeaSEaswaran Raman // At O3, use the value of -locally-hot-callsite-threshold option to populate 2195974d4eeaSEaswaran Raman // Params.LocallyHotCallSiteThreshold. Below O3, this flag has effect only 2196974d4eeaSEaswaran Raman // when it is specified explicitly. 2197974d4eeaSEaswaran Raman if (OptLevel > 2) 2198974d4eeaSEaswaran Raman Params.LocallyHotCallSiteThreshold = LocallyHotCallSiteThreshold; 2199974d4eeaSEaswaran Raman return Params; 22001c57cc2bSEaswaran Raman } 2201