forked from OSchip/llvm-project
1240 lines
51 KiB
C++
1240 lines
51 KiB
C++
//===- Inliner.cpp - Code common to all inliners --------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the mechanics required to implement inlining without
|
|
// missing any calls and updating the call graph. The decisions of which calls
|
|
// are profitable to inline are implemented elsewhere.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Transforms/IPO/Inliner.h"
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include "llvm/ADT/None.h"
|
|
#include "llvm/ADT/Optional.h"
|
|
#include "llvm/ADT/STLExtras.h"
|
|
#include "llvm/ADT/SetVector.h"
|
|
#include "llvm/ADT/SmallPtrSet.h"
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/ADT/Statistic.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Analysis/AliasAnalysis.h"
|
|
#include "llvm/Analysis/AssumptionCache.h"
|
|
#include "llvm/Analysis/BasicAliasAnalysis.h"
|
|
#include "llvm/Analysis/BlockFrequencyInfo.h"
|
|
#include "llvm/Analysis/CGSCCPassManager.h"
|
|
#include "llvm/Analysis/CallGraph.h"
|
|
#include "llvm/Analysis/InlineCost.h"
|
|
#include "llvm/Analysis/LazyCallGraph.h"
|
|
#include "llvm/Analysis/OptimizationRemarkEmitter.h"
|
|
#include "llvm/Analysis/ProfileSummaryInfo.h"
|
|
#include "llvm/Analysis/TargetLibraryInfo.h"
|
|
#include "llvm/Analysis/TargetTransformInfo.h"
|
|
#include "llvm/Transforms/Utils/Local.h"
|
|
#include "llvm/IR/Attributes.h"
|
|
#include "llvm/IR/BasicBlock.h"
|
|
#include "llvm/IR/CallSite.h"
|
|
#include "llvm/IR/DataLayout.h"
|
|
#include "llvm/IR/DebugLoc.h"
|
|
#include "llvm/IR/DerivedTypes.h"
|
|
#include "llvm/IR/DiagnosticInfo.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/InstIterator.h"
|
|
#include "llvm/IR/Instruction.h"
|
|
#include "llvm/IR/Instructions.h"
|
|
#include "llvm/IR/IntrinsicInst.h"
|
|
#include "llvm/IR/Metadata.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/IR/PassManager.h"
|
|
#include "llvm/IR/User.h"
|
|
#include "llvm/IR/Value.h"
|
|
#include "llvm/Pass.h"
|
|
#include "llvm/Support/Casting.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "llvm/Transforms/Utils/Cloning.h"
|
|
#include "llvm/Transforms/Utils/ImportedFunctionsInliningStatistics.h"
|
|
#include "llvm/Transforms/Utils/ModuleUtils.h"
|
|
#include <algorithm>
|
|
#include <cassert>
|
|
#include <functional>
|
|
#include <sstream>
|
|
#include <tuple>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
using namespace llvm;
|
|
|
|
#define DEBUG_TYPE "inline"
|
|
|
|
STATISTIC(NumInlined, "Number of functions inlined");
|
|
STATISTIC(NumCallsDeleted, "Number of call sites deleted, not inlined");
|
|
STATISTIC(NumDeleted, "Number of functions deleted because all callers found");
|
|
STATISTIC(NumMergedAllocas, "Number of allocas merged together");
|
|
|
|
// This weirdly named statistic tracks the number of times that, when attempting
|
|
// to inline a function A into B, we analyze the callers of B in order to see
|
|
// if those would be more profitable and blocked inline steps.
|
|
STATISTIC(NumCallerCallersAnalyzed, "Number of caller-callers analyzed");
|
|
|
|
/// Flag to disable manual alloca merging.
|
|
///
|
|
/// Merging of allocas was originally done as a stack-size saving technique
|
|
/// prior to LLVM's code generator having support for stack coloring based on
|
|
/// lifetime markers. It is now in the process of being removed. To experiment
|
|
/// with disabling it and relying fully on lifetime marker based stack
|
|
/// coloring, you can pass this flag to LLVM.
|
|
static cl::opt<bool>
|
|
DisableInlinedAllocaMerging("disable-inlined-alloca-merging",
|
|
cl::init(false), cl::Hidden);
|
|
|
|
namespace {
|
|
|
|
enum class InlinerFunctionImportStatsOpts {
|
|
No = 0,
|
|
Basic = 1,
|
|
Verbose = 2,
|
|
};
|
|
|
|
} // end anonymous namespace
|
|
|
|
static cl::opt<InlinerFunctionImportStatsOpts> InlinerFunctionImportStats(
|
|
"inliner-function-import-stats",
|
|
cl::init(InlinerFunctionImportStatsOpts::No),
|
|
cl::values(clEnumValN(InlinerFunctionImportStatsOpts::Basic, "basic",
|
|
"basic statistics"),
|
|
clEnumValN(InlinerFunctionImportStatsOpts::Verbose, "verbose",
|
|
"printing of statistics for each inlined function")),
|
|
cl::Hidden, cl::desc("Enable inliner stats for imported functions"));
|
|
|
|
/// Flag to add inline messages as callsite attributes 'inline-remark'.
|
|
static cl::opt<bool>
|
|
InlineRemarkAttribute("inline-remark-attribute", cl::init(false),
|
|
cl::Hidden,
|
|
cl::desc("Enable adding inline-remark attribute to"
|
|
" callsites processed by inliner but decided"
|
|
" to be not inlined"));
|
|
|
|
LegacyInlinerBase::LegacyInlinerBase(char &ID) : CallGraphSCCPass(ID) {}
|
|
|
|
LegacyInlinerBase::LegacyInlinerBase(char &ID, bool InsertLifetime)
|
|
: CallGraphSCCPass(ID), InsertLifetime(InsertLifetime) {}
|
|
|
|
/// For this class, we declare that we require and preserve the call graph.
|
|
/// If the derived class implements this method, it should
|
|
/// always explicitly call the implementation here.
|
|
void LegacyInlinerBase::getAnalysisUsage(AnalysisUsage &AU) const {
|
|
AU.addRequired<AssumptionCacheTracker>();
|
|
AU.addRequired<ProfileSummaryInfoWrapperPass>();
|
|
AU.addRequired<TargetLibraryInfoWrapperPass>();
|
|
getAAResultsAnalysisUsage(AU);
|
|
CallGraphSCCPass::getAnalysisUsage(AU);
|
|
}
|
|
|
|
using InlinedArrayAllocasTy = DenseMap<ArrayType *, std::vector<AllocaInst *>>;
|
|
|
|
/// Look at all of the allocas that we inlined through this call site. If we
|
|
/// have already inlined other allocas through other calls into this function,
|
|
/// then we know that they have disjoint lifetimes and that we can merge them.
|
|
///
|
|
/// There are many heuristics possible for merging these allocas, and the
|
|
/// different options have different tradeoffs. One thing that we *really*
|
|
/// don't want to hurt is SRoA: once inlining happens, often allocas are no
|
|
/// longer address taken and so they can be promoted.
|
|
///
|
|
/// Our "solution" for that is to only merge allocas whose outermost type is an
|
|
/// array type. These are usually not promoted because someone is using a
|
|
/// variable index into them. These are also often the most important ones to
|
|
/// merge.
|
|
///
|
|
/// A better solution would be to have real memory lifetime markers in the IR
|
|
/// and not have the inliner do any merging of allocas at all. This would
|
|
/// allow the backend to do proper stack slot coloring of all allocas that
|
|
/// *actually make it to the backend*, which is really what we want.
|
|
///
|
|
/// Because we don't have this information, we do this simple and useful hack.
|
|
static void mergeInlinedArrayAllocas(
|
|
Function *Caller, InlineFunctionInfo &IFI,
|
|
InlinedArrayAllocasTy &InlinedArrayAllocas, int InlineHistory) {
|
|
SmallPtrSet<AllocaInst *, 16> UsedAllocas;
|
|
|
|
// When processing our SCC, check to see if CS was inlined from some other
|
|
// call site. For example, if we're processing "A" in this code:
|
|
// A() { B() }
|
|
// B() { x = alloca ... C() }
|
|
// C() { y = alloca ... }
|
|
// Assume that C was not inlined into B initially, and so we're processing A
|
|
// and decide to inline B into A. Doing this makes an alloca available for
|
|
// reuse and makes a callsite (C) available for inlining. When we process
|
|
// the C call site we don't want to do any alloca merging between X and Y
|
|
// because their scopes are not disjoint. We could make this smarter by
|
|
// keeping track of the inline history for each alloca in the
|
|
// InlinedArrayAllocas but this isn't likely to be a significant win.
|
|
if (InlineHistory != -1) // Only do merging for top-level call sites in SCC.
|
|
return;
|
|
|
|
// Loop over all the allocas we have so far and see if they can be merged with
|
|
// a previously inlined alloca. If not, remember that we had it.
|
|
for (unsigned AllocaNo = 0, e = IFI.StaticAllocas.size(); AllocaNo != e;
|
|
++AllocaNo) {
|
|
AllocaInst *AI = IFI.StaticAllocas[AllocaNo];
|
|
|
|
// Don't bother trying to merge array allocations (they will usually be
|
|
// canonicalized to be an allocation *of* an array), or allocations whose
|
|
// type is not itself an array (because we're afraid of pessimizing SRoA).
|
|
ArrayType *ATy = dyn_cast<ArrayType>(AI->getAllocatedType());
|
|
if (!ATy || AI->isArrayAllocation())
|
|
continue;
|
|
|
|
// Get the list of all available allocas for this array type.
|
|
std::vector<AllocaInst *> &AllocasForType = InlinedArrayAllocas[ATy];
|
|
|
|
// Loop over the allocas in AllocasForType to see if we can reuse one. Note
|
|
// that we have to be careful not to reuse the same "available" alloca for
|
|
// multiple different allocas that we just inlined, we use the 'UsedAllocas'
|
|
// set to keep track of which "available" allocas are being used by this
|
|
// function. Also, AllocasForType can be empty of course!
|
|
bool MergedAwayAlloca = false;
|
|
for (AllocaInst *AvailableAlloca : AllocasForType) {
|
|
unsigned Align1 = AI->getAlignment(),
|
|
Align2 = AvailableAlloca->getAlignment();
|
|
|
|
// The available alloca has to be in the right function, not in some other
|
|
// function in this SCC.
|
|
if (AvailableAlloca->getParent() != AI->getParent())
|
|
continue;
|
|
|
|
// If the inlined function already uses this alloca then we can't reuse
|
|
// it.
|
|
if (!UsedAllocas.insert(AvailableAlloca).second)
|
|
continue;
|
|
|
|
// Otherwise, we *can* reuse it, RAUW AI into AvailableAlloca and declare
|
|
// success!
|
|
LLVM_DEBUG(dbgs() << " ***MERGED ALLOCA: " << *AI
|
|
<< "\n\t\tINTO: " << *AvailableAlloca << '\n');
|
|
|
|
// Move affected dbg.declare calls immediately after the new alloca to
|
|
// avoid the situation when a dbg.declare precedes its alloca.
|
|
if (auto *L = LocalAsMetadata::getIfExists(AI))
|
|
if (auto *MDV = MetadataAsValue::getIfExists(AI->getContext(), L))
|
|
for (User *U : MDV->users())
|
|
if (DbgDeclareInst *DDI = dyn_cast<DbgDeclareInst>(U))
|
|
DDI->moveBefore(AvailableAlloca->getNextNode());
|
|
|
|
AI->replaceAllUsesWith(AvailableAlloca);
|
|
|
|
if (Align1 != Align2) {
|
|
if (!Align1 || !Align2) {
|
|
const DataLayout &DL = Caller->getParent()->getDataLayout();
|
|
unsigned TypeAlign = DL.getABITypeAlignment(AI->getAllocatedType());
|
|
|
|
Align1 = Align1 ? Align1 : TypeAlign;
|
|
Align2 = Align2 ? Align2 : TypeAlign;
|
|
}
|
|
|
|
if (Align1 > Align2)
|
|
AvailableAlloca->setAlignment(MaybeAlign(AI->getAlignment()));
|
|
}
|
|
|
|
AI->eraseFromParent();
|
|
MergedAwayAlloca = true;
|
|
++NumMergedAllocas;
|
|
IFI.StaticAllocas[AllocaNo] = nullptr;
|
|
break;
|
|
}
|
|
|
|
// If we already nuked the alloca, we're done with it.
|
|
if (MergedAwayAlloca)
|
|
continue;
|
|
|
|
// If we were unable to merge away the alloca either because there are no
|
|
// allocas of the right type available or because we reused them all
|
|
// already, remember that this alloca came from an inlined function and mark
|
|
// it used so we don't reuse it for other allocas from this inline
|
|
// operation.
|
|
AllocasForType.push_back(AI);
|
|
UsedAllocas.insert(AI);
|
|
}
|
|
}
|
|
|
|
/// If it is possible to inline the specified call site,
|
|
/// do so and update the CallGraph for this operation.
|
|
///
|
|
/// This function also does some basic book-keeping to update the IR. The
|
|
/// InlinedArrayAllocas map keeps track of any allocas that are already
|
|
/// available from other functions inlined into the caller. If we are able to
|
|
/// inline this call site we attempt to reuse already available allocas or add
|
|
/// any new allocas to the set if not possible.
|
|
static InlineResult InlineCallIfPossible(
|
|
CallSite CS, InlineFunctionInfo &IFI,
|
|
InlinedArrayAllocasTy &InlinedArrayAllocas, int InlineHistory,
|
|
bool InsertLifetime, function_ref<AAResults &(Function &)> &AARGetter,
|
|
ImportedFunctionsInliningStatistics &ImportedFunctionsStats) {
|
|
Function *Callee = CS.getCalledFunction();
|
|
Function *Caller = CS.getCaller();
|
|
|
|
AAResults &AAR = AARGetter(*Callee);
|
|
|
|
// Try to inline the function. Get the list of static allocas that were
|
|
// inlined.
|
|
InlineResult IR = InlineFunction(CS, IFI, &AAR, InsertLifetime);
|
|
if (!IR)
|
|
return IR;
|
|
|
|
if (InlinerFunctionImportStats != InlinerFunctionImportStatsOpts::No)
|
|
ImportedFunctionsStats.recordInline(*Caller, *Callee);
|
|
|
|
AttributeFuncs::mergeAttributesForInlining(*Caller, *Callee);
|
|
|
|
if (!DisableInlinedAllocaMerging)
|
|
mergeInlinedArrayAllocas(Caller, IFI, InlinedArrayAllocas, InlineHistory);
|
|
|
|
return IR; // success
|
|
}
|
|
|
|
/// Return true if inlining of CS can block the caller from being
|
|
/// inlined which is proved to be more beneficial. \p IC is the
|
|
/// estimated inline cost associated with callsite \p CS.
|
|
/// \p TotalSecondaryCost will be set to the estimated cost of inlining the
|
|
/// caller if \p CS is suppressed for inlining.
|
|
static bool
|
|
shouldBeDeferred(Function *Caller, CallSite CS, InlineCost IC,
|
|
int &TotalSecondaryCost,
|
|
function_ref<InlineCost(CallSite CS)> GetInlineCost) {
|
|
// For now we only handle local or inline functions.
|
|
if (!Caller->hasLocalLinkage() && !Caller->hasLinkOnceODRLinkage())
|
|
return false;
|
|
// If the cost of inlining CS is non-positive, it is not going to prevent the
|
|
// caller from being inlined into its callers and hence we don't need to
|
|
// defer.
|
|
if (IC.getCost() <= 0)
|
|
return false;
|
|
// Try to detect the case where the current inlining candidate caller (call
|
|
// it B) is a static or linkonce-ODR function and is an inlining candidate
|
|
// elsewhere, and the current candidate callee (call it C) is large enough
|
|
// that inlining it into B would make B too big to inline later. In these
|
|
// circumstances it may be best not to inline C into B, but to inline B into
|
|
// its callers.
|
|
//
|
|
// This only applies to static and linkonce-ODR functions because those are
|
|
// expected to be available for inlining in the translation units where they
|
|
// are used. Thus we will always have the opportunity to make local inlining
|
|
// decisions. Importantly the linkonce-ODR linkage covers inline functions
|
|
// and templates in C++.
|
|
//
|
|
// FIXME: All of this logic should be sunk into getInlineCost. It relies on
|
|
// the internal implementation of the inline cost metrics rather than
|
|
// treating them as truly abstract units etc.
|
|
TotalSecondaryCost = 0;
|
|
// The candidate cost to be imposed upon the current function.
|
|
int CandidateCost = IC.getCost() - 1;
|
|
// If the caller has local linkage and can be inlined to all its callers, we
|
|
// can apply a huge negative bonus to TotalSecondaryCost.
|
|
bool ApplyLastCallBonus = Caller->hasLocalLinkage() && !Caller->hasOneUse();
|
|
// This bool tracks what happens if we DO inline C into B.
|
|
bool inliningPreventsSomeOuterInline = false;
|
|
for (User *U : Caller->users()) {
|
|
// If the caller will not be removed (either because it does not have a
|
|
// local linkage or because the LastCallToStaticBonus has been already
|
|
// applied), then we can exit the loop early.
|
|
if (!ApplyLastCallBonus && TotalSecondaryCost >= IC.getCost())
|
|
return false;
|
|
CallSite CS2(U);
|
|
|
|
// If this isn't a call to Caller (it could be some other sort
|
|
// of reference) skip it. Such references will prevent the caller
|
|
// from being removed.
|
|
if (!CS2 || CS2.getCalledFunction() != Caller) {
|
|
ApplyLastCallBonus = false;
|
|
continue;
|
|
}
|
|
|
|
InlineCost IC2 = GetInlineCost(CS2);
|
|
++NumCallerCallersAnalyzed;
|
|
if (!IC2) {
|
|
ApplyLastCallBonus = false;
|
|
continue;
|
|
}
|
|
if (IC2.isAlways())
|
|
continue;
|
|
|
|
// See if inlining of the original callsite would erase the cost delta of
|
|
// this callsite. We subtract off the penalty for the call instruction,
|
|
// which we would be deleting.
|
|
if (IC2.getCostDelta() <= CandidateCost) {
|
|
inliningPreventsSomeOuterInline = true;
|
|
TotalSecondaryCost += IC2.getCost();
|
|
}
|
|
}
|
|
// If all outer calls to Caller would get inlined, the cost for the last
|
|
// one is set very low by getInlineCost, in anticipation that Caller will
|
|
// be removed entirely. We did not account for this above unless there
|
|
// is only one caller of Caller.
|
|
if (ApplyLastCallBonus)
|
|
TotalSecondaryCost -= InlineConstants::LastCallToStaticBonus;
|
|
|
|
if (inliningPreventsSomeOuterInline && TotalSecondaryCost < IC.getCost())
|
|
return true;
|
|
|
|
return false;
|
|
}
|
|
|
|
static std::basic_ostream<char> &operator<<(std::basic_ostream<char> &R,
|
|
const ore::NV &Arg) {
|
|
return R << Arg.Val;
|
|
}
|
|
|
|
template <class RemarkT>
|
|
RemarkT &operator<<(RemarkT &&R, const InlineCost &IC) {
|
|
using namespace ore;
|
|
if (IC.isAlways()) {
|
|
R << "(cost=always)";
|
|
} else if (IC.isNever()) {
|
|
R << "(cost=never)";
|
|
} else {
|
|
R << "(cost=" << ore::NV("Cost", IC.getCost())
|
|
<< ", threshold=" << ore::NV("Threshold", IC.getThreshold()) << ")";
|
|
}
|
|
if (const char *Reason = IC.getReason())
|
|
R << ": " << ore::NV("Reason", Reason);
|
|
return R;
|
|
}
|
|
|
|
static std::string inlineCostStr(const InlineCost &IC) {
|
|
std::stringstream Remark;
|
|
Remark << IC;
|
|
return Remark.str();
|
|
}
|
|
|
|
/// Return the cost only if the inliner should attempt to inline at the given
|
|
/// CallSite. If we return the cost, we will emit an optimisation remark later
|
|
/// using that cost, so we won't do so from this function.
|
|
static Optional<InlineCost>
|
|
shouldInline(CallSite CS, function_ref<InlineCost(CallSite CS)> GetInlineCost,
|
|
OptimizationRemarkEmitter &ORE) {
|
|
using namespace ore;
|
|
|
|
InlineCost IC = GetInlineCost(CS);
|
|
Instruction *Call = CS.getInstruction();
|
|
Function *Callee = CS.getCalledFunction();
|
|
Function *Caller = CS.getCaller();
|
|
|
|
if (IC.isAlways()) {
|
|
LLVM_DEBUG(dbgs() << " Inlining " << inlineCostStr(IC)
|
|
<< ", Call: " << *CS.getInstruction() << "\n");
|
|
return IC;
|
|
}
|
|
|
|
if (IC.isNever()) {
|
|
LLVM_DEBUG(dbgs() << " NOT Inlining " << inlineCostStr(IC)
|
|
<< ", Call: " << *CS.getInstruction() << "\n");
|
|
ORE.emit([&]() {
|
|
return OptimizationRemarkMissed(DEBUG_TYPE, "NeverInline", Call)
|
|
<< NV("Callee", Callee) << " not inlined into "
|
|
<< NV("Caller", Caller) << " because it should never be inlined "
|
|
<< IC;
|
|
});
|
|
return IC;
|
|
}
|
|
|
|
if (!IC) {
|
|
LLVM_DEBUG(dbgs() << " NOT Inlining " << inlineCostStr(IC)
|
|
<< ", Call: " << *CS.getInstruction() << "\n");
|
|
ORE.emit([&]() {
|
|
return OptimizationRemarkMissed(DEBUG_TYPE, "TooCostly", Call)
|
|
<< NV("Callee", Callee) << " not inlined into "
|
|
<< NV("Caller", Caller) << " because too costly to inline " << IC;
|
|
});
|
|
return IC;
|
|
}
|
|
|
|
int TotalSecondaryCost = 0;
|
|
if (shouldBeDeferred(Caller, CS, IC, TotalSecondaryCost, GetInlineCost)) {
|
|
LLVM_DEBUG(dbgs() << " NOT Inlining: " << *CS.getInstruction()
|
|
<< " Cost = " << IC.getCost()
|
|
<< ", outer Cost = " << TotalSecondaryCost << '\n');
|
|
ORE.emit([&]() {
|
|
return OptimizationRemarkMissed(DEBUG_TYPE, "IncreaseCostInOtherContexts",
|
|
Call)
|
|
<< "Not inlining. Cost of inlining " << NV("Callee", Callee)
|
|
<< " increases the cost of inlining " << NV("Caller", Caller)
|
|
<< " in other contexts";
|
|
});
|
|
|
|
// IC does not bool() to false, so get an InlineCost that will.
|
|
// This will not be inspected to make an error message.
|
|
return None;
|
|
}
|
|
|
|
LLVM_DEBUG(dbgs() << " Inlining " << inlineCostStr(IC)
|
|
<< ", Call: " << *CS.getInstruction() << '\n');
|
|
return IC;
|
|
}
|
|
|
|
/// Return true if the specified inline history ID
|
|
/// indicates an inline history that includes the specified function.
|
|
static bool InlineHistoryIncludes(
|
|
Function *F, int InlineHistoryID,
|
|
const SmallVectorImpl<std::pair<Function *, int>> &InlineHistory) {
|
|
while (InlineHistoryID != -1) {
|
|
assert(unsigned(InlineHistoryID) < InlineHistory.size() &&
|
|
"Invalid inline history ID");
|
|
if (InlineHistory[InlineHistoryID].first == F)
|
|
return true;
|
|
InlineHistoryID = InlineHistory[InlineHistoryID].second;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool LegacyInlinerBase::doInitialization(CallGraph &CG) {
|
|
if (InlinerFunctionImportStats != InlinerFunctionImportStatsOpts::No)
|
|
ImportedFunctionsStats.setModuleInfo(CG.getModule());
|
|
return false; // No changes to CallGraph.
|
|
}
|
|
|
|
bool LegacyInlinerBase::runOnSCC(CallGraphSCC &SCC) {
|
|
if (skipSCC(SCC))
|
|
return false;
|
|
return inlineCalls(SCC);
|
|
}
|
|
|
|
static void emit_inlined_into(OptimizationRemarkEmitter &ORE, DebugLoc &DLoc,
|
|
const BasicBlock *Block, const Function &Callee,
|
|
const Function &Caller, const InlineCost &IC) {
|
|
ORE.emit([&]() {
|
|
bool AlwaysInline = IC.isAlways();
|
|
StringRef RemarkName = AlwaysInline ? "AlwaysInline" : "Inlined";
|
|
return OptimizationRemark(DEBUG_TYPE, RemarkName, DLoc, Block)
|
|
<< ore::NV("Callee", &Callee) << " inlined into "
|
|
<< ore::NV("Caller", &Caller) << " with " << IC;
|
|
});
|
|
}
|
|
|
|
static void setInlineRemark(CallSite &CS, StringRef message) {
|
|
if (!InlineRemarkAttribute)
|
|
return;
|
|
|
|
Attribute attr = Attribute::get(CS->getContext(), "inline-remark", message);
|
|
CS.addAttribute(AttributeList::FunctionIndex, attr);
|
|
}
|
|
|
|
static bool
|
|
inlineCallsImpl(CallGraphSCC &SCC, CallGraph &CG,
|
|
std::function<AssumptionCache &(Function &)> GetAssumptionCache,
|
|
ProfileSummaryInfo *PSI,
|
|
std::function<TargetLibraryInfo &(Function &)> GetTLI,
|
|
bool InsertLifetime,
|
|
function_ref<InlineCost(CallSite CS)> GetInlineCost,
|
|
function_ref<AAResults &(Function &)> AARGetter,
|
|
ImportedFunctionsInliningStatistics &ImportedFunctionsStats) {
|
|
SmallPtrSet<Function *, 8> SCCFunctions;
|
|
LLVM_DEBUG(dbgs() << "Inliner visiting SCC:");
|
|
for (CallGraphNode *Node : SCC) {
|
|
Function *F = Node->getFunction();
|
|
if (F)
|
|
SCCFunctions.insert(F);
|
|
LLVM_DEBUG(dbgs() << " " << (F ? F->getName() : "INDIRECTNODE"));
|
|
}
|
|
|
|
// Scan through and identify all call sites ahead of time so that we only
|
|
// inline call sites in the original functions, not call sites that result
|
|
// from inlining other functions.
|
|
SmallVector<std::pair<CallSite, int>, 16> CallSites;
|
|
|
|
// When inlining a callee produces new call sites, we want to keep track of
|
|
// the fact that they were inlined from the callee. This allows us to avoid
|
|
// infinite inlining in some obscure cases. To represent this, we use an
|
|
// index into the InlineHistory vector.
|
|
SmallVector<std::pair<Function *, int>, 8> InlineHistory;
|
|
|
|
for (CallGraphNode *Node : SCC) {
|
|
Function *F = Node->getFunction();
|
|
if (!F || F->isDeclaration())
|
|
continue;
|
|
|
|
OptimizationRemarkEmitter ORE(F);
|
|
for (BasicBlock &BB : *F)
|
|
for (Instruction &I : BB) {
|
|
CallSite CS(cast<Value>(&I));
|
|
// If this isn't a call, or it is a call to an intrinsic, it can
|
|
// never be inlined.
|
|
if (!CS || isa<IntrinsicInst>(I))
|
|
continue;
|
|
|
|
// If this is a direct call to an external function, we can never inline
|
|
// it. If it is an indirect call, inlining may resolve it to be a
|
|
// direct call, so we keep it.
|
|
if (Function *Callee = CS.getCalledFunction())
|
|
if (Callee->isDeclaration()) {
|
|
using namespace ore;
|
|
|
|
setInlineRemark(CS, "unavailable definition");
|
|
ORE.emit([&]() {
|
|
return OptimizationRemarkMissed(DEBUG_TYPE, "NoDefinition", &I)
|
|
<< NV("Callee", Callee) << " will not be inlined into "
|
|
<< NV("Caller", CS.getCaller())
|
|
<< " because its definition is unavailable"
|
|
<< setIsVerbose();
|
|
});
|
|
continue;
|
|
}
|
|
|
|
CallSites.push_back(std::make_pair(CS, -1));
|
|
}
|
|
}
|
|
|
|
LLVM_DEBUG(dbgs() << ": " << CallSites.size() << " call sites.\n");
|
|
|
|
// If there are no calls in this function, exit early.
|
|
if (CallSites.empty())
|
|
return false;
|
|
|
|
// Now that we have all of the call sites, move the ones to functions in the
|
|
// current SCC to the end of the list.
|
|
unsigned FirstCallInSCC = CallSites.size();
|
|
for (unsigned i = 0; i < FirstCallInSCC; ++i)
|
|
if (Function *F = CallSites[i].first.getCalledFunction())
|
|
if (SCCFunctions.count(F))
|
|
std::swap(CallSites[i--], CallSites[--FirstCallInSCC]);
|
|
|
|
InlinedArrayAllocasTy InlinedArrayAllocas;
|
|
InlineFunctionInfo InlineInfo(&CG, &GetAssumptionCache, PSI);
|
|
|
|
// Now that we have all of the call sites, loop over them and inline them if
|
|
// it looks profitable to do so.
|
|
bool Changed = false;
|
|
bool LocalChange;
|
|
do {
|
|
LocalChange = false;
|
|
// Iterate over the outer loop because inlining functions can cause indirect
|
|
// calls to become direct calls.
|
|
// CallSites may be modified inside so ranged for loop can not be used.
|
|
for (unsigned CSi = 0; CSi != CallSites.size(); ++CSi) {
|
|
CallSite CS = CallSites[CSi].first;
|
|
|
|
Function *Caller = CS.getCaller();
|
|
Function *Callee = CS.getCalledFunction();
|
|
|
|
// We can only inline direct calls to non-declarations.
|
|
if (!Callee || Callee->isDeclaration())
|
|
continue;
|
|
|
|
Instruction *Instr = CS.getInstruction();
|
|
|
|
bool IsTriviallyDead =
|
|
isInstructionTriviallyDead(Instr, &GetTLI(*Caller));
|
|
|
|
int InlineHistoryID;
|
|
if (!IsTriviallyDead) {
|
|
// If this call site was obtained by inlining another function, verify
|
|
// that the include path for the function did not include the callee
|
|
// itself. If so, we'd be recursively inlining the same function,
|
|
// which would provide the same callsites, which would cause us to
|
|
// infinitely inline.
|
|
InlineHistoryID = CallSites[CSi].second;
|
|
if (InlineHistoryID != -1 &&
|
|
InlineHistoryIncludes(Callee, InlineHistoryID, InlineHistory)) {
|
|
setInlineRemark(CS, "recursive");
|
|
continue;
|
|
}
|
|
}
|
|
|
|
// FIXME for new PM: because of the old PM we currently generate ORE and
|
|
// in turn BFI on demand. With the new PM, the ORE dependency should
|
|
// just become a regular analysis dependency.
|
|
OptimizationRemarkEmitter ORE(Caller);
|
|
|
|
Optional<InlineCost> OIC = shouldInline(CS, GetInlineCost, ORE);
|
|
// If the policy determines that we should inline this function,
|
|
// delete the call instead.
|
|
if (!OIC.hasValue()) {
|
|
setInlineRemark(CS, "deferred");
|
|
continue;
|
|
}
|
|
|
|
if (!OIC.getValue()) {
|
|
// shouldInline() call returned a negative inline cost that explains
|
|
// why this callsite should not be inlined.
|
|
setInlineRemark(CS, inlineCostStr(*OIC));
|
|
continue;
|
|
}
|
|
|
|
// If this call site is dead and it is to a readonly function, we should
|
|
// just delete the call instead of trying to inline it, regardless of
|
|
// size. This happens because IPSCCP propagates the result out of the
|
|
// call and then we're left with the dead call.
|
|
if (IsTriviallyDead) {
|
|
LLVM_DEBUG(dbgs() << " -> Deleting dead call: " << *Instr << "\n");
|
|
// Update the call graph by deleting the edge from Callee to Caller.
|
|
setInlineRemark(CS, "trivially dead");
|
|
CG[Caller]->removeCallEdgeFor(*cast<CallBase>(CS.getInstruction()));
|
|
Instr->eraseFromParent();
|
|
++NumCallsDeleted;
|
|
} else {
|
|
// Get DebugLoc to report. CS will be invalid after Inliner.
|
|
DebugLoc DLoc = CS->getDebugLoc();
|
|
BasicBlock *Block = CS.getParent();
|
|
|
|
// Attempt to inline the function.
|
|
using namespace ore;
|
|
|
|
InlineResult IR = InlineCallIfPossible(
|
|
CS, InlineInfo, InlinedArrayAllocas, InlineHistoryID,
|
|
InsertLifetime, AARGetter, ImportedFunctionsStats);
|
|
if (!IR) {
|
|
setInlineRemark(CS, std::string(IR) + "; " + inlineCostStr(*OIC));
|
|
ORE.emit([&]() {
|
|
return OptimizationRemarkMissed(DEBUG_TYPE, "NotInlined", DLoc,
|
|
Block)
|
|
<< NV("Callee", Callee) << " will not be inlined into "
|
|
<< NV("Caller", Caller) << ": " << NV("Reason", IR.message);
|
|
});
|
|
continue;
|
|
}
|
|
++NumInlined;
|
|
|
|
emit_inlined_into(ORE, DLoc, Block, *Callee, *Caller, *OIC);
|
|
|
|
// If inlining this function gave us any new call sites, throw them
|
|
// onto our worklist to process. They are useful inline candidates.
|
|
if (!InlineInfo.InlinedCalls.empty()) {
|
|
// Create a new inline history entry for this, so that we remember
|
|
// that these new callsites came about due to inlining Callee.
|
|
int NewHistoryID = InlineHistory.size();
|
|
InlineHistory.push_back(std::make_pair(Callee, InlineHistoryID));
|
|
|
|
for (Value *Ptr : InlineInfo.InlinedCalls)
|
|
CallSites.push_back(std::make_pair(CallSite(Ptr), NewHistoryID));
|
|
}
|
|
}
|
|
|
|
// If we inlined or deleted the last possible call site to the function,
|
|
// delete the function body now.
|
|
if (Callee && Callee->use_empty() && Callee->hasLocalLinkage() &&
|
|
// TODO: Can remove if in SCC now.
|
|
!SCCFunctions.count(Callee) &&
|
|
// The function may be apparently dead, but if there are indirect
|
|
// callgraph references to the node, we cannot delete it yet, this
|
|
// could invalidate the CGSCC iterator.
|
|
CG[Callee]->getNumReferences() == 0) {
|
|
LLVM_DEBUG(dbgs() << " -> Deleting dead function: "
|
|
<< Callee->getName() << "\n");
|
|
CallGraphNode *CalleeNode = CG[Callee];
|
|
|
|
// Remove any call graph edges from the callee to its callees.
|
|
CalleeNode->removeAllCalledFunctions();
|
|
|
|
// Removing the node for callee from the call graph and delete it.
|
|
delete CG.removeFunctionFromModule(CalleeNode);
|
|
++NumDeleted;
|
|
}
|
|
|
|
// Remove this call site from the list. If possible, use
|
|
// swap/pop_back for efficiency, but do not use it if doing so would
|
|
// move a call site to a function in this SCC before the
|
|
// 'FirstCallInSCC' barrier.
|
|
if (SCC.isSingular()) {
|
|
CallSites[CSi] = CallSites.back();
|
|
CallSites.pop_back();
|
|
} else {
|
|
CallSites.erase(CallSites.begin() + CSi);
|
|
}
|
|
--CSi;
|
|
|
|
Changed = true;
|
|
LocalChange = true;
|
|
}
|
|
} while (LocalChange);
|
|
|
|
return Changed;
|
|
}
|
|
|
|
bool LegacyInlinerBase::inlineCalls(CallGraphSCC &SCC) {
|
|
CallGraph &CG = getAnalysis<CallGraphWrapperPass>().getCallGraph();
|
|
ACT = &getAnalysis<AssumptionCacheTracker>();
|
|
PSI = &getAnalysis<ProfileSummaryInfoWrapperPass>().getPSI();
|
|
auto GetTLI = [&](Function &F) -> TargetLibraryInfo & {
|
|
return getAnalysis<TargetLibraryInfoWrapperPass>().getTLI(F);
|
|
};
|
|
auto GetAssumptionCache = [&](Function &F) -> AssumptionCache & {
|
|
return ACT->getAssumptionCache(F);
|
|
};
|
|
return inlineCallsImpl(
|
|
SCC, CG, GetAssumptionCache, PSI, GetTLI, InsertLifetime,
|
|
[this](CallSite CS) { return getInlineCost(CS); }, LegacyAARGetter(*this),
|
|
ImportedFunctionsStats);
|
|
}
|
|
|
|
/// Remove now-dead linkonce functions at the end of
|
|
/// processing to avoid breaking the SCC traversal.
|
|
bool LegacyInlinerBase::doFinalization(CallGraph &CG) {
|
|
if (InlinerFunctionImportStats != InlinerFunctionImportStatsOpts::No)
|
|
ImportedFunctionsStats.dump(InlinerFunctionImportStats ==
|
|
InlinerFunctionImportStatsOpts::Verbose);
|
|
return removeDeadFunctions(CG);
|
|
}
|
|
|
|
/// Remove dead functions that are not included in DNR (Do Not Remove) list.
|
|
bool LegacyInlinerBase::removeDeadFunctions(CallGraph &CG,
|
|
bool AlwaysInlineOnly) {
|
|
SmallVector<CallGraphNode *, 16> FunctionsToRemove;
|
|
SmallVector<Function *, 16> DeadFunctionsInComdats;
|
|
|
|
auto RemoveCGN = [&](CallGraphNode *CGN) {
|
|
// Remove any call graph edges from the function to its callees.
|
|
CGN->removeAllCalledFunctions();
|
|
|
|
// Remove any edges from the external node to the function's call graph
|
|
// node. These edges might have been made irrelegant due to
|
|
// optimization of the program.
|
|
CG.getExternalCallingNode()->removeAnyCallEdgeTo(CGN);
|
|
|
|
// Removing the node for callee from the call graph and delete it.
|
|
FunctionsToRemove.push_back(CGN);
|
|
};
|
|
|
|
// Scan for all of the functions, looking for ones that should now be removed
|
|
// from the program. Insert the dead ones in the FunctionsToRemove set.
|
|
for (const auto &I : CG) {
|
|
CallGraphNode *CGN = I.second.get();
|
|
Function *F = CGN->getFunction();
|
|
if (!F || F->isDeclaration())
|
|
continue;
|
|
|
|
// Handle the case when this function is called and we only want to care
|
|
// about always-inline functions. This is a bit of a hack to share code
|
|
// between here and the InlineAlways pass.
|
|
if (AlwaysInlineOnly && !F->hasFnAttribute(Attribute::AlwaysInline))
|
|
continue;
|
|
|
|
// If the only remaining users of the function are dead constants, remove
|
|
// them.
|
|
F->removeDeadConstantUsers();
|
|
|
|
if (!F->isDefTriviallyDead())
|
|
continue;
|
|
|
|
// It is unsafe to drop a function with discardable linkage from a COMDAT
|
|
// without also dropping the other members of the COMDAT.
|
|
// The inliner doesn't visit non-function entities which are in COMDAT
|
|
// groups so it is unsafe to do so *unless* the linkage is local.
|
|
if (!F->hasLocalLinkage()) {
|
|
if (F->hasComdat()) {
|
|
DeadFunctionsInComdats.push_back(F);
|
|
continue;
|
|
}
|
|
}
|
|
|
|
RemoveCGN(CGN);
|
|
}
|
|
if (!DeadFunctionsInComdats.empty()) {
|
|
// Filter out the functions whose comdats remain alive.
|
|
filterDeadComdatFunctions(CG.getModule(), DeadFunctionsInComdats);
|
|
// Remove the rest.
|
|
for (Function *F : DeadFunctionsInComdats)
|
|
RemoveCGN(CG[F]);
|
|
}
|
|
|
|
if (FunctionsToRemove.empty())
|
|
return false;
|
|
|
|
// Now that we know which functions to delete, do so. We didn't want to do
|
|
// this inline, because that would invalidate our CallGraph::iterator
|
|
// objects. :(
|
|
//
|
|
// Note that it doesn't matter that we are iterating over a non-stable order
|
|
// here to do this, it doesn't matter which order the functions are deleted
|
|
// in.
|
|
array_pod_sort(FunctionsToRemove.begin(), FunctionsToRemove.end());
|
|
FunctionsToRemove.erase(
|
|
std::unique(FunctionsToRemove.begin(), FunctionsToRemove.end()),
|
|
FunctionsToRemove.end());
|
|
for (CallGraphNode *CGN : FunctionsToRemove) {
|
|
delete CG.removeFunctionFromModule(CGN);
|
|
++NumDeleted;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
InlinerPass::~InlinerPass() {
|
|
if (ImportedFunctionsStats) {
|
|
assert(InlinerFunctionImportStats != InlinerFunctionImportStatsOpts::No);
|
|
ImportedFunctionsStats->dump(InlinerFunctionImportStats ==
|
|
InlinerFunctionImportStatsOpts::Verbose);
|
|
}
|
|
}
|
|
|
|
PreservedAnalyses InlinerPass::run(LazyCallGraph::SCC &InitialC,
|
|
CGSCCAnalysisManager &AM, LazyCallGraph &CG,
|
|
CGSCCUpdateResult &UR) {
|
|
const ModuleAnalysisManager &MAM =
|
|
AM.getResult<ModuleAnalysisManagerCGSCCProxy>(InitialC, CG).getManager();
|
|
bool Changed = false;
|
|
|
|
assert(InitialC.size() > 0 && "Cannot handle an empty SCC!");
|
|
Module &M = *InitialC.begin()->getFunction().getParent();
|
|
ProfileSummaryInfo *PSI = MAM.getCachedResult<ProfileSummaryAnalysis>(M);
|
|
|
|
if (!ImportedFunctionsStats &&
|
|
InlinerFunctionImportStats != InlinerFunctionImportStatsOpts::No) {
|
|
ImportedFunctionsStats =
|
|
std::make_unique<ImportedFunctionsInliningStatistics>();
|
|
ImportedFunctionsStats->setModuleInfo(M);
|
|
}
|
|
|
|
// We use a single common worklist for calls across the entire SCC. We
|
|
// process these in-order and append new calls introduced during inlining to
|
|
// the end.
|
|
//
|
|
// Note that this particular order of processing is actually critical to
|
|
// avoid very bad behaviors. Consider *highly connected* call graphs where
|
|
// each function contains a small amonut of code and a couple of calls to
|
|
// other functions. Because the LLVM inliner is fundamentally a bottom-up
|
|
// inliner, it can handle gracefully the fact that these all appear to be
|
|
// reasonable inlining candidates as it will flatten things until they become
|
|
// too big to inline, and then move on and flatten another batch.
|
|
//
|
|
// However, when processing call edges *within* an SCC we cannot rely on this
|
|
// bottom-up behavior. As a consequence, with heavily connected *SCCs* of
|
|
// functions we can end up incrementally inlining N calls into each of
|
|
// N functions because each incremental inlining decision looks good and we
|
|
// don't have a topological ordering to prevent explosions.
|
|
//
|
|
// To compensate for this, we don't process transitive edges made immediate
|
|
// by inlining until we've done one pass of inlining across the entire SCC.
|
|
// Large, highly connected SCCs still lead to some amount of code bloat in
|
|
// this model, but it is uniformly spread across all the functions in the SCC
|
|
// and eventually they all become too large to inline, rather than
|
|
// incrementally maknig a single function grow in a super linear fashion.
|
|
SmallVector<std::pair<CallSite, int>, 16> Calls;
|
|
|
|
FunctionAnalysisManager &FAM =
|
|
AM.getResult<FunctionAnalysisManagerCGSCCProxy>(InitialC, CG)
|
|
.getManager();
|
|
|
|
// Populate the initial list of calls in this SCC.
|
|
for (auto &N : InitialC) {
|
|
auto &ORE =
|
|
FAM.getResult<OptimizationRemarkEmitterAnalysis>(N.getFunction());
|
|
// We want to generally process call sites top-down in order for
|
|
// simplifications stemming from replacing the call with the returned value
|
|
// after inlining to be visible to subsequent inlining decisions.
|
|
// FIXME: Using instructions sequence is a really bad way to do this.
|
|
// Instead we should do an actual RPO walk of the function body.
|
|
for (Instruction &I : instructions(N.getFunction()))
|
|
if (auto CS = CallSite(&I))
|
|
if (Function *Callee = CS.getCalledFunction()) {
|
|
if (!Callee->isDeclaration())
|
|
Calls.push_back({CS, -1});
|
|
else if (!isa<IntrinsicInst>(I)) {
|
|
using namespace ore;
|
|
setInlineRemark(CS, "unavailable definition");
|
|
ORE.emit([&]() {
|
|
return OptimizationRemarkMissed(DEBUG_TYPE, "NoDefinition", &I)
|
|
<< NV("Callee", Callee) << " will not be inlined into "
|
|
<< NV("Caller", CS.getCaller())
|
|
<< " because its definition is unavailable"
|
|
<< setIsVerbose();
|
|
});
|
|
}
|
|
}
|
|
}
|
|
if (Calls.empty())
|
|
return PreservedAnalyses::all();
|
|
|
|
// Capture updatable variables for the current SCC and RefSCC.
|
|
auto *C = &InitialC;
|
|
auto *RC = &C->getOuterRefSCC();
|
|
|
|
// When inlining a callee produces new call sites, we want to keep track of
|
|
// the fact that they were inlined from the callee. This allows us to avoid
|
|
// infinite inlining in some obscure cases. To represent this, we use an
|
|
// index into the InlineHistory vector.
|
|
SmallVector<std::pair<Function *, int>, 16> InlineHistory;
|
|
|
|
// Track a set vector of inlined callees so that we can augment the caller
|
|
// with all of their edges in the call graph before pruning out the ones that
|
|
// got simplified away.
|
|
SmallSetVector<Function *, 4> InlinedCallees;
|
|
|
|
// Track the dead functions to delete once finished with inlining calls. We
|
|
// defer deleting these to make it easier to handle the call graph updates.
|
|
SmallVector<Function *, 4> DeadFunctions;
|
|
|
|
// Loop forward over all of the calls. Note that we cannot cache the size as
|
|
// inlining can introduce new calls that need to be processed.
|
|
for (int i = 0; i < (int)Calls.size(); ++i) {
|
|
// We expect the calls to typically be batched with sequences of calls that
|
|
// have the same caller, so we first set up some shared infrastructure for
|
|
// this caller. We also do any pruning we can at this layer on the caller
|
|
// alone.
|
|
Function &F = *Calls[i].first.getCaller();
|
|
LazyCallGraph::Node &N = *CG.lookup(F);
|
|
if (CG.lookupSCC(N) != C)
|
|
continue;
|
|
if (F.hasOptNone()) {
|
|
setInlineRemark(Calls[i].first, "optnone attribute");
|
|
continue;
|
|
}
|
|
|
|
LLVM_DEBUG(dbgs() << "Inlining calls in: " << F.getName() << "\n");
|
|
|
|
// Get a FunctionAnalysisManager via a proxy for this particular node. We
|
|
// do this each time we visit a node as the SCC may have changed and as
|
|
// we're going to mutate this particular function we want to make sure the
|
|
// proxy is in place to forward any invalidation events. We can use the
|
|
// manager we get here for looking up results for functions other than this
|
|
// node however because those functions aren't going to be mutated by this
|
|
// pass.
|
|
FunctionAnalysisManager &FAM =
|
|
AM.getResult<FunctionAnalysisManagerCGSCCProxy>(*C, CG)
|
|
.getManager();
|
|
|
|
// Get the remarks emission analysis for the caller.
|
|
auto &ORE = FAM.getResult<OptimizationRemarkEmitterAnalysis>(F);
|
|
|
|
std::function<AssumptionCache &(Function &)> GetAssumptionCache =
|
|
[&](Function &F) -> AssumptionCache & {
|
|
return FAM.getResult<AssumptionAnalysis>(F);
|
|
};
|
|
auto GetBFI = [&](Function &F) -> BlockFrequencyInfo & {
|
|
return FAM.getResult<BlockFrequencyAnalysis>(F);
|
|
};
|
|
|
|
auto GetInlineCost = [&](CallSite CS) {
|
|
Function &Callee = *CS.getCalledFunction();
|
|
auto &CalleeTTI = FAM.getResult<TargetIRAnalysis>(Callee);
|
|
bool RemarksEnabled =
|
|
Callee.getContext().getDiagHandlerPtr()->isMissedOptRemarkEnabled(
|
|
DEBUG_TYPE);
|
|
return getInlineCost(cast<CallBase>(*CS.getInstruction()), Params,
|
|
CalleeTTI, GetAssumptionCache, {GetBFI}, PSI,
|
|
RemarksEnabled ? &ORE : nullptr);
|
|
};
|
|
|
|
// Now process as many calls as we have within this caller in the sequnece.
|
|
// We bail out as soon as the caller has to change so we can update the
|
|
// call graph and prepare the context of that new caller.
|
|
bool DidInline = false;
|
|
for (; i < (int)Calls.size() && Calls[i].first.getCaller() == &F; ++i) {
|
|
int InlineHistoryID;
|
|
CallSite CS;
|
|
std::tie(CS, InlineHistoryID) = Calls[i];
|
|
Function &Callee = *CS.getCalledFunction();
|
|
|
|
if (InlineHistoryID != -1 &&
|
|
InlineHistoryIncludes(&Callee, InlineHistoryID, InlineHistory)) {
|
|
setInlineRemark(CS, "recursive");
|
|
continue;
|
|
}
|
|
|
|
// Check if this inlining may repeat breaking an SCC apart that has
|
|
// already been split once before. In that case, inlining here may
|
|
// trigger infinite inlining, much like is prevented within the inliner
|
|
// itself by the InlineHistory above, but spread across CGSCC iterations
|
|
// and thus hidden from the full inline history.
|
|
if (CG.lookupSCC(*CG.lookup(Callee)) == C &&
|
|
UR.InlinedInternalEdges.count({&N, C})) {
|
|
LLVM_DEBUG(dbgs() << "Skipping inlining internal SCC edge from a node "
|
|
"previously split out of this SCC by inlining: "
|
|
<< F.getName() << " -> " << Callee.getName() << "\n");
|
|
setInlineRemark(CS, "recursive SCC split");
|
|
continue;
|
|
}
|
|
|
|
Optional<InlineCost> OIC = shouldInline(CS, GetInlineCost, ORE);
|
|
// Check whether we want to inline this callsite.
|
|
if (!OIC.hasValue()) {
|
|
setInlineRemark(CS, "deferred");
|
|
continue;
|
|
}
|
|
|
|
if (!OIC.getValue()) {
|
|
// shouldInline() call returned a negative inline cost that explains
|
|
// why this callsite should not be inlined.
|
|
setInlineRemark(CS, inlineCostStr(*OIC));
|
|
continue;
|
|
}
|
|
|
|
// Setup the data structure used to plumb customization into the
|
|
// `InlineFunction` routine.
|
|
InlineFunctionInfo IFI(
|
|
/*cg=*/nullptr, &GetAssumptionCache, PSI,
|
|
&FAM.getResult<BlockFrequencyAnalysis>(*(CS.getCaller())),
|
|
&FAM.getResult<BlockFrequencyAnalysis>(Callee));
|
|
|
|
// Get DebugLoc to report. CS will be invalid after Inliner.
|
|
DebugLoc DLoc = CS->getDebugLoc();
|
|
BasicBlock *Block = CS.getParent();
|
|
|
|
using namespace ore;
|
|
|
|
InlineResult IR = InlineFunction(CS, IFI);
|
|
if (!IR) {
|
|
setInlineRemark(CS, std::string(IR) + "; " + inlineCostStr(*OIC));
|
|
ORE.emit([&]() {
|
|
return OptimizationRemarkMissed(DEBUG_TYPE, "NotInlined", DLoc, Block)
|
|
<< NV("Callee", &Callee) << " will not be inlined into "
|
|
<< NV("Caller", &F) << ": " << NV("Reason", IR.message);
|
|
});
|
|
continue;
|
|
}
|
|
DidInline = true;
|
|
InlinedCallees.insert(&Callee);
|
|
|
|
++NumInlined;
|
|
|
|
emit_inlined_into(ORE, DLoc, Block, Callee, F, *OIC);
|
|
|
|
// Add any new callsites to defined functions to the worklist.
|
|
if (!IFI.InlinedCallSites.empty()) {
|
|
int NewHistoryID = InlineHistory.size();
|
|
InlineHistory.push_back({&Callee, InlineHistoryID});
|
|
for (CallSite &CS : reverse(IFI.InlinedCallSites))
|
|
if (Function *NewCallee = CS.getCalledFunction())
|
|
if (!NewCallee->isDeclaration())
|
|
Calls.push_back({CS, NewHistoryID});
|
|
}
|
|
|
|
if (InlinerFunctionImportStats != InlinerFunctionImportStatsOpts::No)
|
|
ImportedFunctionsStats->recordInline(F, Callee);
|
|
|
|
// Merge the attributes based on the inlining.
|
|
AttributeFuncs::mergeAttributesForInlining(F, Callee);
|
|
|
|
// For local functions, check whether this makes the callee trivially
|
|
// dead. In that case, we can drop the body of the function eagerly
|
|
// which may reduce the number of callers of other functions to one,
|
|
// changing inline cost thresholds.
|
|
if (Callee.hasLocalLinkage()) {
|
|
// To check this we also need to nuke any dead constant uses (perhaps
|
|
// made dead by this operation on other functions).
|
|
Callee.removeDeadConstantUsers();
|
|
if (Callee.use_empty() && !CG.isLibFunction(Callee)) {
|
|
Calls.erase(
|
|
std::remove_if(Calls.begin() + i + 1, Calls.end(),
|
|
[&Callee](const std::pair<CallSite, int> &Call) {
|
|
return Call.first.getCaller() == &Callee;
|
|
}),
|
|
Calls.end());
|
|
// Clear the body and queue the function itself for deletion when we
|
|
// finish inlining and call graph updates.
|
|
// Note that after this point, it is an error to do anything other
|
|
// than use the callee's address or delete it.
|
|
Callee.dropAllReferences();
|
|
assert(find(DeadFunctions, &Callee) == DeadFunctions.end() &&
|
|
"Cannot put cause a function to become dead twice!");
|
|
DeadFunctions.push_back(&Callee);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Back the call index up by one to put us in a good position to go around
|
|
// the outer loop.
|
|
--i;
|
|
|
|
if (!DidInline)
|
|
continue;
|
|
Changed = true;
|
|
|
|
// Add all the inlined callees' edges as ref edges to the caller. These are
|
|
// by definition trivial edges as we always have *some* transitive ref edge
|
|
// chain. While in some cases these edges are direct calls inside the
|
|
// callee, they have to be modeled in the inliner as reference edges as
|
|
// there may be a reference edge anywhere along the chain from the current
|
|
// caller to the callee that causes the whole thing to appear like
|
|
// a (transitive) reference edge that will require promotion to a call edge
|
|
// below.
|
|
for (Function *InlinedCallee : InlinedCallees) {
|
|
LazyCallGraph::Node &CalleeN = *CG.lookup(*InlinedCallee);
|
|
for (LazyCallGraph::Edge &E : *CalleeN)
|
|
RC->insertTrivialRefEdge(N, E.getNode());
|
|
}
|
|
|
|
// At this point, since we have made changes we have at least removed
|
|
// a call instruction. However, in the process we do some incremental
|
|
// simplification of the surrounding code. This simplification can
|
|
// essentially do all of the same things as a function pass and we can
|
|
// re-use the exact same logic for updating the call graph to reflect the
|
|
// change.
|
|
LazyCallGraph::SCC *OldC = C;
|
|
C = &updateCGAndAnalysisManagerForFunctionPass(CG, *C, N, AM, UR);
|
|
LLVM_DEBUG(dbgs() << "Updated inlining SCC: " << *C << "\n");
|
|
RC = &C->getOuterRefSCC();
|
|
|
|
// If this causes an SCC to split apart into multiple smaller SCCs, there
|
|
// is a subtle risk we need to prepare for. Other transformations may
|
|
// expose an "infinite inlining" opportunity later, and because of the SCC
|
|
// mutation, we will revisit this function and potentially re-inline. If we
|
|
// do, and that re-inlining also has the potentially to mutate the SCC
|
|
// structure, the infinite inlining problem can manifest through infinite
|
|
// SCC splits and merges. To avoid this, we capture the originating caller
|
|
// node and the SCC containing the call edge. This is a slight over
|
|
// approximation of the possible inlining decisions that must be avoided,
|
|
// but is relatively efficient to store. We use C != OldC to know when
|
|
// a new SCC is generated and the original SCC may be generated via merge
|
|
// in later iterations.
|
|
//
|
|
// It is also possible that even if no new SCC is generated
|
|
// (i.e., C == OldC), the original SCC could be split and then merged
|
|
// into the same one as itself. and the original SCC will be added into
|
|
// UR.CWorklist again, we want to catch such cases too.
|
|
//
|
|
// FIXME: This seems like a very heavyweight way of retaining the inline
|
|
// history, we should look for a more efficient way of tracking it.
|
|
if ((C != OldC || UR.CWorklist.count(OldC)) &&
|
|
llvm::any_of(InlinedCallees, [&](Function *Callee) {
|
|
return CG.lookupSCC(*CG.lookup(*Callee)) == OldC;
|
|
})) {
|
|
LLVM_DEBUG(dbgs() << "Inlined an internal call edge and split an SCC, "
|
|
"retaining this to avoid infinite inlining.\n");
|
|
UR.InlinedInternalEdges.insert({&N, OldC});
|
|
}
|
|
InlinedCallees.clear();
|
|
}
|
|
|
|
// Now that we've finished inlining all of the calls across this SCC, delete
|
|
// all of the trivially dead functions, updating the call graph and the CGSCC
|
|
// pass manager in the process.
|
|
//
|
|
// Note that this walks a pointer set which has non-deterministic order but
|
|
// that is OK as all we do is delete things and add pointers to unordered
|
|
// sets.
|
|
for (Function *DeadF : DeadFunctions) {
|
|
// Get the necessary information out of the call graph and nuke the
|
|
// function there. Also, cclear out any cached analyses.
|
|
auto &DeadC = *CG.lookupSCC(*CG.lookup(*DeadF));
|
|
FunctionAnalysisManager &FAM =
|
|
AM.getResult<FunctionAnalysisManagerCGSCCProxy>(DeadC, CG)
|
|
.getManager();
|
|
FAM.clear(*DeadF, DeadF->getName());
|
|
AM.clear(DeadC, DeadC.getName());
|
|
auto &DeadRC = DeadC.getOuterRefSCC();
|
|
CG.removeDeadFunction(*DeadF);
|
|
|
|
// Mark the relevant parts of the call graph as invalid so we don't visit
|
|
// them.
|
|
UR.InvalidatedSCCs.insert(&DeadC);
|
|
UR.InvalidatedRefSCCs.insert(&DeadRC);
|
|
|
|
// And delete the actual function from the module.
|
|
M.getFunctionList().erase(DeadF);
|
|
++NumDeleted;
|
|
}
|
|
|
|
if (!Changed)
|
|
return PreservedAnalyses::all();
|
|
|
|
// Even if we change the IR, we update the core CGSCC data structures and so
|
|
// can preserve the proxy to the function analysis manager.
|
|
PreservedAnalyses PA;
|
|
PA.preserve<FunctionAnalysisManagerCGSCCProxy>();
|
|
return PA;
|
|
}
|