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