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