17adc3a2bSChandler Carruth //===- InlineCost.cpp - Cost analysis for inliner -------------------------===// 27adc3a2bSChandler Carruth // 32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information. 52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 67adc3a2bSChandler Carruth // 77adc3a2bSChandler Carruth //===----------------------------------------------------------------------===// 87adc3a2bSChandler Carruth // 97adc3a2bSChandler Carruth // This file implements inline cost analysis. 107adc3a2bSChandler Carruth // 117adc3a2bSChandler Carruth //===----------------------------------------------------------------------===// 127adc3a2bSChandler Carruth 137adc3a2bSChandler Carruth #include "llvm/Analysis/InlineCost.h" 147adc3a2bSChandler Carruth #include "llvm/ADT/STLExtras.h" 157adc3a2bSChandler Carruth #include "llvm/ADT/SetVector.h" 167adc3a2bSChandler Carruth #include "llvm/ADT/SmallPtrSet.h" 177adc3a2bSChandler Carruth #include "llvm/ADT/SmallVector.h" 187adc3a2bSChandler Carruth #include "llvm/ADT/Statistic.h" 19aec2fa35SDaniel Jasper #include "llvm/Analysis/AssumptionCache.h" 2012585b01SEaswaran Raman #include "llvm/Analysis/BlockFrequencyInfo.h" 214c1a1d3cSReid Kleckner #include "llvm/Analysis/CFG.h" 227adc3a2bSChandler Carruth #include "llvm/Analysis/CodeMetrics.h" 237adc3a2bSChandler Carruth #include "llvm/Analysis/ConstantFolding.h" 247adc3a2bSChandler Carruth #include "llvm/Analysis/InstructionSimplify.h" 25ba9f245bSDavid Green #include "llvm/Analysis/LoopInfo.h" 2671069cf6SEaswaran Raman #include "llvm/Analysis/ProfileSummaryInfo.h" 27f9ca75f1STeresa Johnson #include "llvm/Analysis/TargetLibraryInfo.h" 287adc3a2bSChandler Carruth #include "llvm/Analysis/TargetTransformInfo.h" 29a4461515SHaicheng Wu #include "llvm/Analysis/ValueTracking.h" 30432a3883SNico Weber #include "llvm/Config/llvm-config.h" 31c965fd94SKirill Naumov #include "llvm/IR/AssemblyAnnotationWriter.h" 327adc3a2bSChandler Carruth #include "llvm/IR/CallingConv.h" 337adc3a2bSChandler Carruth #include "llvm/IR/DataLayout.h" 34ba9f245bSDavid Green #include "llvm/IR/Dominators.h" 357adc3a2bSChandler Carruth #include "llvm/IR/GetElementPtrTypeIterator.h" 367adc3a2bSChandler Carruth #include "llvm/IR/GlobalAlias.h" 377adc3a2bSChandler Carruth #include "llvm/IR/InstVisitor.h" 387adc3a2bSChandler Carruth #include "llvm/IR/IntrinsicInst.h" 397adc3a2bSChandler Carruth #include "llvm/IR/Operator.h" 407cebf0afSCraig Topper #include "llvm/IR/PatternMatch.h" 414c1a1d3cSReid Kleckner #include "llvm/Support/CommandLine.h" 427adc3a2bSChandler Carruth #include "llvm/Support/Debug.h" 43c965fd94SKirill Naumov #include "llvm/Support/FormattedStream.h" 447adc3a2bSChandler Carruth #include "llvm/Support/raw_ostream.h" 457adc3a2bSChandler Carruth 467adc3a2bSChandler Carruth using namespace llvm; 477adc3a2bSChandler Carruth 487adc3a2bSChandler Carruth #define DEBUG_TYPE "inline-cost" 497adc3a2bSChandler Carruth 507adc3a2bSChandler Carruth STATISTIC(NumCallsAnalyzed, "Number of call sites analyzed"); 517adc3a2bSChandler Carruth 527f37a802STeresa Johnson static cl::opt<int> 537f37a802STeresa Johnson DefaultThreshold("inlinedefault-threshold", cl::Hidden, cl::init(225), 547f37a802STeresa Johnson cl::ZeroOrMore, 557f37a802STeresa Johnson cl::desc("Default amount of inlining to perform")); 567f37a802STeresa Johnson 571022b5ebSKirill Naumov static cl::opt<bool> PrintInstructionComments( 581022b5ebSKirill Naumov "print-instruction-comments", cl::Hidden, cl::init(false), 591022b5ebSKirill Naumov cl::desc("Prints comments for instruction based on inline cost analysis")); 60c965fd94SKirill Naumov 611c57cc2bSEaswaran Raman static cl::opt<int> InlineThreshold( 62f4bb2f0dSEaswaran Raman "inline-threshold", cl::Hidden, cl::init(225), cl::ZeroOrMore, 63f4bb2f0dSEaswaran Raman cl::desc("Control the amount of inlining to perform (default = 225)")); 64f4bb2f0dSEaswaran Raman 65f4bb2f0dSEaswaran Raman static cl::opt<int> HintThreshold( 66fd3e7a93SDavid Callahan "inlinehint-threshold", cl::Hidden, cl::init(325), cl::ZeroOrMore, 67f4bb2f0dSEaswaran Raman cl::desc("Threshold for inlining functions with inline hint")); 68f4bb2f0dSEaswaran Raman 6912585b01SEaswaran Raman static cl::opt<int> 7012585b01SEaswaran Raman ColdCallSiteThreshold("inline-cold-callsite-threshold", cl::Hidden, 71fd3e7a93SDavid Callahan cl::init(45), cl::ZeroOrMore, 7212585b01SEaswaran Raman cl::desc("Threshold for inlining cold callsites")); 7312585b01SEaswaran Raman 749895c701SKazu Hirata static cl::opt<bool> InlineEnableCostBenefitAnalysis( 759895c701SKazu Hirata "inline-enable-cost-benefit-analysis", cl::Hidden, cl::init(false), 769895c701SKazu Hirata cl::desc("Enable the cost-benefit analysis for the inliner")); 779895c701SKazu Hirata 789895c701SKazu Hirata static cl::opt<int> InlineSavingsMultiplier( 799895c701SKazu Hirata "inline-savings-multiplier", cl::Hidden, cl::init(8), cl::ZeroOrMore, 809895c701SKazu Hirata cl::desc("Multiplier to multiply cycle savings by during inlining")); 819895c701SKazu Hirata 829895c701SKazu Hirata static cl::opt<int> 839895c701SKazu Hirata InlineSizeAllowance("inline-size-allowance", cl::Hidden, cl::init(100), 849895c701SKazu Hirata cl::ZeroOrMore, 859895c701SKazu Hirata cl::desc("The maximum size of a callee that get's " 869895c701SKazu Hirata "inlined without sufficient cycle savings")); 879895c701SKazu Hirata 88f4bb2f0dSEaswaran Raman // We introduce this threshold to help performance of instrumentation based 89f4bb2f0dSEaswaran Raman // PGO before we actually hook up inliner with analysis passes such as BPI and 90f4bb2f0dSEaswaran Raman // BFI. 91f4bb2f0dSEaswaran Raman static cl::opt<int> ColdThreshold( 92fd3e7a93SDavid Callahan "inlinecold-threshold", cl::Hidden, cl::init(45), cl::ZeroOrMore, 93f4bb2f0dSEaswaran Raman cl::desc("Threshold for inlining functions with cold attribute")); 94f4bb2f0dSEaswaran Raman 95de39cb93SDehao Chen static cl::opt<int> 96de39cb93SDehao Chen HotCallSiteThreshold("hot-callsite-threshold", cl::Hidden, cl::init(3000), 97de39cb93SDehao Chen cl::ZeroOrMore, 98de39cb93SDehao Chen cl::desc("Threshold for hot callsites ")); 99de39cb93SDehao Chen 100974d4eeaSEaswaran Raman static cl::opt<int> LocallyHotCallSiteThreshold( 101974d4eeaSEaswaran Raman "locally-hot-callsite-threshold", cl::Hidden, cl::init(525), cl::ZeroOrMore, 102974d4eeaSEaswaran Raman cl::desc("Threshold for locally hot callsites ")); 103974d4eeaSEaswaran Raman 104c5fa6358SEaswaran Raman static cl::opt<int> ColdCallSiteRelFreq( 105c5fa6358SEaswaran Raman "cold-callsite-rel-freq", cl::Hidden, cl::init(2), cl::ZeroOrMore, 10602a2bb2fSHiroshi Inoue cl::desc("Maximum block frequency, expressed as a percentage of caller's " 107c5fa6358SEaswaran Raman "entry frequency, for a callsite to be cold in the absence of " 108c5fa6358SEaswaran Raman "profile information.")); 109c5fa6358SEaswaran Raman 110974d4eeaSEaswaran Raman static cl::opt<int> HotCallSiteRelFreq( 111974d4eeaSEaswaran Raman "hot-callsite-rel-freq", cl::Hidden, cl::init(60), cl::ZeroOrMore, 112ff77cc75SEaswaran Raman cl::desc("Minimum block frequency, expressed as a multiple of caller's " 113974d4eeaSEaswaran Raman "entry frequency, for a callsite to be hot in the absence of " 114974d4eeaSEaswaran Raman "profile information.")); 115974d4eeaSEaswaran Raman 11646c03668SPhilipp Krones static cl::opt<int> CallPenalty( 11746c03668SPhilipp Krones "inline-call-penalty", cl::Hidden, cl::init(25), 11846c03668SPhilipp Krones cl::desc("Call penalty that is applied per callsite when inlining")); 11946c03668SPhilipp Krones 1204924bb00SEaswaran Raman static cl::opt<bool> OptComputeFullInlineCost( 121fd3e7a93SDavid Callahan "inline-cost-full", cl::Hidden, cl::init(false), cl::ZeroOrMore, 1220812c5beSHaicheng Wu cl::desc("Compute the full inline cost of a call site even when the cost " 1230812c5beSHaicheng Wu "exceeds the threshold.")); 1240812c5beSHaicheng Wu 125ed03d948SMehdi Amini static cl::opt<bool> InlineCallerSupersetNoBuiltin( 126ed03d948SMehdi Amini "inline-caller-superset-nobuiltin", cl::Hidden, cl::init(true), 127ed03d948SMehdi Amini cl::ZeroOrMore, 128ed03d948SMehdi Amini cl::desc("Allow inlining when caller has a superset of callee's nobuiltin " 129f9ca75f1STeresa Johnson "attributes.")); 130f9ca75f1STeresa Johnson 131d48c7859SKirill Naumov static cl::opt<bool> DisableGEPConstOperand( 132d48c7859SKirill Naumov "disable-gep-const-evaluation", cl::Hidden, cl::init(false), 133d48c7859SKirill Naumov cl::desc("Disables evaluation of GetElementPtr with constant operands")); 134d48c7859SKirill Naumov 1357adc3a2bSChandler Carruth namespace { 13606408758SMircea Trofin class InlineCostCallAnalyzer; 137c965fd94SKirill Naumov 1385c97507eSDaniil Suchkov /// This function behaves more like CallBase::hasFnAttr: when it looks for the 1395c97507eSDaniil Suchkov /// requested attribute, it check both the call instruction and the called 1405c97507eSDaniil Suchkov /// function (if it's available and operand bundles don't prohibit that). 1415c97507eSDaniil Suchkov Attribute getFnAttr(CallBase &CB, StringRef AttrKind) { 1425c97507eSDaniil Suchkov Attribute CallAttr = CB.getFnAttr(AttrKind); 1435c97507eSDaniil Suchkov if (CallAttr.isValid()) 1445c97507eSDaniil Suchkov return CallAttr; 1455c97507eSDaniil Suchkov 1465c97507eSDaniil Suchkov // Operand bundles override attributes on the called function, but don't 1475c97507eSDaniil Suchkov // override attributes directly present on the call instruction. 1485c97507eSDaniil Suchkov if (!CB.isFnAttrDisallowedByOpBundle(AttrKind)) 1495c97507eSDaniil Suchkov if (const Function *F = CB.getCalledFunction()) 1505c97507eSDaniil Suchkov return F->getFnAttribute(AttrKind); 1515c97507eSDaniil Suchkov 1525c97507eSDaniil Suchkov return {}; 1535c97507eSDaniil Suchkov } 1545c97507eSDaniil Suchkov 1555c97507eSDaniil Suchkov Optional<int> getStringFnAttrAsInt(CallBase &CB, StringRef AttrKind) { 1565c97507eSDaniil Suchkov Attribute Attr = getFnAttr(CB, AttrKind); 1575c97507eSDaniil Suchkov int AttrValue; 1585c97507eSDaniil Suchkov if (Attr.getValueAsString().getAsInteger(10, AttrValue)) 1595c97507eSDaniil Suchkov return None; 1605c97507eSDaniil Suchkov return AttrValue; 1615c97507eSDaniil Suchkov } 1625c97507eSDaniil Suchkov 163c965fd94SKirill Naumov // This struct is used to store information about inline cost of a 164c965fd94SKirill Naumov // particular instruction 165c965fd94SKirill Naumov struct InstructionCostDetail { 166b4924f01SMircea Trofin int CostBefore = 0; 167b4924f01SMircea Trofin int CostAfter = 0; 168b4924f01SMircea Trofin int ThresholdBefore = 0; 169b4924f01SMircea Trofin int ThresholdAfter = 0; 170b4924f01SMircea Trofin 171b4924f01SMircea Trofin int getThresholdDelta() const { return ThresholdAfter - ThresholdBefore; } 172b4924f01SMircea Trofin 173b4924f01SMircea Trofin int getCostDelta() const { return CostAfter - CostBefore; } 174b4924f01SMircea Trofin 175b4924f01SMircea Trofin bool hasThresholdChanged() const { return ThresholdAfter != ThresholdBefore; } 176c965fd94SKirill Naumov }; 177c965fd94SKirill Naumov 1781022b5ebSKirill Naumov class InlineCostAnnotationWriter : public AssemblyAnnotationWriter { 1791022b5ebSKirill Naumov private: 1801022b5ebSKirill Naumov InlineCostCallAnalyzer *const ICCA; 181c965fd94SKirill Naumov 1821022b5ebSKirill Naumov public: 1831022b5ebSKirill Naumov InlineCostAnnotationWriter(InlineCostCallAnalyzer *ICCA) : ICCA(ICCA) {} 184c965fd94SKirill Naumov virtual void emitInstructionAnnot(const Instruction *I, 1851022b5ebSKirill Naumov formatted_raw_ostream &OS) override; 186c965fd94SKirill Naumov }; 187c965fd94SKirill Naumov 18808e2386dSMircea Trofin /// Carry out call site analysis, in order to evaluate inlinability. 18908e2386dSMircea Trofin /// NOTE: the type is currently used as implementation detail of functions such 19008e2386dSMircea Trofin /// as llvm::getInlineCost. Note the function_ref constructor parameters - the 19108e2386dSMircea Trofin /// expectation is that they come from the outer scope, from the wrapper 19208e2386dSMircea Trofin /// functions. If we want to support constructing CallAnalyzer objects where 19308e2386dSMircea Trofin /// lambdas are provided inline at construction, or where the object needs to 19408e2386dSMircea Trofin /// otherwise survive past the scope of the provided functions, we need to 19508e2386dSMircea Trofin /// revisit the argument types. 1967adc3a2bSChandler Carruth class CallAnalyzer : public InstVisitor<CallAnalyzer, bool> { 1977adc3a2bSChandler Carruth typedef InstVisitor<CallAnalyzer, bool> Base; 1987adc3a2bSChandler Carruth friend class InstVisitor<CallAnalyzer, bool>; 1997adc3a2bSChandler Carruth 20006408758SMircea Trofin protected: 20106408758SMircea Trofin virtual ~CallAnalyzer() {} 2027adc3a2bSChandler Carruth /// The TargetTransformInfo available for this compilation. 2037adc3a2bSChandler Carruth const TargetTransformInfo &TTI; 2047adc3a2bSChandler Carruth 205aec2fa35SDaniel Jasper /// Getter for the cache of @llvm.assume intrinsics. 20608e2386dSMircea Trofin function_ref<AssumptionCache &(Function &)> GetAssumptionCache; 207aec2fa35SDaniel Jasper 20812585b01SEaswaran Raman /// Getter for BlockFrequencyInfo 20908e2386dSMircea Trofin function_ref<BlockFrequencyInfo &(Function &)> GetBFI; 21012585b01SEaswaran Raman 21171069cf6SEaswaran Raman /// Profile summary information. 21271069cf6SEaswaran Raman ProfileSummaryInfo *PSI; 21371069cf6SEaswaran Raman 214f3d122cdSPiotr Padlewski /// The called function. 2157adc3a2bSChandler Carruth Function &F; 2167adc3a2bSChandler Carruth 21785be8ca8SEric Christopher // Cache the DataLayout since we use it a lot. 21885be8ca8SEric Christopher const DataLayout &DL; 21985be8ca8SEric Christopher 2200812c5beSHaicheng Wu /// The OptimizationRemarkEmitter available for this compilation. 2210812c5beSHaicheng Wu OptimizationRemarkEmitter *ORE; 2220812c5beSHaicheng Wu 223f3d122cdSPiotr Padlewski /// The candidate callsite being analyzed. Please do not use this to do 224f3d122cdSPiotr Padlewski /// analysis in the caller function; we want the inline cost query to be 225f3d122cdSPiotr Padlewski /// easily cacheable. Instead, use the cover function paramHasAttr. 226652168a9SFedor Sergeev CallBase &CandidateCall; 2277adc3a2bSChandler Carruth 22806408758SMircea Trofin /// Extension points for handling callsite features. 2299895c701SKazu Hirata // Called before a basic block was analyzed. 2309895c701SKazu Hirata virtual void onBlockStart(const BasicBlock *BB) {} 2319895c701SKazu Hirata 23206408758SMircea Trofin /// Called after a basic block was analyzed. 23306408758SMircea Trofin virtual void onBlockAnalyzed(const BasicBlock *BB) {} 2341c57cc2bSEaswaran Raman 235c965fd94SKirill Naumov /// Called before an instruction was analyzed 236c965fd94SKirill Naumov virtual void onInstructionAnalysisStart(const Instruction *I) {} 237c965fd94SKirill Naumov 238c965fd94SKirill Naumov /// Called after an instruction was analyzed 239c965fd94SKirill Naumov virtual void onInstructionAnalysisFinish(const Instruction *I) {} 240c965fd94SKirill Naumov 24106408758SMircea Trofin /// Called at the end of the analysis of the callsite. Return the outcome of 24206408758SMircea Trofin /// the analysis, i.e. 'InlineResult(true)' if the inlining may happen, or 24306408758SMircea Trofin /// the reason it can't. 2445466597fSMircea Trofin virtual InlineResult finalizeAnalysis() { return InlineResult::success(); } 24506408758SMircea Trofin /// Called when we're about to start processing a basic block, and every time 24606408758SMircea Trofin /// we are done processing an instruction. Return true if there is no point in 24706408758SMircea Trofin /// continuing the analysis (e.g. we've determined already the call site is 24806408758SMircea Trofin /// too expensive to inline) 24906408758SMircea Trofin virtual bool shouldStop() { return false; } 250eeae45dcSFedor Sergeev 25106408758SMircea Trofin /// Called before the analysis of the callee body starts (with callsite 25206408758SMircea Trofin /// contexts propagated). It checks callsite-specific information. Return a 25306408758SMircea Trofin /// reason analysis can't continue if that's the case, or 'true' if it may 25406408758SMircea Trofin /// continue. 2555466597fSMircea Trofin virtual InlineResult onAnalysisStart() { return InlineResult::success(); } 25606408758SMircea Trofin /// Called if the analysis engine decides SROA cannot be done for the given 25706408758SMircea Trofin /// alloca. 25806408758SMircea Trofin virtual void onDisableSROA(AllocaInst *Arg) {} 25906408758SMircea Trofin 26006408758SMircea Trofin /// Called the analysis engine determines load elimination won't happen. 26106408758SMircea Trofin virtual void onDisableLoadElimination() {} 26206408758SMircea Trofin 2635c97507eSDaniil Suchkov /// Called when we visit a CallBase, before the analysis starts. Return false 2645c97507eSDaniil Suchkov /// to stop further processing of the instruction. 2655c97507eSDaniil Suchkov virtual bool onCallBaseVisitStart(CallBase &Call) { return true; } 2665c97507eSDaniil Suchkov 26706408758SMircea Trofin /// Called to account for a call. 26806408758SMircea Trofin virtual void onCallPenalty() {} 26906408758SMircea Trofin 27006408758SMircea Trofin /// Called to account for the expectation the inlining would result in a load 27106408758SMircea Trofin /// elimination. 27206408758SMircea Trofin virtual void onLoadEliminationOpportunity() {} 27306408758SMircea Trofin 27406408758SMircea Trofin /// Called to account for the cost of argument setup for the Call in the 27506408758SMircea Trofin /// callee's body (not the callsite currently under analysis). 27606408758SMircea Trofin virtual void onCallArgumentSetup(const CallBase &Call) {} 27706408758SMircea Trofin 27806408758SMircea Trofin /// Called to account for a load relative intrinsic. 27906408758SMircea Trofin virtual void onLoadRelativeIntrinsic() {} 28006408758SMircea Trofin 28106408758SMircea Trofin /// Called to account for a lowered call. 28206408758SMircea Trofin virtual void onLoweredCall(Function *F, CallBase &Call, bool IsIndirectCall) { 28306408758SMircea Trofin } 28406408758SMircea Trofin 28506408758SMircea Trofin /// Account for a jump table of given size. Return false to stop further 28606408758SMircea Trofin /// processing the switch instruction 28706408758SMircea Trofin virtual bool onJumpTable(unsigned JumpTableSize) { return true; } 28806408758SMircea Trofin 28906408758SMircea Trofin /// Account for a case cluster of given size. Return false to stop further 29006408758SMircea Trofin /// processing of the instruction. 29106408758SMircea Trofin virtual bool onCaseCluster(unsigned NumCaseCluster) { return true; } 29206408758SMircea Trofin 29306408758SMircea Trofin /// Called at the end of processing a switch instruction, with the given 29406408758SMircea Trofin /// number of case clusters. 29506408758SMircea Trofin virtual void onFinalizeSwitch(unsigned JumpTableSize, 29606408758SMircea Trofin unsigned NumCaseCluster) {} 29706408758SMircea Trofin 29806408758SMircea Trofin /// Called to account for any other instruction not specifically accounted 29906408758SMircea Trofin /// for. 30014a16faeSMircea Trofin virtual void onMissedSimplification() {} 30106408758SMircea Trofin 30206408758SMircea Trofin /// Start accounting potential benefits due to SROA for the given alloca. 30306408758SMircea Trofin virtual void onInitializeSROAArg(AllocaInst *Arg) {} 30406408758SMircea Trofin 30506408758SMircea Trofin /// Account SROA savings for the AllocaInst value. 30606408758SMircea Trofin virtual void onAggregateSROAUse(AllocaInst *V) {} 30706408758SMircea Trofin 30806408758SMircea Trofin bool handleSROA(Value *V, bool DoNotDisable) { 30906408758SMircea Trofin // Check for SROA candidates in comparisons. 31006408758SMircea Trofin if (auto *SROAArg = getSROAArgForValueOrNull(V)) { 31106408758SMircea Trofin if (DoNotDisable) { 31206408758SMircea Trofin onAggregateSROAUse(SROAArg); 31306408758SMircea Trofin return true; 31406408758SMircea Trofin } 31506408758SMircea Trofin disableSROAForArg(SROAArg); 31606408758SMircea Trofin } 31706408758SMircea Trofin return false; 31806408758SMircea Trofin } 3197adc3a2bSChandler Carruth 320eeae45dcSFedor Sergeev bool IsCallerRecursive = false; 321eeae45dcSFedor Sergeev bool IsRecursiveCall = false; 322eeae45dcSFedor Sergeev bool ExposesReturnsTwice = false; 323eeae45dcSFedor Sergeev bool HasDynamicAlloca = false; 324eeae45dcSFedor Sergeev bool ContainsNoDuplicateCall = false; 325eeae45dcSFedor Sergeev bool HasReturn = false; 326eeae45dcSFedor Sergeev bool HasIndirectBr = false; 327eeae45dcSFedor Sergeev bool HasUninlineableIntrinsic = false; 328eeae45dcSFedor Sergeev bool InitsVargArgs = false; 3297adc3a2bSChandler Carruth 3307adc3a2bSChandler Carruth /// Number of bytes allocated statically by the callee. 331eeae45dcSFedor Sergeev uint64_t AllocatedSize = 0; 332eeae45dcSFedor Sergeev unsigned NumInstructions = 0; 333eeae45dcSFedor Sergeev unsigned NumVectorInstructions = 0; 334eeae45dcSFedor Sergeev 335f3d122cdSPiotr Padlewski /// While we walk the potentially-inlined instructions, we build up and 336f3d122cdSPiotr Padlewski /// maintain a mapping of simplified values specific to this callsite. The 337f3d122cdSPiotr Padlewski /// idea is to propagate any special information we have about arguments to 338f3d122cdSPiotr Padlewski /// this call through the inlinable section of the function, and account for 339f3d122cdSPiotr Padlewski /// likely simplifications post-inlining. The most important aspect we track 340f3d122cdSPiotr Padlewski /// is CFG altering simplifications -- when we prove a basic block dead, that 341f3d122cdSPiotr Padlewski /// can cause dramatic shifts in the cost of inlining a function. 3427adc3a2bSChandler Carruth DenseMap<Value *, Constant *> SimplifiedValues; 3437adc3a2bSChandler Carruth 344f3d122cdSPiotr Padlewski /// Keep track of the values which map back (through function arguments) to 345f3d122cdSPiotr Padlewski /// allocas on the caller stack which could be simplified through SROA. 34606408758SMircea Trofin DenseMap<Value *, AllocaInst *> SROAArgValues; 347338a6016SMircea Trofin 34806408758SMircea Trofin /// Keep track of Allocas for which we believe we may get SROA optimization. 3492e42cc7aSMircea Trofin DenseSet<AllocaInst *> EnabledSROAAllocas; 3507adc3a2bSChandler Carruth 351f3d122cdSPiotr Padlewski /// Keep track of values which map to a pointer base and constant offset. 3527adc3a2bSChandler Carruth DenseMap<Value *, std::pair<Value *, APInt>> ConstantOffsetPtrs; 3537adc3a2bSChandler Carruth 3543739e14aSHaicheng Wu /// Keep track of dead blocks due to the constant arguments. 3553739e14aSHaicheng Wu SetVector<BasicBlock *> DeadBlocks; 3563739e14aSHaicheng Wu 3573739e14aSHaicheng Wu /// The mapping of the blocks to their known unique successors due to the 3583739e14aSHaicheng Wu /// constant arguments. 3593739e14aSHaicheng Wu DenseMap<BasicBlock *, BasicBlock *> KnownSuccessors; 3603739e14aSHaicheng Wu 361a4461515SHaicheng Wu /// Model the elimination of repeated loads that is expected to happen 362a4461515SHaicheng Wu /// whenever we simplify away the stores that would otherwise cause them to be 363a4461515SHaicheng Wu /// loads. 364a4461515SHaicheng Wu bool EnableLoadElimination; 365f7fff46aSWenlei He 366f7fff46aSWenlei He /// Whether we allow inlining for recursive call. 367f7fff46aSWenlei He bool AllowRecursiveCall; 368f7fff46aSWenlei He 369a4461515SHaicheng Wu SmallPtrSet<Value *, 16> LoadAddrSet; 37006408758SMircea Trofin 37106408758SMircea Trofin AllocaInst *getSROAArgForValueOrNull(Value *V) const { 37206408758SMircea Trofin auto It = SROAArgValues.find(V); 3732e42cc7aSMircea Trofin if (It == SROAArgValues.end() || EnabledSROAAllocas.count(It->second) == 0) 37406408758SMircea Trofin return nullptr; 37506408758SMircea Trofin return It->second; 37606408758SMircea Trofin } 377a4461515SHaicheng Wu 3787adc3a2bSChandler Carruth // Custom simplification helper routines. 3797adc3a2bSChandler Carruth bool isAllocaDerivedArg(Value *V); 38006408758SMircea Trofin void disableSROAForArg(AllocaInst *SROAArg); 3817adc3a2bSChandler Carruth void disableSROA(Value *V); 3823739e14aSHaicheng Wu void findDeadBlocks(BasicBlock *CurrBB, BasicBlock *NextBB); 383a4461515SHaicheng Wu void disableLoadElimination(); 384201b191bSHaicheng Wu bool isGEPFree(GetElementPtrInst &GEP); 385d3558b5dSEvgeny Astigeevich bool canFoldInboundsGEP(GetElementPtrInst &I); 3867adc3a2bSChandler Carruth bool accumulateGEPOffset(GEPOperator &GEP, APInt &Offset); 387652168a9SFedor Sergeev bool simplifyCallSite(Function *F, CallBase &Call); 388617f6364SEaswaran Raman template <typename Callable> 389617f6364SEaswaran Raman bool simplifyInstruction(Instruction &I, Callable Evaluate); 3909697f935SNick Desaulniers bool simplifyIntrinsicCallIsConstant(CallBase &CB); 3917adc3a2bSChandler Carruth ConstantInt *stripAndComputeInBoundsConstantOffsets(Value *&V); 3927adc3a2bSChandler Carruth 3937adc3a2bSChandler Carruth /// Return true if the given argument to the function being considered for 3947adc3a2bSChandler Carruth /// inlining has the given attribute set either at the call site or the 3957adc3a2bSChandler Carruth /// function declaration. Primarily used to inspect call site specific 3967adc3a2bSChandler Carruth /// attributes since these can be more precise than the ones on the callee 3977adc3a2bSChandler Carruth /// itself. 3987adc3a2bSChandler Carruth bool paramHasAttr(Argument *A, Attribute::AttrKind Attr); 3997adc3a2bSChandler Carruth 4007adc3a2bSChandler Carruth /// Return true if the given value is known non null within the callee if 4017adc3a2bSChandler Carruth /// inlined through this particular callsite. 4027adc3a2bSChandler Carruth bool isKnownNonNullInCallee(Value *V); 4037adc3a2bSChandler Carruth 404652168a9SFedor Sergeev /// Return true if size growth is allowed when inlining the callee at \p Call. 405652168a9SFedor Sergeev bool allowSizeGrowth(CallBase &Call); 4069a3fc17aSEaswaran Raman 4077adc3a2bSChandler Carruth // Custom analysis routines. 408c0aa4b75SDavid Bolvansky InlineResult analyzeBlock(BasicBlock *BB, 409c0aa4b75SDavid Bolvansky SmallPtrSetImpl<const Value *> &EphValues); 4107adc3a2bSChandler Carruth 4117adc3a2bSChandler Carruth // Disable several entry points to the visitor so we don't accidentally use 4127adc3a2bSChandler Carruth // them by declaring but not defining them here. 413567556aaSChad Rosier void visit(Module *); 414567556aaSChad Rosier void visit(Module &); 415567556aaSChad Rosier void visit(Function *); 416567556aaSChad Rosier void visit(Function &); 417567556aaSChad Rosier void visit(BasicBlock *); 418567556aaSChad Rosier void visit(BasicBlock &); 4197adc3a2bSChandler Carruth 4207adc3a2bSChandler Carruth // Provide base case for our instruction visit. 4217adc3a2bSChandler Carruth bool visitInstruction(Instruction &I); 4227adc3a2bSChandler Carruth 4237adc3a2bSChandler Carruth // Our visit overrides. 4247adc3a2bSChandler Carruth bool visitAlloca(AllocaInst &I); 4257adc3a2bSChandler Carruth bool visitPHI(PHINode &I); 4267adc3a2bSChandler Carruth bool visitGetElementPtr(GetElementPtrInst &I); 4277adc3a2bSChandler Carruth bool visitBitCast(BitCastInst &I); 4287adc3a2bSChandler Carruth bool visitPtrToInt(PtrToIntInst &I); 4297adc3a2bSChandler Carruth bool visitIntToPtr(IntToPtrInst &I); 4307adc3a2bSChandler Carruth bool visitCastInst(CastInst &I); 4317adc3a2bSChandler Carruth bool visitCmpInst(CmpInst &I); 4327adc3a2bSChandler Carruth bool visitSub(BinaryOperator &I); 4337adc3a2bSChandler Carruth bool visitBinaryOperator(BinaryOperator &I); 4346cda33baSCraig Topper bool visitFNeg(UnaryOperator &I); 4357adc3a2bSChandler Carruth bool visitLoad(LoadInst &I); 4367adc3a2bSChandler Carruth bool visitStore(StoreInst &I); 4377adc3a2bSChandler Carruth bool visitExtractValue(ExtractValueInst &I); 4387adc3a2bSChandler Carruth bool visitInsertValue(InsertValueInst &I); 439652168a9SFedor Sergeev bool visitCallBase(CallBase &Call); 4407adc3a2bSChandler Carruth bool visitReturnInst(ReturnInst &RI); 4417adc3a2bSChandler Carruth bool visitBranchInst(BranchInst &BI); 4423ec848bcSHaicheng Wu bool visitSelectInst(SelectInst &SI); 4437adc3a2bSChandler Carruth bool visitSwitchInst(SwitchInst &SI); 4447adc3a2bSChandler Carruth bool visitIndirectBrInst(IndirectBrInst &IBI); 4457adc3a2bSChandler Carruth bool visitResumeInst(ResumeInst &RI); 4467adc3a2bSChandler Carruth bool visitCleanupReturnInst(CleanupReturnInst &RI); 4477adc3a2bSChandler Carruth bool visitCatchReturnInst(CatchReturnInst &RI); 4487adc3a2bSChandler Carruth bool visitUnreachableInst(UnreachableInst &I); 4497adc3a2bSChandler Carruth 4507adc3a2bSChandler Carruth public: 4517b639f50SJacob Hegna CallAnalyzer(Function &Callee, CallBase &Call, const TargetTransformInfo &TTI, 45208e2386dSMircea Trofin function_ref<AssumptionCache &(Function &)> GetAssumptionCache, 45308e2386dSMircea Trofin function_ref<BlockFrequencyInfo &(Function &)> GetBFI = nullptr, 45408e2386dSMircea Trofin ProfileSummaryInfo *PSI = nullptr, 45508e2386dSMircea Trofin OptimizationRemarkEmitter *ORE = nullptr) 45612585b01SEaswaran Raman : TTI(TTI), GetAssumptionCache(GetAssumptionCache), GetBFI(GetBFI), 4570812c5beSHaicheng Wu PSI(PSI), F(Callee), DL(F.getParent()->getDataLayout()), ORE(ORE), 458f7fff46aSWenlei He CandidateCall(Call), EnableLoadElimination(true), 459f7fff46aSWenlei He AllowRecursiveCall(false) {} 4607adc3a2bSChandler Carruth 46193ac81ccSMircea Trofin InlineResult analyze(); 4627adc3a2bSChandler Carruth 4637f094f7fSKirill Naumov Optional<Constant *> getSimplifiedValue(Instruction *I) { 4647f094f7fSKirill Naumov if (SimplifiedValues.find(I) != SimplifiedValues.end()) 4657f094f7fSKirill Naumov return SimplifiedValues[I]; 4667f094f7fSKirill Naumov return None; 4677f094f7fSKirill Naumov } 4687f094f7fSKirill Naumov 4697adc3a2bSChandler Carruth // Keep a bunch of stats about the cost savings found so we can print them 4707adc3a2bSChandler Carruth // out when debugging. 471eeae45dcSFedor Sergeev unsigned NumConstantArgs = 0; 472eeae45dcSFedor Sergeev unsigned NumConstantOffsetPtrArgs = 0; 473eeae45dcSFedor Sergeev unsigned NumAllocaArgs = 0; 474eeae45dcSFedor Sergeev unsigned NumConstantPtrCmps = 0; 475eeae45dcSFedor Sergeev unsigned NumConstantPtrDiffs = 0; 476eeae45dcSFedor Sergeev unsigned NumInstructionsSimplified = 0; 47776aab66dSMircea Trofin 47876aab66dSMircea Trofin void dump(); 47976aab66dSMircea Trofin }; 480338a6016SMircea Trofin 48199f00635SJacob Hegna // Considering forming a binary search, we should find the number of nodes 48299f00635SJacob Hegna // which is same as the number of comparisons when lowered. For a given 48399f00635SJacob Hegna // number of clusters, n, we can define a recursive function, f(n), to find 48499f00635SJacob Hegna // the number of nodes in the tree. The recursion is : 48599f00635SJacob Hegna // f(n) = 1 + f(n/2) + f (n - n/2), when n > 3, 48699f00635SJacob Hegna // and f(n) = n, when n <= 3. 48799f00635SJacob Hegna // This will lead a binary tree where the leaf should be either f(2) or f(3) 48899f00635SJacob Hegna // when n > 3. So, the number of comparisons from leaves should be n, while 48999f00635SJacob Hegna // the number of non-leaf should be : 49099f00635SJacob Hegna // 2^(log2(n) - 1) - 1 49199f00635SJacob Hegna // = 2^log2(n) * 2^-1 - 1 49299f00635SJacob Hegna // = n / 2 - 1. 49399f00635SJacob Hegna // Considering comparisons from leaf and non-leaf nodes, we can estimate the 49499f00635SJacob Hegna // number of comparisons in a simple closed form : 49599f00635SJacob Hegna // n + n / 2 - 1 = n * 3 / 2 - 1 49699f00635SJacob Hegna int64_t getExpectedNumberOfCompare(int NumCaseCluster) { 49799f00635SJacob Hegna return 3 * static_cast<int64_t>(NumCaseCluster) / 2 - 1; 49899f00635SJacob Hegna } 49999f00635SJacob Hegna 50006408758SMircea Trofin /// FIXME: if it is necessary to derive from InlineCostCallAnalyzer, note 50106408758SMircea Trofin /// the FIXME in onLoweredCall, when instantiating an InlineCostCallAnalyzer 50206408758SMircea Trofin class InlineCostCallAnalyzer final : public CallAnalyzer { 50306408758SMircea Trofin const int CostUpperBound = INT_MAX - InlineConstants::InstrCost - 1; 50406408758SMircea Trofin const bool ComputeFullInlineCost; 50506408758SMircea Trofin int LoadEliminationCost = 0; 50606408758SMircea Trofin /// Bonus to be applied when percentage of vector instructions in callee is 50706408758SMircea Trofin /// high (see more details in updateThreshold). 50806408758SMircea Trofin int VectorBonus = 0; 50906408758SMircea Trofin /// Bonus to be applied when the callee has only one reachable basic block. 51006408758SMircea Trofin int SingleBBBonus = 0; 51106408758SMircea Trofin 51206408758SMircea Trofin /// Tunable parameters that control the analysis. 51306408758SMircea Trofin const InlineParams &Params; 51406408758SMircea Trofin 5151022b5ebSKirill Naumov // This DenseMap stores the delta change in cost and threshold after 5161022b5ebSKirill Naumov // accounting for the given instruction. The map is filled only with the 5171022b5ebSKirill Naumov // flag PrintInstructionComments on. 5181022b5ebSKirill Naumov DenseMap<const Instruction *, InstructionCostDetail> InstructionCostDetailMap; 5191022b5ebSKirill Naumov 52006408758SMircea Trofin /// Upper bound for the inlining cost. Bonuses are being applied to account 52106408758SMircea Trofin /// for speculative "expected profit" of the inlining decision. 52206408758SMircea Trofin int Threshold = 0; 52306408758SMircea Trofin 52406408758SMircea Trofin /// Attempt to evaluate indirect calls to boost its inline cost. 52506408758SMircea Trofin const bool BoostIndirectCalls; 52606408758SMircea Trofin 5278a4013edSMircea Trofin /// Ignore the threshold when finalizing analysis. 5288a4013edSMircea Trofin const bool IgnoreThreshold; 5298a4013edSMircea Trofin 5309895c701SKazu Hirata // True if the cost-benefit-analysis-based inliner is enabled. 5319895c701SKazu Hirata const bool CostBenefitAnalysisEnabled; 5329895c701SKazu Hirata 53306408758SMircea Trofin /// Inlining cost measured in abstract units, accounts for all the 53406408758SMircea Trofin /// instructions expected to be executed for a given function invocation. 53506408758SMircea Trofin /// Instructions that are statically proven to be dead based on call-site 53606408758SMircea Trofin /// arguments are not counted here. 53706408758SMircea Trofin int Cost = 0; 53806408758SMircea Trofin 5399895c701SKazu Hirata // The cumulative cost at the beginning of the basic block being analyzed. At 5409895c701SKazu Hirata // the end of analyzing each basic block, "Cost - CostAtBBStart" represents 5419895c701SKazu Hirata // the size of that basic block. 5429895c701SKazu Hirata int CostAtBBStart = 0; 5439895c701SKazu Hirata 5449895c701SKazu Hirata // The static size of live but cold basic blocks. This is "static" in the 5459895c701SKazu Hirata // sense that it's not weighted by profile counts at all. 5469895c701SKazu Hirata int ColdSize = 0; 5479895c701SKazu Hirata 5486869e6c1SWenlei He // Whether inlining is decided by cost-benefit analysis. 5496869e6c1SWenlei He bool DecidedByCostBenefit = false; 5506869e6c1SWenlei He 5514bdfea2cSLiqiang Tao // The cost-benefit pair computed by cost-benefit analysis. 5524bdfea2cSLiqiang Tao Optional<CostBenefitPair> CostBenefit = None; 5534bdfea2cSLiqiang Tao 55406408758SMircea Trofin bool SingleBB = true; 55506408758SMircea Trofin 55606408758SMircea Trofin unsigned SROACostSavings = 0; 55706408758SMircea Trofin unsigned SROACostSavingsLost = 0; 55806408758SMircea Trofin 55906408758SMircea Trofin /// The mapping of caller Alloca values to their accumulated cost savings. If 56006408758SMircea Trofin /// we have to disable SROA for one of the allocas, this tells us how much 56106408758SMircea Trofin /// cost must be added. 56206408758SMircea Trofin DenseMap<AllocaInst *, int> SROAArgCosts; 56306408758SMircea Trofin 56406408758SMircea Trofin /// Return true if \p Call is a cold callsite. 56506408758SMircea Trofin bool isColdCallSite(CallBase &Call, BlockFrequencyInfo *CallerBFI); 56606408758SMircea Trofin 56706408758SMircea Trofin /// Update Threshold based on callsite properties such as callee 56806408758SMircea Trofin /// attributes and callee hotness for PGO builds. The Callee is explicitly 56906408758SMircea Trofin /// passed to support analyzing indirect calls whose target is inferred by 57006408758SMircea Trofin /// analysis. 57106408758SMircea Trofin void updateThreshold(CallBase &Call, Function &Callee); 57206408758SMircea Trofin /// Return a higher threshold if \p Call is a hot callsite. 57306408758SMircea Trofin Optional<int> getHotCallSiteThreshold(CallBase &Call, 57406408758SMircea Trofin BlockFrequencyInfo *CallerBFI); 57506408758SMircea Trofin 57606408758SMircea Trofin /// Handle a capped 'int' increment for Cost. 57706408758SMircea Trofin void addCost(int64_t Inc, int64_t UpperBound = INT_MAX) { 57806408758SMircea Trofin assert(UpperBound > 0 && UpperBound <= INT_MAX && "invalid upper bound"); 579cfc4def8SJacob Hegna Cost = std::min<int>(UpperBound, Cost + Inc); 58006408758SMircea Trofin } 58106408758SMircea Trofin 58206408758SMircea Trofin void onDisableSROA(AllocaInst *Arg) override { 58306408758SMircea Trofin auto CostIt = SROAArgCosts.find(Arg); 58406408758SMircea Trofin if (CostIt == SROAArgCosts.end()) 58506408758SMircea Trofin return; 58606408758SMircea Trofin addCost(CostIt->second); 58706408758SMircea Trofin SROACostSavings -= CostIt->second; 58806408758SMircea Trofin SROACostSavingsLost += CostIt->second; 58906408758SMircea Trofin SROAArgCosts.erase(CostIt); 59006408758SMircea Trofin } 59106408758SMircea Trofin 59206408758SMircea Trofin void onDisableLoadElimination() override { 59306408758SMircea Trofin addCost(LoadEliminationCost); 59406408758SMircea Trofin LoadEliminationCost = 0; 59506408758SMircea Trofin } 5965c97507eSDaniil Suchkov 5975c97507eSDaniil Suchkov bool onCallBaseVisitStart(CallBase &Call) override { 5985c97507eSDaniil Suchkov if (Optional<int> AttrCallThresholdBonus = 5995c97507eSDaniil Suchkov getStringFnAttrAsInt(Call, "call-threshold-bonus")) 6005c97507eSDaniil Suchkov Threshold += *AttrCallThresholdBonus; 6015c97507eSDaniil Suchkov 6025c97507eSDaniil Suchkov if (Optional<int> AttrCallCost = 6035c97507eSDaniil Suchkov getStringFnAttrAsInt(Call, "call-inline-cost")) { 6045c97507eSDaniil Suchkov addCost(*AttrCallCost); 6055c97507eSDaniil Suchkov // Prevent further processing of the call since we want to override its 6065c97507eSDaniil Suchkov // inline cost, not just add to it. 6075c97507eSDaniil Suchkov return false; 6085c97507eSDaniil Suchkov } 6095c97507eSDaniil Suchkov return true; 6105c97507eSDaniil Suchkov } 6115c97507eSDaniil Suchkov 61246c03668SPhilipp Krones void onCallPenalty() override { addCost(CallPenalty); } 61306408758SMircea Trofin void onCallArgumentSetup(const CallBase &Call) override { 61406408758SMircea Trofin // Pay the price of the argument setup. We account for the average 1 61506408758SMircea Trofin // instruction per call argument setup here. 61606408758SMircea Trofin addCost(Call.arg_size() * InlineConstants::InstrCost); 61706408758SMircea Trofin } 61806408758SMircea Trofin void onLoadRelativeIntrinsic() override { 61906408758SMircea Trofin // This is normally lowered to 4 LLVM instructions. 62006408758SMircea Trofin addCost(3 * InlineConstants::InstrCost); 62106408758SMircea Trofin } 62206408758SMircea Trofin void onLoweredCall(Function *F, CallBase &Call, 62306408758SMircea Trofin bool IsIndirectCall) override { 62406408758SMircea Trofin // We account for the average 1 instruction per call argument setup here. 62506408758SMircea Trofin addCost(Call.arg_size() * InlineConstants::InstrCost); 62606408758SMircea Trofin 62706408758SMircea Trofin // If we have a constant that we are calling as a function, we can peer 62806408758SMircea Trofin // through it and see the function target. This happens not infrequently 62906408758SMircea Trofin // during devirtualization and so we want to give it a hefty bonus for 63006408758SMircea Trofin // inlining, but cap that bonus in the event that inlining wouldn't pan out. 63106408758SMircea Trofin // Pretend to inline the function, with a custom threshold. 63206408758SMircea Trofin if (IsIndirectCall && BoostIndirectCalls) { 63306408758SMircea Trofin auto IndirectCallParams = Params; 63406408758SMircea Trofin IndirectCallParams.DefaultThreshold = 63506408758SMircea Trofin InlineConstants::IndirectCallThreshold; 63606408758SMircea Trofin /// FIXME: if InlineCostCallAnalyzer is derived from, this may need 63706408758SMircea Trofin /// to instantiate the derived class. 63808e2386dSMircea Trofin InlineCostCallAnalyzer CA(*F, Call, IndirectCallParams, TTI, 63908e2386dSMircea Trofin GetAssumptionCache, GetBFI, PSI, ORE, false); 6405466597fSMircea Trofin if (CA.analyze().isSuccess()) { 64106408758SMircea Trofin // We were able to inline the indirect call! Subtract the cost from the 64206408758SMircea Trofin // threshold to get the bonus we want to apply, but don't go below zero. 64306408758SMircea Trofin Cost -= std::max(0, CA.getThreshold() - CA.getCost()); 64406408758SMircea Trofin } 64506408758SMircea Trofin } else 64606408758SMircea Trofin // Otherwise simply add the cost for merely making the call. 64746c03668SPhilipp Krones addCost(CallPenalty); 64806408758SMircea Trofin } 64906408758SMircea Trofin 65006408758SMircea Trofin void onFinalizeSwitch(unsigned JumpTableSize, 65106408758SMircea Trofin unsigned NumCaseCluster) override { 65206408758SMircea Trofin // If suitable for a jump table, consider the cost for the table size and 65306408758SMircea Trofin // branch to destination. 65406408758SMircea Trofin // Maximum valid cost increased in this function. 65506408758SMircea Trofin if (JumpTableSize) { 656cfc4def8SJacob Hegna int64_t JTCost = 657cfc4def8SJacob Hegna static_cast<int64_t>(JumpTableSize) * InlineConstants::InstrCost + 65806408758SMircea Trofin 4 * InlineConstants::InstrCost; 65906408758SMircea Trofin 660cfc4def8SJacob Hegna addCost(JTCost, static_cast<int64_t>(CostUpperBound)); 66106408758SMircea Trofin return; 66206408758SMircea Trofin } 66399f00635SJacob Hegna 66406408758SMircea Trofin if (NumCaseCluster <= 3) { 66506408758SMircea Trofin // Suppose a comparison includes one compare and one conditional branch. 66606408758SMircea Trofin addCost(NumCaseCluster * 2 * InlineConstants::InstrCost); 66706408758SMircea Trofin return; 66806408758SMircea Trofin } 66906408758SMircea Trofin 67099f00635SJacob Hegna int64_t ExpectedNumberOfCompare = 67199f00635SJacob Hegna getExpectedNumberOfCompare(NumCaseCluster); 67206408758SMircea Trofin int64_t SwitchCost = 67306408758SMircea Trofin ExpectedNumberOfCompare * 2 * InlineConstants::InstrCost; 67406408758SMircea Trofin 675cfc4def8SJacob Hegna addCost(SwitchCost, static_cast<int64_t>(CostUpperBound)); 67606408758SMircea Trofin } 67714a16faeSMircea Trofin void onMissedSimplification() override { 67806408758SMircea Trofin addCost(InlineConstants::InstrCost); 67906408758SMircea Trofin } 68006408758SMircea Trofin 68106408758SMircea Trofin void onInitializeSROAArg(AllocaInst *Arg) override { 68206408758SMircea Trofin assert(Arg != nullptr && 68306408758SMircea Trofin "Should not initialize SROA costs for null value."); 68406408758SMircea Trofin SROAArgCosts[Arg] = 0; 68506408758SMircea Trofin } 68606408758SMircea Trofin 68706408758SMircea Trofin void onAggregateSROAUse(AllocaInst *SROAArg) override { 68806408758SMircea Trofin auto CostIt = SROAArgCosts.find(SROAArg); 68906408758SMircea Trofin assert(CostIt != SROAArgCosts.end() && 69006408758SMircea Trofin "expected this argument to have a cost"); 69106408758SMircea Trofin CostIt->second += InlineConstants::InstrCost; 69206408758SMircea Trofin SROACostSavings += InlineConstants::InstrCost; 69306408758SMircea Trofin } 69406408758SMircea Trofin 6959895c701SKazu Hirata void onBlockStart(const BasicBlock *BB) override { CostAtBBStart = Cost; } 6969895c701SKazu Hirata 69706408758SMircea Trofin void onBlockAnalyzed(const BasicBlock *BB) override { 6989895c701SKazu Hirata if (CostBenefitAnalysisEnabled) { 6999895c701SKazu Hirata // Keep track of the static size of live but cold basic blocks. For now, 7009895c701SKazu Hirata // we define a cold basic block to be one that's never executed. 7019895c701SKazu Hirata assert(GetBFI && "GetBFI must be available"); 7029895c701SKazu Hirata BlockFrequencyInfo *BFI = &(GetBFI(F)); 7039895c701SKazu Hirata assert(BFI && "BFI must be available"); 7049895c701SKazu Hirata auto ProfileCount = BFI->getBlockProfileCount(BB); 7059895c701SKazu Hirata assert(ProfileCount.hasValue()); 7069895c701SKazu Hirata if (ProfileCount.getValue() == 0) 7079895c701SKazu Hirata ColdSize += Cost - CostAtBBStart; 7089895c701SKazu Hirata } 7099895c701SKazu Hirata 71006408758SMircea Trofin auto *TI = BB->getTerminator(); 71106408758SMircea Trofin // If we had any successors at this point, than post-inlining is likely to 71206408758SMircea Trofin // have them as well. Note that we assume any basic blocks which existed 71306408758SMircea Trofin // due to branches or switches which folded above will also fold after 71406408758SMircea Trofin // inlining. 71506408758SMircea Trofin if (SingleBB && TI->getNumSuccessors() > 1) { 71606408758SMircea Trofin // Take off the bonus we applied to the threshold. 71706408758SMircea Trofin Threshold -= SingleBBBonus; 71806408758SMircea Trofin SingleBB = false; 71906408758SMircea Trofin } 72006408758SMircea Trofin } 7215466597fSMircea Trofin 722c965fd94SKirill Naumov void onInstructionAnalysisStart(const Instruction *I) override { 723c965fd94SKirill Naumov // This function is called to store the initial cost of inlining before 724c965fd94SKirill Naumov // the given instruction was assessed. 7251022b5ebSKirill Naumov if (!PrintInstructionComments) 726c965fd94SKirill Naumov return; 7271022b5ebSKirill Naumov InstructionCostDetailMap[I].CostBefore = Cost; 7281022b5ebSKirill Naumov InstructionCostDetailMap[I].ThresholdBefore = Threshold; 729c965fd94SKirill Naumov } 730c965fd94SKirill Naumov 731c965fd94SKirill Naumov void onInstructionAnalysisFinish(const Instruction *I) override { 732c965fd94SKirill Naumov // This function is called to find new values of cost and threshold after 733c965fd94SKirill Naumov // the instruction has been assessed. 7341022b5ebSKirill Naumov if (!PrintInstructionComments) 735c965fd94SKirill Naumov return; 7361022b5ebSKirill Naumov InstructionCostDetailMap[I].CostAfter = Cost; 7371022b5ebSKirill Naumov InstructionCostDetailMap[I].ThresholdAfter = Threshold; 738c965fd94SKirill Naumov } 739c965fd94SKirill Naumov 7409895c701SKazu Hirata bool isCostBenefitAnalysisEnabled() { 7419895c701SKazu Hirata if (!PSI || !PSI->hasProfileSummary()) 7429895c701SKazu Hirata return false; 7439895c701SKazu Hirata 7449895c701SKazu Hirata if (!GetBFI) 7459895c701SKazu Hirata return false; 7469895c701SKazu Hirata 7479d375a40SKazu Hirata if (InlineEnableCostBenefitAnalysis.getNumOccurrences()) { 7489d375a40SKazu Hirata // Honor the explicit request from the user. 7499d375a40SKazu Hirata if (!InlineEnableCostBenefitAnalysis) 7509d375a40SKazu Hirata return false; 7519d375a40SKazu Hirata } else { 7529d375a40SKazu Hirata // Otherwise, require instrumentation profile. 7539d375a40SKazu Hirata if (!PSI->hasInstrumentationProfile()) 7549d375a40SKazu Hirata return false; 7559d375a40SKazu Hirata } 7569d375a40SKazu Hirata 7579895c701SKazu Hirata auto *Caller = CandidateCall.getParent()->getParent(); 7589895c701SKazu Hirata if (!Caller->getEntryCount()) 7599895c701SKazu Hirata return false; 7609895c701SKazu Hirata 7619895c701SKazu Hirata BlockFrequencyInfo *CallerBFI = &(GetBFI(*Caller)); 7629895c701SKazu Hirata if (!CallerBFI) 7639895c701SKazu Hirata return false; 7649895c701SKazu Hirata 7659895c701SKazu Hirata // For now, limit to hot call site. 7669895c701SKazu Hirata if (!PSI->isHotCallSite(CandidateCall, CallerBFI)) 7679895c701SKazu Hirata return false; 7689895c701SKazu Hirata 7693c775d93SKazu Hirata // Make sure we have a nonzero entry count. 7703c775d93SKazu Hirata auto EntryCount = F.getEntryCount(); 7713c775d93SKazu Hirata if (!EntryCount || !EntryCount.getCount()) 7729895c701SKazu Hirata return false; 7739895c701SKazu Hirata 7749895c701SKazu Hirata BlockFrequencyInfo *CalleeBFI = &(GetBFI(F)); 7759895c701SKazu Hirata if (!CalleeBFI) 7769895c701SKazu Hirata return false; 7779895c701SKazu Hirata 7789895c701SKazu Hirata return true; 7799895c701SKazu Hirata } 7809895c701SKazu Hirata 7819895c701SKazu Hirata // Determine whether we should inline the given call site, taking into account 7829895c701SKazu Hirata // both the size cost and the cycle savings. Return None if we don't have 7839895c701SKazu Hirata // suficient profiling information to determine. 7849895c701SKazu Hirata Optional<bool> costBenefitAnalysis() { 7859895c701SKazu Hirata if (!CostBenefitAnalysisEnabled) 7869895c701SKazu Hirata return None; 7879895c701SKazu Hirata 7889895c701SKazu Hirata // buildInlinerPipeline in the pass builder sets HotCallSiteThreshold to 0 7899895c701SKazu Hirata // for the prelink phase of the AutoFDO + ThinLTO build. Honor the logic by 7909895c701SKazu Hirata // falling back to the cost-based metric. 7919895c701SKazu Hirata // TODO: Improve this hacky condition. 7929895c701SKazu Hirata if (Threshold == 0) 7939895c701SKazu Hirata return None; 7949895c701SKazu Hirata 7959895c701SKazu Hirata assert(GetBFI); 7969895c701SKazu Hirata BlockFrequencyInfo *CalleeBFI = &(GetBFI(F)); 7979895c701SKazu Hirata assert(CalleeBFI); 7989895c701SKazu Hirata 7999895c701SKazu Hirata // The cycle savings expressed as the sum of InlineConstants::InstrCost 8009895c701SKazu Hirata // multiplied by the estimated dynamic count of each instruction we can 8019895c701SKazu Hirata // avoid. Savings come from the call site cost, such as argument setup and 8029895c701SKazu Hirata // the call instruction, as well as the instructions that are folded. 8039895c701SKazu Hirata // 8049895c701SKazu Hirata // We use 128-bit APInt here to avoid potential overflow. This variable 8059895c701SKazu Hirata // should stay well below 10^^24 (or 2^^80) in practice. This "worst" case 8069895c701SKazu Hirata // assumes that we can avoid or fold a billion instructions, each with a 8079895c701SKazu Hirata // profile count of 10^^15 -- roughly the number of cycles for a 24-hour 8089895c701SKazu Hirata // period on a 4GHz machine. 8099895c701SKazu Hirata APInt CycleSavings(128, 0); 8109895c701SKazu Hirata 8119895c701SKazu Hirata for (auto &BB : F) { 8129895c701SKazu Hirata APInt CurrentSavings(128, 0); 8139895c701SKazu Hirata for (auto &I : BB) { 8149895c701SKazu Hirata if (BranchInst *BI = dyn_cast<BranchInst>(&I)) { 8159895c701SKazu Hirata // Count a conditional branch as savings if it becomes unconditional. 8169895c701SKazu Hirata if (BI->isConditional() && 8179895c701SKazu Hirata dyn_cast_or_null<ConstantInt>( 8189895c701SKazu Hirata SimplifiedValues.lookup(BI->getCondition()))) { 8199895c701SKazu Hirata CurrentSavings += InlineConstants::InstrCost; 8209895c701SKazu Hirata } 8219895c701SKazu Hirata } else if (Value *V = dyn_cast<Value>(&I)) { 8229895c701SKazu Hirata // Count an instruction as savings if we can fold it. 8239895c701SKazu Hirata if (SimplifiedValues.count(V)) { 8249895c701SKazu Hirata CurrentSavings += InlineConstants::InstrCost; 8259895c701SKazu Hirata } 8269895c701SKazu Hirata } 8279895c701SKazu Hirata } 8289895c701SKazu Hirata 8299895c701SKazu Hirata auto ProfileCount = CalleeBFI->getBlockProfileCount(&BB); 8309895c701SKazu Hirata assert(ProfileCount.hasValue()); 8319895c701SKazu Hirata CurrentSavings *= ProfileCount.getValue(); 8329895c701SKazu Hirata CycleSavings += CurrentSavings; 8339895c701SKazu Hirata } 8349895c701SKazu Hirata 8359895c701SKazu Hirata // Compute the cycle savings per call. 8369895c701SKazu Hirata auto EntryProfileCount = F.getEntryCount(); 8373c775d93SKazu Hirata assert(EntryProfileCount.hasValue() && EntryProfileCount.getCount()); 8389895c701SKazu Hirata auto EntryCount = EntryProfileCount.getCount(); 8399895c701SKazu Hirata CycleSavings += EntryCount / 2; 8409895c701SKazu Hirata CycleSavings = CycleSavings.udiv(EntryCount); 8419895c701SKazu Hirata 8429895c701SKazu Hirata // Compute the total savings for the call site. 8439895c701SKazu Hirata auto *CallerBB = CandidateCall.getParent(); 8449895c701SKazu Hirata BlockFrequencyInfo *CallerBFI = &(GetBFI(*(CallerBB->getParent()))); 8459895c701SKazu Hirata CycleSavings += getCallsiteCost(this->CandidateCall, DL); 8469895c701SKazu Hirata CycleSavings *= CallerBFI->getBlockProfileCount(CallerBB).getValue(); 8479895c701SKazu Hirata 8489895c701SKazu Hirata // Remove the cost of the cold basic blocks. 8499895c701SKazu Hirata int Size = Cost - ColdSize; 8509895c701SKazu Hirata 8519895c701SKazu Hirata // Allow tiny callees to be inlined regardless of whether they meet the 8529895c701SKazu Hirata // savings threshold. 8539895c701SKazu Hirata Size = Size > InlineSizeAllowance ? Size - InlineSizeAllowance : 1; 8549895c701SKazu Hirata 8554bdfea2cSLiqiang Tao CostBenefit.emplace(APInt(128, Size), CycleSavings); 8564bdfea2cSLiqiang Tao 8579895c701SKazu Hirata // Return true if the savings justify the cost of inlining. Specifically, 8589895c701SKazu Hirata // we evaluate the following inequality: 8599895c701SKazu Hirata // 8609895c701SKazu Hirata // CycleSavings PSI->getOrCompHotCountThreshold() 8619895c701SKazu Hirata // -------------- >= ----------------------------------- 8629895c701SKazu Hirata // Size InlineSavingsMultiplier 8639895c701SKazu Hirata // 8649895c701SKazu Hirata // Note that the left hand side is specific to a call site. The right hand 8659895c701SKazu Hirata // side is a constant for the entire executable. 8669895c701SKazu Hirata APInt LHS = CycleSavings; 8679895c701SKazu Hirata LHS *= InlineSavingsMultiplier; 8689895c701SKazu Hirata APInt RHS(128, PSI->getOrCompHotCountThreshold()); 8699895c701SKazu Hirata RHS *= Size; 8709895c701SKazu Hirata return LHS.uge(RHS); 8719895c701SKazu Hirata } 8729895c701SKazu Hirata 87306408758SMircea Trofin InlineResult finalizeAnalysis() override { 87406408758SMircea Trofin // Loops generally act a lot like calls in that they act like barriers to 87506408758SMircea Trofin // movement, require a certain amount of setup, etc. So when optimising for 87606408758SMircea Trofin // size, we penalise any call sites that perform loops. We do this after all 87706408758SMircea Trofin // other costs here, so will likely only be dealing with relatively small 87806408758SMircea Trofin // functions (and hence DT and LI will hopefully be cheap). 87906408758SMircea Trofin auto *Caller = CandidateCall.getFunction(); 88006408758SMircea Trofin if (Caller->hasMinSize()) { 88106408758SMircea Trofin DominatorTree DT(F); 88206408758SMircea Trofin LoopInfo LI(DT); 88306408758SMircea Trofin int NumLoops = 0; 88406408758SMircea Trofin for (Loop *L : LI) { 88506408758SMircea Trofin // Ignore loops that will not be executed 88606408758SMircea Trofin if (DeadBlocks.count(L->getHeader())) 88706408758SMircea Trofin continue; 88806408758SMircea Trofin NumLoops++; 88906408758SMircea Trofin } 89046c03668SPhilipp Krones addCost(NumLoops * InlineConstants::LoopPenalty); 89106408758SMircea Trofin } 89206408758SMircea Trofin 89306408758SMircea Trofin // We applied the maximum possible vector bonus at the beginning. Now, 89406408758SMircea Trofin // subtract the excess bonus, if any, from the Threshold before 89506408758SMircea Trofin // comparing against Cost. 89606408758SMircea Trofin if (NumVectorInstructions <= NumInstructions / 10) 89706408758SMircea Trofin Threshold -= VectorBonus; 89806408758SMircea Trofin else if (NumVectorInstructions <= NumInstructions / 2) 89906408758SMircea Trofin Threshold -= VectorBonus / 2; 90006408758SMircea Trofin 9015c97507eSDaniil Suchkov if (Optional<int> AttrCost = 9025c97507eSDaniil Suchkov getStringFnAttrAsInt(CandidateCall, "function-inline-cost")) 9035c97507eSDaniil Suchkov Cost = *AttrCost; 9045c97507eSDaniil Suchkov 9055c97507eSDaniil Suchkov if (Optional<int> AttrThreshold = 9065c97507eSDaniil Suchkov getStringFnAttrAsInt(CandidateCall, "function-inline-threshold")) 9075c97507eSDaniil Suchkov Threshold = *AttrThreshold; 9085c97507eSDaniil Suchkov 9099895c701SKazu Hirata if (auto Result = costBenefitAnalysis()) { 9106869e6c1SWenlei He DecidedByCostBenefit = true; 9119895c701SKazu Hirata if (Result.getValue()) 9129895c701SKazu Hirata return InlineResult::success(); 9139895c701SKazu Hirata else 9149895c701SKazu Hirata return InlineResult::failure("Cost over threshold."); 9159895c701SKazu Hirata } 9169895c701SKazu Hirata 9178a4013edSMircea Trofin if (IgnoreThreshold || Cost < std::max(1, Threshold)) 9185466597fSMircea Trofin return InlineResult::success(); 9195466597fSMircea Trofin return InlineResult::failure("Cost over threshold."); 92006408758SMircea Trofin } 92106408758SMircea Trofin bool shouldStop() override { 92206408758SMircea Trofin // Bail out the moment we cross the threshold. This means we'll under-count 92306408758SMircea Trofin // the cost, but only when undercounting doesn't matter. 9248a4013edSMircea Trofin return !IgnoreThreshold && Cost >= Threshold && !ComputeFullInlineCost; 92506408758SMircea Trofin } 92606408758SMircea Trofin 92706408758SMircea Trofin void onLoadEliminationOpportunity() override { 92806408758SMircea Trofin LoadEliminationCost += InlineConstants::InstrCost; 92906408758SMircea Trofin } 93006408758SMircea Trofin 93106408758SMircea Trofin InlineResult onAnalysisStart() override { 93206408758SMircea Trofin // Perform some tweaks to the cost and threshold based on the direct 93306408758SMircea Trofin // callsite information. 93406408758SMircea Trofin 93506408758SMircea Trofin // We want to more aggressively inline vector-dense kernels, so up the 93606408758SMircea Trofin // threshold, and we'll lower it if the % of vector instructions gets too 93706408758SMircea Trofin // low. Note that these bonuses are some what arbitrary and evolved over 93806408758SMircea Trofin // time by accident as much as because they are principled bonuses. 93906408758SMircea Trofin // 94006408758SMircea Trofin // FIXME: It would be nice to remove all such bonuses. At least it would be 94106408758SMircea Trofin // nice to base the bonus values on something more scientific. 94206408758SMircea Trofin assert(NumInstructions == 0); 94306408758SMircea Trofin assert(NumVectorInstructions == 0); 94406408758SMircea Trofin 94506408758SMircea Trofin // Update the threshold based on callsite properties 94606408758SMircea Trofin updateThreshold(CandidateCall, F); 94706408758SMircea Trofin 94806408758SMircea Trofin // While Threshold depends on commandline options that can take negative 94906408758SMircea Trofin // values, we want to enforce the invariant that the computed threshold and 95006408758SMircea Trofin // bonuses are non-negative. 95106408758SMircea Trofin assert(Threshold >= 0); 95206408758SMircea Trofin assert(SingleBBBonus >= 0); 95306408758SMircea Trofin assert(VectorBonus >= 0); 95406408758SMircea Trofin 95506408758SMircea Trofin // Speculatively apply all possible bonuses to Threshold. If cost exceeds 95606408758SMircea Trofin // this Threshold any time, and cost cannot decrease, we can stop processing 95706408758SMircea Trofin // the rest of the function body. 95806408758SMircea Trofin Threshold += (SingleBBBonus + VectorBonus); 95906408758SMircea Trofin 96006408758SMircea Trofin // Give out bonuses for the callsite, as the instructions setting them up 96106408758SMircea Trofin // will be gone after inlining. 96206408758SMircea Trofin addCost(-getCallsiteCost(this->CandidateCall, DL)); 96306408758SMircea Trofin 96406408758SMircea Trofin // If this function uses the coldcc calling convention, prefer not to inline 96506408758SMircea Trofin // it. 96606408758SMircea Trofin if (F.getCallingConv() == CallingConv::Cold) 96706408758SMircea Trofin Cost += InlineConstants::ColdccPenalty; 96806408758SMircea Trofin 96906408758SMircea Trofin // Check if we're done. This can happen due to bonuses and penalties. 97006408758SMircea Trofin if (Cost >= Threshold && !ComputeFullInlineCost) 9715466597fSMircea Trofin return InlineResult::failure("high cost"); 97206408758SMircea Trofin 9735466597fSMircea Trofin return InlineResult::success(); 97406408758SMircea Trofin } 97506408758SMircea Trofin 97606408758SMircea Trofin public: 97706408758SMircea Trofin InlineCostCallAnalyzer( 97808e2386dSMircea Trofin Function &Callee, CallBase &Call, const InlineParams &Params, 97906408758SMircea Trofin const TargetTransformInfo &TTI, 98008e2386dSMircea Trofin function_ref<AssumptionCache &(Function &)> GetAssumptionCache, 98108e2386dSMircea Trofin function_ref<BlockFrequencyInfo &(Function &)> GetBFI = nullptr, 98208e2386dSMircea Trofin ProfileSummaryInfo *PSI = nullptr, 98308e2386dSMircea Trofin OptimizationRemarkEmitter *ORE = nullptr, bool BoostIndirect = true, 9848a4013edSMircea Trofin bool IgnoreThreshold = false) 98508e2386dSMircea Trofin : CallAnalyzer(Callee, Call, TTI, GetAssumptionCache, GetBFI, PSI, ORE), 98606408758SMircea Trofin ComputeFullInlineCost(OptComputeFullInlineCost || 98765cd3cbbSKazu Hirata Params.ComputeFullInlineCost || ORE || 98865cd3cbbSKazu Hirata isCostBenefitAnalysisEnabled()), 98906408758SMircea Trofin Params(Params), Threshold(Params.DefaultThreshold), 9901022b5ebSKirill Naumov BoostIndirectCalls(BoostIndirect), IgnoreThreshold(IgnoreThreshold), 9919895c701SKazu Hirata CostBenefitAnalysisEnabled(isCostBenefitAnalysisEnabled()), 992f7fff46aSWenlei He Writer(this) { 993f7fff46aSWenlei He AllowRecursiveCall = Params.AllowRecursiveCall.getValue(); 994f7fff46aSWenlei He } 995c965fd94SKirill Naumov 9961022b5ebSKirill Naumov /// Annotation Writer for instruction details 9971022b5ebSKirill Naumov InlineCostAnnotationWriter Writer; 998c965fd94SKirill Naumov 99906408758SMircea Trofin void dump(); 100006408758SMircea Trofin 1001ca899bf9SKirill Naumov // Prints the same analysis as dump(), but its definition is not dependent 1002ca899bf9SKirill Naumov // on the build. 1003ca899bf9SKirill Naumov void print(); 1004ca899bf9SKirill Naumov 10051022b5ebSKirill Naumov Optional<InstructionCostDetail> getCostDetails(const Instruction *I) { 10061022b5ebSKirill Naumov if (InstructionCostDetailMap.find(I) != InstructionCostDetailMap.end()) 10071022b5ebSKirill Naumov return InstructionCostDetailMap[I]; 10081022b5ebSKirill Naumov return None; 10091022b5ebSKirill Naumov } 10101022b5ebSKirill Naumov 101106408758SMircea Trofin virtual ~InlineCostCallAnalyzer() {} 1012cfc4def8SJacob Hegna int getThreshold() const { return Threshold; } 1013cfc4def8SJacob Hegna int getCost() const { return Cost; } 10144bdfea2cSLiqiang Tao Optional<CostBenefitPair> getCostBenefitPair() { return CostBenefit; } 1015cfc4def8SJacob Hegna bool wasDecidedByCostBenefit() const { return DecidedByCostBenefit; } 101606408758SMircea Trofin }; 101799f00635SJacob Hegna 101899f00635SJacob Hegna class InlineCostFeaturesAnalyzer final : public CallAnalyzer { 101999f00635SJacob Hegna private: 102099f00635SJacob Hegna InlineCostFeatures Cost = {}; 102199f00635SJacob Hegna 102299f00635SJacob Hegna // FIXME: These constants are taken from the heuristic-based cost visitor. 102399f00635SJacob Hegna // These should be removed entirely in a later revision to avoid reliance on 102499f00635SJacob Hegna // heuristics in the ML inliner. 102599f00635SJacob Hegna static constexpr int JTCostMultiplier = 4; 102699f00635SJacob Hegna static constexpr int CaseClusterCostMultiplier = 2; 102799f00635SJacob Hegna static constexpr int SwitchCostMultiplier = 2; 102899f00635SJacob Hegna 102999f00635SJacob Hegna // FIXME: These are taken from the heuristic-based cost visitor: we should 103099f00635SJacob Hegna // eventually abstract these to the CallAnalyzer to avoid duplication. 103199f00635SJacob Hegna unsigned SROACostSavingOpportunities = 0; 103299f00635SJacob Hegna int VectorBonus = 0; 103399f00635SJacob Hegna int SingleBBBonus = 0; 103499f00635SJacob Hegna int Threshold = 5; 103599f00635SJacob Hegna 103699f00635SJacob Hegna DenseMap<AllocaInst *, unsigned> SROACosts; 103799f00635SJacob Hegna 103899f00635SJacob Hegna void increment(InlineCostFeatureIndex Feature, int64_t Delta = 1) { 103999f00635SJacob Hegna Cost[static_cast<size_t>(Feature)] += Delta; 104099f00635SJacob Hegna } 104199f00635SJacob Hegna 104299f00635SJacob Hegna void set(InlineCostFeatureIndex Feature, int64_t Value) { 104399f00635SJacob Hegna Cost[static_cast<size_t>(Feature)] = Value; 104499f00635SJacob Hegna } 104599f00635SJacob Hegna 104699f00635SJacob Hegna void onDisableSROA(AllocaInst *Arg) override { 104799f00635SJacob Hegna auto CostIt = SROACosts.find(Arg); 104899f00635SJacob Hegna if (CostIt == SROACosts.end()) 104999f00635SJacob Hegna return; 105099f00635SJacob Hegna 105199f00635SJacob Hegna increment(InlineCostFeatureIndex::SROALosses, CostIt->second); 105299f00635SJacob Hegna SROACostSavingOpportunities -= CostIt->second; 105399f00635SJacob Hegna SROACosts.erase(CostIt); 105499f00635SJacob Hegna } 105599f00635SJacob Hegna 105699f00635SJacob Hegna void onDisableLoadElimination() override { 105799f00635SJacob Hegna set(InlineCostFeatureIndex::LoadElimination, 1); 105899f00635SJacob Hegna } 105999f00635SJacob Hegna 106099f00635SJacob Hegna void onCallPenalty() override { 106146c03668SPhilipp Krones increment(InlineCostFeatureIndex::CallPenalty, CallPenalty); 106299f00635SJacob Hegna } 106399f00635SJacob Hegna 106499f00635SJacob Hegna void onCallArgumentSetup(const CallBase &Call) override { 106599f00635SJacob Hegna increment(InlineCostFeatureIndex::CallArgumentSetup, 106699f00635SJacob Hegna Call.arg_size() * InlineConstants::InstrCost); 106799f00635SJacob Hegna } 106899f00635SJacob Hegna 106999f00635SJacob Hegna void onLoadRelativeIntrinsic() override { 107099f00635SJacob Hegna increment(InlineCostFeatureIndex::LoadRelativeIntrinsic, 107199f00635SJacob Hegna 3 * InlineConstants::InstrCost); 107299f00635SJacob Hegna } 107399f00635SJacob Hegna 107499f00635SJacob Hegna void onLoweredCall(Function *F, CallBase &Call, 107599f00635SJacob Hegna bool IsIndirectCall) override { 107699f00635SJacob Hegna increment(InlineCostFeatureIndex::LoweredCallArgSetup, 107799f00635SJacob Hegna Call.arg_size() * InlineConstants::InstrCost); 107899f00635SJacob Hegna 107999f00635SJacob Hegna if (IsIndirectCall) { 108099f00635SJacob Hegna InlineParams IndirectCallParams = {/* DefaultThreshold*/ 0, 108199f00635SJacob Hegna /*HintThreshold*/ {}, 108299f00635SJacob Hegna /*ColdThreshold*/ {}, 108399f00635SJacob Hegna /*OptSizeThreshold*/ {}, 108499f00635SJacob Hegna /*OptMinSizeThreshold*/ {}, 108599f00635SJacob Hegna /*HotCallSiteThreshold*/ {}, 108699f00635SJacob Hegna /*LocallyHotCallSiteThreshold*/ {}, 108799f00635SJacob Hegna /*ColdCallSiteThreshold*/ {}, 108899f00635SJacob Hegna /*ComputeFullInlineCost*/ true, 108999f00635SJacob Hegna /*EnableDeferral*/ true}; 109099f00635SJacob Hegna IndirectCallParams.DefaultThreshold = 109199f00635SJacob Hegna InlineConstants::IndirectCallThreshold; 109299f00635SJacob Hegna 109399f00635SJacob Hegna InlineCostCallAnalyzer CA(*F, Call, IndirectCallParams, TTI, 109499f00635SJacob Hegna GetAssumptionCache, GetBFI, PSI, ORE, false, 109599f00635SJacob Hegna true); 109699f00635SJacob Hegna if (CA.analyze().isSuccess()) { 109799f00635SJacob Hegna increment(InlineCostFeatureIndex::NestedInlineCostEstimate, 109899f00635SJacob Hegna CA.getCost()); 109999f00635SJacob Hegna increment(InlineCostFeatureIndex::NestedInlines, 1); 110099f00635SJacob Hegna } 110199f00635SJacob Hegna } else { 110299f00635SJacob Hegna onCallPenalty(); 110399f00635SJacob Hegna } 110499f00635SJacob Hegna } 110599f00635SJacob Hegna 110699f00635SJacob Hegna void onFinalizeSwitch(unsigned JumpTableSize, 110799f00635SJacob Hegna unsigned NumCaseCluster) override { 110899f00635SJacob Hegna 110999f00635SJacob Hegna if (JumpTableSize) { 111099f00635SJacob Hegna int64_t JTCost = 111199f00635SJacob Hegna static_cast<int64_t>(JumpTableSize) * InlineConstants::InstrCost + 111299f00635SJacob Hegna JTCostMultiplier * InlineConstants::InstrCost; 111399f00635SJacob Hegna increment(InlineCostFeatureIndex::JumpTablePenalty, JTCost); 111499f00635SJacob Hegna return; 111599f00635SJacob Hegna } 111699f00635SJacob Hegna 111799f00635SJacob Hegna if (NumCaseCluster <= 3) { 111899f00635SJacob Hegna increment(InlineCostFeatureIndex::CaseClusterPenalty, 111999f00635SJacob Hegna NumCaseCluster * CaseClusterCostMultiplier * 112099f00635SJacob Hegna InlineConstants::InstrCost); 112199f00635SJacob Hegna return; 112299f00635SJacob Hegna } 112399f00635SJacob Hegna 112499f00635SJacob Hegna int64_t ExpectedNumberOfCompare = 112599f00635SJacob Hegna getExpectedNumberOfCompare(NumCaseCluster); 112699f00635SJacob Hegna 112799f00635SJacob Hegna int64_t SwitchCost = ExpectedNumberOfCompare * SwitchCostMultiplier * 112899f00635SJacob Hegna InlineConstants::InstrCost; 112999f00635SJacob Hegna increment(InlineCostFeatureIndex::SwitchPenalty, SwitchCost); 113099f00635SJacob Hegna } 113199f00635SJacob Hegna 113299f00635SJacob Hegna void onMissedSimplification() override { 113399f00635SJacob Hegna increment(InlineCostFeatureIndex::UnsimplifiedCommonInstructions, 113499f00635SJacob Hegna InlineConstants::InstrCost); 113599f00635SJacob Hegna } 113699f00635SJacob Hegna 113799f00635SJacob Hegna void onInitializeSROAArg(AllocaInst *Arg) override { SROACosts[Arg] = 0; } 113899f00635SJacob Hegna void onAggregateSROAUse(AllocaInst *Arg) override { 113999f00635SJacob Hegna SROACosts.find(Arg)->second += InlineConstants::InstrCost; 114099f00635SJacob Hegna SROACostSavingOpportunities += InlineConstants::InstrCost; 114199f00635SJacob Hegna } 114299f00635SJacob Hegna 114399f00635SJacob Hegna void onBlockAnalyzed(const BasicBlock *BB) override { 114499f00635SJacob Hegna if (BB->getTerminator()->getNumSuccessors() > 1) 114599f00635SJacob Hegna set(InlineCostFeatureIndex::IsMultipleBlocks, 1); 114699f00635SJacob Hegna Threshold -= SingleBBBonus; 114799f00635SJacob Hegna } 114899f00635SJacob Hegna 114999f00635SJacob Hegna InlineResult finalizeAnalysis() override { 115099f00635SJacob Hegna auto *Caller = CandidateCall.getFunction(); 115199f00635SJacob Hegna if (Caller->hasMinSize()) { 115299f00635SJacob Hegna DominatorTree DT(F); 115399f00635SJacob Hegna LoopInfo LI(DT); 115499f00635SJacob Hegna for (Loop *L : LI) { 115599f00635SJacob Hegna // Ignore loops that will not be executed 115699f00635SJacob Hegna if (DeadBlocks.count(L->getHeader())) 115799f00635SJacob Hegna continue; 115899f00635SJacob Hegna increment(InlineCostFeatureIndex::NumLoops, 115946c03668SPhilipp Krones InlineConstants::LoopPenalty); 116099f00635SJacob Hegna } 116199f00635SJacob Hegna } 116299f00635SJacob Hegna set(InlineCostFeatureIndex::DeadBlocks, DeadBlocks.size()); 116399f00635SJacob Hegna set(InlineCostFeatureIndex::SimplifiedInstructions, 116499f00635SJacob Hegna NumInstructionsSimplified); 116599f00635SJacob Hegna set(InlineCostFeatureIndex::ConstantArgs, NumConstantArgs); 116699f00635SJacob Hegna set(InlineCostFeatureIndex::ConstantOffsetPtrArgs, 116799f00635SJacob Hegna NumConstantOffsetPtrArgs); 116899f00635SJacob Hegna set(InlineCostFeatureIndex::SROASavings, SROACostSavingOpportunities); 116999f00635SJacob Hegna 117099f00635SJacob Hegna if (NumVectorInstructions <= NumInstructions / 10) 11711f3e90e1SJacob Hegna Threshold -= VectorBonus; 117299f00635SJacob Hegna else if (NumVectorInstructions <= NumInstructions / 2) 11731f3e90e1SJacob Hegna Threshold -= VectorBonus / 2; 117499f00635SJacob Hegna 117599f00635SJacob Hegna set(InlineCostFeatureIndex::Threshold, Threshold); 117699f00635SJacob Hegna 117799f00635SJacob Hegna return InlineResult::success(); 117899f00635SJacob Hegna } 117999f00635SJacob Hegna 118099f00635SJacob Hegna bool shouldStop() override { return false; } 118199f00635SJacob Hegna 118299f00635SJacob Hegna void onLoadEliminationOpportunity() override { 118399f00635SJacob Hegna increment(InlineCostFeatureIndex::LoadElimination, 1); 118499f00635SJacob Hegna } 118599f00635SJacob Hegna 118699f00635SJacob Hegna InlineResult onAnalysisStart() override { 118799f00635SJacob Hegna increment(InlineCostFeatureIndex::CallSiteCost, 118899f00635SJacob Hegna -1 * getCallsiteCost(this->CandidateCall, DL)); 118999f00635SJacob Hegna 119099f00635SJacob Hegna set(InlineCostFeatureIndex::ColdCcPenalty, 119199f00635SJacob Hegna (F.getCallingConv() == CallingConv::Cold)); 119299f00635SJacob Hegna 119399f00635SJacob Hegna // FIXME: we shouldn't repeat this logic in both the Features and Cost 119499f00635SJacob Hegna // analyzer - instead, we should abstract it to a common method in the 119599f00635SJacob Hegna // CallAnalyzer 119699f00635SJacob Hegna int SingleBBBonusPercent = 50; 119799f00635SJacob Hegna int VectorBonusPercent = TTI.getInlinerVectorBonusPercent(); 119899f00635SJacob Hegna Threshold += TTI.adjustInliningThreshold(&CandidateCall); 119999f00635SJacob Hegna Threshold *= TTI.getInliningThresholdMultiplier(); 120099f00635SJacob Hegna SingleBBBonus = Threshold * SingleBBBonusPercent / 100; 120199f00635SJacob Hegna VectorBonus = Threshold * VectorBonusPercent / 100; 120299f00635SJacob Hegna Threshold += (SingleBBBonus + VectorBonus); 120399f00635SJacob Hegna 120499f00635SJacob Hegna return InlineResult::success(); 120599f00635SJacob Hegna } 120699f00635SJacob Hegna 120799f00635SJacob Hegna public: 120899f00635SJacob Hegna InlineCostFeaturesAnalyzer( 120999f00635SJacob Hegna const TargetTransformInfo &TTI, 121099f00635SJacob Hegna function_ref<AssumptionCache &(Function &)> &GetAssumptionCache, 121199f00635SJacob Hegna function_ref<BlockFrequencyInfo &(Function &)> GetBFI, 121299f00635SJacob Hegna ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE, Function &Callee, 121399f00635SJacob Hegna CallBase &Call) 121499f00635SJacob Hegna : CallAnalyzer(Callee, Call, TTI, GetAssumptionCache, GetBFI, PSI) {} 121599f00635SJacob Hegna 121699f00635SJacob Hegna const InlineCostFeatures &features() const { return Cost; } 121799f00635SJacob Hegna }; 121899f00635SJacob Hegna 12197adc3a2bSChandler Carruth } // namespace 12207adc3a2bSChandler Carruth 12215f8f34e4SAdrian Prantl /// Test whether the given value is an Alloca-derived function argument. 12227adc3a2bSChandler Carruth bool CallAnalyzer::isAllocaDerivedArg(Value *V) { 12237adc3a2bSChandler Carruth return SROAArgValues.count(V); 12247adc3a2bSChandler Carruth } 12257adc3a2bSChandler Carruth 122606408758SMircea Trofin void CallAnalyzer::disableSROAForArg(AllocaInst *SROAArg) { 122706408758SMircea Trofin onDisableSROA(SROAArg); 12282e42cc7aSMircea Trofin EnabledSROAAllocas.erase(SROAArg); 1229338a6016SMircea Trofin disableLoadElimination(); 1230338a6016SMircea Trofin } 1231c965fd94SKirill Naumov 12327b639f50SJacob Hegna void InlineCostAnnotationWriter::emitInstructionAnnot( 12337b639f50SJacob Hegna const Instruction *I, formatted_raw_ostream &OS) { 1234c965fd94SKirill Naumov // The cost of inlining of the given instruction is printed always. 1235c965fd94SKirill Naumov // The threshold delta is printed only when it is non-zero. It happens 1236c965fd94SKirill Naumov // when we decided to give a bonus at a particular instruction. 12371022b5ebSKirill Naumov Optional<InstructionCostDetail> Record = ICCA->getCostDetails(I); 12381022b5ebSKirill Naumov if (!Record) 12391022b5ebSKirill Naumov OS << "; No analysis for the instruction"; 12401022b5ebSKirill Naumov else { 12411022b5ebSKirill Naumov OS << "; cost before = " << Record->CostBefore 12421022b5ebSKirill Naumov << ", cost after = " << Record->CostAfter 12431022b5ebSKirill Naumov << ", threshold before = " << Record->ThresholdBefore 12441022b5ebSKirill Naumov << ", threshold after = " << Record->ThresholdAfter << ", "; 12451022b5ebSKirill Naumov OS << "cost delta = " << Record->getCostDelta(); 12461022b5ebSKirill Naumov if (Record->hasThresholdChanged()) 12471022b5ebSKirill Naumov OS << ", threshold delta = " << Record->getThresholdDelta(); 12480383253cSKirill Naumov } 12497f094f7fSKirill Naumov auto C = ICCA->getSimplifiedValue(const_cast<Instruction *>(I)); 12507f094f7fSKirill Naumov if (C) { 12517f094f7fSKirill Naumov OS << ", simplified to "; 12527f094f7fSKirill Naumov C.getValue()->print(OS, true); 12537f094f7fSKirill Naumov } 1254c965fd94SKirill Naumov OS << "\n"; 1255c965fd94SKirill Naumov } 1256c965fd94SKirill Naumov 12575f8f34e4SAdrian Prantl /// If 'V' maps to a SROA candidate, disable SROA for it. 12587adc3a2bSChandler Carruth void CallAnalyzer::disableSROA(Value *V) { 125906408758SMircea Trofin if (auto *SROAArg = getSROAArgForValueOrNull(V)) { 126006408758SMircea Trofin disableSROAForArg(SROAArg); 1261338a6016SMircea Trofin } 12627adc3a2bSChandler Carruth } 12637adc3a2bSChandler Carruth 1264a4461515SHaicheng Wu void CallAnalyzer::disableLoadElimination() { 1265a4461515SHaicheng Wu if (EnableLoadElimination) { 126606408758SMircea Trofin onDisableLoadElimination(); 1267a4461515SHaicheng Wu EnableLoadElimination = false; 1268a4461515SHaicheng Wu } 1269a4461515SHaicheng Wu } 1270a4461515SHaicheng Wu 12715f8f34e4SAdrian Prantl /// Accumulate a constant GEP offset into an APInt if possible. 12727adc3a2bSChandler Carruth /// 12737adc3a2bSChandler Carruth /// Returns false if unable to compute the offset for any reason. Respects any 12747adc3a2bSChandler Carruth /// simplified values known during the analysis of this callsite. 12757adc3a2bSChandler Carruth bool CallAnalyzer::accumulateGEPOffset(GEPOperator &GEP, APInt &Offset) { 1276945b7e5aSElena Demikhovsky unsigned IntPtrWidth = DL.getIndexTypeSizeInBits(GEP.getType()); 12777adc3a2bSChandler Carruth assert(IntPtrWidth == Offset.getBitWidth()); 12787adc3a2bSChandler Carruth 12797adc3a2bSChandler Carruth for (gep_type_iterator GTI = gep_type_begin(GEP), GTE = gep_type_end(GEP); 12807adc3a2bSChandler Carruth GTI != GTE; ++GTI) { 12817adc3a2bSChandler Carruth ConstantInt *OpC = dyn_cast<ConstantInt>(GTI.getOperand()); 12827adc3a2bSChandler Carruth if (!OpC) 12837adc3a2bSChandler Carruth if (Constant *SimpleOp = SimplifiedValues.lookup(GTI.getOperand())) 12847adc3a2bSChandler Carruth OpC = dyn_cast<ConstantInt>(SimpleOp); 12857adc3a2bSChandler Carruth if (!OpC) 12867adc3a2bSChandler Carruth return false; 1287567556aaSChad Rosier if (OpC->isZero()) 1288567556aaSChad Rosier continue; 12897adc3a2bSChandler Carruth 12907adc3a2bSChandler Carruth // Handle a struct index, which adds its field offset to the pointer. 1291ab85225bSPeter Collingbourne if (StructType *STy = GTI.getStructTypeOrNull()) { 12927adc3a2bSChandler Carruth unsigned ElementIdx = OpC->getZExtValue(); 12937adc3a2bSChandler Carruth const StructLayout *SL = DL.getStructLayout(STy); 12947adc3a2bSChandler Carruth Offset += APInt(IntPtrWidth, SL->getElementOffset(ElementIdx)); 12957adc3a2bSChandler Carruth continue; 12967adc3a2bSChandler Carruth } 12977adc3a2bSChandler Carruth 12987adc3a2bSChandler Carruth APInt TypeSize(IntPtrWidth, DL.getTypeAllocSize(GTI.getIndexedType())); 12997adc3a2bSChandler Carruth Offset += OpC->getValue().sextOrTrunc(IntPtrWidth) * TypeSize; 13007adc3a2bSChandler Carruth } 13017adc3a2bSChandler Carruth return true; 13027adc3a2bSChandler Carruth } 13037adc3a2bSChandler Carruth 13045f8f34e4SAdrian Prantl /// Use TTI to check whether a GEP is free. 1305201b191bSHaicheng Wu /// 1306201b191bSHaicheng Wu /// Respects any simplified values known during the analysis of this callsite. 1307201b191bSHaicheng Wu bool CallAnalyzer::isGEPFree(GetElementPtrInst &GEP) { 130861c1bd5aSEvgeny Astigeevich SmallVector<Value *, 4> Operands; 130961c1bd5aSEvgeny Astigeevich Operands.push_back(GEP.getOperand(0)); 131091aff1d8SNick Desaulniers for (const Use &Op : GEP.indices()) 131191aff1d8SNick Desaulniers if (Constant *SimpleOp = SimplifiedValues.lookup(Op)) 131261c1bd5aSEvgeny Astigeevich Operands.push_back(SimpleOp); 1313201b191bSHaicheng Wu else 131491aff1d8SNick Desaulniers Operands.push_back(Op); 13153ccbd4f3SSander de Smalen return TTI.getUserCost(&GEP, Operands, 13163ccbd4f3SSander de Smalen TargetTransformInfo::TCK_SizeAndLatency) == 13173ccbd4f3SSander de Smalen TargetTransformInfo::TCC_Free; 1318201b191bSHaicheng Wu } 1319201b191bSHaicheng Wu 13207adc3a2bSChandler Carruth bool CallAnalyzer::visitAlloca(AllocaInst &I) { 1321a3a798d4SArthur Eubanks disableSROA(I.getOperand(0)); 1322a3a798d4SArthur Eubanks 13237adc3a2bSChandler Carruth // Check whether inlining will turn a dynamic alloca into a static 13240f153424SSanjay Patel // alloca and handle that case. 13257adc3a2bSChandler Carruth if (I.isArrayAllocation()) { 13260f153424SSanjay Patel Constant *Size = SimplifiedValues.lookup(I.getArraySize()); 13270f153424SSanjay Patel if (auto *AllocSize = dyn_cast_or_null<ConstantInt>(Size)) { 1328090c108dSAmara Emerson // Sometimes a dynamic alloca could be converted into a static alloca 1329090c108dSAmara Emerson // after this constant prop, and become a huge static alloca on an 1330090c108dSAmara Emerson // unconditional CFG path. Avoid inlining if this is going to happen above 1331090c108dSAmara Emerson // a threshold. 1332090c108dSAmara Emerson // FIXME: If the threshold is removed or lowered too much, we could end up 1333090c108dSAmara Emerson // being too pessimistic and prevent inlining non-problematic code. This 1334090c108dSAmara Emerson // could result in unintended perf regressions. A better overall strategy 1335090c108dSAmara Emerson // is needed to track stack usage during inlining. 13367adc3a2bSChandler Carruth Type *Ty = I.getAllocatedType(); 133722eb80a1SEaswaran Raman AllocatedSize = SaturatingMultiplyAdd( 13387b639f50SJacob Hegna AllocSize->getLimitedValue(), 13397b639f50SJacob Hegna DL.getTypeAllocSize(Ty).getKnownMinSize(), AllocatedSize); 1340a3a798d4SArthur Eubanks if (AllocatedSize > InlineConstants::MaxSimplifiedDynamicAllocaToInline) 1341090c108dSAmara Emerson HasDynamicAlloca = true; 1342090c108dSAmara Emerson return false; 1343090c108dSAmara Emerson } 13447adc3a2bSChandler Carruth } 13457adc3a2bSChandler Carruth 13467adc3a2bSChandler Carruth // Accumulate the allocated size. 13477adc3a2bSChandler Carruth if (I.isStaticAlloca()) { 13487adc3a2bSChandler Carruth Type *Ty = I.getAllocatedType(); 13495a956c85SMircea Trofin AllocatedSize = 13502ccde3c9SCullen Rhodes SaturatingAdd(DL.getTypeAllocSize(Ty).getKnownMinSize(), AllocatedSize); 13517adc3a2bSChandler Carruth } 13527adc3a2bSChandler Carruth 13537adc3a2bSChandler Carruth // FIXME: This is overly conservative. Dynamic allocas are inefficient for 13547adc3a2bSChandler Carruth // a variety of reasons, and so we would like to not inline them into 13557adc3a2bSChandler Carruth // functions which don't currently have a dynamic alloca. This simply 13567adc3a2bSChandler Carruth // disables inlining altogether in the presence of a dynamic alloca. 1357a3a798d4SArthur Eubanks if (!I.isStaticAlloca()) 13587adc3a2bSChandler Carruth HasDynamicAlloca = true; 1359a3a798d4SArthur Eubanks 13607adc3a2bSChandler Carruth return false; 13617adc3a2bSChandler Carruth } 13627adc3a2bSChandler Carruth 13637adc3a2bSChandler Carruth bool CallAnalyzer::visitPHI(PHINode &I) { 13647adc3a2bSChandler Carruth // FIXME: We need to propagate SROA *disabling* through phi nodes, even 13657adc3a2bSChandler Carruth // though we don't want to propagate it's bonuses. The idea is to disable 13667adc3a2bSChandler Carruth // SROA if it *might* be used in an inappropriate manner. 13677adc3a2bSChandler Carruth 13687adc3a2bSChandler Carruth // Phi nodes are always zero-cost. 136977f32994SBjorn Pettersson // FIXME: Pointer sizes may differ between different address spaces, so do we 137077f32994SBjorn Pettersson // need to use correct address space in the call to getPointerSizeInBits here? 137177f32994SBjorn Pettersson // Or could we skip the getPointerSizeInBits call completely? As far as I can 137277f32994SBjorn Pettersson // see the ZeroOffset is used as a dummy value, so we can probably use any 137377f32994SBjorn Pettersson // bit width for the ZeroOffset? 1374735f4671SChris Lattner APInt ZeroOffset = APInt::getZero(DL.getPointerSizeInBits(0)); 13753739e14aSHaicheng Wu bool CheckSROA = I.getType()->isPointerTy(); 13763739e14aSHaicheng Wu 13773739e14aSHaicheng Wu // Track the constant or pointer with constant offset we've seen so far. 13783739e14aSHaicheng Wu Constant *FirstC = nullptr; 13793739e14aSHaicheng Wu std::pair<Value *, APInt> FirstBaseAndOffset = {nullptr, ZeroOffset}; 13803739e14aSHaicheng Wu Value *FirstV = nullptr; 13813739e14aSHaicheng Wu 13823739e14aSHaicheng Wu for (unsigned i = 0, e = I.getNumIncomingValues(); i != e; ++i) { 13833739e14aSHaicheng Wu BasicBlock *Pred = I.getIncomingBlock(i); 13843739e14aSHaicheng Wu // If the incoming block is dead, skip the incoming block. 13853739e14aSHaicheng Wu if (DeadBlocks.count(Pred)) 13863739e14aSHaicheng Wu continue; 13873739e14aSHaicheng Wu // If the parent block of phi is not the known successor of the incoming 13883739e14aSHaicheng Wu // block, skip the incoming block. 13893739e14aSHaicheng Wu BasicBlock *KnownSuccessor = KnownSuccessors[Pred]; 13903739e14aSHaicheng Wu if (KnownSuccessor && KnownSuccessor != I.getParent()) 13913739e14aSHaicheng Wu continue; 13923739e14aSHaicheng Wu 13933739e14aSHaicheng Wu Value *V = I.getIncomingValue(i); 13943739e14aSHaicheng Wu // If the incoming value is this phi itself, skip the incoming value. 13953739e14aSHaicheng Wu if (&I == V) 13963739e14aSHaicheng Wu continue; 13973739e14aSHaicheng Wu 13983739e14aSHaicheng Wu Constant *C = dyn_cast<Constant>(V); 13993739e14aSHaicheng Wu if (!C) 14003739e14aSHaicheng Wu C = SimplifiedValues.lookup(V); 14013739e14aSHaicheng Wu 14023739e14aSHaicheng Wu std::pair<Value *, APInt> BaseAndOffset = {nullptr, ZeroOffset}; 14033739e14aSHaicheng Wu if (!C && CheckSROA) 14043739e14aSHaicheng Wu BaseAndOffset = ConstantOffsetPtrs.lookup(V); 14053739e14aSHaicheng Wu 14063739e14aSHaicheng Wu if (!C && !BaseAndOffset.first) 14073739e14aSHaicheng Wu // The incoming value is neither a constant nor a pointer with constant 14083739e14aSHaicheng Wu // offset, exit early. 14093739e14aSHaicheng Wu return true; 14103739e14aSHaicheng Wu 14113739e14aSHaicheng Wu if (FirstC) { 14123739e14aSHaicheng Wu if (FirstC == C) 14133739e14aSHaicheng Wu // If we've seen a constant incoming value before and it is the same 14143739e14aSHaicheng Wu // constant we see this time, continue checking the next incoming value. 14153739e14aSHaicheng Wu continue; 14163739e14aSHaicheng Wu // Otherwise early exit because we either see a different constant or saw 14173739e14aSHaicheng Wu // a constant before but we have a pointer with constant offset this time. 14183739e14aSHaicheng Wu return true; 14193739e14aSHaicheng Wu } 14203739e14aSHaicheng Wu 14213739e14aSHaicheng Wu if (FirstV) { 14223739e14aSHaicheng Wu // The same logic as above, but check pointer with constant offset here. 14233739e14aSHaicheng Wu if (FirstBaseAndOffset == BaseAndOffset) 14243739e14aSHaicheng Wu continue; 14253739e14aSHaicheng Wu return true; 14263739e14aSHaicheng Wu } 14273739e14aSHaicheng Wu 14283739e14aSHaicheng Wu if (C) { 14293739e14aSHaicheng Wu // This is the 1st time we've seen a constant, record it. 14303739e14aSHaicheng Wu FirstC = C; 14313739e14aSHaicheng Wu continue; 14323739e14aSHaicheng Wu } 14333739e14aSHaicheng Wu 14343739e14aSHaicheng Wu // The remaining case is that this is the 1st time we've seen a pointer with 14353739e14aSHaicheng Wu // constant offset, record it. 14363739e14aSHaicheng Wu FirstV = V; 14373739e14aSHaicheng Wu FirstBaseAndOffset = BaseAndOffset; 14383739e14aSHaicheng Wu } 14393739e14aSHaicheng Wu 14403739e14aSHaicheng Wu // Check if we can map phi to a constant. 14413739e14aSHaicheng Wu if (FirstC) { 14423739e14aSHaicheng Wu SimplifiedValues[&I] = FirstC; 14433739e14aSHaicheng Wu return true; 14443739e14aSHaicheng Wu } 14453739e14aSHaicheng Wu 14463739e14aSHaicheng Wu // Check if we can map phi to a pointer with constant offset. 14473739e14aSHaicheng Wu if (FirstBaseAndOffset.first) { 14483739e14aSHaicheng Wu ConstantOffsetPtrs[&I] = FirstBaseAndOffset; 14493739e14aSHaicheng Wu 145006408758SMircea Trofin if (auto *SROAArg = getSROAArgForValueOrNull(FirstV)) 14513739e14aSHaicheng Wu SROAArgValues[&I] = SROAArg; 14523739e14aSHaicheng Wu } 14533739e14aSHaicheng Wu 14547adc3a2bSChandler Carruth return true; 14557adc3a2bSChandler Carruth } 14567adc3a2bSChandler Carruth 14575f8f34e4SAdrian Prantl /// Check we can fold GEPs of constant-offset call site argument pointers. 1458d3558b5dSEvgeny Astigeevich /// This requires target data and inbounds GEPs. 1459d3558b5dSEvgeny Astigeevich /// 1460d3558b5dSEvgeny Astigeevich /// \return true if the specified GEP can be folded. 1461d3558b5dSEvgeny Astigeevich bool CallAnalyzer::canFoldInboundsGEP(GetElementPtrInst &I) { 1462d3558b5dSEvgeny Astigeevich // Check if we have a base + offset for the pointer. 1463d3558b5dSEvgeny Astigeevich std::pair<Value *, APInt> BaseAndOffset = 1464d3558b5dSEvgeny Astigeevich ConstantOffsetPtrs.lookup(I.getPointerOperand()); 1465d3558b5dSEvgeny Astigeevich if (!BaseAndOffset.first) 1466d3558b5dSEvgeny Astigeevich return false; 1467d3558b5dSEvgeny Astigeevich 1468d3558b5dSEvgeny Astigeevich // Check if the offset of this GEP is constant, and if so accumulate it 1469d3558b5dSEvgeny Astigeevich // into Offset. 1470d3558b5dSEvgeny Astigeevich if (!accumulateGEPOffset(cast<GEPOperator>(I), BaseAndOffset.second)) 1471d3558b5dSEvgeny Astigeevich return false; 1472d3558b5dSEvgeny Astigeevich 1473d3558b5dSEvgeny Astigeevich // Add the result as a new mapping to Base + Offset. 1474d3558b5dSEvgeny Astigeevich ConstantOffsetPtrs[&I] = BaseAndOffset; 1475d3558b5dSEvgeny Astigeevich 1476d3558b5dSEvgeny Astigeevich return true; 1477d3558b5dSEvgeny Astigeevich } 1478d3558b5dSEvgeny Astigeevich 14797adc3a2bSChandler Carruth bool CallAnalyzer::visitGetElementPtr(GetElementPtrInst &I) { 148006408758SMircea Trofin auto *SROAArg = getSROAArgForValueOrNull(I.getPointerOperand()); 14817adc3a2bSChandler Carruth 1482a8b9cdc9SEaswaran Raman // Lambda to check whether a GEP's indices are all constant. 1483a8b9cdc9SEaswaran Raman auto IsGEPOffsetConstant = [&](GetElementPtrInst &GEP) { 148491aff1d8SNick Desaulniers for (const Use &Op : GEP.indices()) 148591aff1d8SNick Desaulniers if (!isa<Constant>(Op) && !SimplifiedValues.lookup(Op)) 1486a8b9cdc9SEaswaran Raman return false; 1487a8b9cdc9SEaswaran Raman return true; 1488a8b9cdc9SEaswaran Raman }; 1489a8b9cdc9SEaswaran Raman 1490d48c7859SKirill Naumov if (!DisableGEPConstOperand) 1491d48c7859SKirill Naumov if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 1492d48c7859SKirill Naumov SmallVector<Constant *, 2> Indices; 1493d48c7859SKirill Naumov for (unsigned int Index = 1; Index < COps.size(); ++Index) 1494d48c7859SKirill Naumov Indices.push_back(COps[Index]); 14957b639f50SJacob Hegna return ConstantExpr::getGetElementPtr( 14967b639f50SJacob Hegna I.getSourceElementType(), COps[0], Indices, I.isInBounds()); 1497d48c7859SKirill Naumov })) 1498d48c7859SKirill Naumov return true; 1499d48c7859SKirill Naumov 1500d3558b5dSEvgeny Astigeevich if ((I.isInBounds() && canFoldInboundsGEP(I)) || IsGEPOffsetConstant(I)) { 150106408758SMircea Trofin if (SROAArg) 15027adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 15037adc3a2bSChandler Carruth 15047adc3a2bSChandler Carruth // Constant GEPs are modeled as free. 15057adc3a2bSChandler Carruth return true; 15067adc3a2bSChandler Carruth } 15077adc3a2bSChandler Carruth 15087adc3a2bSChandler Carruth // Variable GEPs will require math and will disable SROA. 150906408758SMircea Trofin if (SROAArg) 151006408758SMircea Trofin disableSROAForArg(SROAArg); 1511201b191bSHaicheng Wu return isGEPFree(I); 15127adc3a2bSChandler Carruth } 15137adc3a2bSChandler Carruth 1514617f6364SEaswaran Raman /// Simplify \p I if its operands are constants and update SimplifiedValues. 1515617f6364SEaswaran Raman /// \p Evaluate is a callable specific to instruction type that evaluates the 1516617f6364SEaswaran Raman /// instruction when all the operands are constants. 1517617f6364SEaswaran Raman template <typename Callable> 1518617f6364SEaswaran Raman bool CallAnalyzer::simplifyInstruction(Instruction &I, Callable Evaluate) { 1519617f6364SEaswaran Raman SmallVector<Constant *, 2> COps; 1520617f6364SEaswaran Raman for (Value *Op : I.operands()) { 1521617f6364SEaswaran Raman Constant *COp = dyn_cast<Constant>(Op); 15227adc3a2bSChandler Carruth if (!COp) 1523617f6364SEaswaran Raman COp = SimplifiedValues.lookup(Op); 1524617f6364SEaswaran Raman if (!COp) 1525617f6364SEaswaran Raman return false; 1526617f6364SEaswaran Raman COps.push_back(COp); 1527617f6364SEaswaran Raman } 1528617f6364SEaswaran Raman auto *C = Evaluate(COps); 1529617f6364SEaswaran Raman if (!C) 1530617f6364SEaswaran Raman return false; 15317adc3a2bSChandler Carruth SimplifiedValues[&I] = C; 15327adc3a2bSChandler Carruth return true; 15337adc3a2bSChandler Carruth } 15347adc3a2bSChandler Carruth 15359697f935SNick Desaulniers /// Try to simplify a call to llvm.is.constant. 15369697f935SNick Desaulniers /// 15379697f935SNick Desaulniers /// Duplicate the argument checking from CallAnalyzer::simplifyCallSite since 15389697f935SNick Desaulniers /// we expect calls of this specific intrinsic to be infrequent. 15399697f935SNick Desaulniers /// 15409697f935SNick Desaulniers /// FIXME: Given that we know CB's parent (F) caller 15419697f935SNick Desaulniers /// (CandidateCall->getParent()->getParent()), we might be able to determine 15429697f935SNick Desaulniers /// whether inlining F into F's caller would change how the call to 15439697f935SNick Desaulniers /// llvm.is.constant would evaluate. 15449697f935SNick Desaulniers bool CallAnalyzer::simplifyIntrinsicCallIsConstant(CallBase &CB) { 15459697f935SNick Desaulniers Value *Arg = CB.getArgOperand(0); 15469697f935SNick Desaulniers auto *C = dyn_cast<Constant>(Arg); 15479697f935SNick Desaulniers 15489697f935SNick Desaulniers if (!C) 15499697f935SNick Desaulniers C = dyn_cast_or_null<Constant>(SimplifiedValues.lookup(Arg)); 15509697f935SNick Desaulniers 15519697f935SNick Desaulniers Type *RT = CB.getFunctionType()->getReturnType(); 15529697f935SNick Desaulniers SimplifiedValues[&CB] = ConstantInt::get(RT, C ? 1 : 0); 15539697f935SNick Desaulniers return true; 15549697f935SNick Desaulniers } 15559697f935SNick Desaulniers 1556617f6364SEaswaran Raman bool CallAnalyzer::visitBitCast(BitCastInst &I) { 1557617f6364SEaswaran Raman // Propagate constants through bitcasts. 1558617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 1559617f6364SEaswaran Raman return ConstantExpr::getBitCast(COps[0], I.getType()); 1560617f6364SEaswaran Raman })) 1561617f6364SEaswaran Raman return true; 1562617f6364SEaswaran Raman 15637adc3a2bSChandler Carruth // Track base/offsets through casts 1564567556aaSChad Rosier std::pair<Value *, APInt> BaseAndOffset = 1565567556aaSChad Rosier ConstantOffsetPtrs.lookup(I.getOperand(0)); 15667adc3a2bSChandler Carruth // Casts don't change the offset, just wrap it up. 15677adc3a2bSChandler Carruth if (BaseAndOffset.first) 15687adc3a2bSChandler Carruth ConstantOffsetPtrs[&I] = BaseAndOffset; 15697adc3a2bSChandler Carruth 15707adc3a2bSChandler Carruth // Also look for SROA candidates here. 157106408758SMircea Trofin if (auto *SROAArg = getSROAArgForValueOrNull(I.getOperand(0))) 15727adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 15737adc3a2bSChandler Carruth 15747adc3a2bSChandler Carruth // Bitcasts are always zero cost. 15757adc3a2bSChandler Carruth return true; 15767adc3a2bSChandler Carruth } 15777adc3a2bSChandler Carruth 15787adc3a2bSChandler Carruth bool CallAnalyzer::visitPtrToInt(PtrToIntInst &I) { 15797adc3a2bSChandler Carruth // Propagate constants through ptrtoint. 1580617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 1581617f6364SEaswaran Raman return ConstantExpr::getPtrToInt(COps[0], I.getType()); 1582617f6364SEaswaran Raman })) 15837adc3a2bSChandler Carruth return true; 15847adc3a2bSChandler Carruth 15857adc3a2bSChandler Carruth // Track base/offset pairs when converted to a plain integer provided the 15867adc3a2bSChandler Carruth // integer is large enough to represent the pointer. 15877adc3a2bSChandler Carruth unsigned IntegerSize = I.getType()->getScalarSizeInBits(); 158877f32994SBjorn Pettersson unsigned AS = I.getOperand(0)->getType()->getPointerAddressSpace(); 1589faf848acSMikael Holmen if (IntegerSize == DL.getPointerSizeInBits(AS)) { 1590567556aaSChad Rosier std::pair<Value *, APInt> BaseAndOffset = 1591567556aaSChad Rosier ConstantOffsetPtrs.lookup(I.getOperand(0)); 15927adc3a2bSChandler Carruth if (BaseAndOffset.first) 15937adc3a2bSChandler Carruth ConstantOffsetPtrs[&I] = BaseAndOffset; 15947adc3a2bSChandler Carruth } 15957adc3a2bSChandler Carruth 15967adc3a2bSChandler Carruth // This is really weird. Technically, ptrtoint will disable SROA. However, 15977adc3a2bSChandler Carruth // unless that ptrtoint is *used* somewhere in the live basic blocks after 15987adc3a2bSChandler Carruth // inlining, it will be nuked, and SROA should proceed. All of the uses which 15997adc3a2bSChandler Carruth // would block SROA would also block SROA if applied directly to a pointer, 16007adc3a2bSChandler Carruth // and so we can just add the integer in here. The only places where SROA is 16017adc3a2bSChandler Carruth // preserved either cannot fire on an integer, or won't in-and-of themselves 16027adc3a2bSChandler Carruth // disable SROA (ext) w/o some later use that we would see and disable. 160306408758SMircea Trofin if (auto *SROAArg = getSROAArgForValueOrNull(I.getOperand(0))) 16047adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 16057adc3a2bSChandler Carruth 16063ccbd4f3SSander de Smalen return TTI.getUserCost(&I, TargetTransformInfo::TCK_SizeAndLatency) == 16073ccbd4f3SSander de Smalen TargetTransformInfo::TCC_Free; 16087adc3a2bSChandler Carruth } 16097adc3a2bSChandler Carruth 16107adc3a2bSChandler Carruth bool CallAnalyzer::visitIntToPtr(IntToPtrInst &I) { 16117adc3a2bSChandler Carruth // Propagate constants through ptrtoint. 1612617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 1613617f6364SEaswaran Raman return ConstantExpr::getIntToPtr(COps[0], I.getType()); 1614617f6364SEaswaran Raman })) 16157adc3a2bSChandler Carruth return true; 16167adc3a2bSChandler Carruth 16177adc3a2bSChandler Carruth // Track base/offset pairs when round-tripped through a pointer without 16187adc3a2bSChandler Carruth // modifications provided the integer is not too large. 16197adc3a2bSChandler Carruth Value *Op = I.getOperand(0); 16207adc3a2bSChandler Carruth unsigned IntegerSize = Op->getType()->getScalarSizeInBits(); 162177f32994SBjorn Pettersson if (IntegerSize <= DL.getPointerTypeSizeInBits(I.getType())) { 16227adc3a2bSChandler Carruth std::pair<Value *, APInt> BaseAndOffset = ConstantOffsetPtrs.lookup(Op); 16237adc3a2bSChandler Carruth if (BaseAndOffset.first) 16247adc3a2bSChandler Carruth ConstantOffsetPtrs[&I] = BaseAndOffset; 16257adc3a2bSChandler Carruth } 16267adc3a2bSChandler Carruth 16277adc3a2bSChandler Carruth // "Propagate" SROA here in the same manner as we do for ptrtoint above. 162806408758SMircea Trofin if (auto *SROAArg = getSROAArgForValueOrNull(Op)) 16297adc3a2bSChandler Carruth SROAArgValues[&I] = SROAArg; 16307adc3a2bSChandler Carruth 16313ccbd4f3SSander de Smalen return TTI.getUserCost(&I, TargetTransformInfo::TCK_SizeAndLatency) == 16323ccbd4f3SSander de Smalen TargetTransformInfo::TCC_Free; 16337adc3a2bSChandler Carruth } 16347adc3a2bSChandler Carruth 16357adc3a2bSChandler Carruth bool CallAnalyzer::visitCastInst(CastInst &I) { 1636ab53c5e5SCraig Topper // Propagate constants through casts. 1637617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 1638617f6364SEaswaran Raman return ConstantExpr::getCast(I.getOpcode(), COps[0], I.getType()); 1639617f6364SEaswaran Raman })) 16407adc3a2bSChandler Carruth return true; 16417adc3a2bSChandler Carruth 164210563e16SEric Christopher // Disable SROA in the face of arbitrary casts we don't explicitly list 164310563e16SEric Christopher // elsewhere. 16447adc3a2bSChandler Carruth disableSROA(I.getOperand(0)); 16457adc3a2bSChandler Carruth 164639ed9a60SEli Friedman // If this is a floating-point cast, and the target says this operation 164739ed9a60SEli Friedman // is expensive, this may eventually become a library call. Treat the cost 164839ed9a60SEli Friedman // as such. 164939ed9a60SEli Friedman switch (I.getOpcode()) { 165039ed9a60SEli Friedman case Instruction::FPTrunc: 165139ed9a60SEli Friedman case Instruction::FPExt: 165239ed9a60SEli Friedman case Instruction::UIToFP: 165339ed9a60SEli Friedman case Instruction::SIToFP: 165439ed9a60SEli Friedman case Instruction::FPToUI: 165539ed9a60SEli Friedman case Instruction::FPToSI: 165639ed9a60SEli Friedman if (TTI.getFPOpCost(I.getType()) == TargetTransformInfo::TCC_Expensive) 165706408758SMircea Trofin onCallPenalty(); 16584dc0b1acSReid Kleckner break; 165939ed9a60SEli Friedman default: 166039ed9a60SEli Friedman break; 166139ed9a60SEli Friedman } 166239ed9a60SEli Friedman 16633ccbd4f3SSander de Smalen return TTI.getUserCost(&I, TargetTransformInfo::TCK_SizeAndLatency) == 16643ccbd4f3SSander de Smalen TargetTransformInfo::TCC_Free; 16657adc3a2bSChandler Carruth } 16667adc3a2bSChandler Carruth 16677adc3a2bSChandler Carruth bool CallAnalyzer::paramHasAttr(Argument *A, Attribute::AttrKind Attr) { 1668652168a9SFedor Sergeev return CandidateCall.paramHasAttr(A->getArgNo(), Attr); 16697adc3a2bSChandler Carruth } 16707adc3a2bSChandler Carruth 16717adc3a2bSChandler Carruth bool CallAnalyzer::isKnownNonNullInCallee(Value *V) { 16727adc3a2bSChandler Carruth // Does the *call site* have the NonNull attribute set on an argument? We 16737adc3a2bSChandler Carruth // use the attribute on the call site to memoize any analysis done in the 16747adc3a2bSChandler Carruth // caller. This will also trip if the callee function has a non-null 16757adc3a2bSChandler Carruth // parameter attribute, but that's a less interesting case because hopefully 16767adc3a2bSChandler Carruth // the callee would already have been simplified based on that. 16777adc3a2bSChandler Carruth if (Argument *A = dyn_cast<Argument>(V)) 16787adc3a2bSChandler Carruth if (paramHasAttr(A, Attribute::NonNull)) 16797adc3a2bSChandler Carruth return true; 16807adc3a2bSChandler Carruth 16817adc3a2bSChandler Carruth // Is this an alloca in the caller? This is distinct from the attribute case 16827adc3a2bSChandler Carruth // above because attributes aren't updated within the inliner itself and we 16837adc3a2bSChandler Carruth // always want to catch the alloca derived case. 16847adc3a2bSChandler Carruth if (isAllocaDerivedArg(V)) 16857adc3a2bSChandler Carruth // We can actually predict the result of comparisons between an 16867adc3a2bSChandler Carruth // alloca-derived value and null. Note that this fires regardless of 16877adc3a2bSChandler Carruth // SROA firing. 16887adc3a2bSChandler Carruth return true; 16897adc3a2bSChandler Carruth 16907adc3a2bSChandler Carruth return false; 16917adc3a2bSChandler Carruth } 16927adc3a2bSChandler Carruth 1693652168a9SFedor Sergeev bool CallAnalyzer::allowSizeGrowth(CallBase &Call) { 16949a3fc17aSEaswaran Raman // If the normal destination of the invoke or the parent block of the call 16959a3fc17aSEaswaran Raman // site is unreachable-terminated, there is little point in inlining this 16969a3fc17aSEaswaran Raman // unless there is literally zero cost. 16979a3fc17aSEaswaran Raman // FIXME: Note that it is possible that an unreachable-terminated block has a 16989a3fc17aSEaswaran Raman // hot entry. For example, in below scenario inlining hot_call_X() may be 16999a3fc17aSEaswaran Raman // beneficial : 17009a3fc17aSEaswaran Raman // main() { 17019a3fc17aSEaswaran Raman // hot_call_1(); 17029a3fc17aSEaswaran Raman // ... 17039a3fc17aSEaswaran Raman // hot_call_N() 17049a3fc17aSEaswaran Raman // exit(0); 17059a3fc17aSEaswaran Raman // } 17069a3fc17aSEaswaran Raman // For now, we are not handling this corner case here as it is rare in real 17079a3fc17aSEaswaran Raman // code. In future, we should elaborate this based on BPI and BFI in more 17089a3fc17aSEaswaran Raman // general threshold adjusting heuristics in updateThreshold(). 1709652168a9SFedor Sergeev if (InvokeInst *II = dyn_cast<InvokeInst>(&Call)) { 17109a3fc17aSEaswaran Raman if (isa<UnreachableInst>(II->getNormalDest()->getTerminator())) 17119a3fc17aSEaswaran Raman return false; 1712652168a9SFedor Sergeev } else if (isa<UnreachableInst>(Call.getParent()->getTerminator())) 17139a3fc17aSEaswaran Raman return false; 17149a3fc17aSEaswaran Raman 17159a3fc17aSEaswaran Raman return true; 17169a3fc17aSEaswaran Raman } 17179a3fc17aSEaswaran Raman 171806408758SMircea Trofin bool InlineCostCallAnalyzer::isColdCallSite(CallBase &Call, 1719652168a9SFedor Sergeev BlockFrequencyInfo *CallerBFI) { 1720c5fa6358SEaswaran Raman // If global profile summary is available, then callsite's coldness is 1721c5fa6358SEaswaran Raman // determined based on that. 1722bba762a1SChandler Carruth if (PSI && PSI->hasProfileSummary()) 1723ec73ae11SMircea Trofin return PSI->isColdCallSite(Call, CallerBFI); 1724bba762a1SChandler Carruth 1725bba762a1SChandler Carruth // Otherwise we need BFI to be available. 1726c5fa6358SEaswaran Raman if (!CallerBFI) 1727c5fa6358SEaswaran Raman return false; 1728c5fa6358SEaswaran Raman 1729bba762a1SChandler Carruth // Determine if the callsite is cold relative to caller's entry. We could 1730bba762a1SChandler Carruth // potentially cache the computation of scaled entry frequency, but the added 1731bba762a1SChandler Carruth // complexity is not worth it unless this scaling shows up high in the 1732bba762a1SChandler Carruth // profiles. 1733c5fa6358SEaswaran Raman const BranchProbability ColdProb(ColdCallSiteRelFreq, 100); 1734652168a9SFedor Sergeev auto CallSiteBB = Call.getParent(); 1735c5fa6358SEaswaran Raman auto CallSiteFreq = CallerBFI->getBlockFreq(CallSiteBB); 1736c5fa6358SEaswaran Raman auto CallerEntryFreq = 1737652168a9SFedor Sergeev CallerBFI->getBlockFreq(&(Call.getCaller()->getEntryBlock())); 1738c5fa6358SEaswaran Raman return CallSiteFreq < CallerEntryFreq * ColdProb; 1739c5fa6358SEaswaran Raman } 1740c5fa6358SEaswaran Raman 1741974d4eeaSEaswaran Raman Optional<int> 174206408758SMircea Trofin InlineCostCallAnalyzer::getHotCallSiteThreshold(CallBase &Call, 1743974d4eeaSEaswaran Raman BlockFrequencyInfo *CallerBFI) { 1744bba762a1SChandler Carruth 1745974d4eeaSEaswaran Raman // If global profile summary is available, then callsite's hotness is 1746974d4eeaSEaswaran Raman // determined based on that. 1747ec73ae11SMircea Trofin if (PSI && PSI->hasProfileSummary() && PSI->isHotCallSite(Call, CallerBFI)) 1748bba762a1SChandler Carruth return Params.HotCallSiteThreshold; 1749974d4eeaSEaswaran Raman 1750bba762a1SChandler Carruth // Otherwise we need BFI to be available and to have a locally hot callsite 1751bba762a1SChandler Carruth // threshold. 1752bba762a1SChandler Carruth if (!CallerBFI || !Params.LocallyHotCallSiteThreshold) 1753974d4eeaSEaswaran Raman return None; 1754974d4eeaSEaswaran Raman 1755bba762a1SChandler Carruth // Determine if the callsite is hot relative to caller's entry. We could 1756bba762a1SChandler Carruth // potentially cache the computation of scaled entry frequency, but the added 1757bba762a1SChandler Carruth // complexity is not worth it unless this scaling shows up high in the 1758bba762a1SChandler Carruth // profiles. 1759652168a9SFedor Sergeev auto CallSiteBB = Call.getParent(); 1760974d4eeaSEaswaran Raman auto CallSiteFreq = CallerBFI->getBlockFreq(CallSiteBB).getFrequency(); 1761974d4eeaSEaswaran Raman auto CallerEntryFreq = CallerBFI->getEntryFreq(); 1762974d4eeaSEaswaran Raman if (CallSiteFreq >= CallerEntryFreq * HotCallSiteRelFreq) 1763bba762a1SChandler Carruth return Params.LocallyHotCallSiteThreshold; 1764bba762a1SChandler Carruth 1765bba762a1SChandler Carruth // Otherwise treat it normally. 1766974d4eeaSEaswaran Raman return None; 1767974d4eeaSEaswaran Raman } 1768974d4eeaSEaswaran Raman 176906408758SMircea Trofin void InlineCostCallAnalyzer::updateThreshold(CallBase &Call, Function &Callee) { 17709a3fc17aSEaswaran Raman // If no size growth is allowed for this inlining, set Threshold to 0. 1771652168a9SFedor Sergeev if (!allowSizeGrowth(Call)) { 17729a3fc17aSEaswaran Raman Threshold = 0; 17739a3fc17aSEaswaran Raman return; 17749a3fc17aSEaswaran Raman } 17759a3fc17aSEaswaran Raman 1776652168a9SFedor Sergeev Function *Caller = Call.getCaller(); 17771c57cc2bSEaswaran Raman 17781c57cc2bSEaswaran Raman // return min(A, B) if B is valid. 17791c57cc2bSEaswaran Raman auto MinIfValid = [](int A, Optional<int> B) { 17801c57cc2bSEaswaran Raman return B ? std::min(A, B.getValue()) : A; 17811c57cc2bSEaswaran Raman }; 17821c57cc2bSEaswaran Raman 17830d58fcacSEaswaran Raman // return max(A, B) if B is valid. 17840d58fcacSEaswaran Raman auto MaxIfValid = [](int A, Optional<int> B) { 17850d58fcacSEaswaran Raman return B ? std::max(A, B.getValue()) : A; 17860d58fcacSEaswaran Raman }; 17870d58fcacSEaswaran Raman 178851b809bfSEaswaran Raman // Various bonus percentages. These are multiplied by Threshold to get the 178951b809bfSEaswaran Raman // bonus values. 179051b809bfSEaswaran Raman // SingleBBBonus: This bonus is applied if the callee has a single reachable 179151b809bfSEaswaran Raman // basic block at the given callsite context. This is speculatively applied 179251b809bfSEaswaran Raman // and withdrawn if more than one basic block is seen. 179351b809bfSEaswaran Raman // 179451b809bfSEaswaran Raman // LstCallToStaticBonus: This large bonus is applied to ensure the inlining 179551b809bfSEaswaran Raman // of the last call to a static function as inlining such functions is 179651b809bfSEaswaran Raman // guaranteed to reduce code size. 179751b809bfSEaswaran Raman // 179851b809bfSEaswaran Raman // These bonus percentages may be set to 0 based on properties of the caller 179951b809bfSEaswaran Raman // and the callsite. 180051b809bfSEaswaran Raman int SingleBBBonusPercent = 50; 1801d912a9baSDaniil Fukalov int VectorBonusPercent = TTI.getInlinerVectorBonusPercent(); 180202647f73SJordan Rupprecht int LastCallToStaticBonus = InlineConstants::LastCallToStaticBonus; 180351b809bfSEaswaran Raman 180451b809bfSEaswaran Raman // Lambda to set all the above bonus and bonus percentages to 0. 180551b809bfSEaswaran Raman auto DisallowAllBonuses = [&]() { 180651b809bfSEaswaran Raman SingleBBBonusPercent = 0; 180751b809bfSEaswaran Raman VectorBonusPercent = 0; 180851b809bfSEaswaran Raman LastCallToStaticBonus = 0; 180951b809bfSEaswaran Raman }; 181051b809bfSEaswaran Raman 18111c57cc2bSEaswaran Raman // Use the OptMinSizeThreshold or OptSizeThreshold knob if they are available 18121c57cc2bSEaswaran Raman // and reduce the threshold if the caller has the necessary attribute. 181385bd3978SEvandro Menezes if (Caller->hasMinSize()) { 18141c57cc2bSEaswaran Raman Threshold = MinIfValid(Threshold, Params.OptMinSizeThreshold); 181551b809bfSEaswaran Raman // For minsize, we want to disable the single BB bonus and the vector 181651b809bfSEaswaran Raman // bonuses, but not the last-call-to-static bonus. Inlining the last call to 181751b809bfSEaswaran Raman // a static function will, at the minimum, eliminate the parameter setup and 181851b809bfSEaswaran Raman // call/return instructions. 181951b809bfSEaswaran Raman SingleBBBonusPercent = 0; 182051b809bfSEaswaran Raman VectorBonusPercent = 0; 182185bd3978SEvandro Menezes } else if (Caller->hasOptSize()) 18221c57cc2bSEaswaran Raman Threshold = MinIfValid(Threshold, Params.OptSizeThreshold); 1823f4bb2f0dSEaswaran Raman 1824e08b139dSEaswaran Raman // Adjust the threshold based on inlinehint attribute and profile based 1825e08b139dSEaswaran Raman // hotness information if the caller does not have MinSize attribute. 182685bd3978SEvandro Menezes if (!Caller->hasMinSize()) { 1827e08b139dSEaswaran Raman if (Callee.hasFnAttribute(Attribute::InlineHint)) 18280d58fcacSEaswaran Raman Threshold = MaxIfValid(Threshold, Params.HintThreshold); 1829bba762a1SChandler Carruth 18303cd1479cSEaswaran Raman // FIXME: After switching to the new passmanager, simplify the logic below 1831bba762a1SChandler Carruth // by checking only the callsite hotness/coldness as we will reliably 1832bba762a1SChandler Carruth // have local profile information. 18333cd1479cSEaswaran Raman // 1834bba762a1SChandler Carruth // Callsite hotness and coldness can be determined if sample profile is 1835bba762a1SChandler Carruth // used (which adds hotness metadata to calls) or if caller's 1836bba762a1SChandler Carruth // BlockFrequencyInfo is available. 183708e2386dSMircea Trofin BlockFrequencyInfo *CallerBFI = GetBFI ? &(GetBFI(*Caller)) : nullptr; 1838652168a9SFedor Sergeev auto HotCallSiteThreshold = getHotCallSiteThreshold(Call, CallerBFI); 183985bd3978SEvandro Menezes if (!Caller->hasOptSize() && HotCallSiteThreshold) { 1840d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "Hot callsite.\n"); 1841974d4eeaSEaswaran Raman // FIXME: This should update the threshold only if it exceeds the 1842974d4eeaSEaswaran Raman // current threshold, but AutoFDO + ThinLTO currently relies on this 1843974d4eeaSEaswaran Raman // behavior to prevent inlining of hot callsites during ThinLTO 1844974d4eeaSEaswaran Raman // compile phase. 1845974d4eeaSEaswaran Raman Threshold = HotCallSiteThreshold.getValue(); 1846652168a9SFedor Sergeev } else if (isColdCallSite(Call, CallerBFI)) { 1847d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "Cold callsite.\n"); 184851b809bfSEaswaran Raman // Do not apply bonuses for a cold callsite including the 184951b809bfSEaswaran Raman // LastCallToStatic bonus. While this bonus might result in code size 185051b809bfSEaswaran Raman // reduction, it can cause the size of a non-cold caller to increase 185151b809bfSEaswaran Raman // preventing it from being inlined. 185251b809bfSEaswaran Raman DisallowAllBonuses(); 18533cd1479cSEaswaran Raman Threshold = MinIfValid(Threshold, Params.ColdCallSiteThreshold); 1854bba762a1SChandler Carruth } else if (PSI) { 1855bba762a1SChandler Carruth // Use callee's global profile information only if we have no way of 1856bba762a1SChandler Carruth // determining this via callsite information. 18573cd1479cSEaswaran Raman if (PSI->isFunctionEntryHot(&Callee)) { 1858d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "Hot callee.\n"); 1859e08b139dSEaswaran Raman // If callsite hotness can not be determined, we may still know 1860e08b139dSEaswaran Raman // that the callee is hot and treat it as a weaker hint for threshold 1861e08b139dSEaswaran Raman // increase. 1862e08b139dSEaswaran Raman Threshold = MaxIfValid(Threshold, Params.HintThreshold); 1863e08b139dSEaswaran Raman } else if (PSI->isFunctionEntryCold(&Callee)) { 1864d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "Cold callee.\n"); 186551b809bfSEaswaran Raman // Do not apply bonuses for a cold callee including the 186651b809bfSEaswaran Raman // LastCallToStatic bonus. While this bonus might result in code size 186751b809bfSEaswaran Raman // reduction, it can cause the size of a non-cold caller to increase 186851b809bfSEaswaran Raman // preventing it from being inlined. 186951b809bfSEaswaran Raman DisallowAllBonuses(); 18701c57cc2bSEaswaran Raman Threshold = MinIfValid(Threshold, Params.ColdThreshold); 1871e08b139dSEaswaran Raman } 1872e08b139dSEaswaran Raman } 1873e08b139dSEaswaran Raman } 18748650a4daSJustin Lebar 1875b7b99b07SStanislav Mekhanoshin Threshold += TTI.adjustInliningThreshold(&Call); 1876b7b99b07SStanislav Mekhanoshin 187702647f73SJordan Rupprecht // Finally, take the target-specific inlining threshold multiplier into 187802647f73SJordan Rupprecht // account. 18798650a4daSJustin Lebar Threshold *= TTI.getInliningThresholdMultiplier(); 188051b809bfSEaswaran Raman 188151b809bfSEaswaran Raman SingleBBBonus = Threshold * SingleBBBonusPercent / 100; 188251b809bfSEaswaran Raman VectorBonus = Threshold * VectorBonusPercent / 100; 188351b809bfSEaswaran Raman 188411c8efd4SErik Desjardins bool OnlyOneCallAndLocalLinkage = F.hasLocalLinkage() && F.hasOneLiveUse() && 188511c8efd4SErik Desjardins &F == Call.getCalledFunction(); 188602647f73SJordan Rupprecht // If there is only one call of the function, and it has internal linkage, 188702647f73SJordan Rupprecht // the cost of inlining it drops dramatically. It may seem odd to update 188802647f73SJordan Rupprecht // Cost in updateThreshold, but the bonus depends on the logic in this method. 188902647f73SJordan Rupprecht if (OnlyOneCallAndLocalLinkage) 189002647f73SJordan Rupprecht Cost -= LastCallToStaticBonus; 1891f4bb2f0dSEaswaran Raman } 1892f4bb2f0dSEaswaran Raman 18937adc3a2bSChandler Carruth bool CallAnalyzer::visitCmpInst(CmpInst &I) { 18947adc3a2bSChandler Carruth Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); 18957adc3a2bSChandler Carruth // First try to handle simplified comparisons. 1896617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 1897617f6364SEaswaran Raman return ConstantExpr::getCompare(I.getPredicate(), COps[0], COps[1]); 1898617f6364SEaswaran Raman })) 18997adc3a2bSChandler Carruth return true; 19007adc3a2bSChandler Carruth 19017adc3a2bSChandler Carruth if (I.getOpcode() == Instruction::FCmp) 19027adc3a2bSChandler Carruth return false; 19037adc3a2bSChandler Carruth 19047adc3a2bSChandler Carruth // Otherwise look for a comparison between constant offset pointers with 19057adc3a2bSChandler Carruth // a common base. 19067adc3a2bSChandler Carruth Value *LHSBase, *RHSBase; 19077adc3a2bSChandler Carruth APInt LHSOffset, RHSOffset; 19087adc3a2bSChandler Carruth std::tie(LHSBase, LHSOffset) = ConstantOffsetPtrs.lookup(LHS); 19097adc3a2bSChandler Carruth if (LHSBase) { 19107adc3a2bSChandler Carruth std::tie(RHSBase, RHSOffset) = ConstantOffsetPtrs.lookup(RHS); 19117adc3a2bSChandler Carruth if (RHSBase && LHSBase == RHSBase) { 19127adc3a2bSChandler Carruth // We have common bases, fold the icmp to a constant based on the 19137adc3a2bSChandler Carruth // offsets. 19147adc3a2bSChandler Carruth Constant *CLHS = ConstantInt::get(LHS->getContext(), LHSOffset); 19157adc3a2bSChandler Carruth Constant *CRHS = ConstantInt::get(RHS->getContext(), RHSOffset); 19167adc3a2bSChandler Carruth if (Constant *C = ConstantExpr::getICmp(I.getPredicate(), CLHS, CRHS)) { 19177adc3a2bSChandler Carruth SimplifiedValues[&I] = C; 19187adc3a2bSChandler Carruth ++NumConstantPtrCmps; 19197adc3a2bSChandler Carruth return true; 19207adc3a2bSChandler Carruth } 19217adc3a2bSChandler Carruth } 19227adc3a2bSChandler Carruth } 19237adc3a2bSChandler Carruth 19247adc3a2bSChandler Carruth // If the comparison is an equality comparison with null, we can simplify it 19257adc3a2bSChandler Carruth // if we know the value (argument) can't be null 19267adc3a2bSChandler Carruth if (I.isEquality() && isa<ConstantPointerNull>(I.getOperand(1)) && 19277adc3a2bSChandler Carruth isKnownNonNullInCallee(I.getOperand(0))) { 19287adc3a2bSChandler Carruth bool IsNotEqual = I.getPredicate() == CmpInst::ICMP_NE; 19297adc3a2bSChandler Carruth SimplifiedValues[&I] = IsNotEqual ? ConstantInt::getTrue(I.getType()) 19307adc3a2bSChandler Carruth : ConstantInt::getFalse(I.getType()); 19317adc3a2bSChandler Carruth return true; 19327adc3a2bSChandler Carruth } 193306408758SMircea Trofin return handleSROA(I.getOperand(0), isa<ConstantPointerNull>(I.getOperand(1))); 19347adc3a2bSChandler Carruth } 19357adc3a2bSChandler Carruth 19367adc3a2bSChandler Carruth bool CallAnalyzer::visitSub(BinaryOperator &I) { 19377adc3a2bSChandler Carruth // Try to handle a special case: we can fold computing the difference of two 19387adc3a2bSChandler Carruth // constant-related pointers. 19397adc3a2bSChandler Carruth Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); 19407adc3a2bSChandler Carruth Value *LHSBase, *RHSBase; 19417adc3a2bSChandler Carruth APInt LHSOffset, RHSOffset; 19427adc3a2bSChandler Carruth std::tie(LHSBase, LHSOffset) = ConstantOffsetPtrs.lookup(LHS); 19437adc3a2bSChandler Carruth if (LHSBase) { 19447adc3a2bSChandler Carruth std::tie(RHSBase, RHSOffset) = ConstantOffsetPtrs.lookup(RHS); 19457adc3a2bSChandler Carruth if (RHSBase && LHSBase == RHSBase) { 19467adc3a2bSChandler Carruth // We have common bases, fold the subtract to a constant based on the 19477adc3a2bSChandler Carruth // offsets. 19487adc3a2bSChandler Carruth Constant *CLHS = ConstantInt::get(LHS->getContext(), LHSOffset); 19497adc3a2bSChandler Carruth Constant *CRHS = ConstantInt::get(RHS->getContext(), RHSOffset); 19507adc3a2bSChandler Carruth if (Constant *C = ConstantExpr::getSub(CLHS, CRHS)) { 19517adc3a2bSChandler Carruth SimplifiedValues[&I] = C; 19527adc3a2bSChandler Carruth ++NumConstantPtrDiffs; 19537adc3a2bSChandler Carruth return true; 19547adc3a2bSChandler Carruth } 19557adc3a2bSChandler Carruth } 19567adc3a2bSChandler Carruth } 19577adc3a2bSChandler Carruth 19587adc3a2bSChandler Carruth // Otherwise, fall back to the generic logic for simplifying and handling 19597adc3a2bSChandler Carruth // instructions. 19607adc3a2bSChandler Carruth return Base::visitSub(I); 19617adc3a2bSChandler Carruth } 19627adc3a2bSChandler Carruth 19637adc3a2bSChandler Carruth bool CallAnalyzer::visitBinaryOperator(BinaryOperator &I) { 19647adc3a2bSChandler Carruth Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); 19656d14dfe8SHaicheng Wu Constant *CLHS = dyn_cast<Constant>(LHS); 19666d14dfe8SHaicheng Wu if (!CLHS) 19676d14dfe8SHaicheng Wu CLHS = SimplifiedValues.lookup(LHS); 19686d14dfe8SHaicheng Wu Constant *CRHS = dyn_cast<Constant>(RHS); 19696d14dfe8SHaicheng Wu if (!CRHS) 19706d14dfe8SHaicheng Wu CRHS = SimplifiedValues.lookup(RHS); 19716d14dfe8SHaicheng Wu 19727adc3a2bSChandler Carruth Value *SimpleV = nullptr; 19737adc3a2bSChandler Carruth if (auto FI = dyn_cast<FPMathOperator>(&I)) 19745a956c85SMircea Trofin SimpleV = SimplifyBinOp(I.getOpcode(), CLHS ? CLHS : LHS, CRHS ? CRHS : RHS, 19755a956c85SMircea Trofin FI->getFastMathFlags(), DL); 19767adc3a2bSChandler Carruth else 19776d14dfe8SHaicheng Wu SimpleV = 19786d14dfe8SHaicheng Wu SimplifyBinOp(I.getOpcode(), CLHS ? CLHS : LHS, CRHS ? CRHS : RHS, DL); 19797adc3a2bSChandler Carruth 19806d14dfe8SHaicheng Wu if (Constant *C = dyn_cast_or_null<Constant>(SimpleV)) 19816d14dfe8SHaicheng Wu SimplifiedValues[&I] = C; 19826d14dfe8SHaicheng Wu 19836d14dfe8SHaicheng Wu if (SimpleV) 19847adc3a2bSChandler Carruth return true; 19857adc3a2bSChandler Carruth 19867adc3a2bSChandler Carruth // Disable any SROA on arguments to arbitrary, unsimplified binary operators. 19877adc3a2bSChandler Carruth disableSROA(LHS); 19887adc3a2bSChandler Carruth disableSROA(RHS); 19897adc3a2bSChandler Carruth 199039ed9a60SEli Friedman // If the instruction is floating point, and the target says this operation 199139ed9a60SEli Friedman // is expensive, this may eventually become a library call. Treat the cost 19927cebf0afSCraig Topper // as such. Unless it's fneg which can be implemented with an xor. 19937cebf0afSCraig Topper using namespace llvm::PatternMatch; 199439ed9a60SEli Friedman if (I.getType()->isFloatingPointTy() && 19957cebf0afSCraig Topper TTI.getFPOpCost(I.getType()) == TargetTransformInfo::TCC_Expensive && 19967cebf0afSCraig Topper !match(&I, m_FNeg(m_Value()))) 199706408758SMircea Trofin onCallPenalty(); 199839ed9a60SEli Friedman 19997adc3a2bSChandler Carruth return false; 20007adc3a2bSChandler Carruth } 20017adc3a2bSChandler Carruth 20026cda33baSCraig Topper bool CallAnalyzer::visitFNeg(UnaryOperator &I) { 20036cda33baSCraig Topper Value *Op = I.getOperand(0); 20046cda33baSCraig Topper Constant *COp = dyn_cast<Constant>(Op); 20056cda33baSCraig Topper if (!COp) 20066cda33baSCraig Topper COp = SimplifiedValues.lookup(Op); 20076cda33baSCraig Topper 20085a956c85SMircea Trofin Value *SimpleV = SimplifyFNegInst( 20095a956c85SMircea Trofin COp ? COp : Op, cast<FPMathOperator>(I).getFastMathFlags(), DL); 20106cda33baSCraig Topper 20116cda33baSCraig Topper if (Constant *C = dyn_cast_or_null<Constant>(SimpleV)) 20126cda33baSCraig Topper SimplifiedValues[&I] = C; 20136cda33baSCraig Topper 20146cda33baSCraig Topper if (SimpleV) 20156cda33baSCraig Topper return true; 20166cda33baSCraig Topper 20176cda33baSCraig Topper // Disable any SROA on arguments to arbitrary, unsimplified fneg. 20186cda33baSCraig Topper disableSROA(Op); 20196cda33baSCraig Topper 20206cda33baSCraig Topper return false; 20216cda33baSCraig Topper } 20226cda33baSCraig Topper 20237adc3a2bSChandler Carruth bool CallAnalyzer::visitLoad(LoadInst &I) { 202406408758SMircea Trofin if (handleSROA(I.getPointerOperand(), I.isSimple())) 20257adc3a2bSChandler Carruth return true; 20267adc3a2bSChandler Carruth 2027a4461515SHaicheng Wu // If the data is already loaded from this address and hasn't been clobbered 2028a4461515SHaicheng Wu // by any stores or calls, this load is likely to be redundant and can be 2029a4461515SHaicheng Wu // eliminated. 2030a4461515SHaicheng Wu if (EnableLoadElimination && 2031b3689cabSHaicheng Wu !LoadAddrSet.insert(I.getPointerOperand()).second && I.isUnordered()) { 203206408758SMircea Trofin onLoadEliminationOpportunity(); 2033a4461515SHaicheng Wu return true; 2034a4461515SHaicheng Wu } 2035a4461515SHaicheng Wu 20367adc3a2bSChandler Carruth return false; 20377adc3a2bSChandler Carruth } 20387adc3a2bSChandler Carruth 20397adc3a2bSChandler Carruth bool CallAnalyzer::visitStore(StoreInst &I) { 204006408758SMircea Trofin if (handleSROA(I.getPointerOperand(), I.isSimple())) 20417adc3a2bSChandler Carruth return true; 20427adc3a2bSChandler Carruth 2043a4461515SHaicheng Wu // The store can potentially clobber loads and prevent repeated loads from 2044a4461515SHaicheng Wu // being eliminated. 2045a4461515SHaicheng Wu // FIXME: 2046a4461515SHaicheng Wu // 1. We can probably keep an initial set of eliminatable loads substracted 2047a4461515SHaicheng Wu // from the cost even when we finally see a store. We just need to disable 2048a4461515SHaicheng Wu // *further* accumulation of elimination savings. 2049a4461515SHaicheng Wu // 2. We should probably at some point thread MemorySSA for the callee into 2050a4461515SHaicheng Wu // this and then use that to actually compute *really* precise savings. 2051a4461515SHaicheng Wu disableLoadElimination(); 20527adc3a2bSChandler Carruth return false; 20537adc3a2bSChandler Carruth } 20547adc3a2bSChandler Carruth 20557adc3a2bSChandler Carruth bool CallAnalyzer::visitExtractValue(ExtractValueInst &I) { 20567adc3a2bSChandler Carruth // Constant folding for extract value is trivial. 2057617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 2058617f6364SEaswaran Raman return ConstantExpr::getExtractValue(COps[0], I.getIndices()); 2059617f6364SEaswaran Raman })) 20607adc3a2bSChandler Carruth return true; 20617adc3a2bSChandler Carruth 2062ba5b015bSRoman Lebedev // SROA can't look through these, but they may be free. 2063ba5b015bSRoman Lebedev return Base::visitExtractValue(I); 20647adc3a2bSChandler Carruth } 20657adc3a2bSChandler Carruth 20667adc3a2bSChandler Carruth bool CallAnalyzer::visitInsertValue(InsertValueInst &I) { 20677adc3a2bSChandler Carruth // Constant folding for insert value is trivial. 2068617f6364SEaswaran Raman if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) { 2069617f6364SEaswaran Raman return ConstantExpr::getInsertValue(/*AggregateOperand*/ COps[0], 2070617f6364SEaswaran Raman /*InsertedValueOperand*/ COps[1], 2071617f6364SEaswaran Raman I.getIndices()); 2072617f6364SEaswaran Raman })) 20737adc3a2bSChandler Carruth return true; 20747adc3a2bSChandler Carruth 2075ba5b015bSRoman Lebedev // SROA can't look through these, but they may be free. 2076ba5b015bSRoman Lebedev return Base::visitInsertValue(I); 20777adc3a2bSChandler Carruth } 20787adc3a2bSChandler Carruth 20795f8f34e4SAdrian Prantl /// Try to simplify a call site. 20807adc3a2bSChandler Carruth /// 20817adc3a2bSChandler Carruth /// Takes a concrete function and callsite and tries to actually simplify it by 20827adc3a2bSChandler Carruth /// analyzing the arguments and call itself with instsimplify. Returns true if 20837adc3a2bSChandler Carruth /// it has simplified the callsite to some other entity (a constant), making it 20847adc3a2bSChandler Carruth /// free. 2085652168a9SFedor Sergeev bool CallAnalyzer::simplifyCallSite(Function *F, CallBase &Call) { 20867adc3a2bSChandler Carruth // FIXME: Using the instsimplify logic directly for this is inefficient 20877adc3a2bSChandler Carruth // because we have to continually rebuild the argument list even when no 20887adc3a2bSChandler Carruth // simplifications can be performed. Until that is fixed with remapping 20897adc3a2bSChandler Carruth // inside of instsimplify, directly constant fold calls here. 2090652168a9SFedor Sergeev if (!canConstantFoldCallTo(&Call, F)) 20917adc3a2bSChandler Carruth return false; 20927adc3a2bSChandler Carruth 20937adc3a2bSChandler Carruth // Try to re-map the arguments to constants. 20947adc3a2bSChandler Carruth SmallVector<Constant *, 4> ConstantArgs; 2095652168a9SFedor Sergeev ConstantArgs.reserve(Call.arg_size()); 2096652168a9SFedor Sergeev for (Value *I : Call.args()) { 2097652168a9SFedor Sergeev Constant *C = dyn_cast<Constant>(I); 20987adc3a2bSChandler Carruth if (!C) 2099652168a9SFedor Sergeev C = dyn_cast_or_null<Constant>(SimplifiedValues.lookup(I)); 21007adc3a2bSChandler Carruth if (!C) 21017adc3a2bSChandler Carruth return false; // This argument doesn't map to a constant. 21027adc3a2bSChandler Carruth 21037adc3a2bSChandler Carruth ConstantArgs.push_back(C); 21047adc3a2bSChandler Carruth } 2105652168a9SFedor Sergeev if (Constant *C = ConstantFoldCall(&Call, F, ConstantArgs)) { 2106652168a9SFedor Sergeev SimplifiedValues[&Call] = C; 21077adc3a2bSChandler Carruth return true; 21087adc3a2bSChandler Carruth } 21097adc3a2bSChandler Carruth 21107adc3a2bSChandler Carruth return false; 21117adc3a2bSChandler Carruth } 21127adc3a2bSChandler Carruth 2113652168a9SFedor Sergeev bool CallAnalyzer::visitCallBase(CallBase &Call) { 21145c97507eSDaniil Suchkov if (!onCallBaseVisitStart(Call)) 21155c97507eSDaniil Suchkov return true; 21165c97507eSDaniil Suchkov 2117652168a9SFedor Sergeev if (Call.hasFnAttr(Attribute::ReturnsTwice) && 21187adc3a2bSChandler Carruth !F.hasFnAttribute(Attribute::ReturnsTwice)) { 21197adc3a2bSChandler Carruth // This aborts the entire analysis. 21207adc3a2bSChandler Carruth ExposesReturnsTwice = true; 21217adc3a2bSChandler Carruth return false; 21227adc3a2bSChandler Carruth } 2123652168a9SFedor Sergeev if (isa<CallInst>(Call) && cast<CallInst>(Call).cannotDuplicate()) 21247adc3a2bSChandler Carruth ContainsNoDuplicateCall = true; 21257adc3a2bSChandler Carruth 2126825debe8SEhud Katz Value *Callee = Call.getCalledOperand(); 2127825debe8SEhud Katz Function *F = dyn_cast_or_null<Function>(Callee); 2128825debe8SEhud Katz bool IsIndirectCall = !F; 2129825debe8SEhud Katz if (IsIndirectCall) { 2130825debe8SEhud Katz // Check if this happens to be an indirect function call to a known function 2131825debe8SEhud Katz // in this inline context. If not, we've done all we can. 2132825debe8SEhud Katz F = dyn_cast_or_null<Function>(SimplifiedValues.lookup(Callee)); 2133825debe8SEhud Katz if (!F) { 213406408758SMircea Trofin onCallArgumentSetup(Call); 2135825debe8SEhud Katz 2136825debe8SEhud Katz if (!Call.onlyReadsMemory()) 2137825debe8SEhud Katz disableLoadElimination(); 2138825debe8SEhud Katz return Base::visitCallBase(Call); 2139825debe8SEhud Katz } 2140825debe8SEhud Katz } 2141825debe8SEhud Katz 2142825debe8SEhud Katz assert(F && "Expected a call to a known function"); 2143825debe8SEhud Katz 21447adc3a2bSChandler Carruth // When we have a concrete function, first try to simplify it directly. 2145652168a9SFedor Sergeev if (simplifyCallSite(F, Call)) 21467adc3a2bSChandler Carruth return true; 21477adc3a2bSChandler Carruth 21487adc3a2bSChandler Carruth // Next check if it is an intrinsic we know about. 21497adc3a2bSChandler Carruth // FIXME: Lift this into part of the InstVisitor. 2150652168a9SFedor Sergeev if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(&Call)) { 21517adc3a2bSChandler Carruth switch (II->getIntrinsicID()) { 21527adc3a2bSChandler Carruth default: 2153652168a9SFedor Sergeev if (!Call.onlyReadsMemory() && !isAssumeLikeIntrinsic(II)) 2154a4461515SHaicheng Wu disableLoadElimination(); 2155652168a9SFedor Sergeev return Base::visitCallBase(Call); 21567adc3a2bSChandler Carruth 21577dd8dbf4SPeter Collingbourne case Intrinsic::load_relative: 215806408758SMircea Trofin onLoadRelativeIntrinsic(); 21597dd8dbf4SPeter Collingbourne return false; 21607dd8dbf4SPeter Collingbourne 21617adc3a2bSChandler Carruth case Intrinsic::memset: 21627adc3a2bSChandler Carruth case Intrinsic::memcpy: 21637adc3a2bSChandler Carruth case Intrinsic::memmove: 2164a4461515SHaicheng Wu disableLoadElimination(); 21657adc3a2bSChandler Carruth // SROA can usually chew through these intrinsics, but they aren't free. 21667adc3a2bSChandler Carruth return false; 21674296ea72SVitaly Buka case Intrinsic::icall_branch_funnel: 21687adc3a2bSChandler Carruth case Intrinsic::localescape: 21694296ea72SVitaly Buka HasUninlineableIntrinsic = true; 21707adc3a2bSChandler Carruth return false; 217180788d80SFlorian Hahn case Intrinsic::vastart: 217277beee41SSameer AbuAsal InitsVargArgs = true; 217380788d80SFlorian Hahn return false; 2174269b335bSArthur Eubanks case Intrinsic::launder_invariant_group: 2175269b335bSArthur Eubanks case Intrinsic::strip_invariant_group: 2176269b335bSArthur Eubanks if (auto *SROAArg = getSROAArgForValueOrNull(II->getOperand(0))) 2177269b335bSArthur Eubanks SROAArgValues[II] = SROAArg; 2178269b335bSArthur Eubanks return true; 21799697f935SNick Desaulniers case Intrinsic::is_constant: 21809697f935SNick Desaulniers return simplifyIntrinsicCallIsConstant(Call); 21817adc3a2bSChandler Carruth } 21827adc3a2bSChandler Carruth } 21837adc3a2bSChandler Carruth 2184652168a9SFedor Sergeev if (F == Call.getFunction()) { 21857adc3a2bSChandler Carruth // This flag will fully abort the analysis, so don't bother with anything 21867adc3a2bSChandler Carruth // else. 21877adc3a2bSChandler Carruth IsRecursiveCall = true; 2188f7fff46aSWenlei He if (!AllowRecursiveCall) 21897adc3a2bSChandler Carruth return false; 21907adc3a2bSChandler Carruth } 21917adc3a2bSChandler Carruth 21927adc3a2bSChandler Carruth if (TTI.isLoweredToCall(F)) { 219306408758SMircea Trofin onLoweredCall(F, Call, IsIndirectCall); 2194825debe8SEhud Katz } 21957adc3a2bSChandler Carruth 2196825debe8SEhud Katz if (!(Call.onlyReadsMemory() || (IsIndirectCall && F->onlyReadsMemory()))) 2197a4461515SHaicheng Wu disableLoadElimination(); 2198652168a9SFedor Sergeev return Base::visitCallBase(Call); 21997adc3a2bSChandler Carruth } 22007adc3a2bSChandler Carruth 22017adc3a2bSChandler Carruth bool CallAnalyzer::visitReturnInst(ReturnInst &RI) { 22027adc3a2bSChandler Carruth // At least one return instruction will be free after inlining. 22037adc3a2bSChandler Carruth bool Free = !HasReturn; 22047adc3a2bSChandler Carruth HasReturn = true; 22057adc3a2bSChandler Carruth return Free; 22067adc3a2bSChandler Carruth } 22077adc3a2bSChandler Carruth 22087adc3a2bSChandler Carruth bool CallAnalyzer::visitBranchInst(BranchInst &BI) { 22097adc3a2bSChandler Carruth // We model unconditional branches as essentially free -- they really 22107adc3a2bSChandler Carruth // shouldn't exist at all, but handling them makes the behavior of the 22117adc3a2bSChandler Carruth // inliner more regular and predictable. Interestingly, conditional branches 22127adc3a2bSChandler Carruth // which will fold away are also free. 22137adc3a2bSChandler Carruth return BI.isUnconditional() || isa<ConstantInt>(BI.getCondition()) || 22147adc3a2bSChandler Carruth dyn_cast_or_null<ConstantInt>( 22157adc3a2bSChandler Carruth SimplifiedValues.lookup(BI.getCondition())); 22167adc3a2bSChandler Carruth } 22177adc3a2bSChandler Carruth 22183ec848bcSHaicheng Wu bool CallAnalyzer::visitSelectInst(SelectInst &SI) { 22193ec848bcSHaicheng Wu bool CheckSROA = SI.getType()->isPointerTy(); 22203ec848bcSHaicheng Wu Value *TrueVal = SI.getTrueValue(); 22213ec848bcSHaicheng Wu Value *FalseVal = SI.getFalseValue(); 22223ec848bcSHaicheng Wu 22233ec848bcSHaicheng Wu Constant *TrueC = dyn_cast<Constant>(TrueVal); 22243ec848bcSHaicheng Wu if (!TrueC) 22253ec848bcSHaicheng Wu TrueC = SimplifiedValues.lookup(TrueVal); 22263ec848bcSHaicheng Wu Constant *FalseC = dyn_cast<Constant>(FalseVal); 22273ec848bcSHaicheng Wu if (!FalseC) 22283ec848bcSHaicheng Wu FalseC = SimplifiedValues.lookup(FalseVal); 22293ec848bcSHaicheng Wu Constant *CondC = 22303ec848bcSHaicheng Wu dyn_cast_or_null<Constant>(SimplifiedValues.lookup(SI.getCondition())); 22313ec848bcSHaicheng Wu 22323ec848bcSHaicheng Wu if (!CondC) { 22333ec848bcSHaicheng Wu // Select C, X, X => X 22343ec848bcSHaicheng Wu if (TrueC == FalseC && TrueC) { 22353ec848bcSHaicheng Wu SimplifiedValues[&SI] = TrueC; 22363ec848bcSHaicheng Wu return true; 22373ec848bcSHaicheng Wu } 22383ec848bcSHaicheng Wu 22393ec848bcSHaicheng Wu if (!CheckSROA) 22403ec848bcSHaicheng Wu return Base::visitSelectInst(SI); 22413ec848bcSHaicheng Wu 22423ec848bcSHaicheng Wu std::pair<Value *, APInt> TrueBaseAndOffset = 22433ec848bcSHaicheng Wu ConstantOffsetPtrs.lookup(TrueVal); 22443ec848bcSHaicheng Wu std::pair<Value *, APInt> FalseBaseAndOffset = 22453ec848bcSHaicheng Wu ConstantOffsetPtrs.lookup(FalseVal); 22463ec848bcSHaicheng Wu if (TrueBaseAndOffset == FalseBaseAndOffset && TrueBaseAndOffset.first) { 22473ec848bcSHaicheng Wu ConstantOffsetPtrs[&SI] = TrueBaseAndOffset; 22483ec848bcSHaicheng Wu 224906408758SMircea Trofin if (auto *SROAArg = getSROAArgForValueOrNull(TrueVal)) 22503ec848bcSHaicheng Wu SROAArgValues[&SI] = SROAArg; 22513ec848bcSHaicheng Wu return true; 22523ec848bcSHaicheng Wu } 22533ec848bcSHaicheng Wu 22543ec848bcSHaicheng Wu return Base::visitSelectInst(SI); 22553ec848bcSHaicheng Wu } 22563ec848bcSHaicheng Wu 22573ec848bcSHaicheng Wu // Select condition is a constant. 22587b639f50SJacob Hegna Value *SelectedV = CondC->isAllOnesValue() ? TrueVal 22597b639f50SJacob Hegna : (CondC->isNullValue()) ? FalseVal 22607b639f50SJacob Hegna : nullptr; 22613ec848bcSHaicheng Wu if (!SelectedV) { 22623ec848bcSHaicheng Wu // Condition is a vector constant that is not all 1s or all 0s. If all 22633ec848bcSHaicheng Wu // operands are constants, ConstantExpr::getSelect() can handle the cases 22643ec848bcSHaicheng Wu // such as select vectors. 22653ec848bcSHaicheng Wu if (TrueC && FalseC) { 22663ec848bcSHaicheng Wu if (auto *C = ConstantExpr::getSelect(CondC, TrueC, FalseC)) { 22673ec848bcSHaicheng Wu SimplifiedValues[&SI] = C; 22683ec848bcSHaicheng Wu return true; 22693ec848bcSHaicheng Wu } 22703ec848bcSHaicheng Wu } 22713ec848bcSHaicheng Wu return Base::visitSelectInst(SI); 22723ec848bcSHaicheng Wu } 22733ec848bcSHaicheng Wu 22743ec848bcSHaicheng Wu // Condition is either all 1s or all 0s. SI can be simplified. 22753ec848bcSHaicheng Wu if (Constant *SelectedC = dyn_cast<Constant>(SelectedV)) { 22763ec848bcSHaicheng Wu SimplifiedValues[&SI] = SelectedC; 22773ec848bcSHaicheng Wu return true; 22783ec848bcSHaicheng Wu } 22793ec848bcSHaicheng Wu 22803ec848bcSHaicheng Wu if (!CheckSROA) 22813ec848bcSHaicheng Wu return true; 22823ec848bcSHaicheng Wu 22833ec848bcSHaicheng Wu std::pair<Value *, APInt> BaseAndOffset = 22843ec848bcSHaicheng Wu ConstantOffsetPtrs.lookup(SelectedV); 22853ec848bcSHaicheng Wu if (BaseAndOffset.first) { 22863ec848bcSHaicheng Wu ConstantOffsetPtrs[&SI] = BaseAndOffset; 22873ec848bcSHaicheng Wu 228806408758SMircea Trofin if (auto *SROAArg = getSROAArgForValueOrNull(SelectedV)) 22893ec848bcSHaicheng Wu SROAArgValues[&SI] = SROAArg; 22903ec848bcSHaicheng Wu } 22913ec848bcSHaicheng Wu 22923ec848bcSHaicheng Wu return true; 22933ec848bcSHaicheng Wu } 22943ec848bcSHaicheng Wu 22957adc3a2bSChandler Carruth bool CallAnalyzer::visitSwitchInst(SwitchInst &SI) { 22967adc3a2bSChandler Carruth // We model unconditional switches as free, see the comments on handling 22977adc3a2bSChandler Carruth // branches. 22987adc3a2bSChandler Carruth if (isa<ConstantInt>(SI.getCondition())) 22997adc3a2bSChandler Carruth return true; 23007adc3a2bSChandler Carruth if (Value *V = SimplifiedValues.lookup(SI.getCondition())) 23017adc3a2bSChandler Carruth if (isa<ConstantInt>(V)) 23027adc3a2bSChandler Carruth return true; 23037adc3a2bSChandler Carruth 23047ad02eeeSEric Christopher // Assume the most general case where the switch is lowered into 2305919f9e8dSJun Bum Lim // either a jump table, bit test, or a balanced binary tree consisting of 2306919f9e8dSJun Bum Lim // case clusters without merging adjacent clusters with the same 2307919f9e8dSJun Bum Lim // destination. We do not consider the switches that are lowered with a mix 2308919f9e8dSJun Bum Lim // of jump table/bit test/binary search tree. The cost of the switch is 2309919f9e8dSJun Bum Lim // proportional to the size of the tree or the size of jump table range. 23102960d41eSJun Bum Lim // 23112960d41eSJun Bum Lim // NB: We convert large switches which are just used to initialize large phi 2312472462c4SBjorn Pettersson // nodes to lookup tables instead in simplifycfg, so this shouldn't prevent 23132960d41eSJun Bum Lim // inlining those. It will prevent inlining in cases where the optimization 23142960d41eSJun Bum Lim // does not (yet) fire. 2315919f9e8dSJun Bum Lim 2316919f9e8dSJun Bum Lim unsigned JumpTableSize = 0; 231708e2386dSMircea Trofin BlockFrequencyInfo *BFI = GetBFI ? &(GetBFI(F)) : nullptr; 2318919f9e8dSJun Bum Lim unsigned NumCaseCluster = 23190d987e41SHiroshi Yamauchi TTI.getEstimatedNumberOfCaseClusters(SI, JumpTableSize, PSI, BFI); 2320919f9e8dSJun Bum Lim 232106408758SMircea Trofin onFinalizeSwitch(JumpTableSize, NumCaseCluster); 2322919f9e8dSJun Bum Lim return false; 2323919f9e8dSJun Bum Lim } 2324919f9e8dSJun Bum Lim 23257adc3a2bSChandler Carruth bool CallAnalyzer::visitIndirectBrInst(IndirectBrInst &IBI) { 23267adc3a2bSChandler Carruth // We never want to inline functions that contain an indirectbr. This is 23277adc3a2bSChandler Carruth // incorrect because all the blockaddress's (in static global initializers 23287adc3a2bSChandler Carruth // for example) would be referring to the original function, and this 23297adc3a2bSChandler Carruth // indirect jump would jump from the inlined copy of the function into the 23307adc3a2bSChandler Carruth // original function which is extremely undefined behavior. 23317adc3a2bSChandler Carruth // FIXME: This logic isn't really right; we can safely inline functions with 23327adc3a2bSChandler Carruth // indirectbr's as long as no other function or global references the 23337adc3a2bSChandler Carruth // blockaddress of a block within the current function. 23347adc3a2bSChandler Carruth HasIndirectBr = true; 23357adc3a2bSChandler Carruth return false; 23367adc3a2bSChandler Carruth } 23377adc3a2bSChandler Carruth 23387adc3a2bSChandler Carruth bool CallAnalyzer::visitResumeInst(ResumeInst &RI) { 23397adc3a2bSChandler Carruth // FIXME: It's not clear that a single instruction is an accurate model for 23407adc3a2bSChandler Carruth // the inline cost of a resume instruction. 23417adc3a2bSChandler Carruth return false; 23427adc3a2bSChandler Carruth } 23437adc3a2bSChandler Carruth 23447adc3a2bSChandler Carruth bool CallAnalyzer::visitCleanupReturnInst(CleanupReturnInst &CRI) { 23457adc3a2bSChandler Carruth // FIXME: It's not clear that a single instruction is an accurate model for 23467adc3a2bSChandler Carruth // the inline cost of a cleanupret instruction. 23477adc3a2bSChandler Carruth return false; 23487adc3a2bSChandler Carruth } 23497adc3a2bSChandler Carruth 23507adc3a2bSChandler Carruth bool CallAnalyzer::visitCatchReturnInst(CatchReturnInst &CRI) { 23517adc3a2bSChandler Carruth // FIXME: It's not clear that a single instruction is an accurate model for 23528220bcc5SJoseph Tremoulet // the inline cost of a catchret instruction. 23537adc3a2bSChandler Carruth return false; 23547adc3a2bSChandler Carruth } 23557adc3a2bSChandler Carruth 23567adc3a2bSChandler Carruth bool CallAnalyzer::visitUnreachableInst(UnreachableInst &I) { 23577adc3a2bSChandler Carruth // FIXME: It might be reasonably to discount the cost of instructions leading 23587adc3a2bSChandler Carruth // to unreachable as they have the lowest possible impact on both runtime and 23597adc3a2bSChandler Carruth // code size. 23607adc3a2bSChandler Carruth return true; // No actual code is needed for unreachable. 23617adc3a2bSChandler Carruth } 23627adc3a2bSChandler Carruth 23637adc3a2bSChandler Carruth bool CallAnalyzer::visitInstruction(Instruction &I) { 23647adc3a2bSChandler Carruth // Some instructions are free. All of the free intrinsics can also be 23657adc3a2bSChandler Carruth // handled by SROA, etc. 23663ccbd4f3SSander de Smalen if (TTI.getUserCost(&I, TargetTransformInfo::TCK_SizeAndLatency) == 23673ccbd4f3SSander de Smalen TargetTransformInfo::TCC_Free) 23687adc3a2bSChandler Carruth return true; 23697adc3a2bSChandler Carruth 23707adc3a2bSChandler Carruth // We found something we don't understand or can't handle. Mark any SROA-able 23717adc3a2bSChandler Carruth // values in the operand list as no longer viable. 237291aff1d8SNick Desaulniers for (const Use &Op : I.operands()) 237391aff1d8SNick Desaulniers disableSROA(Op); 23747adc3a2bSChandler Carruth 23757adc3a2bSChandler Carruth return false; 23767adc3a2bSChandler Carruth } 23777adc3a2bSChandler Carruth 23785f8f34e4SAdrian Prantl /// Analyze a basic block for its contribution to the inline cost. 23797adc3a2bSChandler Carruth /// 23807adc3a2bSChandler Carruth /// This method walks the analyzer over every instruction in the given basic 23817adc3a2bSChandler Carruth /// block and accounts for their cost during inlining at this callsite. It 23827adc3a2bSChandler Carruth /// aborts early if the threshold has been exceeded or an impossible to inline 23837adc3a2bSChandler Carruth /// construct has been detected. It returns false if inlining is no longer 23847adc3a2bSChandler Carruth /// viable, and true if inlining remains viable. 2385c0aa4b75SDavid Bolvansky InlineResult 2386c0aa4b75SDavid Bolvansky CallAnalyzer::analyzeBlock(BasicBlock *BB, 23877adc3a2bSChandler Carruth SmallPtrSetImpl<const Value *> &EphValues) { 238891aff1d8SNick Desaulniers for (Instruction &I : *BB) { 23897adc3a2bSChandler Carruth // FIXME: Currently, the number of instructions in a function regardless of 23907adc3a2bSChandler Carruth // our ability to simplify them during inline to constants or dead code, 23917adc3a2bSChandler Carruth // are actually used by the vector bonus heuristic. As long as that's true, 23927adc3a2bSChandler Carruth // we have to special case debug intrinsics here to prevent differences in 23937adc3a2bSChandler Carruth // inlining due to debug symbols. Eventually, the number of unsimplified 23947adc3a2bSChandler Carruth // instructions shouldn't factor into the cost computation, but until then, 23957adc3a2bSChandler Carruth // hack around it here. 2396*098a0d8fSHongtao Yu // Similarly, skip pseudo-probes. 2397*098a0d8fSHongtao Yu if (I.isDebugOrPseudoInst()) 2398f3c44569SHongtao Yu continue; 2399f3c44569SHongtao Yu 24007adc3a2bSChandler Carruth // Skip ephemeral values. 240191aff1d8SNick Desaulniers if (EphValues.count(&I)) 24027adc3a2bSChandler Carruth continue; 24037adc3a2bSChandler Carruth 24047adc3a2bSChandler Carruth ++NumInstructions; 240591aff1d8SNick Desaulniers if (isa<ExtractElementInst>(I) || I.getType()->isVectorTy()) 24067adc3a2bSChandler Carruth ++NumVectorInstructions; 24077adc3a2bSChandler Carruth 24087adc3a2bSChandler Carruth // If the instruction simplified to a constant, there is no cost to this 24097adc3a2bSChandler Carruth // instruction. Visit the instructions using our InstVisitor to account for 24107adc3a2bSChandler Carruth // all of the per-instruction logic. The visit tree returns true if we 24117adc3a2bSChandler Carruth // consumed the instruction in any way, and false if the instruction's base 24127adc3a2bSChandler Carruth // cost should count against inlining. 241391aff1d8SNick Desaulniers onInstructionAnalysisStart(&I); 2414c965fd94SKirill Naumov 241591aff1d8SNick Desaulniers if (Base::visit(&I)) 24167adc3a2bSChandler Carruth ++NumInstructionsSimplified; 24177adc3a2bSChandler Carruth else 241814a16faeSMircea Trofin onMissedSimplification(); 24197adc3a2bSChandler Carruth 242091aff1d8SNick Desaulniers onInstructionAnalysisFinish(&I); 24210812c5beSHaicheng Wu using namespace ore; 24227adc3a2bSChandler Carruth // If the visit this instruction detected an uninlinable pattern, abort. 24235466597fSMircea Trofin InlineResult IR = InlineResult::success(); 2424f7fff46aSWenlei He if (IsRecursiveCall && !AllowRecursiveCall) 24255466597fSMircea Trofin IR = InlineResult::failure("recursive"); 2426c0aa4b75SDavid Bolvansky else if (ExposesReturnsTwice) 24275466597fSMircea Trofin IR = InlineResult::failure("exposes returns twice"); 2428c0aa4b75SDavid Bolvansky else if (HasDynamicAlloca) 24295466597fSMircea Trofin IR = InlineResult::failure("dynamic alloca"); 2430c0aa4b75SDavid Bolvansky else if (HasIndirectBr) 24315466597fSMircea Trofin IR = InlineResult::failure("indirect branch"); 2432c0aa4b75SDavid Bolvansky else if (HasUninlineableIntrinsic) 24335466597fSMircea Trofin IR = InlineResult::failure("uninlinable intrinsic"); 243477beee41SSameer AbuAsal else if (InitsVargArgs) 24355466597fSMircea Trofin IR = InlineResult::failure("varargs"); 24365466597fSMircea Trofin if (!IR.isSuccess()) { 24370812c5beSHaicheng Wu if (ORE) 24389590658fSVivek Pandya ORE->emit([&]() { 24399590658fSVivek Pandya return OptimizationRemarkMissed(DEBUG_TYPE, "NeverInline", 2440652168a9SFedor Sergeev &CandidateCall) 2441c0aa4b75SDavid Bolvansky << NV("Callee", &F) << " has uninlinable pattern (" 24425466597fSMircea Trofin << NV("InlineResult", IR.getFailureReason()) 2443c0aa4b75SDavid Bolvansky << ") and cost is not fully computed"; 24449590658fSVivek Pandya }); 2445c0aa4b75SDavid Bolvansky return IR; 24460812c5beSHaicheng Wu } 24477adc3a2bSChandler Carruth 24487adc3a2bSChandler Carruth // If the caller is a recursive function then we don't want to inline 24497adc3a2bSChandler Carruth // functions which allocate a lot of stack space because it would increase 24507adc3a2bSChandler Carruth // the caller stack usage dramatically. 24517adc3a2bSChandler Carruth if (IsCallerRecursive && 24520812c5beSHaicheng Wu AllocatedSize > InlineConstants::TotalAllocaSizeRecursiveCaller) { 24535466597fSMircea Trofin auto IR = 24545466597fSMircea Trofin InlineResult::failure("recursive and allocates too much stack space"); 24550812c5beSHaicheng Wu if (ORE) 24569590658fSVivek Pandya ORE->emit([&]() { 24579590658fSVivek Pandya return OptimizationRemarkMissed(DEBUG_TYPE, "NeverInline", 2458652168a9SFedor Sergeev &CandidateCall) 24595466597fSMircea Trofin << NV("Callee", &F) << " is " 24605466597fSMircea Trofin << NV("InlineResult", IR.getFailureReason()) 2461c0aa4b75SDavid Bolvansky << ". Cost is not fully computed"; 24629590658fSVivek Pandya }); 2463c0aa4b75SDavid Bolvansky return IR; 24640812c5beSHaicheng Wu } 24657adc3a2bSChandler Carruth 246606408758SMircea Trofin if (shouldStop()) 24675466597fSMircea Trofin return InlineResult::failure( 24685466597fSMircea Trofin "Call site analysis is not favorable to inlining."); 24697adc3a2bSChandler Carruth } 24707adc3a2bSChandler Carruth 24715466597fSMircea Trofin return InlineResult::success(); 24727adc3a2bSChandler Carruth } 24737adc3a2bSChandler Carruth 24745f8f34e4SAdrian Prantl /// Compute the base pointer and cumulative constant offsets for V. 24757adc3a2bSChandler Carruth /// 24767adc3a2bSChandler Carruth /// This strips all constant offsets off of V, leaving it the base pointer, and 24777adc3a2bSChandler Carruth /// accumulates the total constant offset applied in the returned constant. It 24787adc3a2bSChandler Carruth /// returns 0 if V is not a pointer, and returns the constant '0' if there are 24797adc3a2bSChandler Carruth /// no constant offsets applied. 24807adc3a2bSChandler Carruth ConstantInt *CallAnalyzer::stripAndComputeInBoundsConstantOffsets(Value *&V) { 24817adc3a2bSChandler Carruth if (!V->getType()->isPointerTy()) 24827adc3a2bSChandler Carruth return nullptr; 24837adc3a2bSChandler Carruth 248477f32994SBjorn Pettersson unsigned AS = V->getType()->getPointerAddressSpace(); 2485945b7e5aSElena Demikhovsky unsigned IntPtrWidth = DL.getIndexSizeInBits(AS); 2486735f4671SChris Lattner APInt Offset = APInt::getZero(IntPtrWidth); 24877adc3a2bSChandler Carruth 24887adc3a2bSChandler Carruth // Even though we don't look through PHI nodes, we could be called on an 24897adc3a2bSChandler Carruth // instruction in an unreachable block, which may be on a cycle. 24907adc3a2bSChandler Carruth SmallPtrSet<Value *, 4> Visited; 24917adc3a2bSChandler Carruth Visited.insert(V); 24927adc3a2bSChandler Carruth do { 24937adc3a2bSChandler Carruth if (GEPOperator *GEP = dyn_cast<GEPOperator>(V)) { 24947adc3a2bSChandler Carruth if (!GEP->isInBounds() || !accumulateGEPOffset(*GEP, Offset)) 24957adc3a2bSChandler Carruth return nullptr; 24967adc3a2bSChandler Carruth V = GEP->getPointerOperand(); 24977adc3a2bSChandler Carruth } else if (Operator::getOpcode(V) == Instruction::BitCast) { 24987adc3a2bSChandler Carruth V = cast<Operator>(V)->getOperand(0); 24997adc3a2bSChandler Carruth } else if (GlobalAlias *GA = dyn_cast<GlobalAlias>(V)) { 25005ce32728SSanjoy Das if (GA->isInterposable()) 25017adc3a2bSChandler Carruth break; 25027adc3a2bSChandler Carruth V = GA->getAliasee(); 25037adc3a2bSChandler Carruth } else { 25047adc3a2bSChandler Carruth break; 25057adc3a2bSChandler Carruth } 25067adc3a2bSChandler Carruth assert(V->getType()->isPointerTy() && "Unexpected operand type!"); 25077adc3a2bSChandler Carruth } while (Visited.insert(V).second); 25087adc3a2bSChandler Carruth 250997572775SNicola Zaghen Type *IdxPtrTy = DL.getIndexType(V->getType()); 251097572775SNicola Zaghen return cast<ConstantInt>(ConstantInt::get(IdxPtrTy, Offset)); 25117adc3a2bSChandler Carruth } 25127adc3a2bSChandler Carruth 25135f8f34e4SAdrian Prantl /// Find dead blocks due to deleted CFG edges during inlining. 25143739e14aSHaicheng Wu /// 25153739e14aSHaicheng Wu /// If we know the successor of the current block, \p CurrBB, has to be \p 25163739e14aSHaicheng Wu /// NextBB, the other successors of \p CurrBB are dead if these successors have 25173739e14aSHaicheng Wu /// no live incoming CFG edges. If one block is found to be dead, we can 25183739e14aSHaicheng Wu /// continue growing the dead block list by checking the successors of the dead 25193739e14aSHaicheng Wu /// blocks to see if all their incoming edges are dead or not. 25203739e14aSHaicheng Wu void CallAnalyzer::findDeadBlocks(BasicBlock *CurrBB, BasicBlock *NextBB) { 25213739e14aSHaicheng Wu auto IsEdgeDead = [&](BasicBlock *Pred, BasicBlock *Succ) { 252202a2bb2fSHiroshi Inoue // A CFG edge is dead if the predecessor is dead or the predecessor has a 25233739e14aSHaicheng Wu // known successor which is not the one under exam. 25243739e14aSHaicheng Wu return (DeadBlocks.count(Pred) || 25253739e14aSHaicheng Wu (KnownSuccessors[Pred] && KnownSuccessors[Pred] != Succ)); 25263739e14aSHaicheng Wu }; 25273739e14aSHaicheng Wu 25283739e14aSHaicheng Wu auto IsNewlyDead = [&](BasicBlock *BB) { 25293739e14aSHaicheng Wu // If all the edges to a block are dead, the block is also dead. 25303739e14aSHaicheng Wu return (!DeadBlocks.count(BB) && 25313739e14aSHaicheng Wu llvm::all_of(predecessors(BB), 25323739e14aSHaicheng Wu [&](BasicBlock *P) { return IsEdgeDead(P, BB); })); 25333739e14aSHaicheng Wu }; 25343739e14aSHaicheng Wu 25353739e14aSHaicheng Wu for (BasicBlock *Succ : successors(CurrBB)) { 25363739e14aSHaicheng Wu if (Succ == NextBB || !IsNewlyDead(Succ)) 25373739e14aSHaicheng Wu continue; 25383739e14aSHaicheng Wu SmallVector<BasicBlock *, 4> NewDead; 25393739e14aSHaicheng Wu NewDead.push_back(Succ); 25403739e14aSHaicheng Wu while (!NewDead.empty()) { 25413739e14aSHaicheng Wu BasicBlock *Dead = NewDead.pop_back_val(); 25423739e14aSHaicheng Wu if (DeadBlocks.insert(Dead)) 25433739e14aSHaicheng Wu // Continue growing the dead block lists. 25443739e14aSHaicheng Wu for (BasicBlock *S : successors(Dead)) 25453739e14aSHaicheng Wu if (IsNewlyDead(S)) 25463739e14aSHaicheng Wu NewDead.push_back(S); 25473739e14aSHaicheng Wu } 25483739e14aSHaicheng Wu } 25493739e14aSHaicheng Wu } 25503739e14aSHaicheng Wu 25515f8f34e4SAdrian Prantl /// Analyze a call site for potential inlining. 25527adc3a2bSChandler Carruth /// 25537adc3a2bSChandler Carruth /// Returns true if inlining this call is viable, and false if it is not 25547adc3a2bSChandler Carruth /// viable. It computes the cost and adjusts the threshold based on numerous 25557adc3a2bSChandler Carruth /// factors and heuristics. If this method returns false but the computed cost 25567adc3a2bSChandler Carruth /// is below the computed threshold, then inlining was forcibly disabled by 25577adc3a2bSChandler Carruth /// some artifact of the routine. 255893ac81ccSMircea Trofin InlineResult CallAnalyzer::analyze() { 25597adc3a2bSChandler Carruth ++NumCallsAnalyzed; 25607adc3a2bSChandler Carruth 256106408758SMircea Trofin auto Result = onAnalysisStart(); 25625466597fSMircea Trofin if (!Result.isSuccess()) 256306408758SMircea Trofin return Result; 25647adc3a2bSChandler Carruth 25657adc3a2bSChandler Carruth if (F.empty()) 25665466597fSMircea Trofin return InlineResult::success(); 25677adc3a2bSChandler Carruth 256893ac81ccSMircea Trofin Function *Caller = CandidateCall.getFunction(); 25697adc3a2bSChandler Carruth // Check if the caller function is recursive itself. 25707adc3a2bSChandler Carruth for (User *U : Caller->users()) { 2571652168a9SFedor Sergeev CallBase *Call = dyn_cast<CallBase>(U); 2572652168a9SFedor Sergeev if (Call && Call->getFunction() == Caller) { 25737adc3a2bSChandler Carruth IsCallerRecursive = true; 25747adc3a2bSChandler Carruth break; 25757adc3a2bSChandler Carruth } 25767adc3a2bSChandler Carruth } 25777adc3a2bSChandler Carruth 25787adc3a2bSChandler Carruth // Populate our simplified values by mapping from function arguments to call 25797adc3a2bSChandler Carruth // arguments with known important simplifications. 258093ac81ccSMircea Trofin auto CAI = CandidateCall.arg_begin(); 258191aff1d8SNick Desaulniers for (Argument &FAI : F.args()) { 258293ac81ccSMircea Trofin assert(CAI != CandidateCall.arg_end()); 25837adc3a2bSChandler Carruth if (Constant *C = dyn_cast<Constant>(CAI)) 258491aff1d8SNick Desaulniers SimplifiedValues[&FAI] = C; 25857adc3a2bSChandler Carruth 25867adc3a2bSChandler Carruth Value *PtrArg = *CAI; 25877adc3a2bSChandler Carruth if (ConstantInt *C = stripAndComputeInBoundsConstantOffsets(PtrArg)) { 258891aff1d8SNick Desaulniers ConstantOffsetPtrs[&FAI] = std::make_pair(PtrArg, C->getValue()); 25897adc3a2bSChandler Carruth 25907adc3a2bSChandler Carruth // We can SROA any pointer arguments derived from alloca instructions. 259106408758SMircea Trofin if (auto *SROAArg = dyn_cast<AllocaInst>(PtrArg)) { 259291aff1d8SNick Desaulniers SROAArgValues[&FAI] = SROAArg; 259306408758SMircea Trofin onInitializeSROAArg(SROAArg); 25942e42cc7aSMircea Trofin EnabledSROAAllocas.insert(SROAArg); 25957adc3a2bSChandler Carruth } 25967adc3a2bSChandler Carruth } 259791aff1d8SNick Desaulniers ++CAI; 25987adc3a2bSChandler Carruth } 25997adc3a2bSChandler Carruth NumConstantArgs = SimplifiedValues.size(); 26007adc3a2bSChandler Carruth NumConstantOffsetPtrArgs = ConstantOffsetPtrs.size(); 26017adc3a2bSChandler Carruth NumAllocaArgs = SROAArgValues.size(); 26027adc3a2bSChandler Carruth 26037adc3a2bSChandler Carruth // FIXME: If a caller has multiple calls to a callee, we end up recomputing 26047adc3a2bSChandler Carruth // the ephemeral values multiple times (and they're completely determined by 26057adc3a2bSChandler Carruth // the callee, so this is purely duplicate work). 26067adc3a2bSChandler Carruth SmallPtrSet<const Value *, 32> EphValues; 2607aec2fa35SDaniel Jasper CodeMetrics::collectEphemeralValues(&F, &GetAssumptionCache(F), EphValues); 26087adc3a2bSChandler Carruth 26097adc3a2bSChandler Carruth // The worklist of live basic blocks in the callee *after* inlining. We avoid 26107adc3a2bSChandler Carruth // adding basic blocks of the callee which can be proven to be dead for this 26117adc3a2bSChandler Carruth // particular call site in order to get more accurate cost estimates. This 26127adc3a2bSChandler Carruth // requires a somewhat heavyweight iteration pattern: we need to walk the 26137adc3a2bSChandler Carruth // basic blocks in a breadth-first order as we insert live successors. To 26147adc3a2bSChandler Carruth // accomplish this, prioritizing for small iterations because we exit after 26157adc3a2bSChandler Carruth // crossing our threshold, we use a small-size optimized SetVector. 26167adc3a2bSChandler Carruth typedef SetVector<BasicBlock *, SmallVector<BasicBlock *, 16>, 2617567556aaSChad Rosier SmallPtrSet<BasicBlock *, 16>> 2618567556aaSChad Rosier BBSetVector; 26197adc3a2bSChandler Carruth BBSetVector BBWorklist; 26207adc3a2bSChandler Carruth BBWorklist.insert(&F.getEntryBlock()); 262106408758SMircea Trofin 26227adc3a2bSChandler Carruth // Note that we *must not* cache the size, this loop grows the worklist. 26237adc3a2bSChandler Carruth for (unsigned Idx = 0; Idx != BBWorklist.size(); ++Idx) { 262406408758SMircea Trofin if (shouldStop()) 26257adc3a2bSChandler Carruth break; 26267adc3a2bSChandler Carruth 26277adc3a2bSChandler Carruth BasicBlock *BB = BBWorklist[Idx]; 26287adc3a2bSChandler Carruth if (BB->empty()) 26297adc3a2bSChandler Carruth continue; 26307adc3a2bSChandler Carruth 26319895c701SKazu Hirata onBlockStart(BB); 26329895c701SKazu Hirata 2633639b29b1SNick Desaulniers // Disallow inlining a blockaddress with uses other than strictly callbr. 2634639b29b1SNick Desaulniers // A blockaddress only has defined behavior for an indirect branch in the 2635639b29b1SNick Desaulniers // same function, and we do not currently support inlining indirect 2636639b29b1SNick Desaulniers // branches. But, the inliner may not see an indirect branch that ends up 2637639b29b1SNick Desaulniers // being dead code at a particular call site. If the blockaddress escapes 2638639b29b1SNick Desaulniers // the function, e.g., via a global variable, inlining may lead to an 2639639b29b1SNick Desaulniers // invalid cross-function reference. 2640639b29b1SNick Desaulniers // FIXME: pr/39560: continue relaxing this overt restriction. 26416a84cd3bSNick Desaulniers if (BB->hasAddressTaken()) 2642639b29b1SNick Desaulniers for (User *U : BlockAddress::get(&*BB)->users()) 2643639b29b1SNick Desaulniers if (!isa<CallBrInst>(*U)) 26445466597fSMircea Trofin return InlineResult::failure("blockaddress used outside of callbr"); 26457adc3a2bSChandler Carruth 26467adc3a2bSChandler Carruth // Analyze the cost of this block. If we blow through the threshold, this 26477adc3a2bSChandler Carruth // returns false, and we can bail on out. 2648c0aa4b75SDavid Bolvansky InlineResult IR = analyzeBlock(BB, EphValues); 26495466597fSMircea Trofin if (!IR.isSuccess()) 2650c0aa4b75SDavid Bolvansky return IR; 26517adc3a2bSChandler Carruth 2652edb12a83SChandler Carruth Instruction *TI = BB->getTerminator(); 26537adc3a2bSChandler Carruth 26547adc3a2bSChandler Carruth // Add in the live successors by first checking whether we have terminator 26557adc3a2bSChandler Carruth // that may be simplified based on the values simplified by this call. 26567adc3a2bSChandler Carruth if (BranchInst *BI = dyn_cast<BranchInst>(TI)) { 26577adc3a2bSChandler Carruth if (BI->isConditional()) { 26587adc3a2bSChandler Carruth Value *Cond = BI->getCondition(); 2659567556aaSChad Rosier if (ConstantInt *SimpleCond = 2660567556aaSChad Rosier dyn_cast_or_null<ConstantInt>(SimplifiedValues.lookup(Cond))) { 26613739e14aSHaicheng Wu BasicBlock *NextBB = BI->getSuccessor(SimpleCond->isZero() ? 1 : 0); 26623739e14aSHaicheng Wu BBWorklist.insert(NextBB); 26633739e14aSHaicheng Wu KnownSuccessors[BB] = NextBB; 26643739e14aSHaicheng Wu findDeadBlocks(BB, NextBB); 26657adc3a2bSChandler Carruth continue; 26667adc3a2bSChandler Carruth } 26677adc3a2bSChandler Carruth } 26687adc3a2bSChandler Carruth } else if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) { 26697adc3a2bSChandler Carruth Value *Cond = SI->getCondition(); 2670567556aaSChad Rosier if (ConstantInt *SimpleCond = 2671567556aaSChad Rosier dyn_cast_or_null<ConstantInt>(SimplifiedValues.lookup(Cond))) { 26723739e14aSHaicheng Wu BasicBlock *NextBB = SI->findCaseValue(SimpleCond)->getCaseSuccessor(); 26733739e14aSHaicheng Wu BBWorklist.insert(NextBB); 26743739e14aSHaicheng Wu KnownSuccessors[BB] = NextBB; 26753739e14aSHaicheng Wu findDeadBlocks(BB, NextBB); 26767adc3a2bSChandler Carruth continue; 26777adc3a2bSChandler Carruth } 26787adc3a2bSChandler Carruth } 26797adc3a2bSChandler Carruth 26807adc3a2bSChandler Carruth // If we're unable to select a particular successor, just count all of 26817adc3a2bSChandler Carruth // them. 26827adc3a2bSChandler Carruth for (unsigned TIdx = 0, TSize = TI->getNumSuccessors(); TIdx != TSize; 26837adc3a2bSChandler Carruth ++TIdx) 26847adc3a2bSChandler Carruth BBWorklist.insert(TI->getSuccessor(TIdx)); 26857adc3a2bSChandler Carruth 268606408758SMircea Trofin onBlockAnalyzed(BB); 26877adc3a2bSChandler Carruth } 26887adc3a2bSChandler Carruth 268911c8efd4SErik Desjardins bool OnlyOneCallAndLocalLinkage = F.hasLocalLinkage() && F.hasOneLiveUse() && 269093ac81ccSMircea Trofin &F == CandidateCall.getCalledFunction(); 26917adc3a2bSChandler Carruth // If this is a noduplicate call, we can still inline as long as 26927adc3a2bSChandler Carruth // inlining this would cause the removal of the caller (so the instruction 26937adc3a2bSChandler Carruth // is not actually duplicated, just moved). 26947adc3a2bSChandler Carruth if (!OnlyOneCallAndLocalLinkage && ContainsNoDuplicateCall) 26955466597fSMircea Trofin return InlineResult::failure("noduplicate"); 26967adc3a2bSChandler Carruth 269706408758SMircea Trofin return finalizeAnalysis(); 26987adc3a2bSChandler Carruth } 26997adc3a2bSChandler Carruth 2700ca899bf9SKirill Naumov void InlineCostCallAnalyzer::print() { 27017adc3a2bSChandler Carruth #define DEBUG_PRINT_STAT(x) dbgs() << " " #x ": " << x << "\n" 27021022b5ebSKirill Naumov if (PrintInstructionComments) 2703c965fd94SKirill Naumov F.print(dbgs(), &Writer); 27047adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantArgs); 27057adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantOffsetPtrArgs); 27067adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumAllocaArgs); 27077adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantPtrCmps); 27087adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumConstantPtrDiffs); 27097adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumInstructionsSimplified); 27107adc3a2bSChandler Carruth DEBUG_PRINT_STAT(NumInstructions); 27117adc3a2bSChandler Carruth DEBUG_PRINT_STAT(SROACostSavings); 27127adc3a2bSChandler Carruth DEBUG_PRINT_STAT(SROACostSavingsLost); 2713a4461515SHaicheng Wu DEBUG_PRINT_STAT(LoadEliminationCost); 27147adc3a2bSChandler Carruth DEBUG_PRINT_STAT(ContainsNoDuplicateCall); 27157adc3a2bSChandler Carruth DEBUG_PRINT_STAT(Cost); 27167adc3a2bSChandler Carruth DEBUG_PRINT_STAT(Threshold); 27177adc3a2bSChandler Carruth #undef DEBUG_PRINT_STAT 27187adc3a2bSChandler Carruth } 2719ca899bf9SKirill Naumov 2720ca899bf9SKirill Naumov #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 2721ca899bf9SKirill Naumov /// Dump stats about this call's analysis. 27227b639f50SJacob Hegna LLVM_DUMP_METHOD void InlineCostCallAnalyzer::dump() { print(); } 27237adc3a2bSChandler Carruth #endif 27247adc3a2bSChandler Carruth 27255f8f34e4SAdrian Prantl /// Test that there are no attribute conflicts between Caller and Callee 27267adc3a2bSChandler Carruth /// that prevent inlining. 2727f9ca75f1STeresa Johnson static bool functionsHaveCompatibleAttributes( 2728f9ca75f1STeresa Johnson Function *Caller, Function *Callee, TargetTransformInfo &TTI, 2729f9ca75f1STeresa Johnson function_ref<const TargetLibraryInfo &(Function &)> &GetTLI) { 2730f9ca75f1STeresa Johnson // Note that CalleeTLI must be a copy not a reference. The legacy pass manager 2731f9ca75f1STeresa Johnson // caches the most recently created TLI in the TargetLibraryInfoWrapperPass 2732f9ca75f1STeresa Johnson // object, and always returns the same object (which is overwritten on each 2733f9ca75f1STeresa Johnson // GetTLI call). Therefore we copy the first result. 2734f9ca75f1STeresa Johnson auto CalleeTLI = GetTLI(*Callee); 27357adc3a2bSChandler Carruth return TTI.areInlineCompatible(Caller, Callee) && 2736f9ca75f1STeresa Johnson GetTLI(*Caller).areInlineCompatible(CalleeTLI, 2737ed03d948SMehdi Amini InlineCallerSupersetNoBuiltin) && 27381cb242ebSAkira Hatanaka AttributeFuncs::areInlineCompatible(*Caller, *Callee); 27397adc3a2bSChandler Carruth } 27407adc3a2bSChandler Carruth 2741652168a9SFedor Sergeev int llvm::getCallsiteCost(CallBase &Call, const DataLayout &DL) { 2742351d9b01SXinliang David Li int Cost = 0; 2743652168a9SFedor Sergeev for (unsigned I = 0, E = Call.arg_size(); I != E; ++I) { 2744652168a9SFedor Sergeev if (Call.isByValArgument(I)) { 2745351d9b01SXinliang David Li // We approximate the number of loads and stores needed by dividing the 2746351d9b01SXinliang David Li // size of the byval type by the target's pointer size. 2747652168a9SFedor Sergeev PointerType *PTy = cast<PointerType>(Call.getArgOperand(I)->getType()); 274807a9df59SArthur Eubanks unsigned TypeSize = DL.getTypeSizeInBits(Call.getParamByValType(I)); 274977f32994SBjorn Pettersson unsigned AS = PTy->getAddressSpace(); 275077f32994SBjorn Pettersson unsigned PointerSize = DL.getPointerSizeInBits(AS); 2751351d9b01SXinliang David Li // Ceiling division. 2752351d9b01SXinliang David Li unsigned NumStores = (TypeSize + PointerSize - 1) / PointerSize; 2753351d9b01SXinliang David Li 2754351d9b01SXinliang David Li // If it generates more than 8 stores it is likely to be expanded as an 2755351d9b01SXinliang David Li // inline memcpy so we take that as an upper bound. Otherwise we assume 2756351d9b01SXinliang David Li // one load and one store per word copied. 2757351d9b01SXinliang David Li // FIXME: The maxStoresPerMemcpy setting from the target should be used 2758351d9b01SXinliang David Li // here instead of a magic number of 8, but it's not available via 2759351d9b01SXinliang David Li // DataLayout. 2760351d9b01SXinliang David Li NumStores = std::min(NumStores, 8U); 2761351d9b01SXinliang David Li 2762351d9b01SXinliang David Li Cost += 2 * NumStores * InlineConstants::InstrCost; 2763351d9b01SXinliang David Li } else { 2764351d9b01SXinliang David Li // For non-byval arguments subtract off one instruction per call 2765351d9b01SXinliang David Li // argument. 2766351d9b01SXinliang David Li Cost += InlineConstants::InstrCost; 2767351d9b01SXinliang David Li } 2768351d9b01SXinliang David Li } 2769351d9b01SXinliang David Li // The call instruction also disappears after inlining. 277046c03668SPhilipp Krones Cost += InlineConstants::InstrCost + CallPenalty; 2771351d9b01SXinliang David Li return Cost; 2772351d9b01SXinliang David Li } 2773351d9b01SXinliang David Li 2774ab6a6837SSean Silva InlineCost llvm::getInlineCost( 2775652168a9SFedor Sergeev CallBase &Call, const InlineParams &Params, TargetTransformInfo &CalleeTTI, 277608e2386dSMircea Trofin function_ref<AssumptionCache &(Function &)> GetAssumptionCache, 2777f9ca75f1STeresa Johnson function_ref<const TargetLibraryInfo &(Function &)> GetTLI, 277808e2386dSMircea Trofin function_ref<BlockFrequencyInfo &(Function &)> GetBFI, 27790812c5beSHaicheng Wu ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE) { 2780652168a9SFedor Sergeev return getInlineCost(Call, Call.getCalledFunction(), Params, CalleeTTI, 278108e2386dSMircea Trofin GetAssumptionCache, GetTLI, GetBFI, PSI, ORE); 2782b9f7120eSEaswaran Raman } 2783b9f7120eSEaswaran Raman 2784011a07c0SMircea Trofin Optional<int> llvm::getInliningCostEstimate( 27858a4013edSMircea Trofin CallBase &Call, TargetTransformInfo &CalleeTTI, 278608e2386dSMircea Trofin function_ref<AssumptionCache &(Function &)> GetAssumptionCache, 278708e2386dSMircea Trofin function_ref<BlockFrequencyInfo &(Function &)> GetBFI, 27888a4013edSMircea Trofin ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE) { 27898a4013edSMircea Trofin const InlineParams Params = {/* DefaultThreshold*/ 0, 27908a4013edSMircea Trofin /*HintThreshold*/ {}, 27918a4013edSMircea Trofin /*ColdThreshold*/ {}, 27928a4013edSMircea Trofin /*OptSizeThreshold*/ {}, 27938a4013edSMircea Trofin /*OptMinSizeThreshold*/ {}, 27948a4013edSMircea Trofin /*HotCallSiteThreshold*/ {}, 27958a4013edSMircea Trofin /*LocallyHotCallSiteThreshold*/ {}, 27968a4013edSMircea Trofin /*ColdCallSiteThreshold*/ {}, 2797347a599eSKazu Hirata /*ComputeFullInlineCost*/ true, 2798347a599eSKazu Hirata /*EnableDeferral*/ true}; 27998a4013edSMircea Trofin 280008e2386dSMircea Trofin InlineCostCallAnalyzer CA(*Call.getCalledFunction(), Call, Params, CalleeTTI, 280108e2386dSMircea Trofin GetAssumptionCache, GetBFI, PSI, ORE, true, 28028a4013edSMircea Trofin /*IgnoreThreshold*/ true); 28038a4013edSMircea Trofin auto R = CA.analyze(); 28048a4013edSMircea Trofin if (!R.isSuccess()) 28058a4013edSMircea Trofin return None; 28068a4013edSMircea Trofin return CA.getCost(); 28078a4013edSMircea Trofin } 28088a4013edSMircea Trofin 280999f00635SJacob Hegna Optional<InlineCostFeatures> llvm::getInliningCostFeatures( 281099f00635SJacob Hegna CallBase &Call, TargetTransformInfo &CalleeTTI, 281199f00635SJacob Hegna function_ref<AssumptionCache &(Function &)> GetAssumptionCache, 281299f00635SJacob Hegna function_ref<BlockFrequencyInfo &(Function &)> GetBFI, 281399f00635SJacob Hegna ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE) { 281499f00635SJacob Hegna InlineCostFeaturesAnalyzer CFA(CalleeTTI, GetAssumptionCache, GetBFI, PSI, 281599f00635SJacob Hegna ORE, *Call.getCalledFunction(), Call); 281699f00635SJacob Hegna auto R = CFA.analyze(); 281799f00635SJacob Hegna if (!R.isSuccess()) 281899f00635SJacob Hegna return None; 281999f00635SJacob Hegna return CFA.features(); 282099f00635SJacob Hegna } 282199f00635SJacob Hegna 2822201498c6SMircea Trofin Optional<InlineResult> llvm::getAttributeBasedInliningDecision( 2823201498c6SMircea Trofin CallBase &Call, Function *Callee, TargetTransformInfo &CalleeTTI, 2824201498c6SMircea Trofin function_ref<const TargetLibraryInfo &(Function &)> GetTLI) { 2825f4bb2f0dSEaswaran Raman 28267adc3a2bSChandler Carruth // Cannot inline indirect calls. 28277adc3a2bSChandler Carruth if (!Callee) 2828201498c6SMircea Trofin return InlineResult::failure("indirect call"); 28297adc3a2bSChandler Carruth 283031e60b91SXun Li // When callee coroutine function is inlined into caller coroutine function 283131e60b91SXun Li // before coro-split pass, 283231e60b91SXun Li // coro-early pass can not handle this quiet well. 283331e60b91SXun Li // So we won't inline the coroutine function if it have not been unsplited 283431e60b91SXun Li if (Callee->isPresplitCoroutine()) 283531e60b91SXun Li return InlineResult::failure("unsplited coroutine call"); 283631e60b91SXun Li 28373851496eSBjorn Pettersson // Never inline calls with byval arguments that does not have the alloca 28383851496eSBjorn Pettersson // address space. Since byval arguments can be replaced with a copy to an 28393851496eSBjorn Pettersson // alloca, the inlined code would need to be adjusted to handle that the 28403851496eSBjorn Pettersson // argument is in the alloca address space (so it is a little bit complicated 28413851496eSBjorn Pettersson // to solve). 28423851496eSBjorn Pettersson unsigned AllocaAS = Callee->getParent()->getDataLayout().getAllocaAddrSpace(); 2843652168a9SFedor Sergeev for (unsigned I = 0, E = Call.arg_size(); I != E; ++I) 2844652168a9SFedor Sergeev if (Call.isByValArgument(I)) { 2845652168a9SFedor Sergeev PointerType *PTy = cast<PointerType>(Call.getArgOperand(I)->getType()); 28463851496eSBjorn Pettersson if (PTy->getAddressSpace() != AllocaAS) 2847201498c6SMircea Trofin return InlineResult::failure("byval arguments without alloca" 2848c0aa4b75SDavid Bolvansky " address space"); 28493851496eSBjorn Pettersson } 28503851496eSBjorn Pettersson 28517adc3a2bSChandler Carruth // Calls to functions with always-inline attributes should be inlined 28527adc3a2bSChandler Carruth // whenever possible. 2853652168a9SFedor Sergeev if (Call.hasFnAttr(Attribute::AlwaysInline)) { 285415b17d0aSYevgeny Rouban auto IsViable = isInlineViable(*Callee); 28555466597fSMircea Trofin if (IsViable.isSuccess()) 2856201498c6SMircea Trofin return InlineResult::success(); 2857201498c6SMircea Trofin return InlineResult::failure(IsViable.getFailureReason()); 28587adc3a2bSChandler Carruth } 28597adc3a2bSChandler Carruth 28607adc3a2bSChandler Carruth // Never inline functions with conflicting attributes (unless callee has 28617adc3a2bSChandler Carruth // always-inline attribute). 2862652168a9SFedor Sergeev Function *Caller = Call.getCaller(); 2863f9ca75f1STeresa Johnson if (!functionsHaveCompatibleAttributes(Caller, Callee, CalleeTTI, GetTLI)) 2864201498c6SMircea Trofin return InlineResult::failure("conflicting attributes"); 28657adc3a2bSChandler Carruth 28667adc3a2bSChandler Carruth // Don't inline this call if the caller has the optnone attribute. 286785bd3978SEvandro Menezes if (Caller->hasOptNone()) 2868201498c6SMircea Trofin return InlineResult::failure("optnone attribute"); 28697adc3a2bSChandler Carruth 287077eeac3dSManoj Gupta // Don't inline a function that treats null pointer as valid into a caller 287177eeac3dSManoj Gupta // that does not have this attribute. 287277eeac3dSManoj Gupta if (!Caller->nullPointerIsDefined() && Callee->nullPointerIsDefined()) 2873201498c6SMircea Trofin return InlineResult::failure("nullptr definitions incompatible"); 287477eeac3dSManoj Gupta 2875c0aa4b75SDavid Bolvansky // Don't inline functions which can be interposed at link-time. 2876c0aa4b75SDavid Bolvansky if (Callee->isInterposable()) 2877201498c6SMircea Trofin return InlineResult::failure("interposable"); 2878c0aa4b75SDavid Bolvansky 2879c0aa4b75SDavid Bolvansky // Don't inline functions marked noinline. 2880c0aa4b75SDavid Bolvansky if (Callee->hasFnAttribute(Attribute::NoInline)) 2881201498c6SMircea Trofin return InlineResult::failure("noinline function attribute"); 2882c0aa4b75SDavid Bolvansky 2883c0aa4b75SDavid Bolvansky // Don't inline call sites marked noinline. 2884652168a9SFedor Sergeev if (Call.isNoInline()) 2885201498c6SMircea Trofin return InlineResult::failure("noinline call site attribute"); 2886201498c6SMircea Trofin 2887bc044a88SNick Desaulniers // Don't inline functions if one does not have any stack protector attribute 2888bc044a88SNick Desaulniers // but the other does. 2889bc044a88SNick Desaulniers if (Caller->hasStackProtectorFnAttr() && !Callee->hasStackProtectorFnAttr()) 2890bc044a88SNick Desaulniers return InlineResult::failure( 2891bc044a88SNick Desaulniers "stack protected caller but callee requested no stack protector"); 2892bc044a88SNick Desaulniers if (Callee->hasStackProtectorFnAttr() && !Caller->hasStackProtectorFnAttr()) 2893bc044a88SNick Desaulniers return InlineResult::failure( 2894bc044a88SNick Desaulniers "stack protected callee but caller requested no stack protector"); 2895bc044a88SNick Desaulniers 2896201498c6SMircea Trofin return None; 2897201498c6SMircea Trofin } 2898201498c6SMircea Trofin 2899201498c6SMircea Trofin InlineCost llvm::getInlineCost( 2900201498c6SMircea Trofin CallBase &Call, Function *Callee, const InlineParams &Params, 2901201498c6SMircea Trofin TargetTransformInfo &CalleeTTI, 290208e2386dSMircea Trofin function_ref<AssumptionCache &(Function &)> GetAssumptionCache, 2903201498c6SMircea Trofin function_ref<const TargetLibraryInfo &(Function &)> GetTLI, 290408e2386dSMircea Trofin function_ref<BlockFrequencyInfo &(Function &)> GetBFI, 2905201498c6SMircea Trofin ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE) { 2906201498c6SMircea Trofin 2907201498c6SMircea Trofin auto UserDecision = 2908201498c6SMircea Trofin llvm::getAttributeBasedInliningDecision(Call, Callee, CalleeTTI, GetTLI); 2909201498c6SMircea Trofin 2910201498c6SMircea Trofin if (UserDecision.hasValue()) { 2911201498c6SMircea Trofin if (UserDecision->isSuccess()) 2912201498c6SMircea Trofin return llvm::InlineCost::getAlways("always inline attribute"); 2913201498c6SMircea Trofin return llvm::InlineCost::getNever(UserDecision->getFailureReason()); 2914201498c6SMircea Trofin } 29157adc3a2bSChandler Carruth 2916d34e60caSNicola Zaghen LLVM_DEBUG(llvm::dbgs() << " Analyzing call of " << Callee->getName() 2917201498c6SMircea Trofin << "... (caller:" << Call.getCaller()->getName() 2918201498c6SMircea Trofin << ")\n"); 29197adc3a2bSChandler Carruth 292008e2386dSMircea Trofin InlineCostCallAnalyzer CA(*Callee, Call, Params, CalleeTTI, 292108e2386dSMircea Trofin GetAssumptionCache, GetBFI, PSI, ORE); 292293ac81ccSMircea Trofin InlineResult ShouldInline = CA.analyze(); 29237adc3a2bSChandler Carruth 2924d34e60caSNicola Zaghen LLVM_DEBUG(CA.dump()); 29257adc3a2bSChandler Carruth 29266869e6c1SWenlei He // Always make cost benefit based decision explicit. 29276869e6c1SWenlei He // We use always/never here since threshold is not meaningful, 29286869e6c1SWenlei He // as it's not what drives cost-benefit analysis. 29296869e6c1SWenlei He if (CA.wasDecidedByCostBenefit()) { 29306869e6c1SWenlei He if (ShouldInline.isSuccess()) 29314bdfea2cSLiqiang Tao return InlineCost::getAlways("benefit over cost", 29324bdfea2cSLiqiang Tao CA.getCostBenefitPair()); 29336869e6c1SWenlei He else 29344bdfea2cSLiqiang Tao return InlineCost::getNever("cost over benefit", CA.getCostBenefitPair()); 29356869e6c1SWenlei He } 29366869e6c1SWenlei He 29377adc3a2bSChandler Carruth // Check if there was a reason to force inlining or no inlining. 29385466597fSMircea Trofin if (!ShouldInline.isSuccess() && CA.getCost() < CA.getThreshold()) 29395466597fSMircea Trofin return InlineCost::getNever(ShouldInline.getFailureReason()); 29405466597fSMircea Trofin if (ShouldInline.isSuccess() && CA.getCost() >= CA.getThreshold()) 2941c0aa4b75SDavid Bolvansky return InlineCost::getAlways("empty function"); 29427adc3a2bSChandler Carruth 29437adc3a2bSChandler Carruth return llvm::InlineCost::get(CA.getCost(), CA.getThreshold()); 29447adc3a2bSChandler Carruth } 29457adc3a2bSChandler Carruth 294615b17d0aSYevgeny Rouban InlineResult llvm::isInlineViable(Function &F) { 29477adc3a2bSChandler Carruth bool ReturnsTwice = F.hasFnAttribute(Attribute::ReturnsTwice); 294891aff1d8SNick Desaulniers for (BasicBlock &BB : F) { 2949639b29b1SNick Desaulniers // Disallow inlining of functions which contain indirect branches. 295091aff1d8SNick Desaulniers if (isa<IndirectBrInst>(BB.getTerminator())) 29515466597fSMircea Trofin return InlineResult::failure("contains indirect branches"); 295215b17d0aSYevgeny Rouban 2953639b29b1SNick Desaulniers // Disallow inlining of blockaddresses which are used by non-callbr 2954639b29b1SNick Desaulniers // instructions. 295591aff1d8SNick Desaulniers if (BB.hasAddressTaken()) 295691aff1d8SNick Desaulniers for (User *U : BlockAddress::get(&BB)->users()) 2957639b29b1SNick Desaulniers if (!isa<CallBrInst>(*U)) 29585466597fSMircea Trofin return InlineResult::failure("blockaddress used outside of callbr"); 29597adc3a2bSChandler Carruth 296091aff1d8SNick Desaulniers for (auto &II : BB) { 2961652168a9SFedor Sergeev CallBase *Call = dyn_cast<CallBase>(&II); 2962652168a9SFedor Sergeev if (!Call) 29637adc3a2bSChandler Carruth continue; 29647adc3a2bSChandler Carruth 29657adc3a2bSChandler Carruth // Disallow recursive calls. 2966bc044a88SNick Desaulniers Function *Callee = Call->getCalledFunction(); 2967bc044a88SNick Desaulniers if (&F == Callee) 29685466597fSMircea Trofin return InlineResult::failure("recursive call"); 29697adc3a2bSChandler Carruth 29707adc3a2bSChandler Carruth // Disallow calls which expose returns-twice to a function not previously 29717adc3a2bSChandler Carruth // attributed as such. 2972652168a9SFedor Sergeev if (!ReturnsTwice && isa<CallInst>(Call) && 2973652168a9SFedor Sergeev cast<CallInst>(Call)->canReturnTwice()) 29745466597fSMircea Trofin return InlineResult::failure("exposes returns-twice attribute"); 29757adc3a2bSChandler Carruth 2976bc044a88SNick Desaulniers if (Callee) 2977bc044a88SNick Desaulniers switch (Callee->getIntrinsicID()) { 29781636651eSFlorian Hahn default: 29791636651eSFlorian Hahn break; 298006408758SMircea Trofin case llvm::Intrinsic::icall_branch_funnel: 29814296ea72SVitaly Buka // Disallow inlining of @llvm.icall.branch.funnel because current 29824296ea72SVitaly Buka // backend can't separate call targets from call arguments. 29835466597fSMircea Trofin return InlineResult::failure( 29845466597fSMircea Trofin "disallowed inlining of @llvm.icall.branch.funnel"); 298506408758SMircea Trofin case llvm::Intrinsic::localescape: 29867adc3a2bSChandler Carruth // Disallow inlining functions that call @llvm.localescape. Doing this 29877adc3a2bSChandler Carruth // correctly would require major changes to the inliner. 29885466597fSMircea Trofin return InlineResult::failure( 29895466597fSMircea Trofin "disallowed inlining of @llvm.localescape"); 29901636651eSFlorian Hahn case llvm::Intrinsic::vastart: 299106408758SMircea Trofin // Disallow inlining of functions that initialize VarArgs with 299206408758SMircea Trofin // va_start. 29935466597fSMircea Trofin return InlineResult::failure( 29945466597fSMircea Trofin "contains VarArgs initialized with va_start"); 29957adc3a2bSChandler Carruth } 29967adc3a2bSChandler Carruth } 29971636651eSFlorian Hahn } 29987adc3a2bSChandler Carruth 29995466597fSMircea Trofin return InlineResult::success(); 30007adc3a2bSChandler Carruth } 30011c57cc2bSEaswaran Raman 30021c57cc2bSEaswaran Raman // APIs to create InlineParams based on command line flags and/or other 30031c57cc2bSEaswaran Raman // parameters. 30041c57cc2bSEaswaran Raman 30051c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams(int Threshold) { 30061c57cc2bSEaswaran Raman InlineParams Params; 30071c57cc2bSEaswaran Raman 30081c57cc2bSEaswaran Raman // This field is the threshold to use for a callee by default. This is 30091c57cc2bSEaswaran Raman // derived from one or more of: 30101c57cc2bSEaswaran Raman // * optimization or size-optimization levels, 30111c57cc2bSEaswaran Raman // * a value passed to createFunctionInliningPass function, or 30121c57cc2bSEaswaran Raman // * the -inline-threshold flag. 30131c57cc2bSEaswaran Raman // If the -inline-threshold flag is explicitly specified, that is used 30141c57cc2bSEaswaran Raman // irrespective of anything else. 30151c57cc2bSEaswaran Raman if (InlineThreshold.getNumOccurrences() > 0) 30161c57cc2bSEaswaran Raman Params.DefaultThreshold = InlineThreshold; 30171c57cc2bSEaswaran Raman else 30181c57cc2bSEaswaran Raman Params.DefaultThreshold = Threshold; 30191c57cc2bSEaswaran Raman 30201c57cc2bSEaswaran Raman // Set the HintThreshold knob from the -inlinehint-threshold. 30211c57cc2bSEaswaran Raman Params.HintThreshold = HintThreshold; 30221c57cc2bSEaswaran Raman 30231c57cc2bSEaswaran Raman // Set the HotCallSiteThreshold knob from the -hot-callsite-threshold. 30241c57cc2bSEaswaran Raman Params.HotCallSiteThreshold = HotCallSiteThreshold; 30251c57cc2bSEaswaran Raman 3026974d4eeaSEaswaran Raman // If the -locally-hot-callsite-threshold is explicitly specified, use it to 3027974d4eeaSEaswaran Raman // populate LocallyHotCallSiteThreshold. Later, we populate 3028974d4eeaSEaswaran Raman // Params.LocallyHotCallSiteThreshold from -locally-hot-callsite-threshold if 3029974d4eeaSEaswaran Raman // we know that optimization level is O3 (in the getInlineParams variant that 3030974d4eeaSEaswaran Raman // takes the opt and size levels). 3031974d4eeaSEaswaran Raman // FIXME: Remove this check (and make the assignment unconditional) after 3032974d4eeaSEaswaran Raman // addressing size regression issues at O2. 3033974d4eeaSEaswaran Raman if (LocallyHotCallSiteThreshold.getNumOccurrences() > 0) 3034974d4eeaSEaswaran Raman Params.LocallyHotCallSiteThreshold = LocallyHotCallSiteThreshold; 3035974d4eeaSEaswaran Raman 30365a956c85SMircea Trofin // Set the ColdCallSiteThreshold knob from the 30375a956c85SMircea Trofin // -inline-cold-callsite-threshold. 303812585b01SEaswaran Raman Params.ColdCallSiteThreshold = ColdCallSiteThreshold; 303912585b01SEaswaran Raman 30401c57cc2bSEaswaran Raman // Set the OptMinSizeThreshold and OptSizeThreshold params only if the 30411c57cc2bSEaswaran Raman // -inlinehint-threshold commandline option is not explicitly given. If that 30421c57cc2bSEaswaran Raman // option is present, then its value applies even for callees with size and 30431c57cc2bSEaswaran Raman // minsize attributes. 30441c57cc2bSEaswaran Raman // If the -inline-threshold is not specified, set the ColdThreshold from the 30451c57cc2bSEaswaran Raman // -inlinecold-threshold even if it is not explicitly passed. If 30461c57cc2bSEaswaran Raman // -inline-threshold is specified, then -inlinecold-threshold needs to be 30471c57cc2bSEaswaran Raman // explicitly specified to set the ColdThreshold knob 30481c57cc2bSEaswaran Raman if (InlineThreshold.getNumOccurrences() == 0) { 30491c57cc2bSEaswaran Raman Params.OptMinSizeThreshold = InlineConstants::OptMinSizeThreshold; 30501c57cc2bSEaswaran Raman Params.OptSizeThreshold = InlineConstants::OptSizeThreshold; 30511c57cc2bSEaswaran Raman Params.ColdThreshold = ColdThreshold; 30521c57cc2bSEaswaran Raman } else if (ColdThreshold.getNumOccurrences() > 0) { 30531c57cc2bSEaswaran Raman Params.ColdThreshold = ColdThreshold; 30541c57cc2bSEaswaran Raman } 30551c57cc2bSEaswaran Raman return Params; 30561c57cc2bSEaswaran Raman } 30571c57cc2bSEaswaran Raman 30581c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams() { 30597f37a802STeresa Johnson return getInlineParams(DefaultThreshold); 30601c57cc2bSEaswaran Raman } 30611c57cc2bSEaswaran Raman 30621c57cc2bSEaswaran Raman // Compute the default threshold for inlining based on the opt level and the 30631c57cc2bSEaswaran Raman // size opt level. 30641c57cc2bSEaswaran Raman static int computeThresholdFromOptLevels(unsigned OptLevel, 30651c57cc2bSEaswaran Raman unsigned SizeOptLevel) { 30661c57cc2bSEaswaran Raman if (OptLevel > 2) 30671c57cc2bSEaswaran Raman return InlineConstants::OptAggressiveThreshold; 30681c57cc2bSEaswaran Raman if (SizeOptLevel == 1) // -Os 30691c57cc2bSEaswaran Raman return InlineConstants::OptSizeThreshold; 30701c57cc2bSEaswaran Raman if (SizeOptLevel == 2) // -Oz 30711c57cc2bSEaswaran Raman return InlineConstants::OptMinSizeThreshold; 30727f37a802STeresa Johnson return DefaultThreshold; 30731c57cc2bSEaswaran Raman } 30741c57cc2bSEaswaran Raman 30751c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams(unsigned OptLevel, unsigned SizeOptLevel) { 3076974d4eeaSEaswaran Raman auto Params = 3077974d4eeaSEaswaran Raman getInlineParams(computeThresholdFromOptLevels(OptLevel, SizeOptLevel)); 3078974d4eeaSEaswaran Raman // At O3, use the value of -locally-hot-callsite-threshold option to populate 3079974d4eeaSEaswaran Raman // Params.LocallyHotCallSiteThreshold. Below O3, this flag has effect only 3080974d4eeaSEaswaran Raman // when it is specified explicitly. 3081974d4eeaSEaswaran Raman if (OptLevel > 2) 3082974d4eeaSEaswaran Raman Params.LocallyHotCallSiteThreshold = LocallyHotCallSiteThreshold; 3083974d4eeaSEaswaran Raman return Params; 30841c57cc2bSEaswaran Raman } 30856a5d7d49SKirill Naumov 30866a5d7d49SKirill Naumov PreservedAnalyses 30876a5d7d49SKirill Naumov InlineCostAnnotationPrinterPass::run(Function &F, 30886a5d7d49SKirill Naumov FunctionAnalysisManager &FAM) { 30896a5d7d49SKirill Naumov PrintInstructionComments = true; 30907b639f50SJacob Hegna std::function<AssumptionCache &(Function &)> GetAssumptionCache = 30917b639f50SJacob Hegna [&](Function &F) -> AssumptionCache & { 30926a5d7d49SKirill Naumov return FAM.getResult<AssumptionAnalysis>(F); 30936a5d7d49SKirill Naumov }; 30946a5d7d49SKirill Naumov Module *M = F.getParent(); 30956a5d7d49SKirill Naumov ProfileSummaryInfo PSI(*M); 30966a5d7d49SKirill Naumov DataLayout DL(M); 30976a5d7d49SKirill Naumov TargetTransformInfo TTI(DL); 30986a5d7d49SKirill Naumov // FIXME: Redesign the usage of InlineParams to expand the scope of this pass. 30996a5d7d49SKirill Naumov // In the current implementation, the type of InlineParams doesn't matter as 31006a5d7d49SKirill Naumov // the pass serves only for verification of inliner's decisions. 31016a5d7d49SKirill Naumov // We can add a flag which determines InlineParams for this run. Right now, 31026a5d7d49SKirill Naumov // the default InlineParams are used. 31036a5d7d49SKirill Naumov const InlineParams Params = llvm::getInlineParams(); 31046a5d7d49SKirill Naumov for (BasicBlock &BB : F) { 31056a5d7d49SKirill Naumov for (Instruction &I : BB) { 31066a5d7d49SKirill Naumov if (CallInst *CI = dyn_cast<CallInst>(&I)) { 31076a5d7d49SKirill Naumov Function *CalledFunction = CI->getCalledFunction(); 31086a5d7d49SKirill Naumov if (!CalledFunction || CalledFunction->isDeclaration()) 31096a5d7d49SKirill Naumov continue; 31106a5d7d49SKirill Naumov OptimizationRemarkEmitter ORE(CalledFunction); 31116a5d7d49SKirill Naumov InlineCostCallAnalyzer ICCA(*CalledFunction, *CI, Params, TTI, 31126a5d7d49SKirill Naumov GetAssumptionCache, nullptr, &PSI, &ORE); 31136a5d7d49SKirill Naumov ICCA.analyze(); 31146a5d7d49SKirill Naumov OS << " Analyzing call of " << CalledFunction->getName() 31156a5d7d49SKirill Naumov << "... (caller:" << CI->getCaller()->getName() << ")\n"; 31166a5d7d49SKirill Naumov ICCA.print(); 31176a5d7d49SKirill Naumov } 31186a5d7d49SKirill Naumov } 31196a5d7d49SKirill Naumov } 31206a5d7d49SKirill Naumov return PreservedAnalyses::all(); 31216a5d7d49SKirill Naumov } 3122