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 
57*0383253cSKirill Naumov static cl::opt<bool> PrintDebugInstructionDeltas(
58*0383253cSKirill Naumov     "print-instruction-deltas", cl::Hidden, cl::init(false),
59c965fd94SKirill Naumov     cl::desc("Prints deltas of cost and threshold per instruction"));
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 
74f4bb2f0dSEaswaran Raman // We introduce this threshold to help performance of instrumentation based
75f4bb2f0dSEaswaran Raman // PGO before we actually hook up inliner with analysis passes such as BPI and
76f4bb2f0dSEaswaran Raman // BFI.
77f4bb2f0dSEaswaran Raman static cl::opt<int> ColdThreshold(
78fd3e7a93SDavid Callahan     "inlinecold-threshold", cl::Hidden, cl::init(45), cl::ZeroOrMore,
79f4bb2f0dSEaswaran Raman     cl::desc("Threshold for inlining functions with cold attribute"));
80f4bb2f0dSEaswaran Raman 
81de39cb93SDehao Chen static cl::opt<int>
82de39cb93SDehao Chen     HotCallSiteThreshold("hot-callsite-threshold", cl::Hidden, cl::init(3000),
83de39cb93SDehao Chen                          cl::ZeroOrMore,
84de39cb93SDehao Chen                          cl::desc("Threshold for hot callsites "));
85de39cb93SDehao Chen 
86974d4eeaSEaswaran Raman static cl::opt<int> LocallyHotCallSiteThreshold(
87974d4eeaSEaswaran Raman     "locally-hot-callsite-threshold", cl::Hidden, cl::init(525), cl::ZeroOrMore,
88974d4eeaSEaswaran Raman     cl::desc("Threshold for locally hot callsites "));
89974d4eeaSEaswaran Raman 
90c5fa6358SEaswaran Raman static cl::opt<int> ColdCallSiteRelFreq(
91c5fa6358SEaswaran Raman     "cold-callsite-rel-freq", cl::Hidden, cl::init(2), cl::ZeroOrMore,
9202a2bb2fSHiroshi Inoue     cl::desc("Maximum block frequency, expressed as a percentage of caller's "
93c5fa6358SEaswaran Raman              "entry frequency, for a callsite to be cold in the absence of "
94c5fa6358SEaswaran Raman              "profile information."));
95c5fa6358SEaswaran Raman 
96974d4eeaSEaswaran Raman static cl::opt<int> HotCallSiteRelFreq(
97974d4eeaSEaswaran Raman     "hot-callsite-rel-freq", cl::Hidden, cl::init(60), cl::ZeroOrMore,
98ff77cc75SEaswaran Raman     cl::desc("Minimum block frequency, expressed as a multiple of caller's "
99974d4eeaSEaswaran Raman              "entry frequency, for a callsite to be hot in the absence of "
100974d4eeaSEaswaran Raman              "profile information."));
101974d4eeaSEaswaran Raman 
1024924bb00SEaswaran Raman static cl::opt<bool> OptComputeFullInlineCost(
103fd3e7a93SDavid Callahan     "inline-cost-full", cl::Hidden, cl::init(false), cl::ZeroOrMore,
1040812c5beSHaicheng Wu     cl::desc("Compute the full inline cost of a call site even when the cost "
1050812c5beSHaicheng Wu              "exceeds the threshold."));
1060812c5beSHaicheng Wu 
107ed03d948SMehdi Amini static cl::opt<bool> InlineCallerSupersetNoBuiltin(
108ed03d948SMehdi Amini     "inline-caller-superset-nobuiltin", cl::Hidden, cl::init(true),
109ed03d948SMehdi Amini     cl::ZeroOrMore,
110ed03d948SMehdi Amini     cl::desc("Allow inlining when caller has a superset of callee's nobuiltin "
111f9ca75f1STeresa Johnson              "attributes."));
112f9ca75f1STeresa Johnson 
1137adc3a2bSChandler Carruth namespace {
11406408758SMircea Trofin class InlineCostCallAnalyzer;
115c965fd94SKirill Naumov 
116c965fd94SKirill Naumov // This struct is used to store information about inline cost of a
117c965fd94SKirill Naumov // particular instruction
118c965fd94SKirill Naumov struct InstructionCostDetail {
119b4924f01SMircea Trofin   int CostBefore = 0;
120b4924f01SMircea Trofin   int CostAfter = 0;
121b4924f01SMircea Trofin   int ThresholdBefore = 0;
122b4924f01SMircea Trofin   int ThresholdAfter = 0;
123b4924f01SMircea Trofin 
124b4924f01SMircea Trofin   int getThresholdDelta() const { return ThresholdAfter - ThresholdBefore; }
125b4924f01SMircea Trofin 
126b4924f01SMircea Trofin   int getCostDelta() const { return CostAfter - CostBefore; }
127b4924f01SMircea Trofin 
128b4924f01SMircea Trofin   bool hasThresholdChanged() const { return ThresholdAfter != ThresholdBefore; }
129c965fd94SKirill Naumov };
130c965fd94SKirill Naumov 
131c965fd94SKirill Naumov class CostAnnotationWriter : public AssemblyAnnotationWriter {
132c965fd94SKirill Naumov public:
133c965fd94SKirill Naumov   // This DenseMap stores the delta change in cost and threshold after
134c965fd94SKirill Naumov   // accounting for the given instruction.
135c965fd94SKirill Naumov   DenseMap<const Instruction *, InstructionCostDetail> CostThresholdMap;
136c965fd94SKirill Naumov 
137c965fd94SKirill Naumov   virtual void emitInstructionAnnot(const Instruction *I,
138c965fd94SKirill Naumov                                     formatted_raw_ostream &OS);
139c965fd94SKirill Naumov };
140c965fd94SKirill Naumov 
1417adc3a2bSChandler Carruth class CallAnalyzer : public InstVisitor<CallAnalyzer, bool> {
1427adc3a2bSChandler Carruth   typedef InstVisitor<CallAnalyzer, bool> Base;
1437adc3a2bSChandler Carruth   friend class InstVisitor<CallAnalyzer, bool>;
1447adc3a2bSChandler Carruth 
14506408758SMircea Trofin protected:
14606408758SMircea Trofin   virtual ~CallAnalyzer() {}
1477adc3a2bSChandler Carruth   /// The TargetTransformInfo available for this compilation.
1487adc3a2bSChandler Carruth   const TargetTransformInfo &TTI;
1497adc3a2bSChandler Carruth 
150aec2fa35SDaniel Jasper   /// Getter for the cache of @llvm.assume intrinsics.
151aec2fa35SDaniel Jasper   std::function<AssumptionCache &(Function &)> &GetAssumptionCache;
152aec2fa35SDaniel Jasper 
15312585b01SEaswaran Raman   /// Getter for BlockFrequencyInfo
15412585b01SEaswaran Raman   Optional<function_ref<BlockFrequencyInfo &(Function &)>> &GetBFI;
15512585b01SEaswaran Raman 
15671069cf6SEaswaran Raman   /// Profile summary information.
15771069cf6SEaswaran Raman   ProfileSummaryInfo *PSI;
15871069cf6SEaswaran Raman 
159f3d122cdSPiotr Padlewski   /// The called function.
1607adc3a2bSChandler Carruth   Function &F;
1617adc3a2bSChandler Carruth 
16285be8ca8SEric Christopher   // Cache the DataLayout since we use it a lot.
16385be8ca8SEric Christopher   const DataLayout &DL;
16485be8ca8SEric Christopher 
1650812c5beSHaicheng Wu   /// The OptimizationRemarkEmitter available for this compilation.
1660812c5beSHaicheng Wu   OptimizationRemarkEmitter *ORE;
1670812c5beSHaicheng Wu 
168f3d122cdSPiotr Padlewski   /// The candidate callsite being analyzed. Please do not use this to do
169f3d122cdSPiotr Padlewski   /// analysis in the caller function; we want the inline cost query to be
170f3d122cdSPiotr Padlewski   /// easily cacheable. Instead, use the cover function paramHasAttr.
171652168a9SFedor Sergeev   CallBase &CandidateCall;
1727adc3a2bSChandler Carruth 
17306408758SMircea Trofin   /// Extension points for handling callsite features.
17406408758SMircea Trofin   /// Called after a basic block was analyzed.
17506408758SMircea Trofin   virtual void onBlockAnalyzed(const BasicBlock *BB) {}
1761c57cc2bSEaswaran Raman 
177c965fd94SKirill Naumov   /// Called before an instruction was analyzed
178c965fd94SKirill Naumov   virtual void onInstructionAnalysisStart(const Instruction *I) {}
179c965fd94SKirill Naumov 
180c965fd94SKirill Naumov   /// Called after an instruction was analyzed
181c965fd94SKirill Naumov   virtual void onInstructionAnalysisFinish(const Instruction *I) {}
182c965fd94SKirill Naumov 
18306408758SMircea Trofin   /// Called at the end of the analysis of the callsite. Return the outcome of
18406408758SMircea Trofin   /// the analysis, i.e. 'InlineResult(true)' if the inlining may happen, or
18506408758SMircea Trofin   /// the reason it can't.
1865466597fSMircea Trofin   virtual InlineResult finalizeAnalysis() { return InlineResult::success(); }
18706408758SMircea Trofin   /// Called when we're about to start processing a basic block, and every time
18806408758SMircea Trofin   /// we are done processing an instruction. Return true if there is no point in
18906408758SMircea Trofin   /// continuing the analysis (e.g. we've determined already the call site is
19006408758SMircea Trofin   /// too expensive to inline)
19106408758SMircea Trofin   virtual bool shouldStop() { return false; }
192eeae45dcSFedor Sergeev 
19306408758SMircea Trofin   /// Called before the analysis of the callee body starts (with callsite
19406408758SMircea Trofin   /// contexts propagated).  It checks callsite-specific information. Return a
19506408758SMircea Trofin   /// reason analysis can't continue if that's the case, or 'true' if it may
19606408758SMircea Trofin   /// continue.
1975466597fSMircea Trofin   virtual InlineResult onAnalysisStart() { return InlineResult::success(); }
19806408758SMircea Trofin   /// Called if the analysis engine decides SROA cannot be done for the given
19906408758SMircea Trofin   /// alloca.
20006408758SMircea Trofin   virtual void onDisableSROA(AllocaInst *Arg) {}
20106408758SMircea Trofin 
20206408758SMircea Trofin   /// Called the analysis engine determines load elimination won't happen.
20306408758SMircea Trofin   virtual void onDisableLoadElimination() {}
20406408758SMircea Trofin 
20506408758SMircea Trofin   /// Called to account for a call.
20606408758SMircea Trofin   virtual void onCallPenalty() {}
20706408758SMircea Trofin 
20806408758SMircea Trofin   /// Called to account for the expectation the inlining would result in a load
20906408758SMircea Trofin   /// elimination.
21006408758SMircea Trofin   virtual void onLoadEliminationOpportunity() {}
21106408758SMircea Trofin 
21206408758SMircea Trofin   /// Called to account for the cost of argument setup for the Call in the
21306408758SMircea Trofin   /// callee's body (not the callsite currently under analysis).
21406408758SMircea Trofin   virtual void onCallArgumentSetup(const CallBase &Call) {}
21506408758SMircea Trofin 
21606408758SMircea Trofin   /// Called to account for a load relative intrinsic.
21706408758SMircea Trofin   virtual void onLoadRelativeIntrinsic() {}
21806408758SMircea Trofin 
21906408758SMircea Trofin   /// Called to account for a lowered call.
22006408758SMircea Trofin   virtual void onLoweredCall(Function *F, CallBase &Call, bool IsIndirectCall) {
22106408758SMircea Trofin   }
22206408758SMircea Trofin 
22306408758SMircea Trofin   /// Account for a jump table of given size. Return false to stop further
22406408758SMircea Trofin   /// processing the switch instruction
22506408758SMircea Trofin   virtual bool onJumpTable(unsigned JumpTableSize) { return true; }
22606408758SMircea Trofin 
22706408758SMircea Trofin   /// Account for a case cluster of given size. Return false to stop further
22806408758SMircea Trofin   /// processing of the instruction.
22906408758SMircea Trofin   virtual bool onCaseCluster(unsigned NumCaseCluster) { return true; }
23006408758SMircea Trofin 
23106408758SMircea Trofin   /// Called at the end of processing a switch instruction, with the given
23206408758SMircea Trofin   /// number of case clusters.
23306408758SMircea Trofin   virtual void onFinalizeSwitch(unsigned JumpTableSize,
23406408758SMircea Trofin                                 unsigned NumCaseCluster) {}
23506408758SMircea Trofin 
23606408758SMircea Trofin   /// Called to account for any other instruction not specifically accounted
23706408758SMircea Trofin   /// for.
23814a16faeSMircea Trofin   virtual void onMissedSimplification() {}
23906408758SMircea Trofin 
24006408758SMircea Trofin   /// Start accounting potential benefits due to SROA for the given alloca.
24106408758SMircea Trofin   virtual void onInitializeSROAArg(AllocaInst *Arg) {}
24206408758SMircea Trofin 
24306408758SMircea Trofin   /// Account SROA savings for the AllocaInst value.
24406408758SMircea Trofin   virtual void onAggregateSROAUse(AllocaInst *V) {}
24506408758SMircea Trofin 
24606408758SMircea Trofin   bool handleSROA(Value *V, bool DoNotDisable) {
24706408758SMircea Trofin     // Check for SROA candidates in comparisons.
24806408758SMircea Trofin     if (auto *SROAArg = getSROAArgForValueOrNull(V)) {
24906408758SMircea Trofin       if (DoNotDisable) {
25006408758SMircea Trofin         onAggregateSROAUse(SROAArg);
25106408758SMircea Trofin         return true;
25206408758SMircea Trofin       }
25306408758SMircea Trofin       disableSROAForArg(SROAArg);
25406408758SMircea Trofin     }
25506408758SMircea Trofin     return false;
25606408758SMircea Trofin   }
2577adc3a2bSChandler Carruth 
258eeae45dcSFedor Sergeev   bool IsCallerRecursive = false;
259eeae45dcSFedor Sergeev   bool IsRecursiveCall = false;
260eeae45dcSFedor Sergeev   bool ExposesReturnsTwice = false;
261eeae45dcSFedor Sergeev   bool HasDynamicAlloca = false;
262eeae45dcSFedor Sergeev   bool ContainsNoDuplicateCall = false;
263eeae45dcSFedor Sergeev   bool HasReturn = false;
264eeae45dcSFedor Sergeev   bool HasIndirectBr = false;
265eeae45dcSFedor Sergeev   bool HasUninlineableIntrinsic = false;
266eeae45dcSFedor Sergeev   bool InitsVargArgs = false;
2677adc3a2bSChandler Carruth 
2687adc3a2bSChandler Carruth   /// Number of bytes allocated statically by the callee.
269eeae45dcSFedor Sergeev   uint64_t AllocatedSize = 0;
270eeae45dcSFedor Sergeev   unsigned NumInstructions = 0;
271eeae45dcSFedor Sergeev   unsigned NumVectorInstructions = 0;
272eeae45dcSFedor Sergeev 
273f3d122cdSPiotr Padlewski   /// While we walk the potentially-inlined instructions, we build up and
274f3d122cdSPiotr Padlewski   /// maintain a mapping of simplified values specific to this callsite. The
275f3d122cdSPiotr Padlewski   /// idea is to propagate any special information we have about arguments to
276f3d122cdSPiotr Padlewski   /// this call through the inlinable section of the function, and account for
277f3d122cdSPiotr Padlewski   /// likely simplifications post-inlining. The most important aspect we track
278f3d122cdSPiotr Padlewski   /// is CFG altering simplifications -- when we prove a basic block dead, that
279f3d122cdSPiotr Padlewski   /// can cause dramatic shifts in the cost of inlining a function.
2807adc3a2bSChandler Carruth   DenseMap<Value *, Constant *> SimplifiedValues;
2817adc3a2bSChandler Carruth 
282f3d122cdSPiotr Padlewski   /// Keep track of the values which map back (through function arguments) to
283f3d122cdSPiotr Padlewski   /// allocas on the caller stack which could be simplified through SROA.
28406408758SMircea Trofin   DenseMap<Value *, AllocaInst *> SROAArgValues;
285338a6016SMircea Trofin 
28606408758SMircea Trofin   /// Keep track of Allocas for which we believe we may get SROA optimization.
2872e42cc7aSMircea Trofin   DenseSet<AllocaInst *> EnabledSROAAllocas;
2887adc3a2bSChandler Carruth 
289f3d122cdSPiotr Padlewski   /// Keep track of values which map to a pointer base and constant offset.
2907adc3a2bSChandler Carruth   DenseMap<Value *, std::pair<Value *, APInt>> ConstantOffsetPtrs;
2917adc3a2bSChandler Carruth 
2923739e14aSHaicheng Wu   /// Keep track of dead blocks due to the constant arguments.
2933739e14aSHaicheng Wu   SetVector<BasicBlock *> DeadBlocks;
2943739e14aSHaicheng Wu 
2953739e14aSHaicheng Wu   /// The mapping of the blocks to their known unique successors due to the
2963739e14aSHaicheng Wu   /// constant arguments.
2973739e14aSHaicheng Wu   DenseMap<BasicBlock *, BasicBlock *> KnownSuccessors;
2983739e14aSHaicheng Wu 
299a4461515SHaicheng Wu   /// Model the elimination of repeated loads that is expected to happen
300a4461515SHaicheng Wu   /// whenever we simplify away the stores that would otherwise cause them to be
301a4461515SHaicheng Wu   /// loads.
302a4461515SHaicheng Wu   bool EnableLoadElimination;
303a4461515SHaicheng Wu   SmallPtrSet<Value *, 16> LoadAddrSet;
30406408758SMircea Trofin 
30506408758SMircea Trofin   AllocaInst *getSROAArgForValueOrNull(Value *V) const {
30606408758SMircea Trofin     auto It = SROAArgValues.find(V);
3072e42cc7aSMircea Trofin     if (It == SROAArgValues.end() || EnabledSROAAllocas.count(It->second) == 0)
30806408758SMircea Trofin       return nullptr;
30906408758SMircea Trofin     return It->second;
31006408758SMircea Trofin   }
311a4461515SHaicheng Wu 
3127adc3a2bSChandler Carruth   // Custom simplification helper routines.
3137adc3a2bSChandler Carruth   bool isAllocaDerivedArg(Value *V);
31406408758SMircea Trofin   void disableSROAForArg(AllocaInst *SROAArg);
3157adc3a2bSChandler Carruth   void disableSROA(Value *V);
3163739e14aSHaicheng Wu   void findDeadBlocks(BasicBlock *CurrBB, BasicBlock *NextBB);
317a4461515SHaicheng Wu   void disableLoadElimination();
318201b191bSHaicheng Wu   bool isGEPFree(GetElementPtrInst &GEP);
319d3558b5dSEvgeny Astigeevich   bool canFoldInboundsGEP(GetElementPtrInst &I);
3207adc3a2bSChandler Carruth   bool accumulateGEPOffset(GEPOperator &GEP, APInt &Offset);
321652168a9SFedor Sergeev   bool simplifyCallSite(Function *F, CallBase &Call);
322617f6364SEaswaran Raman   template <typename Callable>
323617f6364SEaswaran Raman   bool simplifyInstruction(Instruction &I, Callable Evaluate);
3247adc3a2bSChandler Carruth   ConstantInt *stripAndComputeInBoundsConstantOffsets(Value *&V);
3257adc3a2bSChandler Carruth 
3267adc3a2bSChandler Carruth   /// Return true if the given argument to the function being considered for
3277adc3a2bSChandler Carruth   /// inlining has the given attribute set either at the call site or the
3287adc3a2bSChandler Carruth   /// function declaration.  Primarily used to inspect call site specific
3297adc3a2bSChandler Carruth   /// attributes since these can be more precise than the ones on the callee
3307adc3a2bSChandler Carruth   /// itself.
3317adc3a2bSChandler Carruth   bool paramHasAttr(Argument *A, Attribute::AttrKind Attr);
3327adc3a2bSChandler Carruth 
3337adc3a2bSChandler Carruth   /// Return true if the given value is known non null within the callee if
3347adc3a2bSChandler Carruth   /// inlined through this particular callsite.
3357adc3a2bSChandler Carruth   bool isKnownNonNullInCallee(Value *V);
3367adc3a2bSChandler Carruth 
337652168a9SFedor Sergeev   /// Return true if size growth is allowed when inlining the callee at \p Call.
338652168a9SFedor Sergeev   bool allowSizeGrowth(CallBase &Call);
3399a3fc17aSEaswaran Raman 
3407adc3a2bSChandler Carruth   // Custom analysis routines.
341c0aa4b75SDavid Bolvansky   InlineResult analyzeBlock(BasicBlock *BB,
342c0aa4b75SDavid Bolvansky                             SmallPtrSetImpl<const Value *> &EphValues);
3437adc3a2bSChandler Carruth 
3447adc3a2bSChandler Carruth   // Disable several entry points to the visitor so we don't accidentally use
3457adc3a2bSChandler Carruth   // them by declaring but not defining them here.
346567556aaSChad Rosier   void visit(Module *);
347567556aaSChad Rosier   void visit(Module &);
348567556aaSChad Rosier   void visit(Function *);
349567556aaSChad Rosier   void visit(Function &);
350567556aaSChad Rosier   void visit(BasicBlock *);
351567556aaSChad Rosier   void visit(BasicBlock &);
3527adc3a2bSChandler Carruth 
3537adc3a2bSChandler Carruth   // Provide base case for our instruction visit.
3547adc3a2bSChandler Carruth   bool visitInstruction(Instruction &I);
3557adc3a2bSChandler Carruth 
3567adc3a2bSChandler Carruth   // Our visit overrides.
3577adc3a2bSChandler Carruth   bool visitAlloca(AllocaInst &I);
3587adc3a2bSChandler Carruth   bool visitPHI(PHINode &I);
3597adc3a2bSChandler Carruth   bool visitGetElementPtr(GetElementPtrInst &I);
3607adc3a2bSChandler Carruth   bool visitBitCast(BitCastInst &I);
3617adc3a2bSChandler Carruth   bool visitPtrToInt(PtrToIntInst &I);
3627adc3a2bSChandler Carruth   bool visitIntToPtr(IntToPtrInst &I);
3637adc3a2bSChandler Carruth   bool visitCastInst(CastInst &I);
3647adc3a2bSChandler Carruth   bool visitUnaryInstruction(UnaryInstruction &I);
3657adc3a2bSChandler Carruth   bool visitCmpInst(CmpInst &I);
3667adc3a2bSChandler Carruth   bool visitSub(BinaryOperator &I);
3677adc3a2bSChandler Carruth   bool visitBinaryOperator(BinaryOperator &I);
3686cda33baSCraig Topper   bool visitFNeg(UnaryOperator &I);
3697adc3a2bSChandler Carruth   bool visitLoad(LoadInst &I);
3707adc3a2bSChandler Carruth   bool visitStore(StoreInst &I);
3717adc3a2bSChandler Carruth   bool visitExtractValue(ExtractValueInst &I);
3727adc3a2bSChandler Carruth   bool visitInsertValue(InsertValueInst &I);
373652168a9SFedor Sergeev   bool visitCallBase(CallBase &Call);
3747adc3a2bSChandler Carruth   bool visitReturnInst(ReturnInst &RI);
3757adc3a2bSChandler Carruth   bool visitBranchInst(BranchInst &BI);
3763ec848bcSHaicheng Wu   bool visitSelectInst(SelectInst &SI);
3777adc3a2bSChandler Carruth   bool visitSwitchInst(SwitchInst &SI);
3787adc3a2bSChandler Carruth   bool visitIndirectBrInst(IndirectBrInst &IBI);
3797adc3a2bSChandler Carruth   bool visitResumeInst(ResumeInst &RI);
3807adc3a2bSChandler Carruth   bool visitCleanupReturnInst(CleanupReturnInst &RI);
3817adc3a2bSChandler Carruth   bool visitCatchReturnInst(CatchReturnInst &RI);
3827adc3a2bSChandler Carruth   bool visitUnreachableInst(UnreachableInst &I);
3837adc3a2bSChandler Carruth 
3847adc3a2bSChandler Carruth public:
385ab6a6837SSean Silva   CallAnalyzer(const TargetTransformInfo &TTI,
386aec2fa35SDaniel Jasper                std::function<AssumptionCache &(Function &)> &GetAssumptionCache,
38712585b01SEaswaran Raman                Optional<function_ref<BlockFrequencyInfo &(Function &)>> &GetBFI,
3880812c5beSHaicheng Wu                ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE,
38906408758SMircea Trofin                Function &Callee, CallBase &Call)
39012585b01SEaswaran Raman       : TTI(TTI), GetAssumptionCache(GetAssumptionCache), GetBFI(GetBFI),
3910812c5beSHaicheng Wu         PSI(PSI), F(Callee), DL(F.getParent()->getDataLayout()), ORE(ORE),
39206408758SMircea Trofin         CandidateCall(Call), EnableLoadElimination(true) {}
3937adc3a2bSChandler Carruth 
39493ac81ccSMircea Trofin   InlineResult analyze();
3957adc3a2bSChandler Carruth 
3967adc3a2bSChandler Carruth   // Keep a bunch of stats about the cost savings found so we can print them
3977adc3a2bSChandler Carruth   // out when debugging.
398eeae45dcSFedor Sergeev   unsigned NumConstantArgs = 0;
399eeae45dcSFedor Sergeev   unsigned NumConstantOffsetPtrArgs = 0;
400eeae45dcSFedor Sergeev   unsigned NumAllocaArgs = 0;
401eeae45dcSFedor Sergeev   unsigned NumConstantPtrCmps = 0;
402eeae45dcSFedor Sergeev   unsigned NumConstantPtrDiffs = 0;
403eeae45dcSFedor Sergeev   unsigned NumInstructionsSimplified = 0;
40476aab66dSMircea Trofin 
40576aab66dSMircea Trofin   void dump();
40676aab66dSMircea Trofin };
407338a6016SMircea Trofin 
40806408758SMircea Trofin /// FIXME: if it is necessary to derive from InlineCostCallAnalyzer, note
40906408758SMircea Trofin /// the FIXME in onLoweredCall, when instantiating an InlineCostCallAnalyzer
41006408758SMircea Trofin class InlineCostCallAnalyzer final : public CallAnalyzer {
41106408758SMircea Trofin   const int CostUpperBound = INT_MAX - InlineConstants::InstrCost - 1;
41206408758SMircea Trofin   const bool ComputeFullInlineCost;
41306408758SMircea Trofin   int LoadEliminationCost = 0;
41406408758SMircea Trofin   /// Bonus to be applied when percentage of vector instructions in callee is
41506408758SMircea Trofin   /// high (see more details in updateThreshold).
41606408758SMircea Trofin   int VectorBonus = 0;
41706408758SMircea Trofin   /// Bonus to be applied when the callee has only one reachable basic block.
41806408758SMircea Trofin   int SingleBBBonus = 0;
41906408758SMircea Trofin 
42006408758SMircea Trofin   /// Tunable parameters that control the analysis.
42106408758SMircea Trofin   const InlineParams &Params;
42206408758SMircea Trofin 
42306408758SMircea Trofin   /// Upper bound for the inlining cost. Bonuses are being applied to account
42406408758SMircea Trofin   /// for speculative "expected profit" of the inlining decision.
42506408758SMircea Trofin   int Threshold = 0;
42606408758SMircea Trofin 
42706408758SMircea Trofin   /// Attempt to evaluate indirect calls to boost its inline cost.
42806408758SMircea Trofin   const bool BoostIndirectCalls;
42906408758SMircea Trofin 
4308a4013edSMircea Trofin   /// Ignore the threshold when finalizing analysis.
4318a4013edSMircea Trofin   const bool IgnoreThreshold;
4328a4013edSMircea Trofin 
43306408758SMircea Trofin   /// Inlining cost measured in abstract units, accounts for all the
43406408758SMircea Trofin   /// instructions expected to be executed for a given function invocation.
43506408758SMircea Trofin   /// Instructions that are statically proven to be dead based on call-site
43606408758SMircea Trofin   /// arguments are not counted here.
43706408758SMircea Trofin   int Cost = 0;
43806408758SMircea Trofin 
43906408758SMircea Trofin   bool SingleBB = true;
44006408758SMircea Trofin 
44106408758SMircea Trofin   unsigned SROACostSavings = 0;
44206408758SMircea Trofin   unsigned SROACostSavingsLost = 0;
44306408758SMircea Trofin 
44406408758SMircea Trofin   /// The mapping of caller Alloca values to their accumulated cost savings. If
44506408758SMircea Trofin   /// we have to disable SROA for one of the allocas, this tells us how much
44606408758SMircea Trofin   /// cost must be added.
44706408758SMircea Trofin   DenseMap<AllocaInst *, int> SROAArgCosts;
44806408758SMircea Trofin 
44906408758SMircea Trofin   /// Return true if \p Call is a cold callsite.
45006408758SMircea Trofin   bool isColdCallSite(CallBase &Call, BlockFrequencyInfo *CallerBFI);
45106408758SMircea Trofin 
45206408758SMircea Trofin   /// Update Threshold based on callsite properties such as callee
45306408758SMircea Trofin   /// attributes and callee hotness for PGO builds. The Callee is explicitly
45406408758SMircea Trofin   /// passed to support analyzing indirect calls whose target is inferred by
45506408758SMircea Trofin   /// analysis.
45606408758SMircea Trofin   void updateThreshold(CallBase &Call, Function &Callee);
45706408758SMircea Trofin   /// Return a higher threshold if \p Call is a hot callsite.
45806408758SMircea Trofin   Optional<int> getHotCallSiteThreshold(CallBase &Call,
45906408758SMircea Trofin                                         BlockFrequencyInfo *CallerBFI);
46006408758SMircea Trofin 
46106408758SMircea Trofin   /// Handle a capped 'int' increment for Cost.
46206408758SMircea Trofin   void addCost(int64_t Inc, int64_t UpperBound = INT_MAX) {
46306408758SMircea Trofin     assert(UpperBound > 0 && UpperBound <= INT_MAX && "invalid upper bound");
46406408758SMircea Trofin     Cost = (int)std::min(UpperBound, Cost + Inc);
46506408758SMircea Trofin   }
46606408758SMircea Trofin 
46706408758SMircea Trofin   void onDisableSROA(AllocaInst *Arg) override {
46806408758SMircea Trofin     auto CostIt = SROAArgCosts.find(Arg);
46906408758SMircea Trofin     if (CostIt == SROAArgCosts.end())
47006408758SMircea Trofin       return;
47106408758SMircea Trofin     addCost(CostIt->second);
47206408758SMircea Trofin     SROACostSavings -= CostIt->second;
47306408758SMircea Trofin     SROACostSavingsLost += CostIt->second;
47406408758SMircea Trofin     SROAArgCosts.erase(CostIt);
47506408758SMircea Trofin   }
47606408758SMircea Trofin 
47706408758SMircea Trofin   void onDisableLoadElimination() override {
47806408758SMircea Trofin     addCost(LoadEliminationCost);
47906408758SMircea Trofin     LoadEliminationCost = 0;
48006408758SMircea Trofin   }
48106408758SMircea Trofin   void onCallPenalty() override { addCost(InlineConstants::CallPenalty); }
48206408758SMircea Trofin   void onCallArgumentSetup(const CallBase &Call) override {
48306408758SMircea Trofin     // Pay the price of the argument setup. We account for the average 1
48406408758SMircea Trofin     // instruction per call argument setup here.
48506408758SMircea Trofin     addCost(Call.arg_size() * InlineConstants::InstrCost);
48606408758SMircea Trofin   }
48706408758SMircea Trofin   void onLoadRelativeIntrinsic() override {
48806408758SMircea Trofin     // This is normally lowered to 4 LLVM instructions.
48906408758SMircea Trofin     addCost(3 * InlineConstants::InstrCost);
49006408758SMircea Trofin   }
49106408758SMircea Trofin   void onLoweredCall(Function *F, CallBase &Call,
49206408758SMircea Trofin                      bool IsIndirectCall) override {
49306408758SMircea Trofin     // We account for the average 1 instruction per call argument setup here.
49406408758SMircea Trofin     addCost(Call.arg_size() * InlineConstants::InstrCost);
49506408758SMircea Trofin 
49606408758SMircea Trofin     // If we have a constant that we are calling as a function, we can peer
49706408758SMircea Trofin     // through it and see the function target. This happens not infrequently
49806408758SMircea Trofin     // during devirtualization and so we want to give it a hefty bonus for
49906408758SMircea Trofin     // inlining, but cap that bonus in the event that inlining wouldn't pan out.
50006408758SMircea Trofin     // Pretend to inline the function, with a custom threshold.
50106408758SMircea Trofin     if (IsIndirectCall && BoostIndirectCalls) {
50206408758SMircea Trofin       auto IndirectCallParams = Params;
50306408758SMircea Trofin       IndirectCallParams.DefaultThreshold =
50406408758SMircea Trofin           InlineConstants::IndirectCallThreshold;
50506408758SMircea Trofin       /// FIXME: if InlineCostCallAnalyzer is derived from, this may need
50606408758SMircea Trofin       /// to instantiate the derived class.
50706408758SMircea Trofin       InlineCostCallAnalyzer CA(TTI, GetAssumptionCache, GetBFI, PSI, ORE, *F,
50806408758SMircea Trofin                                 Call, IndirectCallParams, false);
5095466597fSMircea Trofin       if (CA.analyze().isSuccess()) {
51006408758SMircea Trofin         // We were able to inline the indirect call! Subtract the cost from the
51106408758SMircea Trofin         // threshold to get the bonus we want to apply, but don't go below zero.
51206408758SMircea Trofin         Cost -= std::max(0, CA.getThreshold() - CA.getCost());
51306408758SMircea Trofin       }
51406408758SMircea Trofin     } else
51506408758SMircea Trofin       // Otherwise simply add the cost for merely making the call.
51606408758SMircea Trofin       addCost(InlineConstants::CallPenalty);
51706408758SMircea Trofin   }
51806408758SMircea Trofin 
51906408758SMircea Trofin   void onFinalizeSwitch(unsigned JumpTableSize,
52006408758SMircea Trofin                         unsigned NumCaseCluster) override {
52106408758SMircea Trofin     // If suitable for a jump table, consider the cost for the table size and
52206408758SMircea Trofin     // branch to destination.
52306408758SMircea Trofin     // Maximum valid cost increased in this function.
52406408758SMircea Trofin     if (JumpTableSize) {
52506408758SMircea Trofin       int64_t JTCost = (int64_t)JumpTableSize * InlineConstants::InstrCost +
52606408758SMircea Trofin                        4 * InlineConstants::InstrCost;
52706408758SMircea Trofin 
52806408758SMircea Trofin       addCost(JTCost, (int64_t)CostUpperBound);
52906408758SMircea Trofin       return;
53006408758SMircea Trofin     }
53106408758SMircea Trofin     // Considering forming a binary search, we should find the number of nodes
53206408758SMircea Trofin     // which is same as the number of comparisons when lowered. For a given
53306408758SMircea Trofin     // number of clusters, n, we can define a recursive function, f(n), to find
53406408758SMircea Trofin     // the number of nodes in the tree. The recursion is :
53506408758SMircea Trofin     // f(n) = 1 + f(n/2) + f (n - n/2), when n > 3,
53606408758SMircea Trofin     // and f(n) = n, when n <= 3.
53706408758SMircea Trofin     // This will lead a binary tree where the leaf should be either f(2) or f(3)
53806408758SMircea Trofin     // when n > 3.  So, the number of comparisons from leaves should be n, while
53906408758SMircea Trofin     // the number of non-leaf should be :
54006408758SMircea Trofin     //   2^(log2(n) - 1) - 1
54106408758SMircea Trofin     //   = 2^log2(n) * 2^-1 - 1
54206408758SMircea Trofin     //   = n / 2 - 1.
54306408758SMircea Trofin     // Considering comparisons from leaf and non-leaf nodes, we can estimate the
54406408758SMircea Trofin     // number of comparisons in a simple closed form :
54506408758SMircea Trofin     //   n + n / 2 - 1 = n * 3 / 2 - 1
54606408758SMircea Trofin     if (NumCaseCluster <= 3) {
54706408758SMircea Trofin       // Suppose a comparison includes one compare and one conditional branch.
54806408758SMircea Trofin       addCost(NumCaseCluster * 2 * InlineConstants::InstrCost);
54906408758SMircea Trofin       return;
55006408758SMircea Trofin     }
55106408758SMircea Trofin 
55206408758SMircea Trofin     int64_t ExpectedNumberOfCompare = 3 * (int64_t)NumCaseCluster / 2 - 1;
55306408758SMircea Trofin     int64_t SwitchCost =
55406408758SMircea Trofin         ExpectedNumberOfCompare * 2 * InlineConstants::InstrCost;
55506408758SMircea Trofin 
55606408758SMircea Trofin     addCost(SwitchCost, (int64_t)CostUpperBound);
55706408758SMircea Trofin   }
55814a16faeSMircea Trofin   void onMissedSimplification() override {
55906408758SMircea Trofin     addCost(InlineConstants::InstrCost);
56006408758SMircea Trofin   }
56106408758SMircea Trofin 
56206408758SMircea Trofin   void onInitializeSROAArg(AllocaInst *Arg) override {
56306408758SMircea Trofin     assert(Arg != nullptr &&
56406408758SMircea Trofin            "Should not initialize SROA costs for null value.");
56506408758SMircea Trofin     SROAArgCosts[Arg] = 0;
56606408758SMircea Trofin   }
56706408758SMircea Trofin 
56806408758SMircea Trofin   void onAggregateSROAUse(AllocaInst *SROAArg) override {
56906408758SMircea Trofin     auto CostIt = SROAArgCosts.find(SROAArg);
57006408758SMircea Trofin     assert(CostIt != SROAArgCosts.end() &&
57106408758SMircea Trofin            "expected this argument to have a cost");
57206408758SMircea Trofin     CostIt->second += InlineConstants::InstrCost;
57306408758SMircea Trofin     SROACostSavings += InlineConstants::InstrCost;
57406408758SMircea Trofin   }
57506408758SMircea Trofin 
57606408758SMircea Trofin   void onBlockAnalyzed(const BasicBlock *BB) override {
57706408758SMircea Trofin     auto *TI = BB->getTerminator();
57806408758SMircea Trofin     // If we had any successors at this point, than post-inlining is likely to
57906408758SMircea Trofin     // have them as well. Note that we assume any basic blocks which existed
58006408758SMircea Trofin     // due to branches or switches which folded above will also fold after
58106408758SMircea Trofin     // inlining.
58206408758SMircea Trofin     if (SingleBB && TI->getNumSuccessors() > 1) {
58306408758SMircea Trofin       // Take off the bonus we applied to the threshold.
58406408758SMircea Trofin       Threshold -= SingleBBBonus;
58506408758SMircea Trofin       SingleBB = false;
58606408758SMircea Trofin     }
58706408758SMircea Trofin   }
5885466597fSMircea Trofin 
589c965fd94SKirill Naumov   void onInstructionAnalysisStart(const Instruction *I) override {
590c965fd94SKirill Naumov     // This function is called to store the initial cost of inlining before
591c965fd94SKirill Naumov     // the given instruction was assessed.
592c965fd94SKirill Naumov     if (!PrintDebugInstructionDeltas)
593c965fd94SKirill Naumov       return;
594c965fd94SKirill Naumov     Writer.CostThresholdMap[I].CostBefore = Cost;
595c965fd94SKirill Naumov     Writer.CostThresholdMap[I].ThresholdBefore = Threshold;
596c965fd94SKirill Naumov   }
597c965fd94SKirill Naumov 
598c965fd94SKirill Naumov   void onInstructionAnalysisFinish(const Instruction *I) override {
599c965fd94SKirill Naumov     // This function is called to find new values of cost and threshold after
600c965fd94SKirill Naumov     // the instruction has been assessed.
601c965fd94SKirill Naumov     if (!PrintDebugInstructionDeltas)
602c965fd94SKirill Naumov       return;
603c965fd94SKirill Naumov     Writer.CostThresholdMap[I].CostAfter = Cost;
604c965fd94SKirill Naumov     Writer.CostThresholdMap[I].ThresholdAfter = Threshold;
605c965fd94SKirill Naumov   }
606c965fd94SKirill Naumov 
60706408758SMircea Trofin   InlineResult finalizeAnalysis() override {
60806408758SMircea Trofin     // Loops generally act a lot like calls in that they act like barriers to
60906408758SMircea Trofin     // movement, require a certain amount of setup, etc. So when optimising for
61006408758SMircea Trofin     // size, we penalise any call sites that perform loops. We do this after all
61106408758SMircea Trofin     // other costs here, so will likely only be dealing with relatively small
61206408758SMircea Trofin     // functions (and hence DT and LI will hopefully be cheap).
61306408758SMircea Trofin     auto *Caller = CandidateCall.getFunction();
61406408758SMircea Trofin     if (Caller->hasMinSize()) {
61506408758SMircea Trofin       DominatorTree DT(F);
61606408758SMircea Trofin       LoopInfo LI(DT);
61706408758SMircea Trofin       int NumLoops = 0;
61806408758SMircea Trofin       for (Loop *L : LI) {
61906408758SMircea Trofin         // Ignore loops that will not be executed
62006408758SMircea Trofin         if (DeadBlocks.count(L->getHeader()))
62106408758SMircea Trofin           continue;
62206408758SMircea Trofin         NumLoops++;
62306408758SMircea Trofin       }
62406408758SMircea Trofin       addCost(NumLoops * InlineConstants::CallPenalty);
62506408758SMircea Trofin     }
62606408758SMircea Trofin 
62706408758SMircea Trofin     // We applied the maximum possible vector bonus at the beginning. Now,
62806408758SMircea Trofin     // subtract the excess bonus, if any, from the Threshold before
62906408758SMircea Trofin     // comparing against Cost.
63006408758SMircea Trofin     if (NumVectorInstructions <= NumInstructions / 10)
63106408758SMircea Trofin       Threshold -= VectorBonus;
63206408758SMircea Trofin     else if (NumVectorInstructions <= NumInstructions / 2)
63306408758SMircea Trofin       Threshold -= VectorBonus / 2;
63406408758SMircea Trofin 
6358a4013edSMircea Trofin     if (IgnoreThreshold || Cost < std::max(1, Threshold))
6365466597fSMircea Trofin       return InlineResult::success();
6375466597fSMircea Trofin     return InlineResult::failure("Cost over threshold.");
63806408758SMircea Trofin   }
63906408758SMircea Trofin   bool shouldStop() override {
64006408758SMircea Trofin     // Bail out the moment we cross the threshold. This means we'll under-count
64106408758SMircea Trofin     // the cost, but only when undercounting doesn't matter.
6428a4013edSMircea Trofin     return !IgnoreThreshold && Cost >= Threshold && !ComputeFullInlineCost;
64306408758SMircea Trofin   }
64406408758SMircea Trofin 
64506408758SMircea Trofin   void onLoadEliminationOpportunity() override {
64606408758SMircea Trofin     LoadEliminationCost += InlineConstants::InstrCost;
64706408758SMircea Trofin   }
64806408758SMircea Trofin 
64906408758SMircea Trofin   InlineResult onAnalysisStart() override {
65006408758SMircea Trofin     // Perform some tweaks to the cost and threshold based on the direct
65106408758SMircea Trofin     // callsite information.
65206408758SMircea Trofin 
65306408758SMircea Trofin     // We want to more aggressively inline vector-dense kernels, so up the
65406408758SMircea Trofin     // threshold, and we'll lower it if the % of vector instructions gets too
65506408758SMircea Trofin     // low. Note that these bonuses are some what arbitrary and evolved over
65606408758SMircea Trofin     // time by accident as much as because they are principled bonuses.
65706408758SMircea Trofin     //
65806408758SMircea Trofin     // FIXME: It would be nice to remove all such bonuses. At least it would be
65906408758SMircea Trofin     // nice to base the bonus values on something more scientific.
66006408758SMircea Trofin     assert(NumInstructions == 0);
66106408758SMircea Trofin     assert(NumVectorInstructions == 0);
66206408758SMircea Trofin 
66306408758SMircea Trofin     // Update the threshold based on callsite properties
66406408758SMircea Trofin     updateThreshold(CandidateCall, F);
66506408758SMircea Trofin 
66606408758SMircea Trofin     // While Threshold depends on commandline options that can take negative
66706408758SMircea Trofin     // values, we want to enforce the invariant that the computed threshold and
66806408758SMircea Trofin     // bonuses are non-negative.
66906408758SMircea Trofin     assert(Threshold >= 0);
67006408758SMircea Trofin     assert(SingleBBBonus >= 0);
67106408758SMircea Trofin     assert(VectorBonus >= 0);
67206408758SMircea Trofin 
67306408758SMircea Trofin     // Speculatively apply all possible bonuses to Threshold. If cost exceeds
67406408758SMircea Trofin     // this Threshold any time, and cost cannot decrease, we can stop processing
67506408758SMircea Trofin     // the rest of the function body.
67606408758SMircea Trofin     Threshold += (SingleBBBonus + VectorBonus);
67706408758SMircea Trofin 
67806408758SMircea Trofin     // Give out bonuses for the callsite, as the instructions setting them up
67906408758SMircea Trofin     // will be gone after inlining.
68006408758SMircea Trofin     addCost(-getCallsiteCost(this->CandidateCall, DL));
68106408758SMircea Trofin 
68206408758SMircea Trofin     // If this function uses the coldcc calling convention, prefer not to inline
68306408758SMircea Trofin     // it.
68406408758SMircea Trofin     if (F.getCallingConv() == CallingConv::Cold)
68506408758SMircea Trofin       Cost += InlineConstants::ColdccPenalty;
68606408758SMircea Trofin 
68706408758SMircea Trofin     // Check if we're done. This can happen due to bonuses and penalties.
68806408758SMircea Trofin     if (Cost >= Threshold && !ComputeFullInlineCost)
6895466597fSMircea Trofin       return InlineResult::failure("high cost");
69006408758SMircea Trofin 
6915466597fSMircea Trofin     return InlineResult::success();
69206408758SMircea Trofin   }
69306408758SMircea Trofin 
69406408758SMircea Trofin public:
69506408758SMircea Trofin   InlineCostCallAnalyzer(
69606408758SMircea Trofin       const TargetTransformInfo &TTI,
69706408758SMircea Trofin       std::function<AssumptionCache &(Function &)> &GetAssumptionCache,
69806408758SMircea Trofin       Optional<function_ref<BlockFrequencyInfo &(Function &)>> &GetBFI,
69906408758SMircea Trofin       ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE, Function &Callee,
7008a4013edSMircea Trofin       CallBase &Call, const InlineParams &Params, bool BoostIndirect = true,
7018a4013edSMircea Trofin       bool IgnoreThreshold = false)
70206408758SMircea Trofin       : CallAnalyzer(TTI, GetAssumptionCache, GetBFI, PSI, ORE, Callee, Call),
70306408758SMircea Trofin         ComputeFullInlineCost(OptComputeFullInlineCost ||
70406408758SMircea Trofin                               Params.ComputeFullInlineCost || ORE),
70506408758SMircea Trofin         Params(Params), Threshold(Params.DefaultThreshold),
7068a4013edSMircea Trofin         BoostIndirectCalls(BoostIndirect), IgnoreThreshold(IgnoreThreshold) {}
707c965fd94SKirill Naumov 
708c965fd94SKirill Naumov   /// Annotation Writer for cost annotation
709c965fd94SKirill Naumov   CostAnnotationWriter Writer;
710c965fd94SKirill Naumov 
71106408758SMircea Trofin   void dump();
71206408758SMircea Trofin 
71306408758SMircea Trofin   virtual ~InlineCostCallAnalyzer() {}
71406408758SMircea Trofin   int getThreshold() { return Threshold; }
71506408758SMircea Trofin   int getCost() { return Cost; }
71606408758SMircea Trofin };
7177adc3a2bSChandler Carruth } // namespace
7187adc3a2bSChandler Carruth 
7195f8f34e4SAdrian Prantl /// Test whether the given value is an Alloca-derived function argument.
7207adc3a2bSChandler Carruth bool CallAnalyzer::isAllocaDerivedArg(Value *V) {
7217adc3a2bSChandler Carruth   return SROAArgValues.count(V);
7227adc3a2bSChandler Carruth }
7237adc3a2bSChandler Carruth 
72406408758SMircea Trofin void CallAnalyzer::disableSROAForArg(AllocaInst *SROAArg) {
72506408758SMircea Trofin   onDisableSROA(SROAArg);
7262e42cc7aSMircea Trofin   EnabledSROAAllocas.erase(SROAArg);
727338a6016SMircea Trofin   disableLoadElimination();
728338a6016SMircea Trofin }
729c965fd94SKirill Naumov 
730*0383253cSKirill Naumov void CostAnnotationWriter::emitInstructionAnnot(const Instruction *I,
731*0383253cSKirill Naumov                                                 formatted_raw_ostream &OS) {
732c965fd94SKirill Naumov   // The cost of inlining of the given instruction is printed always.
733c965fd94SKirill Naumov   // The threshold delta is printed only when it is non-zero. It happens
734c965fd94SKirill Naumov   // when we decided to give a bonus at a particular instruction.
735*0383253cSKirill Naumov   if (CostThresholdMap.count(I) == 0) {
736*0383253cSKirill Naumov     OS << "; No analysis for the instruction\n";
737*0383253cSKirill Naumov     return;
738*0383253cSKirill Naumov   }
739b4924f01SMircea Trofin   const auto &Record = CostThresholdMap[I];
740b4924f01SMircea Trofin   OS << "; cost before = " << Record.CostBefore
741b4924f01SMircea Trofin      << ", cost after = " << Record.CostAfter
742b4924f01SMircea Trofin      << ", threshold before = " << Record.ThresholdBefore
743b4924f01SMircea Trofin      << ", threshold after = " << Record.ThresholdAfter << ", ";
744b4924f01SMircea Trofin   OS << "cost delta = " << Record.getCostDelta();
745b4924f01SMircea Trofin   if (Record.hasThresholdChanged())
746b4924f01SMircea Trofin     OS << ", threshold delta = " << Record.getThresholdDelta();
747c965fd94SKirill Naumov   OS << "\n";
748c965fd94SKirill Naumov }
749c965fd94SKirill Naumov 
7505f8f34e4SAdrian Prantl /// If 'V' maps to a SROA candidate, disable SROA for it.
7517adc3a2bSChandler Carruth void CallAnalyzer::disableSROA(Value *V) {
75206408758SMircea Trofin   if (auto *SROAArg = getSROAArgForValueOrNull(V)) {
75306408758SMircea Trofin     disableSROAForArg(SROAArg);
754338a6016SMircea Trofin   }
7557adc3a2bSChandler Carruth }
7567adc3a2bSChandler Carruth 
757a4461515SHaicheng Wu void CallAnalyzer::disableLoadElimination() {
758a4461515SHaicheng Wu   if (EnableLoadElimination) {
75906408758SMircea Trofin     onDisableLoadElimination();
760a4461515SHaicheng Wu     EnableLoadElimination = false;
761a4461515SHaicheng Wu   }
762a4461515SHaicheng Wu }
763a4461515SHaicheng Wu 
7645f8f34e4SAdrian Prantl /// Accumulate a constant GEP offset into an APInt if possible.
7657adc3a2bSChandler Carruth ///
7667adc3a2bSChandler Carruth /// Returns false if unable to compute the offset for any reason. Respects any
7677adc3a2bSChandler Carruth /// simplified values known during the analysis of this callsite.
7687adc3a2bSChandler Carruth bool CallAnalyzer::accumulateGEPOffset(GEPOperator &GEP, APInt &Offset) {
769945b7e5aSElena Demikhovsky   unsigned IntPtrWidth = DL.getIndexTypeSizeInBits(GEP.getType());
7707adc3a2bSChandler Carruth   assert(IntPtrWidth == Offset.getBitWidth());
7717adc3a2bSChandler Carruth 
7727adc3a2bSChandler Carruth   for (gep_type_iterator GTI = gep_type_begin(GEP), GTE = gep_type_end(GEP);
7737adc3a2bSChandler Carruth        GTI != GTE; ++GTI) {
7747adc3a2bSChandler Carruth     ConstantInt *OpC = dyn_cast<ConstantInt>(GTI.getOperand());
7757adc3a2bSChandler Carruth     if (!OpC)
7767adc3a2bSChandler Carruth       if (Constant *SimpleOp = SimplifiedValues.lookup(GTI.getOperand()))
7777adc3a2bSChandler Carruth         OpC = dyn_cast<ConstantInt>(SimpleOp);
7787adc3a2bSChandler Carruth     if (!OpC)
7797adc3a2bSChandler Carruth       return false;
780567556aaSChad Rosier     if (OpC->isZero())
781567556aaSChad Rosier       continue;
7827adc3a2bSChandler Carruth 
7837adc3a2bSChandler Carruth     // Handle a struct index, which adds its field offset to the pointer.
784ab85225bSPeter Collingbourne     if (StructType *STy = GTI.getStructTypeOrNull()) {
7857adc3a2bSChandler Carruth       unsigned ElementIdx = OpC->getZExtValue();
7867adc3a2bSChandler Carruth       const StructLayout *SL = DL.getStructLayout(STy);
7877adc3a2bSChandler Carruth       Offset += APInt(IntPtrWidth, SL->getElementOffset(ElementIdx));
7887adc3a2bSChandler Carruth       continue;
7897adc3a2bSChandler Carruth     }
7907adc3a2bSChandler Carruth 
7917adc3a2bSChandler Carruth     APInt TypeSize(IntPtrWidth, DL.getTypeAllocSize(GTI.getIndexedType()));
7927adc3a2bSChandler Carruth     Offset += OpC->getValue().sextOrTrunc(IntPtrWidth) * TypeSize;
7937adc3a2bSChandler Carruth   }
7947adc3a2bSChandler Carruth   return true;
7957adc3a2bSChandler Carruth }
7967adc3a2bSChandler Carruth 
7975f8f34e4SAdrian Prantl /// Use TTI to check whether a GEP is free.
798201b191bSHaicheng Wu ///
799201b191bSHaicheng Wu /// Respects any simplified values known during the analysis of this callsite.
800201b191bSHaicheng Wu bool CallAnalyzer::isGEPFree(GetElementPtrInst &GEP) {
80161c1bd5aSEvgeny Astigeevich   SmallVector<Value *, 4> Operands;
80261c1bd5aSEvgeny Astigeevich   Operands.push_back(GEP.getOperand(0));
803201b191bSHaicheng Wu   for (User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end(); I != E; ++I)
804201b191bSHaicheng Wu     if (Constant *SimpleOp = SimplifiedValues.lookup(*I))
80561c1bd5aSEvgeny Astigeevich       Operands.push_back(SimpleOp);
806201b191bSHaicheng Wu     else
80761c1bd5aSEvgeny Astigeevich       Operands.push_back(*I);
808e9c9329aSSam Parker   return TargetTransformInfo::TCC_Free ==
809*0383253cSKirill Naumov          TTI.getUserCost(&GEP, Operands,
810*0383253cSKirill Naumov                          TargetTransformInfo::TCK_SizeAndLatency);
811201b191bSHaicheng Wu }
812201b191bSHaicheng Wu 
8137adc3a2bSChandler Carruth bool CallAnalyzer::visitAlloca(AllocaInst &I) {
8147adc3a2bSChandler Carruth   // Check whether inlining will turn a dynamic alloca into a static
8150f153424SSanjay Patel   // alloca and handle that case.
8167adc3a2bSChandler Carruth   if (I.isArrayAllocation()) {
8170f153424SSanjay Patel     Constant *Size = SimplifiedValues.lookup(I.getArraySize());
8180f153424SSanjay Patel     if (auto *AllocSize = dyn_cast_or_null<ConstantInt>(Size)) {
8197adc3a2bSChandler Carruth       Type *Ty = I.getAllocatedType();
82022eb80a1SEaswaran Raman       AllocatedSize = SaturatingMultiplyAdd(
821b302561bSGraham Hunter           AllocSize->getLimitedValue(), DL.getTypeAllocSize(Ty).getFixedSize(),
822b302561bSGraham Hunter           AllocatedSize);
8237adc3a2bSChandler Carruth       return Base::visitAlloca(I);
8247adc3a2bSChandler Carruth     }
8257adc3a2bSChandler Carruth   }
8267adc3a2bSChandler Carruth 
8277adc3a2bSChandler Carruth   // Accumulate the allocated size.
8287adc3a2bSChandler Carruth   if (I.isStaticAlloca()) {
8297adc3a2bSChandler Carruth     Type *Ty = I.getAllocatedType();
8305a956c85SMircea Trofin     AllocatedSize =
8315a956c85SMircea Trofin         SaturatingAdd(DL.getTypeAllocSize(Ty).getFixedSize(), AllocatedSize);
8327adc3a2bSChandler Carruth   }
8337adc3a2bSChandler Carruth 
8347adc3a2bSChandler Carruth   // We will happily inline static alloca instructions.
8357adc3a2bSChandler Carruth   if (I.isStaticAlloca())
8367adc3a2bSChandler Carruth     return Base::visitAlloca(I);
8377adc3a2bSChandler Carruth 
8387adc3a2bSChandler Carruth   // FIXME: This is overly conservative. Dynamic allocas are inefficient for
8397adc3a2bSChandler Carruth   // a variety of reasons, and so we would like to not inline them into
8407adc3a2bSChandler Carruth   // functions which don't currently have a dynamic alloca. This simply
8417adc3a2bSChandler Carruth   // disables inlining altogether in the presence of a dynamic alloca.
8427adc3a2bSChandler Carruth   HasDynamicAlloca = true;
8437adc3a2bSChandler Carruth   return false;
8447adc3a2bSChandler Carruth }
8457adc3a2bSChandler Carruth 
8467adc3a2bSChandler Carruth bool CallAnalyzer::visitPHI(PHINode &I) {
8477adc3a2bSChandler Carruth   // FIXME: We need to propagate SROA *disabling* through phi nodes, even
8487adc3a2bSChandler Carruth   // though we don't want to propagate it's bonuses. The idea is to disable
8497adc3a2bSChandler Carruth   // SROA if it *might* be used in an inappropriate manner.
8507adc3a2bSChandler Carruth 
8517adc3a2bSChandler Carruth   // Phi nodes are always zero-cost.
85277f32994SBjorn Pettersson   // FIXME: Pointer sizes may differ between different address spaces, so do we
85377f32994SBjorn Pettersson   // need to use correct address space in the call to getPointerSizeInBits here?
85477f32994SBjorn Pettersson   // Or could we skip the getPointerSizeInBits call completely? As far as I can
85577f32994SBjorn Pettersson   // see the ZeroOffset is used as a dummy value, so we can probably use any
85677f32994SBjorn Pettersson   // bit width for the ZeroOffset?
85777f32994SBjorn Pettersson   APInt ZeroOffset = APInt::getNullValue(DL.getPointerSizeInBits(0));
8583739e14aSHaicheng Wu   bool CheckSROA = I.getType()->isPointerTy();
8593739e14aSHaicheng Wu 
8603739e14aSHaicheng Wu   // Track the constant or pointer with constant offset we've seen so far.
8613739e14aSHaicheng Wu   Constant *FirstC = nullptr;
8623739e14aSHaicheng Wu   std::pair<Value *, APInt> FirstBaseAndOffset = {nullptr, ZeroOffset};
8633739e14aSHaicheng Wu   Value *FirstV = nullptr;
8643739e14aSHaicheng Wu 
8653739e14aSHaicheng Wu   for (unsigned i = 0, e = I.getNumIncomingValues(); i != e; ++i) {
8663739e14aSHaicheng Wu     BasicBlock *Pred = I.getIncomingBlock(i);
8673739e14aSHaicheng Wu     // If the incoming block is dead, skip the incoming block.
8683739e14aSHaicheng Wu     if (DeadBlocks.count(Pred))
8693739e14aSHaicheng Wu       continue;
8703739e14aSHaicheng Wu     // If the parent block of phi is not the known successor of the incoming
8713739e14aSHaicheng Wu     // block, skip the incoming block.
8723739e14aSHaicheng Wu     BasicBlock *KnownSuccessor = KnownSuccessors[Pred];
8733739e14aSHaicheng Wu     if (KnownSuccessor && KnownSuccessor != I.getParent())
8743739e14aSHaicheng Wu       continue;
8753739e14aSHaicheng Wu 
8763739e14aSHaicheng Wu     Value *V = I.getIncomingValue(i);
8773739e14aSHaicheng Wu     // If the incoming value is this phi itself, skip the incoming value.
8783739e14aSHaicheng Wu     if (&I == V)
8793739e14aSHaicheng Wu       continue;
8803739e14aSHaicheng Wu 
8813739e14aSHaicheng Wu     Constant *C = dyn_cast<Constant>(V);
8823739e14aSHaicheng Wu     if (!C)
8833739e14aSHaicheng Wu       C = SimplifiedValues.lookup(V);
8843739e14aSHaicheng Wu 
8853739e14aSHaicheng Wu     std::pair<Value *, APInt> BaseAndOffset = {nullptr, ZeroOffset};
8863739e14aSHaicheng Wu     if (!C && CheckSROA)
8873739e14aSHaicheng Wu       BaseAndOffset = ConstantOffsetPtrs.lookup(V);
8883739e14aSHaicheng Wu 
8893739e14aSHaicheng Wu     if (!C && !BaseAndOffset.first)
8903739e14aSHaicheng Wu       // The incoming value is neither a constant nor a pointer with constant
8913739e14aSHaicheng Wu       // offset, exit early.
8923739e14aSHaicheng Wu       return true;
8933739e14aSHaicheng Wu 
8943739e14aSHaicheng Wu     if (FirstC) {
8953739e14aSHaicheng Wu       if (FirstC == C)
8963739e14aSHaicheng Wu         // If we've seen a constant incoming value before and it is the same
8973739e14aSHaicheng Wu         // constant we see this time, continue checking the next incoming value.
8983739e14aSHaicheng Wu         continue;
8993739e14aSHaicheng Wu       // Otherwise early exit because we either see a different constant or saw
9003739e14aSHaicheng Wu       // a constant before but we have a pointer with constant offset this time.
9013739e14aSHaicheng Wu       return true;
9023739e14aSHaicheng Wu     }
9033739e14aSHaicheng Wu 
9043739e14aSHaicheng Wu     if (FirstV) {
9053739e14aSHaicheng Wu       // The same logic as above, but check pointer with constant offset here.
9063739e14aSHaicheng Wu       if (FirstBaseAndOffset == BaseAndOffset)
9073739e14aSHaicheng Wu         continue;
9083739e14aSHaicheng Wu       return true;
9093739e14aSHaicheng Wu     }
9103739e14aSHaicheng Wu 
9113739e14aSHaicheng Wu     if (C) {
9123739e14aSHaicheng Wu       // This is the 1st time we've seen a constant, record it.
9133739e14aSHaicheng Wu       FirstC = C;
9143739e14aSHaicheng Wu       continue;
9153739e14aSHaicheng Wu     }
9163739e14aSHaicheng Wu 
9173739e14aSHaicheng Wu     // The remaining case is that this is the 1st time we've seen a pointer with
9183739e14aSHaicheng Wu     // constant offset, record it.
9193739e14aSHaicheng Wu     FirstV = V;
9203739e14aSHaicheng Wu     FirstBaseAndOffset = BaseAndOffset;
9213739e14aSHaicheng Wu   }
9223739e14aSHaicheng Wu 
9233739e14aSHaicheng Wu   // Check if we can map phi to a constant.
9243739e14aSHaicheng Wu   if (FirstC) {
9253739e14aSHaicheng Wu     SimplifiedValues[&I] = FirstC;
9263739e14aSHaicheng Wu     return true;
9273739e14aSHaicheng Wu   }
9283739e14aSHaicheng Wu 
9293739e14aSHaicheng Wu   // Check if we can map phi to a pointer with constant offset.
9303739e14aSHaicheng Wu   if (FirstBaseAndOffset.first) {
9313739e14aSHaicheng Wu     ConstantOffsetPtrs[&I] = FirstBaseAndOffset;
9323739e14aSHaicheng Wu 
93306408758SMircea Trofin     if (auto *SROAArg = getSROAArgForValueOrNull(FirstV))
9343739e14aSHaicheng Wu       SROAArgValues[&I] = SROAArg;
9353739e14aSHaicheng Wu   }
9363739e14aSHaicheng Wu 
9377adc3a2bSChandler Carruth   return true;
9387adc3a2bSChandler Carruth }
9397adc3a2bSChandler Carruth 
9405f8f34e4SAdrian Prantl /// Check we can fold GEPs of constant-offset call site argument pointers.
941d3558b5dSEvgeny Astigeevich /// This requires target data and inbounds GEPs.
942d3558b5dSEvgeny Astigeevich ///
943d3558b5dSEvgeny Astigeevich /// \return true if the specified GEP can be folded.
944d3558b5dSEvgeny Astigeevich bool CallAnalyzer::canFoldInboundsGEP(GetElementPtrInst &I) {
945d3558b5dSEvgeny Astigeevich   // Check if we have a base + offset for the pointer.
946d3558b5dSEvgeny Astigeevich   std::pair<Value *, APInt> BaseAndOffset =
947d3558b5dSEvgeny Astigeevich       ConstantOffsetPtrs.lookup(I.getPointerOperand());
948d3558b5dSEvgeny Astigeevich   if (!BaseAndOffset.first)
949d3558b5dSEvgeny Astigeevich     return false;
950d3558b5dSEvgeny Astigeevich 
951d3558b5dSEvgeny Astigeevich   // Check if the offset of this GEP is constant, and if so accumulate it
952d3558b5dSEvgeny Astigeevich   // into Offset.
953d3558b5dSEvgeny Astigeevich   if (!accumulateGEPOffset(cast<GEPOperator>(I), BaseAndOffset.second))
954d3558b5dSEvgeny Astigeevich     return false;
955d3558b5dSEvgeny Astigeevich 
956d3558b5dSEvgeny Astigeevich   // Add the result as a new mapping to Base + Offset.
957d3558b5dSEvgeny Astigeevich   ConstantOffsetPtrs[&I] = BaseAndOffset;
958d3558b5dSEvgeny Astigeevich 
959d3558b5dSEvgeny Astigeevich   return true;
960d3558b5dSEvgeny Astigeevich }
961d3558b5dSEvgeny Astigeevich 
9627adc3a2bSChandler Carruth bool CallAnalyzer::visitGetElementPtr(GetElementPtrInst &I) {
96306408758SMircea Trofin   auto *SROAArg = getSROAArgForValueOrNull(I.getPointerOperand());
9647adc3a2bSChandler Carruth 
965a8b9cdc9SEaswaran Raman   // Lambda to check whether a GEP's indices are all constant.
966a8b9cdc9SEaswaran Raman   auto IsGEPOffsetConstant = [&](GetElementPtrInst &GEP) {
967a8b9cdc9SEaswaran Raman     for (User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end(); I != E; ++I)
968a8b9cdc9SEaswaran Raman       if (!isa<Constant>(*I) && !SimplifiedValues.lookup(*I))
969a8b9cdc9SEaswaran Raman         return false;
970a8b9cdc9SEaswaran Raman     return true;
971a8b9cdc9SEaswaran Raman   };
972a8b9cdc9SEaswaran Raman 
973d3558b5dSEvgeny Astigeevich   if ((I.isInBounds() && canFoldInboundsGEP(I)) || IsGEPOffsetConstant(I)) {
97406408758SMircea Trofin     if (SROAArg)
9757adc3a2bSChandler Carruth       SROAArgValues[&I] = SROAArg;
9767adc3a2bSChandler Carruth 
9777adc3a2bSChandler Carruth     // Constant GEPs are modeled as free.
9787adc3a2bSChandler Carruth     return true;
9797adc3a2bSChandler Carruth   }
9807adc3a2bSChandler Carruth 
9817adc3a2bSChandler Carruth   // Variable GEPs will require math and will disable SROA.
98206408758SMircea Trofin   if (SROAArg)
98306408758SMircea Trofin     disableSROAForArg(SROAArg);
984201b191bSHaicheng Wu   return isGEPFree(I);
9857adc3a2bSChandler Carruth }
9867adc3a2bSChandler Carruth 
987617f6364SEaswaran Raman /// Simplify \p I if its operands are constants and update SimplifiedValues.
988617f6364SEaswaran Raman /// \p Evaluate is a callable specific to instruction type that evaluates the
989617f6364SEaswaran Raman /// instruction when all the operands are constants.
990617f6364SEaswaran Raman template <typename Callable>
991617f6364SEaswaran Raman bool CallAnalyzer::simplifyInstruction(Instruction &I, Callable Evaluate) {
992617f6364SEaswaran Raman   SmallVector<Constant *, 2> COps;
993617f6364SEaswaran Raman   for (Value *Op : I.operands()) {
994617f6364SEaswaran Raman     Constant *COp = dyn_cast<Constant>(Op);
9957adc3a2bSChandler Carruth     if (!COp)
996617f6364SEaswaran Raman       COp = SimplifiedValues.lookup(Op);
997617f6364SEaswaran Raman     if (!COp)
998617f6364SEaswaran Raman       return false;
999617f6364SEaswaran Raman     COps.push_back(COp);
1000617f6364SEaswaran Raman   }
1001617f6364SEaswaran Raman   auto *C = Evaluate(COps);
1002617f6364SEaswaran Raman   if (!C)
1003617f6364SEaswaran Raman     return false;
10047adc3a2bSChandler Carruth   SimplifiedValues[&I] = C;
10057adc3a2bSChandler Carruth   return true;
10067adc3a2bSChandler Carruth }
10077adc3a2bSChandler Carruth 
1008617f6364SEaswaran Raman bool CallAnalyzer::visitBitCast(BitCastInst &I) {
1009617f6364SEaswaran Raman   // Propagate constants through bitcasts.
1010617f6364SEaswaran Raman   if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) {
1011617f6364SEaswaran Raman         return ConstantExpr::getBitCast(COps[0], I.getType());
1012617f6364SEaswaran Raman       }))
1013617f6364SEaswaran Raman     return true;
1014617f6364SEaswaran Raman 
10157adc3a2bSChandler Carruth   // Track base/offsets through casts
1016567556aaSChad Rosier   std::pair<Value *, APInt> BaseAndOffset =
1017567556aaSChad Rosier       ConstantOffsetPtrs.lookup(I.getOperand(0));
10187adc3a2bSChandler Carruth   // Casts don't change the offset, just wrap it up.
10197adc3a2bSChandler Carruth   if (BaseAndOffset.first)
10207adc3a2bSChandler Carruth     ConstantOffsetPtrs[&I] = BaseAndOffset;
10217adc3a2bSChandler Carruth 
10227adc3a2bSChandler Carruth   // Also look for SROA candidates here.
102306408758SMircea Trofin   if (auto *SROAArg = getSROAArgForValueOrNull(I.getOperand(0)))
10247adc3a2bSChandler Carruth     SROAArgValues[&I] = SROAArg;
10257adc3a2bSChandler Carruth 
10267adc3a2bSChandler Carruth   // Bitcasts are always zero cost.
10277adc3a2bSChandler Carruth   return true;
10287adc3a2bSChandler Carruth }
10297adc3a2bSChandler Carruth 
10307adc3a2bSChandler Carruth bool CallAnalyzer::visitPtrToInt(PtrToIntInst &I) {
10317adc3a2bSChandler Carruth   // Propagate constants through ptrtoint.
1032617f6364SEaswaran Raman   if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) {
1033617f6364SEaswaran Raman         return ConstantExpr::getPtrToInt(COps[0], I.getType());
1034617f6364SEaswaran Raman       }))
10357adc3a2bSChandler Carruth     return true;
10367adc3a2bSChandler Carruth 
10377adc3a2bSChandler Carruth   // Track base/offset pairs when converted to a plain integer provided the
10387adc3a2bSChandler Carruth   // integer is large enough to represent the pointer.
10397adc3a2bSChandler Carruth   unsigned IntegerSize = I.getType()->getScalarSizeInBits();
104077f32994SBjorn Pettersson   unsigned AS = I.getOperand(0)->getType()->getPointerAddressSpace();
104177f32994SBjorn Pettersson   if (IntegerSize >= DL.getPointerSizeInBits(AS)) {
1042567556aaSChad Rosier     std::pair<Value *, APInt> BaseAndOffset =
1043567556aaSChad Rosier         ConstantOffsetPtrs.lookup(I.getOperand(0));
10447adc3a2bSChandler Carruth     if (BaseAndOffset.first)
10457adc3a2bSChandler Carruth       ConstantOffsetPtrs[&I] = BaseAndOffset;
10467adc3a2bSChandler Carruth   }
10477adc3a2bSChandler Carruth 
10487adc3a2bSChandler Carruth   // This is really weird. Technically, ptrtoint will disable SROA. However,
10497adc3a2bSChandler Carruth   // unless that ptrtoint is *used* somewhere in the live basic blocks after
10507adc3a2bSChandler Carruth   // inlining, it will be nuked, and SROA should proceed. All of the uses which
10517adc3a2bSChandler Carruth   // would block SROA would also block SROA if applied directly to a pointer,
10527adc3a2bSChandler Carruth   // and so we can just add the integer in here. The only places where SROA is
10537adc3a2bSChandler Carruth   // preserved either cannot fire on an integer, or won't in-and-of themselves
10547adc3a2bSChandler Carruth   // disable SROA (ext) w/o some later use that we would see and disable.
105506408758SMircea Trofin   if (auto *SROAArg = getSROAArgForValueOrNull(I.getOperand(0)))
10567adc3a2bSChandler Carruth     SROAArgValues[&I] = SROAArg;
10577adc3a2bSChandler Carruth 
1058e9c9329aSSam Parker   return TargetTransformInfo::TCC_Free ==
1059e9c9329aSSam Parker          TTI.getUserCost(&I, TargetTransformInfo::TCK_SizeAndLatency);
10607adc3a2bSChandler Carruth }
10617adc3a2bSChandler Carruth 
10627adc3a2bSChandler Carruth bool CallAnalyzer::visitIntToPtr(IntToPtrInst &I) {
10637adc3a2bSChandler Carruth   // Propagate constants through ptrtoint.
1064617f6364SEaswaran Raman   if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) {
1065617f6364SEaswaran Raman         return ConstantExpr::getIntToPtr(COps[0], I.getType());
1066617f6364SEaswaran Raman       }))
10677adc3a2bSChandler Carruth     return true;
10687adc3a2bSChandler Carruth 
10697adc3a2bSChandler Carruth   // Track base/offset pairs when round-tripped through a pointer without
10707adc3a2bSChandler Carruth   // modifications provided the integer is not too large.
10717adc3a2bSChandler Carruth   Value *Op = I.getOperand(0);
10727adc3a2bSChandler Carruth   unsigned IntegerSize = Op->getType()->getScalarSizeInBits();
107377f32994SBjorn Pettersson   if (IntegerSize <= DL.getPointerTypeSizeInBits(I.getType())) {
10747adc3a2bSChandler Carruth     std::pair<Value *, APInt> BaseAndOffset = ConstantOffsetPtrs.lookup(Op);
10757adc3a2bSChandler Carruth     if (BaseAndOffset.first)
10767adc3a2bSChandler Carruth       ConstantOffsetPtrs[&I] = BaseAndOffset;
10777adc3a2bSChandler Carruth   }
10787adc3a2bSChandler Carruth 
10797adc3a2bSChandler Carruth   // "Propagate" SROA here in the same manner as we do for ptrtoint above.
108006408758SMircea Trofin   if (auto *SROAArg = getSROAArgForValueOrNull(Op))
10817adc3a2bSChandler Carruth     SROAArgValues[&I] = SROAArg;
10827adc3a2bSChandler Carruth 
1083e9c9329aSSam Parker   return TargetTransformInfo::TCC_Free ==
1084e9c9329aSSam Parker          TTI.getUserCost(&I, TargetTransformInfo::TCK_SizeAndLatency);
10857adc3a2bSChandler Carruth }
10867adc3a2bSChandler Carruth 
10877adc3a2bSChandler Carruth bool CallAnalyzer::visitCastInst(CastInst &I) {
1088ab53c5e5SCraig Topper   // Propagate constants through casts.
1089617f6364SEaswaran Raman   if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) {
1090617f6364SEaswaran Raman         return ConstantExpr::getCast(I.getOpcode(), COps[0], I.getType());
1091617f6364SEaswaran Raman       }))
10927adc3a2bSChandler Carruth     return true;
10937adc3a2bSChandler Carruth 
10947adc3a2bSChandler Carruth   // Disable SROA in the face of arbitrary casts we don't whitelist elsewhere.
10957adc3a2bSChandler Carruth   disableSROA(I.getOperand(0));
10967adc3a2bSChandler Carruth 
109739ed9a60SEli Friedman   // If this is a floating-point cast, and the target says this operation
109839ed9a60SEli Friedman   // is expensive, this may eventually become a library call. Treat the cost
109939ed9a60SEli Friedman   // as such.
110039ed9a60SEli Friedman   switch (I.getOpcode()) {
110139ed9a60SEli Friedman   case Instruction::FPTrunc:
110239ed9a60SEli Friedman   case Instruction::FPExt:
110339ed9a60SEli Friedman   case Instruction::UIToFP:
110439ed9a60SEli Friedman   case Instruction::SIToFP:
110539ed9a60SEli Friedman   case Instruction::FPToUI:
110639ed9a60SEli Friedman   case Instruction::FPToSI:
110739ed9a60SEli Friedman     if (TTI.getFPOpCost(I.getType()) == TargetTransformInfo::TCC_Expensive)
110806408758SMircea Trofin       onCallPenalty();
11094dc0b1acSReid Kleckner     break;
111039ed9a60SEli Friedman   default:
111139ed9a60SEli Friedman     break;
111239ed9a60SEli Friedman   }
111339ed9a60SEli Friedman 
1114e9c9329aSSam Parker   return TargetTransformInfo::TCC_Free ==
1115e9c9329aSSam Parker          TTI.getUserCost(&I, TargetTransformInfo::TCK_SizeAndLatency);
11167adc3a2bSChandler Carruth }
11177adc3a2bSChandler Carruth 
11187adc3a2bSChandler Carruth bool CallAnalyzer::visitUnaryInstruction(UnaryInstruction &I) {
11197adc3a2bSChandler Carruth   Value *Operand = I.getOperand(0);
1120617f6364SEaswaran Raman   if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) {
1121617f6364SEaswaran Raman         return ConstantFoldInstOperands(&I, COps[0], DL);
1122617f6364SEaswaran Raman       }))
11237adc3a2bSChandler Carruth     return true;
11247adc3a2bSChandler Carruth 
1125ab53c5e5SCraig Topper   // Disable any SROA on the argument to arbitrary unary instructions.
11267adc3a2bSChandler Carruth   disableSROA(Operand);
11277adc3a2bSChandler Carruth 
11287adc3a2bSChandler Carruth   return false;
11297adc3a2bSChandler Carruth }
11307adc3a2bSChandler Carruth 
11317adc3a2bSChandler Carruth bool CallAnalyzer::paramHasAttr(Argument *A, Attribute::AttrKind Attr) {
1132652168a9SFedor Sergeev   return CandidateCall.paramHasAttr(A->getArgNo(), Attr);
11337adc3a2bSChandler Carruth }
11347adc3a2bSChandler Carruth 
11357adc3a2bSChandler Carruth bool CallAnalyzer::isKnownNonNullInCallee(Value *V) {
11367adc3a2bSChandler Carruth   // Does the *call site* have the NonNull attribute set on an argument?  We
11377adc3a2bSChandler Carruth   // use the attribute on the call site to memoize any analysis done in the
11387adc3a2bSChandler Carruth   // caller. This will also trip if the callee function has a non-null
11397adc3a2bSChandler Carruth   // parameter attribute, but that's a less interesting case because hopefully
11407adc3a2bSChandler Carruth   // the callee would already have been simplified based on that.
11417adc3a2bSChandler Carruth   if (Argument *A = dyn_cast<Argument>(V))
11427adc3a2bSChandler Carruth     if (paramHasAttr(A, Attribute::NonNull))
11437adc3a2bSChandler Carruth       return true;
11447adc3a2bSChandler Carruth 
11457adc3a2bSChandler Carruth   // Is this an alloca in the caller?  This is distinct from the attribute case
11467adc3a2bSChandler Carruth   // above because attributes aren't updated within the inliner itself and we
11477adc3a2bSChandler Carruth   // always want to catch the alloca derived case.
11487adc3a2bSChandler Carruth   if (isAllocaDerivedArg(V))
11497adc3a2bSChandler Carruth     // We can actually predict the result of comparisons between an
11507adc3a2bSChandler Carruth     // alloca-derived value and null. Note that this fires regardless of
11517adc3a2bSChandler Carruth     // SROA firing.
11527adc3a2bSChandler Carruth     return true;
11537adc3a2bSChandler Carruth 
11547adc3a2bSChandler Carruth   return false;
11557adc3a2bSChandler Carruth }
11567adc3a2bSChandler Carruth 
1157652168a9SFedor Sergeev bool CallAnalyzer::allowSizeGrowth(CallBase &Call) {
11589a3fc17aSEaswaran Raman   // If the normal destination of the invoke or the parent block of the call
11599a3fc17aSEaswaran Raman   // site is unreachable-terminated, there is little point in inlining this
11609a3fc17aSEaswaran Raman   // unless there is literally zero cost.
11619a3fc17aSEaswaran Raman   // FIXME: Note that it is possible that an unreachable-terminated block has a
11629a3fc17aSEaswaran Raman   // hot entry. For example, in below scenario inlining hot_call_X() may be
11639a3fc17aSEaswaran Raman   // beneficial :
11649a3fc17aSEaswaran Raman   // main() {
11659a3fc17aSEaswaran Raman   //   hot_call_1();
11669a3fc17aSEaswaran Raman   //   ...
11679a3fc17aSEaswaran Raman   //   hot_call_N()
11689a3fc17aSEaswaran Raman   //   exit(0);
11699a3fc17aSEaswaran Raman   // }
11709a3fc17aSEaswaran Raman   // For now, we are not handling this corner case here as it is rare in real
11719a3fc17aSEaswaran Raman   // code. In future, we should elaborate this based on BPI and BFI in more
11729a3fc17aSEaswaran Raman   // general threshold adjusting heuristics in updateThreshold().
1173652168a9SFedor Sergeev   if (InvokeInst *II = dyn_cast<InvokeInst>(&Call)) {
11749a3fc17aSEaswaran Raman     if (isa<UnreachableInst>(II->getNormalDest()->getTerminator()))
11759a3fc17aSEaswaran Raman       return false;
1176652168a9SFedor Sergeev   } else if (isa<UnreachableInst>(Call.getParent()->getTerminator()))
11779a3fc17aSEaswaran Raman     return false;
11789a3fc17aSEaswaran Raman 
11799a3fc17aSEaswaran Raman   return true;
11809a3fc17aSEaswaran Raman }
11819a3fc17aSEaswaran Raman 
118206408758SMircea Trofin bool InlineCostCallAnalyzer::isColdCallSite(CallBase &Call,
1183652168a9SFedor Sergeev                                             BlockFrequencyInfo *CallerBFI) {
1184c5fa6358SEaswaran Raman   // If global profile summary is available, then callsite's coldness is
1185c5fa6358SEaswaran Raman   // determined based on that.
1186bba762a1SChandler Carruth   if (PSI && PSI->hasProfileSummary())
1187ec73ae11SMircea Trofin     return PSI->isColdCallSite(Call, CallerBFI);
1188bba762a1SChandler Carruth 
1189bba762a1SChandler Carruth   // Otherwise we need BFI to be available.
1190c5fa6358SEaswaran Raman   if (!CallerBFI)
1191c5fa6358SEaswaran Raman     return false;
1192c5fa6358SEaswaran Raman 
1193bba762a1SChandler Carruth   // Determine if the callsite is cold relative to caller's entry. We could
1194bba762a1SChandler Carruth   // potentially cache the computation of scaled entry frequency, but the added
1195bba762a1SChandler Carruth   // complexity is not worth it unless this scaling shows up high in the
1196bba762a1SChandler Carruth   // profiles.
1197c5fa6358SEaswaran Raman   const BranchProbability ColdProb(ColdCallSiteRelFreq, 100);
1198652168a9SFedor Sergeev   auto CallSiteBB = Call.getParent();
1199c5fa6358SEaswaran Raman   auto CallSiteFreq = CallerBFI->getBlockFreq(CallSiteBB);
1200c5fa6358SEaswaran Raman   auto CallerEntryFreq =
1201652168a9SFedor Sergeev       CallerBFI->getBlockFreq(&(Call.getCaller()->getEntryBlock()));
1202c5fa6358SEaswaran Raman   return CallSiteFreq < CallerEntryFreq * ColdProb;
1203c5fa6358SEaswaran Raman }
1204c5fa6358SEaswaran Raman 
1205974d4eeaSEaswaran Raman Optional<int>
120606408758SMircea Trofin InlineCostCallAnalyzer::getHotCallSiteThreshold(CallBase &Call,
1207974d4eeaSEaswaran Raman                                                 BlockFrequencyInfo *CallerBFI) {
1208bba762a1SChandler Carruth 
1209974d4eeaSEaswaran Raman   // If global profile summary is available, then callsite's hotness is
1210974d4eeaSEaswaran Raman   // determined based on that.
1211ec73ae11SMircea Trofin   if (PSI && PSI->hasProfileSummary() && PSI->isHotCallSite(Call, CallerBFI))
1212bba762a1SChandler Carruth     return Params.HotCallSiteThreshold;
1213974d4eeaSEaswaran Raman 
1214bba762a1SChandler Carruth   // Otherwise we need BFI to be available and to have a locally hot callsite
1215bba762a1SChandler Carruth   // threshold.
1216bba762a1SChandler Carruth   if (!CallerBFI || !Params.LocallyHotCallSiteThreshold)
1217974d4eeaSEaswaran Raman     return None;
1218974d4eeaSEaswaran Raman 
1219bba762a1SChandler Carruth   // Determine if the callsite is hot relative to caller's entry. We could
1220bba762a1SChandler Carruth   // potentially cache the computation of scaled entry frequency, but the added
1221bba762a1SChandler Carruth   // complexity is not worth it unless this scaling shows up high in the
1222bba762a1SChandler Carruth   // profiles.
1223652168a9SFedor Sergeev   auto CallSiteBB = Call.getParent();
1224974d4eeaSEaswaran Raman   auto CallSiteFreq = CallerBFI->getBlockFreq(CallSiteBB).getFrequency();
1225974d4eeaSEaswaran Raman   auto CallerEntryFreq = CallerBFI->getEntryFreq();
1226974d4eeaSEaswaran Raman   if (CallSiteFreq >= CallerEntryFreq * HotCallSiteRelFreq)
1227bba762a1SChandler Carruth     return Params.LocallyHotCallSiteThreshold;
1228bba762a1SChandler Carruth 
1229bba762a1SChandler Carruth   // Otherwise treat it normally.
1230974d4eeaSEaswaran Raman   return None;
1231974d4eeaSEaswaran Raman }
1232974d4eeaSEaswaran Raman 
123306408758SMircea Trofin void InlineCostCallAnalyzer::updateThreshold(CallBase &Call, Function &Callee) {
12349a3fc17aSEaswaran Raman   // If no size growth is allowed for this inlining, set Threshold to 0.
1235652168a9SFedor Sergeev   if (!allowSizeGrowth(Call)) {
12369a3fc17aSEaswaran Raman     Threshold = 0;
12379a3fc17aSEaswaran Raman     return;
12389a3fc17aSEaswaran Raman   }
12399a3fc17aSEaswaran Raman 
1240652168a9SFedor Sergeev   Function *Caller = Call.getCaller();
12411c57cc2bSEaswaran Raman 
12421c57cc2bSEaswaran Raman   // return min(A, B) if B is valid.
12431c57cc2bSEaswaran Raman   auto MinIfValid = [](int A, Optional<int> B) {
12441c57cc2bSEaswaran Raman     return B ? std::min(A, B.getValue()) : A;
12451c57cc2bSEaswaran Raman   };
12461c57cc2bSEaswaran Raman 
12470d58fcacSEaswaran Raman   // return max(A, B) if B is valid.
12480d58fcacSEaswaran Raman   auto MaxIfValid = [](int A, Optional<int> B) {
12490d58fcacSEaswaran Raman     return B ? std::max(A, B.getValue()) : A;
12500d58fcacSEaswaran Raman   };
12510d58fcacSEaswaran Raman 
125251b809bfSEaswaran Raman   // Various bonus percentages. These are multiplied by Threshold to get the
125351b809bfSEaswaran Raman   // bonus values.
125451b809bfSEaswaran Raman   // SingleBBBonus: This bonus is applied if the callee has a single reachable
125551b809bfSEaswaran Raman   // basic block at the given callsite context. This is speculatively applied
125651b809bfSEaswaran Raman   // and withdrawn if more than one basic block is seen.
125751b809bfSEaswaran Raman   //
125851b809bfSEaswaran Raman   // LstCallToStaticBonus: This large bonus is applied to ensure the inlining
125951b809bfSEaswaran Raman   // of the last call to a static function as inlining such functions is
126051b809bfSEaswaran Raman   // guaranteed to reduce code size.
126151b809bfSEaswaran Raman   //
126251b809bfSEaswaran Raman   // These bonus percentages may be set to 0 based on properties of the caller
126351b809bfSEaswaran Raman   // and the callsite.
126451b809bfSEaswaran Raman   int SingleBBBonusPercent = 50;
1265d912a9baSDaniil Fukalov   int VectorBonusPercent = TTI.getInlinerVectorBonusPercent();
126602647f73SJordan Rupprecht   int LastCallToStaticBonus = InlineConstants::LastCallToStaticBonus;
126751b809bfSEaswaran Raman 
126851b809bfSEaswaran Raman   // Lambda to set all the above bonus and bonus percentages to 0.
126951b809bfSEaswaran Raman   auto DisallowAllBonuses = [&]() {
127051b809bfSEaswaran Raman     SingleBBBonusPercent = 0;
127151b809bfSEaswaran Raman     VectorBonusPercent = 0;
127251b809bfSEaswaran Raman     LastCallToStaticBonus = 0;
127351b809bfSEaswaran Raman   };
127451b809bfSEaswaran Raman 
12751c57cc2bSEaswaran Raman   // Use the OptMinSizeThreshold or OptSizeThreshold knob if they are available
12761c57cc2bSEaswaran Raman   // and reduce the threshold if the caller has the necessary attribute.
127785bd3978SEvandro Menezes   if (Caller->hasMinSize()) {
12781c57cc2bSEaswaran Raman     Threshold = MinIfValid(Threshold, Params.OptMinSizeThreshold);
127951b809bfSEaswaran Raman     // For minsize, we want to disable the single BB bonus and the vector
128051b809bfSEaswaran Raman     // bonuses, but not the last-call-to-static bonus. Inlining the last call to
128151b809bfSEaswaran Raman     // a static function will, at the minimum, eliminate the parameter setup and
128251b809bfSEaswaran Raman     // call/return instructions.
128351b809bfSEaswaran Raman     SingleBBBonusPercent = 0;
128451b809bfSEaswaran Raman     VectorBonusPercent = 0;
128585bd3978SEvandro Menezes   } else if (Caller->hasOptSize())
12861c57cc2bSEaswaran Raman     Threshold = MinIfValid(Threshold, Params.OptSizeThreshold);
1287f4bb2f0dSEaswaran Raman 
1288e08b139dSEaswaran Raman   // Adjust the threshold based on inlinehint attribute and profile based
1289e08b139dSEaswaran Raman   // hotness information if the caller does not have MinSize attribute.
129085bd3978SEvandro Menezes   if (!Caller->hasMinSize()) {
1291e08b139dSEaswaran Raman     if (Callee.hasFnAttribute(Attribute::InlineHint))
12920d58fcacSEaswaran Raman       Threshold = MaxIfValid(Threshold, Params.HintThreshold);
1293bba762a1SChandler Carruth 
12943cd1479cSEaswaran Raman     // FIXME: After switching to the new passmanager, simplify the logic below
1295bba762a1SChandler Carruth     // by checking only the callsite hotness/coldness as we will reliably
1296bba762a1SChandler Carruth     // have local profile information.
12973cd1479cSEaswaran Raman     //
1298bba762a1SChandler Carruth     // Callsite hotness and coldness can be determined if sample profile is
1299bba762a1SChandler Carruth     // used (which adds hotness metadata to calls) or if caller's
1300bba762a1SChandler Carruth     // BlockFrequencyInfo is available.
1301bba762a1SChandler Carruth     BlockFrequencyInfo *CallerBFI = GetBFI ? &((*GetBFI)(*Caller)) : nullptr;
1302652168a9SFedor Sergeev     auto HotCallSiteThreshold = getHotCallSiteThreshold(Call, CallerBFI);
130385bd3978SEvandro Menezes     if (!Caller->hasOptSize() && HotCallSiteThreshold) {
1304d34e60caSNicola Zaghen       LLVM_DEBUG(dbgs() << "Hot callsite.\n");
1305974d4eeaSEaswaran Raman       // FIXME: This should update the threshold only if it exceeds the
1306974d4eeaSEaswaran Raman       // current threshold, but AutoFDO + ThinLTO currently relies on this
1307974d4eeaSEaswaran Raman       // behavior to prevent inlining of hot callsites during ThinLTO
1308974d4eeaSEaswaran Raman       // compile phase.
1309974d4eeaSEaswaran Raman       Threshold = HotCallSiteThreshold.getValue();
1310652168a9SFedor Sergeev     } else if (isColdCallSite(Call, CallerBFI)) {
1311d34e60caSNicola Zaghen       LLVM_DEBUG(dbgs() << "Cold callsite.\n");
131251b809bfSEaswaran Raman       // Do not apply bonuses for a cold callsite including the
131351b809bfSEaswaran Raman       // LastCallToStatic bonus. While this bonus might result in code size
131451b809bfSEaswaran Raman       // reduction, it can cause the size of a non-cold caller to increase
131551b809bfSEaswaran Raman       // preventing it from being inlined.
131651b809bfSEaswaran Raman       DisallowAllBonuses();
13173cd1479cSEaswaran Raman       Threshold = MinIfValid(Threshold, Params.ColdCallSiteThreshold);
1318bba762a1SChandler Carruth     } else if (PSI) {
1319bba762a1SChandler Carruth       // Use callee's global profile information only if we have no way of
1320bba762a1SChandler Carruth       // determining this via callsite information.
13213cd1479cSEaswaran Raman       if (PSI->isFunctionEntryHot(&Callee)) {
1322d34e60caSNicola Zaghen         LLVM_DEBUG(dbgs() << "Hot callee.\n");
1323e08b139dSEaswaran Raman         // If callsite hotness can not be determined, we may still know
1324e08b139dSEaswaran Raman         // that the callee is hot and treat it as a weaker hint for threshold
1325e08b139dSEaswaran Raman         // increase.
1326e08b139dSEaswaran Raman         Threshold = MaxIfValid(Threshold, Params.HintThreshold);
1327e08b139dSEaswaran Raman       } else if (PSI->isFunctionEntryCold(&Callee)) {
1328d34e60caSNicola Zaghen         LLVM_DEBUG(dbgs() << "Cold callee.\n");
132951b809bfSEaswaran Raman         // Do not apply bonuses for a cold callee including the
133051b809bfSEaswaran Raman         // LastCallToStatic bonus. While this bonus might result in code size
133151b809bfSEaswaran Raman         // reduction, it can cause the size of a non-cold caller to increase
133251b809bfSEaswaran Raman         // preventing it from being inlined.
133351b809bfSEaswaran Raman         DisallowAllBonuses();
13341c57cc2bSEaswaran Raman         Threshold = MinIfValid(Threshold, Params.ColdThreshold);
1335e08b139dSEaswaran Raman       }
1336e08b139dSEaswaran Raman     }
1337e08b139dSEaswaran Raman   }
13388650a4daSJustin Lebar 
133902647f73SJordan Rupprecht   // Finally, take the target-specific inlining threshold multiplier into
134002647f73SJordan Rupprecht   // account.
13418650a4daSJustin Lebar   Threshold *= TTI.getInliningThresholdMultiplier();
134251b809bfSEaswaran Raman 
134351b809bfSEaswaran Raman   SingleBBBonus = Threshold * SingleBBBonusPercent / 100;
134451b809bfSEaswaran Raman   VectorBonus = Threshold * VectorBonusPercent / 100;
134551b809bfSEaswaran Raman 
134602647f73SJordan Rupprecht   bool OnlyOneCallAndLocalLinkage =
134702647f73SJordan Rupprecht       F.hasLocalLinkage() && F.hasOneUse() && &F == Call.getCalledFunction();
134802647f73SJordan Rupprecht   // If there is only one call of the function, and it has internal linkage,
134902647f73SJordan Rupprecht   // the cost of inlining it drops dramatically. It may seem odd to update
135002647f73SJordan Rupprecht   // Cost in updateThreshold, but the bonus depends on the logic in this method.
135102647f73SJordan Rupprecht   if (OnlyOneCallAndLocalLinkage)
135202647f73SJordan Rupprecht     Cost -= LastCallToStaticBonus;
1353f4bb2f0dSEaswaran Raman }
1354f4bb2f0dSEaswaran Raman 
13557adc3a2bSChandler Carruth bool CallAnalyzer::visitCmpInst(CmpInst &I) {
13567adc3a2bSChandler Carruth   Value *LHS = I.getOperand(0), *RHS = I.getOperand(1);
13577adc3a2bSChandler Carruth   // First try to handle simplified comparisons.
1358617f6364SEaswaran Raman   if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) {
1359617f6364SEaswaran Raman         return ConstantExpr::getCompare(I.getPredicate(), COps[0], COps[1]);
1360617f6364SEaswaran Raman       }))
13617adc3a2bSChandler Carruth     return true;
13627adc3a2bSChandler Carruth 
13637adc3a2bSChandler Carruth   if (I.getOpcode() == Instruction::FCmp)
13647adc3a2bSChandler Carruth     return false;
13657adc3a2bSChandler Carruth 
13667adc3a2bSChandler Carruth   // Otherwise look for a comparison between constant offset pointers with
13677adc3a2bSChandler Carruth   // a common base.
13687adc3a2bSChandler Carruth   Value *LHSBase, *RHSBase;
13697adc3a2bSChandler Carruth   APInt LHSOffset, RHSOffset;
13707adc3a2bSChandler Carruth   std::tie(LHSBase, LHSOffset) = ConstantOffsetPtrs.lookup(LHS);
13717adc3a2bSChandler Carruth   if (LHSBase) {
13727adc3a2bSChandler Carruth     std::tie(RHSBase, RHSOffset) = ConstantOffsetPtrs.lookup(RHS);
13737adc3a2bSChandler Carruth     if (RHSBase && LHSBase == RHSBase) {
13747adc3a2bSChandler Carruth       // We have common bases, fold the icmp to a constant based on the
13757adc3a2bSChandler Carruth       // offsets.
13767adc3a2bSChandler Carruth       Constant *CLHS = ConstantInt::get(LHS->getContext(), LHSOffset);
13777adc3a2bSChandler Carruth       Constant *CRHS = ConstantInt::get(RHS->getContext(), RHSOffset);
13787adc3a2bSChandler Carruth       if (Constant *C = ConstantExpr::getICmp(I.getPredicate(), CLHS, CRHS)) {
13797adc3a2bSChandler Carruth         SimplifiedValues[&I] = C;
13807adc3a2bSChandler Carruth         ++NumConstantPtrCmps;
13817adc3a2bSChandler Carruth         return true;
13827adc3a2bSChandler Carruth       }
13837adc3a2bSChandler Carruth     }
13847adc3a2bSChandler Carruth   }
13857adc3a2bSChandler Carruth 
13867adc3a2bSChandler Carruth   // If the comparison is an equality comparison with null, we can simplify it
13877adc3a2bSChandler Carruth   // if we know the value (argument) can't be null
13887adc3a2bSChandler Carruth   if (I.isEquality() && isa<ConstantPointerNull>(I.getOperand(1)) &&
13897adc3a2bSChandler Carruth       isKnownNonNullInCallee(I.getOperand(0))) {
13907adc3a2bSChandler Carruth     bool IsNotEqual = I.getPredicate() == CmpInst::ICMP_NE;
13917adc3a2bSChandler Carruth     SimplifiedValues[&I] = IsNotEqual ? ConstantInt::getTrue(I.getType())
13927adc3a2bSChandler Carruth                                       : ConstantInt::getFalse(I.getType());
13937adc3a2bSChandler Carruth     return true;
13947adc3a2bSChandler Carruth   }
139506408758SMircea Trofin   return handleSROA(I.getOperand(0), isa<ConstantPointerNull>(I.getOperand(1)));
13967adc3a2bSChandler Carruth }
13977adc3a2bSChandler Carruth 
13987adc3a2bSChandler Carruth bool CallAnalyzer::visitSub(BinaryOperator &I) {
13997adc3a2bSChandler Carruth   // Try to handle a special case: we can fold computing the difference of two
14007adc3a2bSChandler Carruth   // constant-related pointers.
14017adc3a2bSChandler Carruth   Value *LHS = I.getOperand(0), *RHS = I.getOperand(1);
14027adc3a2bSChandler Carruth   Value *LHSBase, *RHSBase;
14037adc3a2bSChandler Carruth   APInt LHSOffset, RHSOffset;
14047adc3a2bSChandler Carruth   std::tie(LHSBase, LHSOffset) = ConstantOffsetPtrs.lookup(LHS);
14057adc3a2bSChandler Carruth   if (LHSBase) {
14067adc3a2bSChandler Carruth     std::tie(RHSBase, RHSOffset) = ConstantOffsetPtrs.lookup(RHS);
14077adc3a2bSChandler Carruth     if (RHSBase && LHSBase == RHSBase) {
14087adc3a2bSChandler Carruth       // We have common bases, fold the subtract to a constant based on the
14097adc3a2bSChandler Carruth       // offsets.
14107adc3a2bSChandler Carruth       Constant *CLHS = ConstantInt::get(LHS->getContext(), LHSOffset);
14117adc3a2bSChandler Carruth       Constant *CRHS = ConstantInt::get(RHS->getContext(), RHSOffset);
14127adc3a2bSChandler Carruth       if (Constant *C = ConstantExpr::getSub(CLHS, CRHS)) {
14137adc3a2bSChandler Carruth         SimplifiedValues[&I] = C;
14147adc3a2bSChandler Carruth         ++NumConstantPtrDiffs;
14157adc3a2bSChandler Carruth         return true;
14167adc3a2bSChandler Carruth       }
14177adc3a2bSChandler Carruth     }
14187adc3a2bSChandler Carruth   }
14197adc3a2bSChandler Carruth 
14207adc3a2bSChandler Carruth   // Otherwise, fall back to the generic logic for simplifying and handling
14217adc3a2bSChandler Carruth   // instructions.
14227adc3a2bSChandler Carruth   return Base::visitSub(I);
14237adc3a2bSChandler Carruth }
14247adc3a2bSChandler Carruth 
14257adc3a2bSChandler Carruth bool CallAnalyzer::visitBinaryOperator(BinaryOperator &I) {
14267adc3a2bSChandler Carruth   Value *LHS = I.getOperand(0), *RHS = I.getOperand(1);
14276d14dfe8SHaicheng Wu   Constant *CLHS = dyn_cast<Constant>(LHS);
14286d14dfe8SHaicheng Wu   if (!CLHS)
14296d14dfe8SHaicheng Wu     CLHS = SimplifiedValues.lookup(LHS);
14306d14dfe8SHaicheng Wu   Constant *CRHS = dyn_cast<Constant>(RHS);
14316d14dfe8SHaicheng Wu   if (!CRHS)
14326d14dfe8SHaicheng Wu     CRHS = SimplifiedValues.lookup(RHS);
14336d14dfe8SHaicheng Wu 
14347adc3a2bSChandler Carruth   Value *SimpleV = nullptr;
14357adc3a2bSChandler Carruth   if (auto FI = dyn_cast<FPMathOperator>(&I))
14365a956c85SMircea Trofin     SimpleV = SimplifyBinOp(I.getOpcode(), CLHS ? CLHS : LHS, CRHS ? CRHS : RHS,
14375a956c85SMircea Trofin                             FI->getFastMathFlags(), DL);
14387adc3a2bSChandler Carruth   else
14396d14dfe8SHaicheng Wu     SimpleV =
14406d14dfe8SHaicheng Wu         SimplifyBinOp(I.getOpcode(), CLHS ? CLHS : LHS, CRHS ? CRHS : RHS, DL);
14417adc3a2bSChandler Carruth 
14426d14dfe8SHaicheng Wu   if (Constant *C = dyn_cast_or_null<Constant>(SimpleV))
14436d14dfe8SHaicheng Wu     SimplifiedValues[&I] = C;
14446d14dfe8SHaicheng Wu 
14456d14dfe8SHaicheng Wu   if (SimpleV)
14467adc3a2bSChandler Carruth     return true;
14477adc3a2bSChandler Carruth 
14487adc3a2bSChandler Carruth   // Disable any SROA on arguments to arbitrary, unsimplified binary operators.
14497adc3a2bSChandler Carruth   disableSROA(LHS);
14507adc3a2bSChandler Carruth   disableSROA(RHS);
14517adc3a2bSChandler Carruth 
145239ed9a60SEli Friedman   // If the instruction is floating point, and the target says this operation
145339ed9a60SEli Friedman   // is expensive, this may eventually become a library call. Treat the cost
14547cebf0afSCraig Topper   // as such. Unless it's fneg which can be implemented with an xor.
14557cebf0afSCraig Topper   using namespace llvm::PatternMatch;
145639ed9a60SEli Friedman   if (I.getType()->isFloatingPointTy() &&
14577cebf0afSCraig Topper       TTI.getFPOpCost(I.getType()) == TargetTransformInfo::TCC_Expensive &&
14587cebf0afSCraig Topper       !match(&I, m_FNeg(m_Value())))
145906408758SMircea Trofin     onCallPenalty();
146039ed9a60SEli Friedman 
14617adc3a2bSChandler Carruth   return false;
14627adc3a2bSChandler Carruth }
14637adc3a2bSChandler Carruth 
14646cda33baSCraig Topper bool CallAnalyzer::visitFNeg(UnaryOperator &I) {
14656cda33baSCraig Topper   Value *Op = I.getOperand(0);
14666cda33baSCraig Topper   Constant *COp = dyn_cast<Constant>(Op);
14676cda33baSCraig Topper   if (!COp)
14686cda33baSCraig Topper     COp = SimplifiedValues.lookup(Op);
14696cda33baSCraig Topper 
14705a956c85SMircea Trofin   Value *SimpleV = SimplifyFNegInst(
14715a956c85SMircea Trofin       COp ? COp : Op, cast<FPMathOperator>(I).getFastMathFlags(), DL);
14726cda33baSCraig Topper 
14736cda33baSCraig Topper   if (Constant *C = dyn_cast_or_null<Constant>(SimpleV))
14746cda33baSCraig Topper     SimplifiedValues[&I] = C;
14756cda33baSCraig Topper 
14766cda33baSCraig Topper   if (SimpleV)
14776cda33baSCraig Topper     return true;
14786cda33baSCraig Topper 
14796cda33baSCraig Topper   // Disable any SROA on arguments to arbitrary, unsimplified fneg.
14806cda33baSCraig Topper   disableSROA(Op);
14816cda33baSCraig Topper 
14826cda33baSCraig Topper   return false;
14836cda33baSCraig Topper }
14846cda33baSCraig Topper 
14857adc3a2bSChandler Carruth bool CallAnalyzer::visitLoad(LoadInst &I) {
148606408758SMircea Trofin   if (handleSROA(I.getPointerOperand(), I.isSimple()))
14877adc3a2bSChandler Carruth     return true;
14887adc3a2bSChandler Carruth 
1489a4461515SHaicheng Wu   // If the data is already loaded from this address and hasn't been clobbered
1490a4461515SHaicheng Wu   // by any stores or calls, this load is likely to be redundant and can be
1491a4461515SHaicheng Wu   // eliminated.
1492a4461515SHaicheng Wu   if (EnableLoadElimination &&
1493b3689cabSHaicheng Wu       !LoadAddrSet.insert(I.getPointerOperand()).second && I.isUnordered()) {
149406408758SMircea Trofin     onLoadEliminationOpportunity();
1495a4461515SHaicheng Wu     return true;
1496a4461515SHaicheng Wu   }
1497a4461515SHaicheng Wu 
14987adc3a2bSChandler Carruth   return false;
14997adc3a2bSChandler Carruth }
15007adc3a2bSChandler Carruth 
15017adc3a2bSChandler Carruth bool CallAnalyzer::visitStore(StoreInst &I) {
150206408758SMircea Trofin   if (handleSROA(I.getPointerOperand(), I.isSimple()))
15037adc3a2bSChandler Carruth     return true;
15047adc3a2bSChandler Carruth 
1505a4461515SHaicheng Wu   // The store can potentially clobber loads and prevent repeated loads from
1506a4461515SHaicheng Wu   // being eliminated.
1507a4461515SHaicheng Wu   // FIXME:
1508a4461515SHaicheng Wu   // 1. We can probably keep an initial set of eliminatable loads substracted
1509a4461515SHaicheng Wu   // from the cost even when we finally see a store. We just need to disable
1510a4461515SHaicheng Wu   // *further* accumulation of elimination savings.
1511a4461515SHaicheng Wu   // 2. We should probably at some point thread MemorySSA for the callee into
1512a4461515SHaicheng Wu   // this and then use that to actually compute *really* precise savings.
1513a4461515SHaicheng Wu   disableLoadElimination();
15147adc3a2bSChandler Carruth   return false;
15157adc3a2bSChandler Carruth }
15167adc3a2bSChandler Carruth 
15177adc3a2bSChandler Carruth bool CallAnalyzer::visitExtractValue(ExtractValueInst &I) {
15187adc3a2bSChandler Carruth   // Constant folding for extract value is trivial.
1519617f6364SEaswaran Raman   if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) {
1520617f6364SEaswaran Raman         return ConstantExpr::getExtractValue(COps[0], I.getIndices());
1521617f6364SEaswaran Raman       }))
15227adc3a2bSChandler Carruth     return true;
15237adc3a2bSChandler Carruth 
15247adc3a2bSChandler Carruth   // SROA can look through these but give them a cost.
15257adc3a2bSChandler Carruth   return false;
15267adc3a2bSChandler Carruth }
15277adc3a2bSChandler Carruth 
15287adc3a2bSChandler Carruth bool CallAnalyzer::visitInsertValue(InsertValueInst &I) {
15297adc3a2bSChandler Carruth   // Constant folding for insert value is trivial.
1530617f6364SEaswaran Raman   if (simplifyInstruction(I, [&](SmallVectorImpl<Constant *> &COps) {
1531617f6364SEaswaran Raman         return ConstantExpr::getInsertValue(/*AggregateOperand*/ COps[0],
1532617f6364SEaswaran Raman                                             /*InsertedValueOperand*/ COps[1],
1533617f6364SEaswaran Raman                                             I.getIndices());
1534617f6364SEaswaran Raman       }))
15357adc3a2bSChandler Carruth     return true;
15367adc3a2bSChandler Carruth 
15377adc3a2bSChandler Carruth   // SROA can look through these but give them a cost.
15387adc3a2bSChandler Carruth   return false;
15397adc3a2bSChandler Carruth }
15407adc3a2bSChandler Carruth 
15415f8f34e4SAdrian Prantl /// Try to simplify a call site.
15427adc3a2bSChandler Carruth ///
15437adc3a2bSChandler Carruth /// Takes a concrete function and callsite and tries to actually simplify it by
15447adc3a2bSChandler Carruth /// analyzing the arguments and call itself with instsimplify. Returns true if
15457adc3a2bSChandler Carruth /// it has simplified the callsite to some other entity (a constant), making it
15467adc3a2bSChandler Carruth /// free.
1547652168a9SFedor Sergeev bool CallAnalyzer::simplifyCallSite(Function *F, CallBase &Call) {
15487adc3a2bSChandler Carruth   // FIXME: Using the instsimplify logic directly for this is inefficient
15497adc3a2bSChandler Carruth   // because we have to continually rebuild the argument list even when no
15507adc3a2bSChandler Carruth   // simplifications can be performed. Until that is fixed with remapping
15517adc3a2bSChandler Carruth   // inside of instsimplify, directly constant fold calls here.
1552652168a9SFedor Sergeev   if (!canConstantFoldCallTo(&Call, F))
15537adc3a2bSChandler Carruth     return false;
15547adc3a2bSChandler Carruth 
15557adc3a2bSChandler Carruth   // Try to re-map the arguments to constants.
15567adc3a2bSChandler Carruth   SmallVector<Constant *, 4> ConstantArgs;
1557652168a9SFedor Sergeev   ConstantArgs.reserve(Call.arg_size());
1558652168a9SFedor Sergeev   for (Value *I : Call.args()) {
1559652168a9SFedor Sergeev     Constant *C = dyn_cast<Constant>(I);
15607adc3a2bSChandler Carruth     if (!C)
1561652168a9SFedor Sergeev       C = dyn_cast_or_null<Constant>(SimplifiedValues.lookup(I));
15627adc3a2bSChandler Carruth     if (!C)
15637adc3a2bSChandler Carruth       return false; // This argument doesn't map to a constant.
15647adc3a2bSChandler Carruth 
15657adc3a2bSChandler Carruth     ConstantArgs.push_back(C);
15667adc3a2bSChandler Carruth   }
1567652168a9SFedor Sergeev   if (Constant *C = ConstantFoldCall(&Call, F, ConstantArgs)) {
1568652168a9SFedor Sergeev     SimplifiedValues[&Call] = C;
15697adc3a2bSChandler Carruth     return true;
15707adc3a2bSChandler Carruth   }
15717adc3a2bSChandler Carruth 
15727adc3a2bSChandler Carruth   return false;
15737adc3a2bSChandler Carruth }
15747adc3a2bSChandler Carruth 
1575652168a9SFedor Sergeev bool CallAnalyzer::visitCallBase(CallBase &Call) {
1576652168a9SFedor Sergeev   if (Call.hasFnAttr(Attribute::ReturnsTwice) &&
15777adc3a2bSChandler Carruth       !F.hasFnAttribute(Attribute::ReturnsTwice)) {
15787adc3a2bSChandler Carruth     // This aborts the entire analysis.
15797adc3a2bSChandler Carruth     ExposesReturnsTwice = true;
15807adc3a2bSChandler Carruth     return false;
15817adc3a2bSChandler Carruth   }
1582652168a9SFedor Sergeev   if (isa<CallInst>(Call) && cast<CallInst>(Call).cannotDuplicate())
15837adc3a2bSChandler Carruth     ContainsNoDuplicateCall = true;
15847adc3a2bSChandler Carruth 
1585825debe8SEhud Katz   Value *Callee = Call.getCalledOperand();
1586825debe8SEhud Katz   Function *F = dyn_cast_or_null<Function>(Callee);
1587825debe8SEhud Katz   bool IsIndirectCall = !F;
1588825debe8SEhud Katz   if (IsIndirectCall) {
1589825debe8SEhud Katz     // Check if this happens to be an indirect function call to a known function
1590825debe8SEhud Katz     // in this inline context. If not, we've done all we can.
1591825debe8SEhud Katz     F = dyn_cast_or_null<Function>(SimplifiedValues.lookup(Callee));
1592825debe8SEhud Katz     if (!F) {
159306408758SMircea Trofin       onCallArgumentSetup(Call);
1594825debe8SEhud Katz 
1595825debe8SEhud Katz       if (!Call.onlyReadsMemory())
1596825debe8SEhud Katz         disableLoadElimination();
1597825debe8SEhud Katz       return Base::visitCallBase(Call);
1598825debe8SEhud Katz     }
1599825debe8SEhud Katz   }
1600825debe8SEhud Katz 
1601825debe8SEhud Katz   assert(F && "Expected a call to a known function");
1602825debe8SEhud Katz 
16037adc3a2bSChandler Carruth   // When we have a concrete function, first try to simplify it directly.
1604652168a9SFedor Sergeev   if (simplifyCallSite(F, Call))
16057adc3a2bSChandler Carruth     return true;
16067adc3a2bSChandler Carruth 
16077adc3a2bSChandler Carruth   // Next check if it is an intrinsic we know about.
16087adc3a2bSChandler Carruth   // FIXME: Lift this into part of the InstVisitor.
1609652168a9SFedor Sergeev   if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(&Call)) {
16107adc3a2bSChandler Carruth     switch (II->getIntrinsicID()) {
16117adc3a2bSChandler Carruth     default:
1612652168a9SFedor Sergeev       if (!Call.onlyReadsMemory() && !isAssumeLikeIntrinsic(II))
1613a4461515SHaicheng Wu         disableLoadElimination();
1614652168a9SFedor Sergeev       return Base::visitCallBase(Call);
16157adc3a2bSChandler Carruth 
16167dd8dbf4SPeter Collingbourne     case Intrinsic::load_relative:
161706408758SMircea Trofin       onLoadRelativeIntrinsic();
16187dd8dbf4SPeter Collingbourne       return false;
16197dd8dbf4SPeter Collingbourne 
16207adc3a2bSChandler Carruth     case Intrinsic::memset:
16217adc3a2bSChandler Carruth     case Intrinsic::memcpy:
16227adc3a2bSChandler Carruth     case Intrinsic::memmove:
1623a4461515SHaicheng Wu       disableLoadElimination();
16247adc3a2bSChandler Carruth       // SROA can usually chew through these intrinsics, but they aren't free.
16257adc3a2bSChandler Carruth       return false;
16264296ea72SVitaly Buka     case Intrinsic::icall_branch_funnel:
16277adc3a2bSChandler Carruth     case Intrinsic::localescape:
16284296ea72SVitaly Buka       HasUninlineableIntrinsic = true;
16297adc3a2bSChandler Carruth       return false;
163080788d80SFlorian Hahn     case Intrinsic::vastart:
163177beee41SSameer AbuAsal       InitsVargArgs = true;
163280788d80SFlorian Hahn       return false;
16337adc3a2bSChandler Carruth     }
16347adc3a2bSChandler Carruth   }
16357adc3a2bSChandler Carruth 
1636652168a9SFedor Sergeev   if (F == Call.getFunction()) {
16377adc3a2bSChandler Carruth     // This flag will fully abort the analysis, so don't bother with anything
16387adc3a2bSChandler Carruth     // else.
16397adc3a2bSChandler Carruth     IsRecursiveCall = true;
16407adc3a2bSChandler Carruth     return false;
16417adc3a2bSChandler Carruth   }
16427adc3a2bSChandler Carruth 
16437adc3a2bSChandler Carruth   if (TTI.isLoweredToCall(F)) {
164406408758SMircea Trofin     onLoweredCall(F, Call, IsIndirectCall);
1645825debe8SEhud Katz   }
16467adc3a2bSChandler Carruth 
1647825debe8SEhud Katz   if (!(Call.onlyReadsMemory() || (IsIndirectCall && F->onlyReadsMemory())))
1648a4461515SHaicheng Wu     disableLoadElimination();
1649652168a9SFedor Sergeev   return Base::visitCallBase(Call);
16507adc3a2bSChandler Carruth }
16517adc3a2bSChandler Carruth 
16527adc3a2bSChandler Carruth bool CallAnalyzer::visitReturnInst(ReturnInst &RI) {
16537adc3a2bSChandler Carruth   // At least one return instruction will be free after inlining.
16547adc3a2bSChandler Carruth   bool Free = !HasReturn;
16557adc3a2bSChandler Carruth   HasReturn = true;
16567adc3a2bSChandler Carruth   return Free;
16577adc3a2bSChandler Carruth }
16587adc3a2bSChandler Carruth 
16597adc3a2bSChandler Carruth bool CallAnalyzer::visitBranchInst(BranchInst &BI) {
16607adc3a2bSChandler Carruth   // We model unconditional branches as essentially free -- they really
16617adc3a2bSChandler Carruth   // shouldn't exist at all, but handling them makes the behavior of the
16627adc3a2bSChandler Carruth   // inliner more regular and predictable. Interestingly, conditional branches
16637adc3a2bSChandler Carruth   // which will fold away are also free.
16647adc3a2bSChandler Carruth   return BI.isUnconditional() || isa<ConstantInt>(BI.getCondition()) ||
16657adc3a2bSChandler Carruth          dyn_cast_or_null<ConstantInt>(
16667adc3a2bSChandler Carruth              SimplifiedValues.lookup(BI.getCondition()));
16677adc3a2bSChandler Carruth }
16687adc3a2bSChandler Carruth 
16693ec848bcSHaicheng Wu bool CallAnalyzer::visitSelectInst(SelectInst &SI) {
16703ec848bcSHaicheng Wu   bool CheckSROA = SI.getType()->isPointerTy();
16713ec848bcSHaicheng Wu   Value *TrueVal = SI.getTrueValue();
16723ec848bcSHaicheng Wu   Value *FalseVal = SI.getFalseValue();
16733ec848bcSHaicheng Wu 
16743ec848bcSHaicheng Wu   Constant *TrueC = dyn_cast<Constant>(TrueVal);
16753ec848bcSHaicheng Wu   if (!TrueC)
16763ec848bcSHaicheng Wu     TrueC = SimplifiedValues.lookup(TrueVal);
16773ec848bcSHaicheng Wu   Constant *FalseC = dyn_cast<Constant>(FalseVal);
16783ec848bcSHaicheng Wu   if (!FalseC)
16793ec848bcSHaicheng Wu     FalseC = SimplifiedValues.lookup(FalseVal);
16803ec848bcSHaicheng Wu   Constant *CondC =
16813ec848bcSHaicheng Wu       dyn_cast_or_null<Constant>(SimplifiedValues.lookup(SI.getCondition()));
16823ec848bcSHaicheng Wu 
16833ec848bcSHaicheng Wu   if (!CondC) {
16843ec848bcSHaicheng Wu     // Select C, X, X => X
16853ec848bcSHaicheng Wu     if (TrueC == FalseC && TrueC) {
16863ec848bcSHaicheng Wu       SimplifiedValues[&SI] = TrueC;
16873ec848bcSHaicheng Wu       return true;
16883ec848bcSHaicheng Wu     }
16893ec848bcSHaicheng Wu 
16903ec848bcSHaicheng Wu     if (!CheckSROA)
16913ec848bcSHaicheng Wu       return Base::visitSelectInst(SI);
16923ec848bcSHaicheng Wu 
16933ec848bcSHaicheng Wu     std::pair<Value *, APInt> TrueBaseAndOffset =
16943ec848bcSHaicheng Wu         ConstantOffsetPtrs.lookup(TrueVal);
16953ec848bcSHaicheng Wu     std::pair<Value *, APInt> FalseBaseAndOffset =
16963ec848bcSHaicheng Wu         ConstantOffsetPtrs.lookup(FalseVal);
16973ec848bcSHaicheng Wu     if (TrueBaseAndOffset == FalseBaseAndOffset && TrueBaseAndOffset.first) {
16983ec848bcSHaicheng Wu       ConstantOffsetPtrs[&SI] = TrueBaseAndOffset;
16993ec848bcSHaicheng Wu 
170006408758SMircea Trofin       if (auto *SROAArg = getSROAArgForValueOrNull(TrueVal))
17013ec848bcSHaicheng Wu         SROAArgValues[&SI] = SROAArg;
17023ec848bcSHaicheng Wu       return true;
17033ec848bcSHaicheng Wu     }
17043ec848bcSHaicheng Wu 
17053ec848bcSHaicheng Wu     return Base::visitSelectInst(SI);
17063ec848bcSHaicheng Wu   }
17073ec848bcSHaicheng Wu 
17083ec848bcSHaicheng Wu   // Select condition is a constant.
17093ec848bcSHaicheng Wu   Value *SelectedV = CondC->isAllOnesValue()
17103ec848bcSHaicheng Wu                          ? TrueVal
17113ec848bcSHaicheng Wu                          : (CondC->isNullValue()) ? FalseVal : nullptr;
17123ec848bcSHaicheng Wu   if (!SelectedV) {
17133ec848bcSHaicheng Wu     // Condition is a vector constant that is not all 1s or all 0s.  If all
17143ec848bcSHaicheng Wu     // operands are constants, ConstantExpr::getSelect() can handle the cases
17153ec848bcSHaicheng Wu     // such as select vectors.
17163ec848bcSHaicheng Wu     if (TrueC && FalseC) {
17173ec848bcSHaicheng Wu       if (auto *C = ConstantExpr::getSelect(CondC, TrueC, FalseC)) {
17183ec848bcSHaicheng Wu         SimplifiedValues[&SI] = C;
17193ec848bcSHaicheng Wu         return true;
17203ec848bcSHaicheng Wu       }
17213ec848bcSHaicheng Wu     }
17223ec848bcSHaicheng Wu     return Base::visitSelectInst(SI);
17233ec848bcSHaicheng Wu   }
17243ec848bcSHaicheng Wu 
17253ec848bcSHaicheng Wu   // Condition is either all 1s or all 0s. SI can be simplified.
17263ec848bcSHaicheng Wu   if (Constant *SelectedC = dyn_cast<Constant>(SelectedV)) {
17273ec848bcSHaicheng Wu     SimplifiedValues[&SI] = SelectedC;
17283ec848bcSHaicheng Wu     return true;
17293ec848bcSHaicheng Wu   }
17303ec848bcSHaicheng Wu 
17313ec848bcSHaicheng Wu   if (!CheckSROA)
17323ec848bcSHaicheng Wu     return true;
17333ec848bcSHaicheng Wu 
17343ec848bcSHaicheng Wu   std::pair<Value *, APInt> BaseAndOffset =
17353ec848bcSHaicheng Wu       ConstantOffsetPtrs.lookup(SelectedV);
17363ec848bcSHaicheng Wu   if (BaseAndOffset.first) {
17373ec848bcSHaicheng Wu     ConstantOffsetPtrs[&SI] = BaseAndOffset;
17383ec848bcSHaicheng Wu 
173906408758SMircea Trofin     if (auto *SROAArg = getSROAArgForValueOrNull(SelectedV))
17403ec848bcSHaicheng Wu       SROAArgValues[&SI] = SROAArg;
17413ec848bcSHaicheng Wu   }
17423ec848bcSHaicheng Wu 
17433ec848bcSHaicheng Wu   return true;
17443ec848bcSHaicheng Wu }
17453ec848bcSHaicheng Wu 
17467adc3a2bSChandler Carruth bool CallAnalyzer::visitSwitchInst(SwitchInst &SI) {
17477adc3a2bSChandler Carruth   // We model unconditional switches as free, see the comments on handling
17487adc3a2bSChandler Carruth   // branches.
17497adc3a2bSChandler Carruth   if (isa<ConstantInt>(SI.getCondition()))
17507adc3a2bSChandler Carruth     return true;
17517adc3a2bSChandler Carruth   if (Value *V = SimplifiedValues.lookup(SI.getCondition()))
17527adc3a2bSChandler Carruth     if (isa<ConstantInt>(V))
17537adc3a2bSChandler Carruth       return true;
17547adc3a2bSChandler Carruth 
17557ad02eeeSEric Christopher   // Assume the most general case where the switch is lowered into
1756919f9e8dSJun Bum Lim   // either a jump table, bit test, or a balanced binary tree consisting of
1757919f9e8dSJun Bum Lim   // case clusters without merging adjacent clusters with the same
1758919f9e8dSJun Bum Lim   // destination. We do not consider the switches that are lowered with a mix
1759919f9e8dSJun Bum Lim   // of jump table/bit test/binary search tree. The cost of the switch is
1760919f9e8dSJun Bum Lim   // proportional to the size of the tree or the size of jump table range.
17612960d41eSJun Bum Lim   //
17622960d41eSJun Bum Lim   // NB: We convert large switches which are just used to initialize large phi
17632960d41eSJun Bum Lim   // nodes to lookup tables instead in simplify-cfg, so this shouldn't prevent
17642960d41eSJun Bum Lim   // inlining those. It will prevent inlining in cases where the optimization
17652960d41eSJun Bum Lim   // does not (yet) fire.
1766919f9e8dSJun Bum Lim 
1767919f9e8dSJun Bum Lim   unsigned JumpTableSize = 0;
17680d987e41SHiroshi Yamauchi   BlockFrequencyInfo *BFI = GetBFI ? &((*GetBFI)(F)) : nullptr;
1769919f9e8dSJun Bum Lim   unsigned NumCaseCluster =
17700d987e41SHiroshi Yamauchi       TTI.getEstimatedNumberOfCaseClusters(SI, JumpTableSize, PSI, BFI);
1771919f9e8dSJun Bum Lim 
177206408758SMircea Trofin   onFinalizeSwitch(JumpTableSize, NumCaseCluster);
1773919f9e8dSJun Bum Lim   return false;
1774919f9e8dSJun Bum Lim }
1775919f9e8dSJun Bum Lim 
17767adc3a2bSChandler Carruth bool CallAnalyzer::visitIndirectBrInst(IndirectBrInst &IBI) {
17777adc3a2bSChandler Carruth   // We never want to inline functions that contain an indirectbr.  This is
17787adc3a2bSChandler Carruth   // incorrect because all the blockaddress's (in static global initializers
17797adc3a2bSChandler Carruth   // for example) would be referring to the original function, and this
17807adc3a2bSChandler Carruth   // indirect jump would jump from the inlined copy of the function into the
17817adc3a2bSChandler Carruth   // original function which is extremely undefined behavior.
17827adc3a2bSChandler Carruth   // FIXME: This logic isn't really right; we can safely inline functions with
17837adc3a2bSChandler Carruth   // indirectbr's as long as no other function or global references the
17847adc3a2bSChandler Carruth   // blockaddress of a block within the current function.
17857adc3a2bSChandler Carruth   HasIndirectBr = true;
17867adc3a2bSChandler Carruth   return false;
17877adc3a2bSChandler Carruth }
17887adc3a2bSChandler Carruth 
17897adc3a2bSChandler Carruth bool CallAnalyzer::visitResumeInst(ResumeInst &RI) {
17907adc3a2bSChandler Carruth   // FIXME: It's not clear that a single instruction is an accurate model for
17917adc3a2bSChandler Carruth   // the inline cost of a resume instruction.
17927adc3a2bSChandler Carruth   return false;
17937adc3a2bSChandler Carruth }
17947adc3a2bSChandler Carruth 
17957adc3a2bSChandler Carruth bool CallAnalyzer::visitCleanupReturnInst(CleanupReturnInst &CRI) {
17967adc3a2bSChandler Carruth   // FIXME: It's not clear that a single instruction is an accurate model for
17977adc3a2bSChandler Carruth   // the inline cost of a cleanupret instruction.
17987adc3a2bSChandler Carruth   return false;
17997adc3a2bSChandler Carruth }
18007adc3a2bSChandler Carruth 
18017adc3a2bSChandler Carruth bool CallAnalyzer::visitCatchReturnInst(CatchReturnInst &CRI) {
18027adc3a2bSChandler Carruth   // FIXME: It's not clear that a single instruction is an accurate model for
18038220bcc5SJoseph Tremoulet   // the inline cost of a catchret instruction.
18047adc3a2bSChandler Carruth   return false;
18057adc3a2bSChandler Carruth }
18067adc3a2bSChandler Carruth 
18077adc3a2bSChandler Carruth bool CallAnalyzer::visitUnreachableInst(UnreachableInst &I) {
18087adc3a2bSChandler Carruth   // FIXME: It might be reasonably to discount the cost of instructions leading
18097adc3a2bSChandler Carruth   // to unreachable as they have the lowest possible impact on both runtime and
18107adc3a2bSChandler Carruth   // code size.
18117adc3a2bSChandler Carruth   return true; // No actual code is needed for unreachable.
18127adc3a2bSChandler Carruth }
18137adc3a2bSChandler Carruth 
18147adc3a2bSChandler Carruth bool CallAnalyzer::visitInstruction(Instruction &I) {
18157adc3a2bSChandler Carruth   // Some instructions are free. All of the free intrinsics can also be
18167adc3a2bSChandler Carruth   // handled by SROA, etc.
1817e9c9329aSSam Parker   if (TargetTransformInfo::TCC_Free ==
1818e9c9329aSSam Parker       TTI.getUserCost(&I, TargetTransformInfo::TCK_SizeAndLatency))
18197adc3a2bSChandler Carruth     return true;
18207adc3a2bSChandler Carruth 
18217adc3a2bSChandler Carruth   // We found something we don't understand or can't handle. Mark any SROA-able
18227adc3a2bSChandler Carruth   // values in the operand list as no longer viable.
18237adc3a2bSChandler Carruth   for (User::op_iterator OI = I.op_begin(), OE = I.op_end(); OI != OE; ++OI)
18247adc3a2bSChandler Carruth     disableSROA(*OI);
18257adc3a2bSChandler Carruth 
18267adc3a2bSChandler Carruth   return false;
18277adc3a2bSChandler Carruth }
18287adc3a2bSChandler Carruth 
18295f8f34e4SAdrian Prantl /// Analyze a basic block for its contribution to the inline cost.
18307adc3a2bSChandler Carruth ///
18317adc3a2bSChandler Carruth /// This method walks the analyzer over every instruction in the given basic
18327adc3a2bSChandler Carruth /// block and accounts for their cost during inlining at this callsite. It
18337adc3a2bSChandler Carruth /// aborts early if the threshold has been exceeded or an impossible to inline
18347adc3a2bSChandler Carruth /// construct has been detected. It returns false if inlining is no longer
18357adc3a2bSChandler Carruth /// viable, and true if inlining remains viable.
1836c0aa4b75SDavid Bolvansky InlineResult
1837c0aa4b75SDavid Bolvansky CallAnalyzer::analyzeBlock(BasicBlock *BB,
18387adc3a2bSChandler Carruth                            SmallPtrSetImpl<const Value *> &EphValues) {
18397adc3a2bSChandler Carruth   for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I) {
18407adc3a2bSChandler Carruth     // FIXME: Currently, the number of instructions in a function regardless of
18417adc3a2bSChandler Carruth     // our ability to simplify them during inline to constants or dead code,
18427adc3a2bSChandler Carruth     // are actually used by the vector bonus heuristic. As long as that's true,
18437adc3a2bSChandler Carruth     // we have to special case debug intrinsics here to prevent differences in
18447adc3a2bSChandler Carruth     // inlining due to debug symbols. Eventually, the number of unsimplified
18457adc3a2bSChandler Carruth     // instructions shouldn't factor into the cost computation, but until then,
18467adc3a2bSChandler Carruth     // hack around it here.
18477adc3a2bSChandler Carruth     if (isa<DbgInfoIntrinsic>(I))
18487adc3a2bSChandler Carruth       continue;
18497adc3a2bSChandler Carruth 
18507adc3a2bSChandler Carruth     // Skip ephemeral values.
18515a82c916SDuncan P. N. Exon Smith     if (EphValues.count(&*I))
18527adc3a2bSChandler Carruth       continue;
18537adc3a2bSChandler Carruth 
18547adc3a2bSChandler Carruth     ++NumInstructions;
18557adc3a2bSChandler Carruth     if (isa<ExtractElementInst>(I) || I->getType()->isVectorTy())
18567adc3a2bSChandler Carruth       ++NumVectorInstructions;
18577adc3a2bSChandler Carruth 
18587adc3a2bSChandler Carruth     // If the instruction simplified to a constant, there is no cost to this
18597adc3a2bSChandler Carruth     // instruction. Visit the instructions using our InstVisitor to account for
18607adc3a2bSChandler Carruth     // all of the per-instruction logic. The visit tree returns true if we
18617adc3a2bSChandler Carruth     // consumed the instruction in any way, and false if the instruction's base
18627adc3a2bSChandler Carruth     // cost should count against inlining.
1863c965fd94SKirill Naumov     onInstructionAnalysisStart(&*I);
1864c965fd94SKirill Naumov 
18655a82c916SDuncan P. N. Exon Smith     if (Base::visit(&*I))
18667adc3a2bSChandler Carruth       ++NumInstructionsSimplified;
18677adc3a2bSChandler Carruth     else
186814a16faeSMircea Trofin       onMissedSimplification();
18697adc3a2bSChandler Carruth 
1870c965fd94SKirill Naumov     onInstructionAnalysisFinish(&*I);
18710812c5beSHaicheng Wu     using namespace ore;
18727adc3a2bSChandler Carruth     // If the visit this instruction detected an uninlinable pattern, abort.
18735466597fSMircea Trofin     InlineResult IR = InlineResult::success();
1874c0aa4b75SDavid Bolvansky     if (IsRecursiveCall)
18755466597fSMircea Trofin       IR = InlineResult::failure("recursive");
1876c0aa4b75SDavid Bolvansky     else if (ExposesReturnsTwice)
18775466597fSMircea Trofin       IR = InlineResult::failure("exposes returns twice");
1878c0aa4b75SDavid Bolvansky     else if (HasDynamicAlloca)
18795466597fSMircea Trofin       IR = InlineResult::failure("dynamic alloca");
1880c0aa4b75SDavid Bolvansky     else if (HasIndirectBr)
18815466597fSMircea Trofin       IR = InlineResult::failure("indirect branch");
1882c0aa4b75SDavid Bolvansky     else if (HasUninlineableIntrinsic)
18835466597fSMircea Trofin       IR = InlineResult::failure("uninlinable intrinsic");
188477beee41SSameer AbuAsal     else if (InitsVargArgs)
18855466597fSMircea Trofin       IR = InlineResult::failure("varargs");
18865466597fSMircea Trofin     if (!IR.isSuccess()) {
18870812c5beSHaicheng Wu       if (ORE)
18889590658fSVivek Pandya         ORE->emit([&]() {
18899590658fSVivek Pandya           return OptimizationRemarkMissed(DEBUG_TYPE, "NeverInline",
1890652168a9SFedor Sergeev                                           &CandidateCall)
1891c0aa4b75SDavid Bolvansky                  << NV("Callee", &F) << " has uninlinable pattern ("
18925466597fSMircea Trofin                  << NV("InlineResult", IR.getFailureReason())
1893c0aa4b75SDavid Bolvansky                  << ") and cost is not fully computed";
18949590658fSVivek Pandya         });
1895c0aa4b75SDavid Bolvansky       return IR;
18960812c5beSHaicheng Wu     }
18977adc3a2bSChandler Carruth 
18987adc3a2bSChandler Carruth     // If the caller is a recursive function then we don't want to inline
18997adc3a2bSChandler Carruth     // functions which allocate a lot of stack space because it would increase
19007adc3a2bSChandler Carruth     // the caller stack usage dramatically.
19017adc3a2bSChandler Carruth     if (IsCallerRecursive &&
19020812c5beSHaicheng Wu         AllocatedSize > InlineConstants::TotalAllocaSizeRecursiveCaller) {
19035466597fSMircea Trofin       auto IR =
19045466597fSMircea Trofin           InlineResult::failure("recursive and allocates too much stack space");
19050812c5beSHaicheng Wu       if (ORE)
19069590658fSVivek Pandya         ORE->emit([&]() {
19079590658fSVivek Pandya           return OptimizationRemarkMissed(DEBUG_TYPE, "NeverInline",
1908652168a9SFedor Sergeev                                           &CandidateCall)
19095466597fSMircea Trofin                  << NV("Callee", &F) << " is "
19105466597fSMircea Trofin                  << NV("InlineResult", IR.getFailureReason())
1911c0aa4b75SDavid Bolvansky                  << ". Cost is not fully computed";
19129590658fSVivek Pandya         });
1913c0aa4b75SDavid Bolvansky       return IR;
19140812c5beSHaicheng Wu     }
19157adc3a2bSChandler Carruth 
191606408758SMircea Trofin     if (shouldStop())
19175466597fSMircea Trofin       return InlineResult::failure(
19185466597fSMircea Trofin           "Call site analysis is not favorable to inlining.");
19197adc3a2bSChandler Carruth   }
19207adc3a2bSChandler Carruth 
19215466597fSMircea Trofin   return InlineResult::success();
19227adc3a2bSChandler Carruth }
19237adc3a2bSChandler Carruth 
19245f8f34e4SAdrian Prantl /// Compute the base pointer and cumulative constant offsets for V.
19257adc3a2bSChandler Carruth ///
19267adc3a2bSChandler Carruth /// This strips all constant offsets off of V, leaving it the base pointer, and
19277adc3a2bSChandler Carruth /// accumulates the total constant offset applied in the returned constant. It
19287adc3a2bSChandler Carruth /// returns 0 if V is not a pointer, and returns the constant '0' if there are
19297adc3a2bSChandler Carruth /// no constant offsets applied.
19307adc3a2bSChandler Carruth ConstantInt *CallAnalyzer::stripAndComputeInBoundsConstantOffsets(Value *&V) {
19317adc3a2bSChandler Carruth   if (!V->getType()->isPointerTy())
19327adc3a2bSChandler Carruth     return nullptr;
19337adc3a2bSChandler Carruth 
193477f32994SBjorn Pettersson   unsigned AS = V->getType()->getPointerAddressSpace();
1935945b7e5aSElena Demikhovsky   unsigned IntPtrWidth = DL.getIndexSizeInBits(AS);
19367adc3a2bSChandler Carruth   APInt Offset = APInt::getNullValue(IntPtrWidth);
19377adc3a2bSChandler Carruth 
19387adc3a2bSChandler Carruth   // Even though we don't look through PHI nodes, we could be called on an
19397adc3a2bSChandler Carruth   // instruction in an unreachable block, which may be on a cycle.
19407adc3a2bSChandler Carruth   SmallPtrSet<Value *, 4> Visited;
19417adc3a2bSChandler Carruth   Visited.insert(V);
19427adc3a2bSChandler Carruth   do {
19437adc3a2bSChandler Carruth     if (GEPOperator *GEP = dyn_cast<GEPOperator>(V)) {
19447adc3a2bSChandler Carruth       if (!GEP->isInBounds() || !accumulateGEPOffset(*GEP, Offset))
19457adc3a2bSChandler Carruth         return nullptr;
19467adc3a2bSChandler Carruth       V = GEP->getPointerOperand();
19477adc3a2bSChandler Carruth     } else if (Operator::getOpcode(V) == Instruction::BitCast) {
19487adc3a2bSChandler Carruth       V = cast<Operator>(V)->getOperand(0);
19497adc3a2bSChandler Carruth     } else if (GlobalAlias *GA = dyn_cast<GlobalAlias>(V)) {
19505ce32728SSanjoy Das       if (GA->isInterposable())
19517adc3a2bSChandler Carruth         break;
19527adc3a2bSChandler Carruth       V = GA->getAliasee();
19537adc3a2bSChandler Carruth     } else {
19547adc3a2bSChandler Carruth       break;
19557adc3a2bSChandler Carruth     }
19567adc3a2bSChandler Carruth     assert(V->getType()->isPointerTy() && "Unexpected operand type!");
19577adc3a2bSChandler Carruth   } while (Visited.insert(V).second);
19587adc3a2bSChandler Carruth 
195997572775SNicola Zaghen   Type *IdxPtrTy = DL.getIndexType(V->getType());
196097572775SNicola Zaghen   return cast<ConstantInt>(ConstantInt::get(IdxPtrTy, Offset));
19617adc3a2bSChandler Carruth }
19627adc3a2bSChandler Carruth 
19635f8f34e4SAdrian Prantl /// Find dead blocks due to deleted CFG edges during inlining.
19643739e14aSHaicheng Wu ///
19653739e14aSHaicheng Wu /// If we know the successor of the current block, \p CurrBB, has to be \p
19663739e14aSHaicheng Wu /// NextBB, the other successors of \p CurrBB are dead if these successors have
19673739e14aSHaicheng Wu /// no live incoming CFG edges.  If one block is found to be dead, we can
19683739e14aSHaicheng Wu /// continue growing the dead block list by checking the successors of the dead
19693739e14aSHaicheng Wu /// blocks to see if all their incoming edges are dead or not.
19703739e14aSHaicheng Wu void CallAnalyzer::findDeadBlocks(BasicBlock *CurrBB, BasicBlock *NextBB) {
19713739e14aSHaicheng Wu   auto IsEdgeDead = [&](BasicBlock *Pred, BasicBlock *Succ) {
197202a2bb2fSHiroshi Inoue     // A CFG edge is dead if the predecessor is dead or the predecessor has a
19733739e14aSHaicheng Wu     // known successor which is not the one under exam.
19743739e14aSHaicheng Wu     return (DeadBlocks.count(Pred) ||
19753739e14aSHaicheng Wu             (KnownSuccessors[Pred] && KnownSuccessors[Pred] != Succ));
19763739e14aSHaicheng Wu   };
19773739e14aSHaicheng Wu 
19783739e14aSHaicheng Wu   auto IsNewlyDead = [&](BasicBlock *BB) {
19793739e14aSHaicheng Wu     // If all the edges to a block are dead, the block is also dead.
19803739e14aSHaicheng Wu     return (!DeadBlocks.count(BB) &&
19813739e14aSHaicheng Wu             llvm::all_of(predecessors(BB),
19823739e14aSHaicheng Wu                          [&](BasicBlock *P) { return IsEdgeDead(P, BB); }));
19833739e14aSHaicheng Wu   };
19843739e14aSHaicheng Wu 
19853739e14aSHaicheng Wu   for (BasicBlock *Succ : successors(CurrBB)) {
19863739e14aSHaicheng Wu     if (Succ == NextBB || !IsNewlyDead(Succ))
19873739e14aSHaicheng Wu       continue;
19883739e14aSHaicheng Wu     SmallVector<BasicBlock *, 4> NewDead;
19893739e14aSHaicheng Wu     NewDead.push_back(Succ);
19903739e14aSHaicheng Wu     while (!NewDead.empty()) {
19913739e14aSHaicheng Wu       BasicBlock *Dead = NewDead.pop_back_val();
19923739e14aSHaicheng Wu       if (DeadBlocks.insert(Dead))
19933739e14aSHaicheng Wu         // Continue growing the dead block lists.
19943739e14aSHaicheng Wu         for (BasicBlock *S : successors(Dead))
19953739e14aSHaicheng Wu           if (IsNewlyDead(S))
19963739e14aSHaicheng Wu             NewDead.push_back(S);
19973739e14aSHaicheng Wu     }
19983739e14aSHaicheng Wu   }
19993739e14aSHaicheng Wu }
20003739e14aSHaicheng Wu 
20015f8f34e4SAdrian Prantl /// Analyze a call site for potential inlining.
20027adc3a2bSChandler Carruth ///
20037adc3a2bSChandler Carruth /// Returns true if inlining this call is viable, and false if it is not
20047adc3a2bSChandler Carruth /// viable. It computes the cost and adjusts the threshold based on numerous
20057adc3a2bSChandler Carruth /// factors and heuristics. If this method returns false but the computed cost
20067adc3a2bSChandler Carruth /// is below the computed threshold, then inlining was forcibly disabled by
20077adc3a2bSChandler Carruth /// some artifact of the routine.
200893ac81ccSMircea Trofin InlineResult CallAnalyzer::analyze() {
20097adc3a2bSChandler Carruth   ++NumCallsAnalyzed;
20107adc3a2bSChandler Carruth 
201106408758SMircea Trofin   auto Result = onAnalysisStart();
20125466597fSMircea Trofin   if (!Result.isSuccess())
201306408758SMircea Trofin     return Result;
20147adc3a2bSChandler Carruth 
20157adc3a2bSChandler Carruth   if (F.empty())
20165466597fSMircea Trofin     return InlineResult::success();
20177adc3a2bSChandler Carruth 
201893ac81ccSMircea Trofin   Function *Caller = CandidateCall.getFunction();
20197adc3a2bSChandler Carruth   // Check if the caller function is recursive itself.
20207adc3a2bSChandler Carruth   for (User *U : Caller->users()) {
2021652168a9SFedor Sergeev     CallBase *Call = dyn_cast<CallBase>(U);
2022652168a9SFedor Sergeev     if (Call && Call->getFunction() == Caller) {
20237adc3a2bSChandler Carruth       IsCallerRecursive = true;
20247adc3a2bSChandler Carruth       break;
20257adc3a2bSChandler Carruth     }
20267adc3a2bSChandler Carruth   }
20277adc3a2bSChandler Carruth 
20287adc3a2bSChandler Carruth   // Populate our simplified values by mapping from function arguments to call
20297adc3a2bSChandler Carruth   // arguments with known important simplifications.
203093ac81ccSMircea Trofin   auto CAI = CandidateCall.arg_begin();
20317adc3a2bSChandler Carruth   for (Function::arg_iterator FAI = F.arg_begin(), FAE = F.arg_end();
20327adc3a2bSChandler Carruth        FAI != FAE; ++FAI, ++CAI) {
203393ac81ccSMircea Trofin     assert(CAI != CandidateCall.arg_end());
20347adc3a2bSChandler Carruth     if (Constant *C = dyn_cast<Constant>(CAI))
20355a82c916SDuncan P. N. Exon Smith       SimplifiedValues[&*FAI] = C;
20367adc3a2bSChandler Carruth 
20377adc3a2bSChandler Carruth     Value *PtrArg = *CAI;
20387adc3a2bSChandler Carruth     if (ConstantInt *C = stripAndComputeInBoundsConstantOffsets(PtrArg)) {
20395a82c916SDuncan P. N. Exon Smith       ConstantOffsetPtrs[&*FAI] = std::make_pair(PtrArg, C->getValue());
20407adc3a2bSChandler Carruth 
20417adc3a2bSChandler Carruth       // We can SROA any pointer arguments derived from alloca instructions.
204206408758SMircea Trofin       if (auto *SROAArg = dyn_cast<AllocaInst>(PtrArg)) {
204306408758SMircea Trofin         SROAArgValues[&*FAI] = SROAArg;
204406408758SMircea Trofin         onInitializeSROAArg(SROAArg);
20452e42cc7aSMircea Trofin         EnabledSROAAllocas.insert(SROAArg);
20467adc3a2bSChandler Carruth       }
20477adc3a2bSChandler Carruth     }
20487adc3a2bSChandler Carruth   }
20497adc3a2bSChandler Carruth   NumConstantArgs = SimplifiedValues.size();
20507adc3a2bSChandler Carruth   NumConstantOffsetPtrArgs = ConstantOffsetPtrs.size();
20517adc3a2bSChandler Carruth   NumAllocaArgs = SROAArgValues.size();
20527adc3a2bSChandler Carruth 
20537adc3a2bSChandler Carruth   // FIXME: If a caller has multiple calls to a callee, we end up recomputing
20547adc3a2bSChandler Carruth   // the ephemeral values multiple times (and they're completely determined by
20557adc3a2bSChandler Carruth   // the callee, so this is purely duplicate work).
20567adc3a2bSChandler Carruth   SmallPtrSet<const Value *, 32> EphValues;
2057aec2fa35SDaniel Jasper   CodeMetrics::collectEphemeralValues(&F, &GetAssumptionCache(F), EphValues);
20587adc3a2bSChandler Carruth 
20597adc3a2bSChandler Carruth   // The worklist of live basic blocks in the callee *after* inlining. We avoid
20607adc3a2bSChandler Carruth   // adding basic blocks of the callee which can be proven to be dead for this
20617adc3a2bSChandler Carruth   // particular call site in order to get more accurate cost estimates. This
20627adc3a2bSChandler Carruth   // requires a somewhat heavyweight iteration pattern: we need to walk the
20637adc3a2bSChandler Carruth   // basic blocks in a breadth-first order as we insert live successors. To
20647adc3a2bSChandler Carruth   // accomplish this, prioritizing for small iterations because we exit after
20657adc3a2bSChandler Carruth   // crossing our threshold, we use a small-size optimized SetVector.
20667adc3a2bSChandler Carruth   typedef SetVector<BasicBlock *, SmallVector<BasicBlock *, 16>,
2067567556aaSChad Rosier                     SmallPtrSet<BasicBlock *, 16>>
2068567556aaSChad Rosier       BBSetVector;
20697adc3a2bSChandler Carruth   BBSetVector BBWorklist;
20707adc3a2bSChandler Carruth   BBWorklist.insert(&F.getEntryBlock());
207106408758SMircea Trofin 
20727adc3a2bSChandler Carruth   // Note that we *must not* cache the size, this loop grows the worklist.
20737adc3a2bSChandler Carruth   for (unsigned Idx = 0; Idx != BBWorklist.size(); ++Idx) {
207406408758SMircea Trofin     if (shouldStop())
20757adc3a2bSChandler Carruth       break;
20767adc3a2bSChandler Carruth 
20777adc3a2bSChandler Carruth     BasicBlock *BB = BBWorklist[Idx];
20787adc3a2bSChandler Carruth     if (BB->empty())
20797adc3a2bSChandler Carruth       continue;
20807adc3a2bSChandler Carruth 
2081639b29b1SNick Desaulniers     // Disallow inlining a blockaddress with uses other than strictly callbr.
2082639b29b1SNick Desaulniers     // A blockaddress only has defined behavior for an indirect branch in the
2083639b29b1SNick Desaulniers     // same function, and we do not currently support inlining indirect
2084639b29b1SNick Desaulniers     // branches.  But, the inliner may not see an indirect branch that ends up
2085639b29b1SNick Desaulniers     // being dead code at a particular call site. If the blockaddress escapes
2086639b29b1SNick Desaulniers     // the function, e.g., via a global variable, inlining may lead to an
2087639b29b1SNick Desaulniers     // invalid cross-function reference.
2088639b29b1SNick Desaulniers     // FIXME: pr/39560: continue relaxing this overt restriction.
20896a84cd3bSNick Desaulniers     if (BB->hasAddressTaken())
2090639b29b1SNick Desaulniers       for (User *U : BlockAddress::get(&*BB)->users())
2091639b29b1SNick Desaulniers         if (!isa<CallBrInst>(*U))
20925466597fSMircea Trofin           return InlineResult::failure("blockaddress used outside of callbr");
20937adc3a2bSChandler Carruth 
20947adc3a2bSChandler Carruth     // Analyze the cost of this block. If we blow through the threshold, this
20957adc3a2bSChandler Carruth     // returns false, and we can bail on out.
2096c0aa4b75SDavid Bolvansky     InlineResult IR = analyzeBlock(BB, EphValues);
20975466597fSMircea Trofin     if (!IR.isSuccess())
2098c0aa4b75SDavid Bolvansky       return IR;
20997adc3a2bSChandler Carruth 
2100edb12a83SChandler Carruth     Instruction *TI = BB->getTerminator();
21017adc3a2bSChandler Carruth 
21027adc3a2bSChandler Carruth     // Add in the live successors by first checking whether we have terminator
21037adc3a2bSChandler Carruth     // that may be simplified based on the values simplified by this call.
21047adc3a2bSChandler Carruth     if (BranchInst *BI = dyn_cast<BranchInst>(TI)) {
21057adc3a2bSChandler Carruth       if (BI->isConditional()) {
21067adc3a2bSChandler Carruth         Value *Cond = BI->getCondition();
2107567556aaSChad Rosier         if (ConstantInt *SimpleCond =
2108567556aaSChad Rosier                 dyn_cast_or_null<ConstantInt>(SimplifiedValues.lookup(Cond))) {
21093739e14aSHaicheng Wu           BasicBlock *NextBB = BI->getSuccessor(SimpleCond->isZero() ? 1 : 0);
21103739e14aSHaicheng Wu           BBWorklist.insert(NextBB);
21113739e14aSHaicheng Wu           KnownSuccessors[BB] = NextBB;
21123739e14aSHaicheng Wu           findDeadBlocks(BB, NextBB);
21137adc3a2bSChandler Carruth           continue;
21147adc3a2bSChandler Carruth         }
21157adc3a2bSChandler Carruth       }
21167adc3a2bSChandler Carruth     } else if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) {
21177adc3a2bSChandler Carruth       Value *Cond = SI->getCondition();
2118567556aaSChad Rosier       if (ConstantInt *SimpleCond =
2119567556aaSChad Rosier               dyn_cast_or_null<ConstantInt>(SimplifiedValues.lookup(Cond))) {
21203739e14aSHaicheng Wu         BasicBlock *NextBB = SI->findCaseValue(SimpleCond)->getCaseSuccessor();
21213739e14aSHaicheng Wu         BBWorklist.insert(NextBB);
21223739e14aSHaicheng Wu         KnownSuccessors[BB] = NextBB;
21233739e14aSHaicheng Wu         findDeadBlocks(BB, NextBB);
21247adc3a2bSChandler Carruth         continue;
21257adc3a2bSChandler Carruth       }
21267adc3a2bSChandler Carruth     }
21277adc3a2bSChandler Carruth 
21287adc3a2bSChandler Carruth     // If we're unable to select a particular successor, just count all of
21297adc3a2bSChandler Carruth     // them.
21307adc3a2bSChandler Carruth     for (unsigned TIdx = 0, TSize = TI->getNumSuccessors(); TIdx != TSize;
21317adc3a2bSChandler Carruth          ++TIdx)
21327adc3a2bSChandler Carruth       BBWorklist.insert(TI->getSuccessor(TIdx));
21337adc3a2bSChandler Carruth 
213406408758SMircea Trofin     onBlockAnalyzed(BB);
21357adc3a2bSChandler Carruth   }
21367adc3a2bSChandler Carruth 
213793ac81ccSMircea Trofin   bool OnlyOneCallAndLocalLinkage = F.hasLocalLinkage() && F.hasOneUse() &&
213893ac81ccSMircea Trofin                                     &F == CandidateCall.getCalledFunction();
21397adc3a2bSChandler Carruth   // If this is a noduplicate call, we can still inline as long as
21407adc3a2bSChandler Carruth   // inlining this would cause the removal of the caller (so the instruction
21417adc3a2bSChandler Carruth   // is not actually duplicated, just moved).
21427adc3a2bSChandler Carruth   if (!OnlyOneCallAndLocalLinkage && ContainsNoDuplicateCall)
21435466597fSMircea Trofin     return InlineResult::failure("noduplicate");
21447adc3a2bSChandler Carruth 
214506408758SMircea Trofin   return finalizeAnalysis();
21467adc3a2bSChandler Carruth }
21477adc3a2bSChandler Carruth 
2148615eb470SAaron Ballman #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
21495f8f34e4SAdrian Prantl /// Dump stats about this call's analysis.
215006408758SMircea Trofin LLVM_DUMP_METHOD void InlineCostCallAnalyzer::dump() {
21517adc3a2bSChandler Carruth #define DEBUG_PRINT_STAT(x) dbgs() << "      " #x ": " << x << "\n"
2152c965fd94SKirill Naumov   if (PrintDebugInstructionDeltas)
2153c965fd94SKirill Naumov     F.print(dbgs(), &Writer);
21547adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumConstantArgs);
21557adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumConstantOffsetPtrArgs);
21567adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumAllocaArgs);
21577adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumConstantPtrCmps);
21587adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumConstantPtrDiffs);
21597adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumInstructionsSimplified);
21607adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(NumInstructions);
21617adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(SROACostSavings);
21627adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(SROACostSavingsLost);
2163a4461515SHaicheng Wu   DEBUG_PRINT_STAT(LoadEliminationCost);
21647adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(ContainsNoDuplicateCall);
21657adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(Cost);
21667adc3a2bSChandler Carruth   DEBUG_PRINT_STAT(Threshold);
21677adc3a2bSChandler Carruth #undef DEBUG_PRINT_STAT
21687adc3a2bSChandler Carruth }
21697adc3a2bSChandler Carruth #endif
21707adc3a2bSChandler Carruth 
21715f8f34e4SAdrian Prantl /// Test that there are no attribute conflicts between Caller and Callee
21727adc3a2bSChandler Carruth ///        that prevent inlining.
2173f9ca75f1STeresa Johnson static bool functionsHaveCompatibleAttributes(
2174f9ca75f1STeresa Johnson     Function *Caller, Function *Callee, TargetTransformInfo &TTI,
2175f9ca75f1STeresa Johnson     function_ref<const TargetLibraryInfo &(Function &)> &GetTLI) {
2176f9ca75f1STeresa Johnson   // Note that CalleeTLI must be a copy not a reference. The legacy pass manager
2177f9ca75f1STeresa Johnson   // caches the most recently created TLI in the TargetLibraryInfoWrapperPass
2178f9ca75f1STeresa Johnson   // object, and always returns the same object (which is overwritten on each
2179f9ca75f1STeresa Johnson   // GetTLI call). Therefore we copy the first result.
2180f9ca75f1STeresa Johnson   auto CalleeTLI = GetTLI(*Callee);
21817adc3a2bSChandler Carruth   return TTI.areInlineCompatible(Caller, Callee) &&
2182f9ca75f1STeresa Johnson          GetTLI(*Caller).areInlineCompatible(CalleeTLI,
2183ed03d948SMehdi Amini                                              InlineCallerSupersetNoBuiltin) &&
21841cb242ebSAkira Hatanaka          AttributeFuncs::areInlineCompatible(*Caller, *Callee);
21857adc3a2bSChandler Carruth }
21867adc3a2bSChandler Carruth 
2187652168a9SFedor Sergeev int llvm::getCallsiteCost(CallBase &Call, const DataLayout &DL) {
2188351d9b01SXinliang David Li   int Cost = 0;
2189652168a9SFedor Sergeev   for (unsigned I = 0, E = Call.arg_size(); I != E; ++I) {
2190652168a9SFedor Sergeev     if (Call.isByValArgument(I)) {
2191351d9b01SXinliang David Li       // We approximate the number of loads and stores needed by dividing the
2192351d9b01SXinliang David Li       // size of the byval type by the target's pointer size.
2193652168a9SFedor Sergeev       PointerType *PTy = cast<PointerType>(Call.getArgOperand(I)->getType());
2194351d9b01SXinliang David Li       unsigned TypeSize = DL.getTypeSizeInBits(PTy->getElementType());
219577f32994SBjorn Pettersson       unsigned AS = PTy->getAddressSpace();
219677f32994SBjorn Pettersson       unsigned PointerSize = DL.getPointerSizeInBits(AS);
2197351d9b01SXinliang David Li       // Ceiling division.
2198351d9b01SXinliang David Li       unsigned NumStores = (TypeSize + PointerSize - 1) / PointerSize;
2199351d9b01SXinliang David Li 
2200351d9b01SXinliang David Li       // If it generates more than 8 stores it is likely to be expanded as an
2201351d9b01SXinliang David Li       // inline memcpy so we take that as an upper bound. Otherwise we assume
2202351d9b01SXinliang David Li       // one load and one store per word copied.
2203351d9b01SXinliang David Li       // FIXME: The maxStoresPerMemcpy setting from the target should be used
2204351d9b01SXinliang David Li       // here instead of a magic number of 8, but it's not available via
2205351d9b01SXinliang David Li       // DataLayout.
2206351d9b01SXinliang David Li       NumStores = std::min(NumStores, 8U);
2207351d9b01SXinliang David Li 
2208351d9b01SXinliang David Li       Cost += 2 * NumStores * InlineConstants::InstrCost;
2209351d9b01SXinliang David Li     } else {
2210351d9b01SXinliang David Li       // For non-byval arguments subtract off one instruction per call
2211351d9b01SXinliang David Li       // argument.
2212351d9b01SXinliang David Li       Cost += InlineConstants::InstrCost;
2213351d9b01SXinliang David Li     }
2214351d9b01SXinliang David Li   }
2215351d9b01SXinliang David Li   // The call instruction also disappears after inlining.
2216351d9b01SXinliang David Li   Cost += InlineConstants::InstrCost + InlineConstants::CallPenalty;
2217351d9b01SXinliang David Li   return Cost;
2218351d9b01SXinliang David Li }
2219351d9b01SXinliang David Li 
2220ab6a6837SSean Silva InlineCost llvm::getInlineCost(
2221652168a9SFedor Sergeev     CallBase &Call, const InlineParams &Params, TargetTransformInfo &CalleeTTI,
2222aec2fa35SDaniel Jasper     std::function<AssumptionCache &(Function &)> &GetAssumptionCache,
222312585b01SEaswaran Raman     Optional<function_ref<BlockFrequencyInfo &(Function &)>> GetBFI,
2224f9ca75f1STeresa Johnson     function_ref<const TargetLibraryInfo &(Function &)> GetTLI,
22250812c5beSHaicheng Wu     ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE) {
2226652168a9SFedor Sergeev   return getInlineCost(Call, Call.getCalledFunction(), Params, CalleeTTI,
2227f9ca75f1STeresa Johnson                        GetAssumptionCache, GetBFI, GetTLI, PSI, ORE);
2228b9f7120eSEaswaran Raman }
2229b9f7120eSEaswaran Raman 
2230011a07c0SMircea Trofin Optional<int> llvm::getInliningCostEstimate(
22318a4013edSMircea Trofin     CallBase &Call, TargetTransformInfo &CalleeTTI,
22328a4013edSMircea Trofin     std::function<AssumptionCache &(Function &)> &GetAssumptionCache,
22338a4013edSMircea Trofin     Optional<function_ref<BlockFrequencyInfo &(Function &)>> GetBFI,
22348a4013edSMircea Trofin     ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE) {
22358a4013edSMircea Trofin   const InlineParams Params = {/* DefaultThreshold*/ 0,
22368a4013edSMircea Trofin                                /*HintThreshold*/ {},
22378a4013edSMircea Trofin                                /*ColdThreshold*/ {},
22388a4013edSMircea Trofin                                /*OptSizeThreshold*/ {},
22398a4013edSMircea Trofin                                /*OptMinSizeThreshold*/ {},
22408a4013edSMircea Trofin                                /*HotCallSiteThreshold*/ {},
22418a4013edSMircea Trofin                                /*LocallyHotCallSiteThreshold*/ {},
22428a4013edSMircea Trofin                                /*ColdCallSiteThreshold*/ {},
22438a4013edSMircea Trofin                                /* ComputeFullInlineCost*/ true};
22448a4013edSMircea Trofin 
22458a4013edSMircea Trofin   InlineCostCallAnalyzer CA(CalleeTTI, GetAssumptionCache, GetBFI, PSI, ORE,
22468a4013edSMircea Trofin                             *Call.getCalledFunction(), Call, Params, true,
22478a4013edSMircea Trofin                             /*IgnoreThreshold*/ true);
22488a4013edSMircea Trofin   auto R = CA.analyze();
22498a4013edSMircea Trofin   if (!R.isSuccess())
22508a4013edSMircea Trofin     return None;
22518a4013edSMircea Trofin   return CA.getCost();
22528a4013edSMircea Trofin }
22538a4013edSMircea Trofin 
2254201498c6SMircea Trofin Optional<InlineResult> llvm::getAttributeBasedInliningDecision(
2255201498c6SMircea Trofin     CallBase &Call, Function *Callee, TargetTransformInfo &CalleeTTI,
2256201498c6SMircea Trofin     function_ref<const TargetLibraryInfo &(Function &)> GetTLI) {
2257f4bb2f0dSEaswaran Raman 
22587adc3a2bSChandler Carruth   // Cannot inline indirect calls.
22597adc3a2bSChandler Carruth   if (!Callee)
2260201498c6SMircea Trofin     return InlineResult::failure("indirect call");
22617adc3a2bSChandler Carruth 
22623851496eSBjorn Pettersson   // Never inline calls with byval arguments that does not have the alloca
22633851496eSBjorn Pettersson   // address space. Since byval arguments can be replaced with a copy to an
22643851496eSBjorn Pettersson   // alloca, the inlined code would need to be adjusted to handle that the
22653851496eSBjorn Pettersson   // argument is in the alloca address space (so it is a little bit complicated
22663851496eSBjorn Pettersson   // to solve).
22673851496eSBjorn Pettersson   unsigned AllocaAS = Callee->getParent()->getDataLayout().getAllocaAddrSpace();
2268652168a9SFedor Sergeev   for (unsigned I = 0, E = Call.arg_size(); I != E; ++I)
2269652168a9SFedor Sergeev     if (Call.isByValArgument(I)) {
2270652168a9SFedor Sergeev       PointerType *PTy = cast<PointerType>(Call.getArgOperand(I)->getType());
22713851496eSBjorn Pettersson       if (PTy->getAddressSpace() != AllocaAS)
2272201498c6SMircea Trofin         return InlineResult::failure("byval arguments without alloca"
2273c0aa4b75SDavid Bolvansky                                      " address space");
22743851496eSBjorn Pettersson     }
22753851496eSBjorn Pettersson 
22767adc3a2bSChandler Carruth   // Calls to functions with always-inline attributes should be inlined
22777adc3a2bSChandler Carruth   // whenever possible.
2278652168a9SFedor Sergeev   if (Call.hasFnAttr(Attribute::AlwaysInline)) {
227915b17d0aSYevgeny Rouban     auto IsViable = isInlineViable(*Callee);
22805466597fSMircea Trofin     if (IsViable.isSuccess())
2281201498c6SMircea Trofin       return InlineResult::success();
2282201498c6SMircea Trofin     return InlineResult::failure(IsViable.getFailureReason());
22837adc3a2bSChandler Carruth   }
22847adc3a2bSChandler Carruth 
22857adc3a2bSChandler Carruth   // Never inline functions with conflicting attributes (unless callee has
22867adc3a2bSChandler Carruth   // always-inline attribute).
2287652168a9SFedor Sergeev   Function *Caller = Call.getCaller();
2288f9ca75f1STeresa Johnson   if (!functionsHaveCompatibleAttributes(Caller, Callee, CalleeTTI, GetTLI))
2289201498c6SMircea Trofin     return InlineResult::failure("conflicting attributes");
22907adc3a2bSChandler Carruth 
22917adc3a2bSChandler Carruth   // Don't inline this call if the caller has the optnone attribute.
229285bd3978SEvandro Menezes   if (Caller->hasOptNone())
2293201498c6SMircea Trofin     return InlineResult::failure("optnone attribute");
22947adc3a2bSChandler Carruth 
229577eeac3dSManoj Gupta   // Don't inline a function that treats null pointer as valid into a caller
229677eeac3dSManoj Gupta   // that does not have this attribute.
229777eeac3dSManoj Gupta   if (!Caller->nullPointerIsDefined() && Callee->nullPointerIsDefined())
2298201498c6SMircea Trofin     return InlineResult::failure("nullptr definitions incompatible");
229977eeac3dSManoj Gupta 
2300c0aa4b75SDavid Bolvansky   // Don't inline functions which can be interposed at link-time.
2301c0aa4b75SDavid Bolvansky   if (Callee->isInterposable())
2302201498c6SMircea Trofin     return InlineResult::failure("interposable");
2303c0aa4b75SDavid Bolvansky 
2304c0aa4b75SDavid Bolvansky   // Don't inline functions marked noinline.
2305c0aa4b75SDavid Bolvansky   if (Callee->hasFnAttribute(Attribute::NoInline))
2306201498c6SMircea Trofin     return InlineResult::failure("noinline function attribute");
2307c0aa4b75SDavid Bolvansky 
2308c0aa4b75SDavid Bolvansky   // Don't inline call sites marked noinline.
2309652168a9SFedor Sergeev   if (Call.isNoInline())
2310201498c6SMircea Trofin     return InlineResult::failure("noinline call site attribute");
2311201498c6SMircea Trofin 
2312201498c6SMircea Trofin   return None;
2313201498c6SMircea Trofin }
2314201498c6SMircea Trofin 
2315201498c6SMircea Trofin InlineCost llvm::getInlineCost(
2316201498c6SMircea Trofin     CallBase &Call, Function *Callee, const InlineParams &Params,
2317201498c6SMircea Trofin     TargetTransformInfo &CalleeTTI,
2318201498c6SMircea Trofin     std::function<AssumptionCache &(Function &)> &GetAssumptionCache,
2319201498c6SMircea Trofin     Optional<function_ref<BlockFrequencyInfo &(Function &)>> GetBFI,
2320201498c6SMircea Trofin     function_ref<const TargetLibraryInfo &(Function &)> GetTLI,
2321201498c6SMircea Trofin     ProfileSummaryInfo *PSI, OptimizationRemarkEmitter *ORE) {
2322201498c6SMircea Trofin 
2323201498c6SMircea Trofin   auto UserDecision =
2324201498c6SMircea Trofin       llvm::getAttributeBasedInliningDecision(Call, Callee, CalleeTTI, GetTLI);
2325201498c6SMircea Trofin 
2326201498c6SMircea Trofin   if (UserDecision.hasValue()) {
2327201498c6SMircea Trofin     if (UserDecision->isSuccess())
2328201498c6SMircea Trofin       return llvm::InlineCost::getAlways("always inline attribute");
2329201498c6SMircea Trofin     return llvm::InlineCost::getNever(UserDecision->getFailureReason());
2330201498c6SMircea Trofin   }
23317adc3a2bSChandler Carruth 
2332d34e60caSNicola Zaghen   LLVM_DEBUG(llvm::dbgs() << "      Analyzing call of " << Callee->getName()
2333201498c6SMircea Trofin                           << "... (caller:" << Call.getCaller()->getName()
2334201498c6SMircea Trofin                           << ")\n");
23357adc3a2bSChandler Carruth 
233606408758SMircea Trofin   InlineCostCallAnalyzer CA(CalleeTTI, GetAssumptionCache, GetBFI, PSI, ORE,
233706408758SMircea Trofin                             *Callee, Call, Params);
233893ac81ccSMircea Trofin   InlineResult ShouldInline = CA.analyze();
23397adc3a2bSChandler Carruth 
2340d34e60caSNicola Zaghen   LLVM_DEBUG(CA.dump());
23417adc3a2bSChandler Carruth 
23427adc3a2bSChandler Carruth   // Check if there was a reason to force inlining or no inlining.
23435466597fSMircea Trofin   if (!ShouldInline.isSuccess() && CA.getCost() < CA.getThreshold())
23445466597fSMircea Trofin     return InlineCost::getNever(ShouldInline.getFailureReason());
23455466597fSMircea Trofin   if (ShouldInline.isSuccess() && CA.getCost() >= CA.getThreshold())
2346c0aa4b75SDavid Bolvansky     return InlineCost::getAlways("empty function");
23477adc3a2bSChandler Carruth 
23487adc3a2bSChandler Carruth   return llvm::InlineCost::get(CA.getCost(), CA.getThreshold());
23497adc3a2bSChandler Carruth }
23507adc3a2bSChandler Carruth 
235115b17d0aSYevgeny Rouban InlineResult llvm::isInlineViable(Function &F) {
23527adc3a2bSChandler Carruth   bool ReturnsTwice = F.hasFnAttribute(Attribute::ReturnsTwice);
23537adc3a2bSChandler Carruth   for (Function::iterator BI = F.begin(), BE = F.end(); BI != BE; ++BI) {
2354639b29b1SNick Desaulniers     // Disallow inlining of functions which contain indirect branches.
235515b17d0aSYevgeny Rouban     if (isa<IndirectBrInst>(BI->getTerminator()))
23565466597fSMircea Trofin       return InlineResult::failure("contains indirect branches");
235715b17d0aSYevgeny Rouban 
2358639b29b1SNick Desaulniers     // Disallow inlining of blockaddresses which are used by non-callbr
2359639b29b1SNick Desaulniers     // instructions.
23606a84cd3bSNick Desaulniers     if (BI->hasAddressTaken())
2361639b29b1SNick Desaulniers       for (User *U : BlockAddress::get(&*BI)->users())
2362639b29b1SNick Desaulniers         if (!isa<CallBrInst>(*U))
23635466597fSMircea Trofin           return InlineResult::failure("blockaddress used outside of callbr");
23647adc3a2bSChandler Carruth 
23655a82c916SDuncan P. N. Exon Smith     for (auto &II : *BI) {
2366652168a9SFedor Sergeev       CallBase *Call = dyn_cast<CallBase>(&II);
2367652168a9SFedor Sergeev       if (!Call)
23687adc3a2bSChandler Carruth         continue;
23697adc3a2bSChandler Carruth 
23707adc3a2bSChandler Carruth       // Disallow recursive calls.
2371652168a9SFedor Sergeev       if (&F == Call->getCalledFunction())
23725466597fSMircea Trofin         return InlineResult::failure("recursive call");
23737adc3a2bSChandler Carruth 
23747adc3a2bSChandler Carruth       // Disallow calls which expose returns-twice to a function not previously
23757adc3a2bSChandler Carruth       // attributed as such.
2376652168a9SFedor Sergeev       if (!ReturnsTwice && isa<CallInst>(Call) &&
2377652168a9SFedor Sergeev           cast<CallInst>(Call)->canReturnTwice())
23785466597fSMircea Trofin         return InlineResult::failure("exposes returns-twice attribute");
23797adc3a2bSChandler Carruth 
2380652168a9SFedor Sergeev       if (Call->getCalledFunction())
2381652168a9SFedor Sergeev         switch (Call->getCalledFunction()->getIntrinsicID()) {
23821636651eSFlorian Hahn         default:
23831636651eSFlorian Hahn           break;
238406408758SMircea Trofin         case llvm::Intrinsic::icall_branch_funnel:
23854296ea72SVitaly Buka           // Disallow inlining of @llvm.icall.branch.funnel because current
23864296ea72SVitaly Buka           // backend can't separate call targets from call arguments.
23875466597fSMircea Trofin           return InlineResult::failure(
23885466597fSMircea Trofin               "disallowed inlining of @llvm.icall.branch.funnel");
238906408758SMircea Trofin         case llvm::Intrinsic::localescape:
23907adc3a2bSChandler Carruth           // Disallow inlining functions that call @llvm.localescape. Doing this
23917adc3a2bSChandler Carruth           // correctly would require major changes to the inliner.
23925466597fSMircea Trofin           return InlineResult::failure(
23935466597fSMircea Trofin               "disallowed inlining of @llvm.localescape");
23941636651eSFlorian Hahn         case llvm::Intrinsic::vastart:
239506408758SMircea Trofin           // Disallow inlining of functions that initialize VarArgs with
239606408758SMircea Trofin           // va_start.
23975466597fSMircea Trofin           return InlineResult::failure(
23985466597fSMircea Trofin               "contains VarArgs initialized with va_start");
23997adc3a2bSChandler Carruth         }
24007adc3a2bSChandler Carruth     }
24011636651eSFlorian Hahn   }
24027adc3a2bSChandler Carruth 
24035466597fSMircea Trofin   return InlineResult::success();
24047adc3a2bSChandler Carruth }
24051c57cc2bSEaswaran Raman 
24061c57cc2bSEaswaran Raman // APIs to create InlineParams based on command line flags and/or other
24071c57cc2bSEaswaran Raman // parameters.
24081c57cc2bSEaswaran Raman 
24091c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams(int Threshold) {
24101c57cc2bSEaswaran Raman   InlineParams Params;
24111c57cc2bSEaswaran Raman 
24121c57cc2bSEaswaran Raman   // This field is the threshold to use for a callee by default. This is
24131c57cc2bSEaswaran Raman   // derived from one or more of:
24141c57cc2bSEaswaran Raman   //  * optimization or size-optimization levels,
24151c57cc2bSEaswaran Raman   //  * a value passed to createFunctionInliningPass function, or
24161c57cc2bSEaswaran Raman   //  * the -inline-threshold flag.
24171c57cc2bSEaswaran Raman   //  If the -inline-threshold flag is explicitly specified, that is used
24181c57cc2bSEaswaran Raman   //  irrespective of anything else.
24191c57cc2bSEaswaran Raman   if (InlineThreshold.getNumOccurrences() > 0)
24201c57cc2bSEaswaran Raman     Params.DefaultThreshold = InlineThreshold;
24211c57cc2bSEaswaran Raman   else
24221c57cc2bSEaswaran Raman     Params.DefaultThreshold = Threshold;
24231c57cc2bSEaswaran Raman 
24241c57cc2bSEaswaran Raman   // Set the HintThreshold knob from the -inlinehint-threshold.
24251c57cc2bSEaswaran Raman   Params.HintThreshold = HintThreshold;
24261c57cc2bSEaswaran Raman 
24271c57cc2bSEaswaran Raman   // Set the HotCallSiteThreshold knob from the -hot-callsite-threshold.
24281c57cc2bSEaswaran Raman   Params.HotCallSiteThreshold = HotCallSiteThreshold;
24291c57cc2bSEaswaran Raman 
2430974d4eeaSEaswaran Raman   // If the -locally-hot-callsite-threshold is explicitly specified, use it to
2431974d4eeaSEaswaran Raman   // populate LocallyHotCallSiteThreshold. Later, we populate
2432974d4eeaSEaswaran Raman   // Params.LocallyHotCallSiteThreshold from -locally-hot-callsite-threshold if
2433974d4eeaSEaswaran Raman   // we know that optimization level is O3 (in the getInlineParams variant that
2434974d4eeaSEaswaran Raman   // takes the opt and size levels).
2435974d4eeaSEaswaran Raman   // FIXME: Remove this check (and make the assignment unconditional) after
2436974d4eeaSEaswaran Raman   // addressing size regression issues at O2.
2437974d4eeaSEaswaran Raman   if (LocallyHotCallSiteThreshold.getNumOccurrences() > 0)
2438974d4eeaSEaswaran Raman     Params.LocallyHotCallSiteThreshold = LocallyHotCallSiteThreshold;
2439974d4eeaSEaswaran Raman 
24405a956c85SMircea Trofin   // Set the ColdCallSiteThreshold knob from the
24415a956c85SMircea Trofin   // -inline-cold-callsite-threshold.
244212585b01SEaswaran Raman   Params.ColdCallSiteThreshold = ColdCallSiteThreshold;
244312585b01SEaswaran Raman 
24441c57cc2bSEaswaran Raman   // Set the OptMinSizeThreshold and OptSizeThreshold params only if the
24451c57cc2bSEaswaran Raman   // -inlinehint-threshold commandline option is not explicitly given. If that
24461c57cc2bSEaswaran Raman   // option is present, then its value applies even for callees with size and
24471c57cc2bSEaswaran Raman   // minsize attributes.
24481c57cc2bSEaswaran Raman   // If the -inline-threshold is not specified, set the ColdThreshold from the
24491c57cc2bSEaswaran Raman   // -inlinecold-threshold even if it is not explicitly passed. If
24501c57cc2bSEaswaran Raman   // -inline-threshold is specified, then -inlinecold-threshold needs to be
24511c57cc2bSEaswaran Raman   // explicitly specified to set the ColdThreshold knob
24521c57cc2bSEaswaran Raman   if (InlineThreshold.getNumOccurrences() == 0) {
24531c57cc2bSEaswaran Raman     Params.OptMinSizeThreshold = InlineConstants::OptMinSizeThreshold;
24541c57cc2bSEaswaran Raman     Params.OptSizeThreshold = InlineConstants::OptSizeThreshold;
24551c57cc2bSEaswaran Raman     Params.ColdThreshold = ColdThreshold;
24561c57cc2bSEaswaran Raman   } else if (ColdThreshold.getNumOccurrences() > 0) {
24571c57cc2bSEaswaran Raman     Params.ColdThreshold = ColdThreshold;
24581c57cc2bSEaswaran Raman   }
24591c57cc2bSEaswaran Raman   return Params;
24601c57cc2bSEaswaran Raman }
24611c57cc2bSEaswaran Raman 
24621c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams() {
24637f37a802STeresa Johnson   return getInlineParams(DefaultThreshold);
24641c57cc2bSEaswaran Raman }
24651c57cc2bSEaswaran Raman 
24661c57cc2bSEaswaran Raman // Compute the default threshold for inlining based on the opt level and the
24671c57cc2bSEaswaran Raman // size opt level.
24681c57cc2bSEaswaran Raman static int computeThresholdFromOptLevels(unsigned OptLevel,
24691c57cc2bSEaswaran Raman                                          unsigned SizeOptLevel) {
24701c57cc2bSEaswaran Raman   if (OptLevel > 2)
24711c57cc2bSEaswaran Raman     return InlineConstants::OptAggressiveThreshold;
24721c57cc2bSEaswaran Raman   if (SizeOptLevel == 1) // -Os
24731c57cc2bSEaswaran Raman     return InlineConstants::OptSizeThreshold;
24741c57cc2bSEaswaran Raman   if (SizeOptLevel == 2) // -Oz
24751c57cc2bSEaswaran Raman     return InlineConstants::OptMinSizeThreshold;
24767f37a802STeresa Johnson   return DefaultThreshold;
24771c57cc2bSEaswaran Raman }
24781c57cc2bSEaswaran Raman 
24791c57cc2bSEaswaran Raman InlineParams llvm::getInlineParams(unsigned OptLevel, unsigned SizeOptLevel) {
2480974d4eeaSEaswaran Raman   auto Params =
2481974d4eeaSEaswaran Raman       getInlineParams(computeThresholdFromOptLevels(OptLevel, SizeOptLevel));
2482974d4eeaSEaswaran Raman   // At O3, use the value of -locally-hot-callsite-threshold option to populate
2483974d4eeaSEaswaran Raman   // Params.LocallyHotCallSiteThreshold. Below O3, this flag has effect only
2484974d4eeaSEaswaran Raman   // when it is specified explicitly.
2485974d4eeaSEaswaran Raman   if (OptLevel > 2)
2486974d4eeaSEaswaran Raman     Params.LocallyHotCallSiteThreshold = LocallyHotCallSiteThreshold;
2487974d4eeaSEaswaran Raman   return Params;
24881c57cc2bSEaswaran Raman }
2489