forked from OSchip/llvm-project
1353 lines
50 KiB
C++
1353 lines
50 KiB
C++
//===- lib/CodeGen/MachineTraceMetrics.cpp --------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/CodeGen/MachineTraceMetrics.h"
|
|
#include "llvm/ADT/ArrayRef.h"
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include "llvm/ADT/Optional.h"
|
|
#include "llvm/ADT/PostOrderIterator.h"
|
|
#include "llvm/ADT/SmallPtrSet.h"
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/ADT/SparseSet.h"
|
|
#include "llvm/CodeGen/MachineBasicBlock.h"
|
|
#include "llvm/CodeGen/MachineBranchProbabilityInfo.h"
|
|
#include "llvm/CodeGen/MachineFunction.h"
|
|
#include "llvm/CodeGen/MachineInstr.h"
|
|
#include "llvm/CodeGen/MachineLoopInfo.h"
|
|
#include "llvm/CodeGen/MachineOperand.h"
|
|
#include "llvm/CodeGen/MachineRegisterInfo.h"
|
|
#include "llvm/CodeGen/TargetRegisterInfo.h"
|
|
#include "llvm/CodeGen/TargetSchedule.h"
|
|
#include "llvm/CodeGen/TargetSubtargetInfo.h"
|
|
#include "llvm/InitializePasses.h"
|
|
#include "llvm/MC/MCRegisterInfo.h"
|
|
#include "llvm/Pass.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include "llvm/Support/Format.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include <algorithm>
|
|
#include <cassert>
|
|
#include <iterator>
|
|
#include <tuple>
|
|
#include <utility>
|
|
|
|
using namespace llvm;
|
|
|
|
#define DEBUG_TYPE "machine-trace-metrics"
|
|
|
|
char MachineTraceMetrics::ID = 0;
|
|
|
|
char &llvm::MachineTraceMetricsID = MachineTraceMetrics::ID;
|
|
|
|
INITIALIZE_PASS_BEGIN(MachineTraceMetrics, DEBUG_TYPE,
|
|
"Machine Trace Metrics", false, true)
|
|
INITIALIZE_PASS_DEPENDENCY(MachineBranchProbabilityInfo)
|
|
INITIALIZE_PASS_DEPENDENCY(MachineLoopInfo)
|
|
INITIALIZE_PASS_END(MachineTraceMetrics, DEBUG_TYPE,
|
|
"Machine Trace Metrics", false, true)
|
|
|
|
MachineTraceMetrics::MachineTraceMetrics() : MachineFunctionPass(ID) {
|
|
std::fill(std::begin(Ensembles), std::end(Ensembles), nullptr);
|
|
}
|
|
|
|
void MachineTraceMetrics::getAnalysisUsage(AnalysisUsage &AU) const {
|
|
AU.setPreservesAll();
|
|
AU.addRequired<MachineBranchProbabilityInfo>();
|
|
AU.addRequired<MachineLoopInfo>();
|
|
MachineFunctionPass::getAnalysisUsage(AU);
|
|
}
|
|
|
|
bool MachineTraceMetrics::runOnMachineFunction(MachineFunction &Func) {
|
|
MF = &Func;
|
|
const TargetSubtargetInfo &ST = MF->getSubtarget();
|
|
TII = ST.getInstrInfo();
|
|
TRI = ST.getRegisterInfo();
|
|
MRI = &MF->getRegInfo();
|
|
Loops = &getAnalysis<MachineLoopInfo>();
|
|
SchedModel.init(&ST);
|
|
BlockInfo.resize(MF->getNumBlockIDs());
|
|
ProcResourceCycles.resize(MF->getNumBlockIDs() *
|
|
SchedModel.getNumProcResourceKinds());
|
|
return false;
|
|
}
|
|
|
|
void MachineTraceMetrics::releaseMemory() {
|
|
MF = nullptr;
|
|
BlockInfo.clear();
|
|
for (unsigned i = 0; i != TS_NumStrategies; ++i) {
|
|
delete Ensembles[i];
|
|
Ensembles[i] = nullptr;
|
|
}
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Fixed block information
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// The number of instructions in a basic block and the CPU resources used by
|
|
// those instructions don't depend on any given trace strategy.
|
|
|
|
/// Compute the resource usage in basic block MBB.
|
|
const MachineTraceMetrics::FixedBlockInfo*
|
|
MachineTraceMetrics::getResources(const MachineBasicBlock *MBB) {
|
|
assert(MBB && "No basic block");
|
|
FixedBlockInfo *FBI = &BlockInfo[MBB->getNumber()];
|
|
if (FBI->hasResources())
|
|
return FBI;
|
|
|
|
// Compute resource usage in the block.
|
|
FBI->HasCalls = false;
|
|
unsigned InstrCount = 0;
|
|
|
|
// Add up per-processor resource cycles as well.
|
|
unsigned PRKinds = SchedModel.getNumProcResourceKinds();
|
|
SmallVector<unsigned, 32> PRCycles(PRKinds);
|
|
|
|
for (const auto &MI : *MBB) {
|
|
if (MI.isTransient())
|
|
continue;
|
|
++InstrCount;
|
|
if (MI.isCall())
|
|
FBI->HasCalls = true;
|
|
|
|
// Count processor resources used.
|
|
if (!SchedModel.hasInstrSchedModel())
|
|
continue;
|
|
const MCSchedClassDesc *SC = SchedModel.resolveSchedClass(&MI);
|
|
if (!SC->isValid())
|
|
continue;
|
|
|
|
for (TargetSchedModel::ProcResIter
|
|
PI = SchedModel.getWriteProcResBegin(SC),
|
|
PE = SchedModel.getWriteProcResEnd(SC); PI != PE; ++PI) {
|
|
assert(PI->ProcResourceIdx < PRKinds && "Bad processor resource kind");
|
|
PRCycles[PI->ProcResourceIdx] += PI->Cycles;
|
|
}
|
|
}
|
|
FBI->InstrCount = InstrCount;
|
|
|
|
// Scale the resource cycles so they are comparable.
|
|
unsigned PROffset = MBB->getNumber() * PRKinds;
|
|
for (unsigned K = 0; K != PRKinds; ++K)
|
|
ProcResourceCycles[PROffset + K] =
|
|
PRCycles[K] * SchedModel.getResourceFactor(K);
|
|
|
|
return FBI;
|
|
}
|
|
|
|
ArrayRef<unsigned>
|
|
MachineTraceMetrics::getProcResourceCycles(unsigned MBBNum) const {
|
|
assert(BlockInfo[MBBNum].hasResources() &&
|
|
"getResources() must be called before getProcResourceCycles()");
|
|
unsigned PRKinds = SchedModel.getNumProcResourceKinds();
|
|
assert((MBBNum+1) * PRKinds <= ProcResourceCycles.size());
|
|
return makeArrayRef(ProcResourceCycles.data() + MBBNum * PRKinds, PRKinds);
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Ensemble utility functions
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
MachineTraceMetrics::Ensemble::Ensemble(MachineTraceMetrics *ct)
|
|
: MTM(*ct) {
|
|
BlockInfo.resize(MTM.BlockInfo.size());
|
|
unsigned PRKinds = MTM.SchedModel.getNumProcResourceKinds();
|
|
ProcResourceDepths.resize(MTM.BlockInfo.size() * PRKinds);
|
|
ProcResourceHeights.resize(MTM.BlockInfo.size() * PRKinds);
|
|
}
|
|
|
|
// Virtual destructor serves as an anchor.
|
|
MachineTraceMetrics::Ensemble::~Ensemble() = default;
|
|
|
|
const MachineLoop*
|
|
MachineTraceMetrics::Ensemble::getLoopFor(const MachineBasicBlock *MBB) const {
|
|
return MTM.Loops->getLoopFor(MBB);
|
|
}
|
|
|
|
// Update resource-related information in the TraceBlockInfo for MBB.
|
|
// Only update resources related to the trace above MBB.
|
|
void MachineTraceMetrics::Ensemble::
|
|
computeDepthResources(const MachineBasicBlock *MBB) {
|
|
TraceBlockInfo *TBI = &BlockInfo[MBB->getNumber()];
|
|
unsigned PRKinds = MTM.SchedModel.getNumProcResourceKinds();
|
|
unsigned PROffset = MBB->getNumber() * PRKinds;
|
|
|
|
// Compute resources from trace above. The top block is simple.
|
|
if (!TBI->Pred) {
|
|
TBI->InstrDepth = 0;
|
|
TBI->Head = MBB->getNumber();
|
|
std::fill(ProcResourceDepths.begin() + PROffset,
|
|
ProcResourceDepths.begin() + PROffset + PRKinds, 0);
|
|
return;
|
|
}
|
|
|
|
// Compute from the block above. A post-order traversal ensures the
|
|
// predecessor is always computed first.
|
|
unsigned PredNum = TBI->Pred->getNumber();
|
|
TraceBlockInfo *PredTBI = &BlockInfo[PredNum];
|
|
assert(PredTBI->hasValidDepth() && "Trace above has not been computed yet");
|
|
const FixedBlockInfo *PredFBI = MTM.getResources(TBI->Pred);
|
|
TBI->InstrDepth = PredTBI->InstrDepth + PredFBI->InstrCount;
|
|
TBI->Head = PredTBI->Head;
|
|
|
|
// Compute per-resource depths.
|
|
ArrayRef<unsigned> PredPRDepths = getProcResourceDepths(PredNum);
|
|
ArrayRef<unsigned> PredPRCycles = MTM.getProcResourceCycles(PredNum);
|
|
for (unsigned K = 0; K != PRKinds; ++K)
|
|
ProcResourceDepths[PROffset + K] = PredPRDepths[K] + PredPRCycles[K];
|
|
}
|
|
|
|
// Update resource-related information in the TraceBlockInfo for MBB.
|
|
// Only update resources related to the trace below MBB.
|
|
void MachineTraceMetrics::Ensemble::
|
|
computeHeightResources(const MachineBasicBlock *MBB) {
|
|
TraceBlockInfo *TBI = &BlockInfo[MBB->getNumber()];
|
|
unsigned PRKinds = MTM.SchedModel.getNumProcResourceKinds();
|
|
unsigned PROffset = MBB->getNumber() * PRKinds;
|
|
|
|
// Compute resources for the current block.
|
|
TBI->InstrHeight = MTM.getResources(MBB)->InstrCount;
|
|
ArrayRef<unsigned> PRCycles = MTM.getProcResourceCycles(MBB->getNumber());
|
|
|
|
// The trace tail is done.
|
|
if (!TBI->Succ) {
|
|
TBI->Tail = MBB->getNumber();
|
|
llvm::copy(PRCycles, ProcResourceHeights.begin() + PROffset);
|
|
return;
|
|
}
|
|
|
|
// Compute from the block below. A post-order traversal ensures the
|
|
// predecessor is always computed first.
|
|
unsigned SuccNum = TBI->Succ->getNumber();
|
|
TraceBlockInfo *SuccTBI = &BlockInfo[SuccNum];
|
|
assert(SuccTBI->hasValidHeight() && "Trace below has not been computed yet");
|
|
TBI->InstrHeight += SuccTBI->InstrHeight;
|
|
TBI->Tail = SuccTBI->Tail;
|
|
|
|
// Compute per-resource heights.
|
|
ArrayRef<unsigned> SuccPRHeights = getProcResourceHeights(SuccNum);
|
|
for (unsigned K = 0; K != PRKinds; ++K)
|
|
ProcResourceHeights[PROffset + K] = SuccPRHeights[K] + PRCycles[K];
|
|
}
|
|
|
|
// Check if depth resources for MBB are valid and return the TBI.
|
|
// Return NULL if the resources have been invalidated.
|
|
const MachineTraceMetrics::TraceBlockInfo*
|
|
MachineTraceMetrics::Ensemble::
|
|
getDepthResources(const MachineBasicBlock *MBB) const {
|
|
const TraceBlockInfo *TBI = &BlockInfo[MBB->getNumber()];
|
|
return TBI->hasValidDepth() ? TBI : nullptr;
|
|
}
|
|
|
|
// Check if height resources for MBB are valid and return the TBI.
|
|
// Return NULL if the resources have been invalidated.
|
|
const MachineTraceMetrics::TraceBlockInfo*
|
|
MachineTraceMetrics::Ensemble::
|
|
getHeightResources(const MachineBasicBlock *MBB) const {
|
|
const TraceBlockInfo *TBI = &BlockInfo[MBB->getNumber()];
|
|
return TBI->hasValidHeight() ? TBI : nullptr;
|
|
}
|
|
|
|
/// Get an array of processor resource depths for MBB. Indexed by processor
|
|
/// resource kind, this array contains the scaled processor resources consumed
|
|
/// by all blocks preceding MBB in its trace. It does not include instructions
|
|
/// in MBB.
|
|
///
|
|
/// Compare TraceBlockInfo::InstrDepth.
|
|
ArrayRef<unsigned>
|
|
MachineTraceMetrics::Ensemble::
|
|
getProcResourceDepths(unsigned MBBNum) const {
|
|
unsigned PRKinds = MTM.SchedModel.getNumProcResourceKinds();
|
|
assert((MBBNum+1) * PRKinds <= ProcResourceDepths.size());
|
|
return makeArrayRef(ProcResourceDepths.data() + MBBNum * PRKinds, PRKinds);
|
|
}
|
|
|
|
/// Get an array of processor resource heights for MBB. Indexed by processor
|
|
/// resource kind, this array contains the scaled processor resources consumed
|
|
/// by this block and all blocks following it in its trace.
|
|
///
|
|
/// Compare TraceBlockInfo::InstrHeight.
|
|
ArrayRef<unsigned>
|
|
MachineTraceMetrics::Ensemble::
|
|
getProcResourceHeights(unsigned MBBNum) const {
|
|
unsigned PRKinds = MTM.SchedModel.getNumProcResourceKinds();
|
|
assert((MBBNum+1) * PRKinds <= ProcResourceHeights.size());
|
|
return makeArrayRef(ProcResourceHeights.data() + MBBNum * PRKinds, PRKinds);
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Trace Selection Strategies
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// A trace selection strategy is implemented as a sub-class of Ensemble. The
|
|
// trace through a block B is computed by two DFS traversals of the CFG
|
|
// starting from B. One upwards, and one downwards. During the upwards DFS,
|
|
// pickTracePred() is called on the post-ordered blocks. During the downwards
|
|
// DFS, pickTraceSucc() is called in a post-order.
|
|
//
|
|
|
|
// We never allow traces that leave loops, but we do allow traces to enter
|
|
// nested loops. We also never allow traces to contain back-edges.
|
|
//
|
|
// This means that a loop header can never appear above the center block of a
|
|
// trace, except as the trace head. Below the center block, loop exiting edges
|
|
// are banned.
|
|
//
|
|
// Return true if an edge from the From loop to the To loop is leaving a loop.
|
|
// Either of To and From can be null.
|
|
static bool isExitingLoop(const MachineLoop *From, const MachineLoop *To) {
|
|
return From && !From->contains(To);
|
|
}
|
|
|
|
// MinInstrCountEnsemble - Pick the trace that executes the least number of
|
|
// instructions.
|
|
namespace {
|
|
|
|
class MinInstrCountEnsemble : public MachineTraceMetrics::Ensemble {
|
|
const char *getName() const override { return "MinInstr"; }
|
|
const MachineBasicBlock *pickTracePred(const MachineBasicBlock*) override;
|
|
const MachineBasicBlock *pickTraceSucc(const MachineBasicBlock*) override;
|
|
|
|
public:
|
|
MinInstrCountEnsemble(MachineTraceMetrics *mtm)
|
|
: MachineTraceMetrics::Ensemble(mtm) {}
|
|
};
|
|
|
|
} // end anonymous namespace
|
|
|
|
// Select the preferred predecessor for MBB.
|
|
const MachineBasicBlock*
|
|
MinInstrCountEnsemble::pickTracePred(const MachineBasicBlock *MBB) {
|
|
if (MBB->pred_empty())
|
|
return nullptr;
|
|
const MachineLoop *CurLoop = getLoopFor(MBB);
|
|
// Don't leave loops, and never follow back-edges.
|
|
if (CurLoop && MBB == CurLoop->getHeader())
|
|
return nullptr;
|
|
unsigned CurCount = MTM.getResources(MBB)->InstrCount;
|
|
const MachineBasicBlock *Best = nullptr;
|
|
unsigned BestDepth = 0;
|
|
for (const MachineBasicBlock *Pred : MBB->predecessors()) {
|
|
const MachineTraceMetrics::TraceBlockInfo *PredTBI =
|
|
getDepthResources(Pred);
|
|
// Ignore cycles that aren't natural loops.
|
|
if (!PredTBI)
|
|
continue;
|
|
// Pick the predecessor that would give this block the smallest InstrDepth.
|
|
unsigned Depth = PredTBI->InstrDepth + CurCount;
|
|
if (!Best || Depth < BestDepth) {
|
|
Best = Pred;
|
|
BestDepth = Depth;
|
|
}
|
|
}
|
|
return Best;
|
|
}
|
|
|
|
// Select the preferred successor for MBB.
|
|
const MachineBasicBlock*
|
|
MinInstrCountEnsemble::pickTraceSucc(const MachineBasicBlock *MBB) {
|
|
if (MBB->pred_empty())
|
|
return nullptr;
|
|
const MachineLoop *CurLoop = getLoopFor(MBB);
|
|
const MachineBasicBlock *Best = nullptr;
|
|
unsigned BestHeight = 0;
|
|
for (const MachineBasicBlock *Succ : MBB->successors()) {
|
|
// Don't consider back-edges.
|
|
if (CurLoop && Succ == CurLoop->getHeader())
|
|
continue;
|
|
// Don't consider successors exiting CurLoop.
|
|
if (isExitingLoop(CurLoop, getLoopFor(Succ)))
|
|
continue;
|
|
const MachineTraceMetrics::TraceBlockInfo *SuccTBI =
|
|
getHeightResources(Succ);
|
|
// Ignore cycles that aren't natural loops.
|
|
if (!SuccTBI)
|
|
continue;
|
|
// Pick the successor that would give this block the smallest InstrHeight.
|
|
unsigned Height = SuccTBI->InstrHeight;
|
|
if (!Best || Height < BestHeight) {
|
|
Best = Succ;
|
|
BestHeight = Height;
|
|
}
|
|
}
|
|
return Best;
|
|
}
|
|
|
|
// Get an Ensemble sub-class for the requested trace strategy.
|
|
MachineTraceMetrics::Ensemble *
|
|
MachineTraceMetrics::getEnsemble(MachineTraceMetrics::Strategy strategy) {
|
|
assert(strategy < TS_NumStrategies && "Invalid trace strategy enum");
|
|
Ensemble *&E = Ensembles[strategy];
|
|
if (E)
|
|
return E;
|
|
|
|
// Allocate new Ensemble on demand.
|
|
switch (strategy) {
|
|
case TS_MinInstrCount: return (E = new MinInstrCountEnsemble(this));
|
|
default: llvm_unreachable("Invalid trace strategy enum");
|
|
}
|
|
}
|
|
|
|
void MachineTraceMetrics::invalidate(const MachineBasicBlock *MBB) {
|
|
LLVM_DEBUG(dbgs() << "Invalidate traces through " << printMBBReference(*MBB)
|
|
<< '\n');
|
|
BlockInfo[MBB->getNumber()].invalidate();
|
|
for (unsigned i = 0; i != TS_NumStrategies; ++i)
|
|
if (Ensembles[i])
|
|
Ensembles[i]->invalidate(MBB);
|
|
}
|
|
|
|
void MachineTraceMetrics::verifyAnalysis() const {
|
|
if (!MF)
|
|
return;
|
|
#ifndef NDEBUG
|
|
assert(BlockInfo.size() == MF->getNumBlockIDs() && "Outdated BlockInfo size");
|
|
for (unsigned i = 0; i != TS_NumStrategies; ++i)
|
|
if (Ensembles[i])
|
|
Ensembles[i]->verify();
|
|
#endif
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Trace building
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Traces are built by two CFG traversals. To avoid recomputing too much, use a
|
|
// set abstraction that confines the search to the current loop, and doesn't
|
|
// revisit blocks.
|
|
|
|
namespace {
|
|
|
|
struct LoopBounds {
|
|
MutableArrayRef<MachineTraceMetrics::TraceBlockInfo> Blocks;
|
|
SmallPtrSet<const MachineBasicBlock*, 8> Visited;
|
|
const MachineLoopInfo *Loops;
|
|
bool Downward = false;
|
|
|
|
LoopBounds(MutableArrayRef<MachineTraceMetrics::TraceBlockInfo> blocks,
|
|
const MachineLoopInfo *loops) : Blocks(blocks), Loops(loops) {}
|
|
};
|
|
|
|
} // end anonymous namespace
|
|
|
|
// Specialize po_iterator_storage in order to prune the post-order traversal so
|
|
// it is limited to the current loop and doesn't traverse the loop back edges.
|
|
namespace llvm {
|
|
|
|
template<>
|
|
class po_iterator_storage<LoopBounds, true> {
|
|
LoopBounds &LB;
|
|
|
|
public:
|
|
po_iterator_storage(LoopBounds &lb) : LB(lb) {}
|
|
|
|
void finishPostorder(const MachineBasicBlock*) {}
|
|
|
|
bool insertEdge(Optional<const MachineBasicBlock *> From,
|
|
const MachineBasicBlock *To) {
|
|
// Skip already visited To blocks.
|
|
MachineTraceMetrics::TraceBlockInfo &TBI = LB.Blocks[To->getNumber()];
|
|
if (LB.Downward ? TBI.hasValidHeight() : TBI.hasValidDepth())
|
|
return false;
|
|
// From is null once when To is the trace center block.
|
|
if (From) {
|
|
if (const MachineLoop *FromLoop = LB.Loops->getLoopFor(*From)) {
|
|
// Don't follow backedges, don't leave FromLoop when going upwards.
|
|
if ((LB.Downward ? To : *From) == FromLoop->getHeader())
|
|
return false;
|
|
// Don't leave FromLoop.
|
|
if (isExitingLoop(FromLoop, LB.Loops->getLoopFor(To)))
|
|
return false;
|
|
}
|
|
}
|
|
// To is a new block. Mark the block as visited in case the CFG has cycles
|
|
// that MachineLoopInfo didn't recognize as a natural loop.
|
|
return LB.Visited.insert(To).second;
|
|
}
|
|
};
|
|
|
|
} // end namespace llvm
|
|
|
|
/// Compute the trace through MBB.
|
|
void MachineTraceMetrics::Ensemble::computeTrace(const MachineBasicBlock *MBB) {
|
|
LLVM_DEBUG(dbgs() << "Computing " << getName() << " trace through "
|
|
<< printMBBReference(*MBB) << '\n');
|
|
// Set up loop bounds for the backwards post-order traversal.
|
|
LoopBounds Bounds(BlockInfo, MTM.Loops);
|
|
|
|
// Run an upwards post-order search for the trace start.
|
|
Bounds.Downward = false;
|
|
Bounds.Visited.clear();
|
|
for (auto I : inverse_post_order_ext(MBB, Bounds)) {
|
|
LLVM_DEBUG(dbgs() << " pred for " << printMBBReference(*I) << ": ");
|
|
TraceBlockInfo &TBI = BlockInfo[I->getNumber()];
|
|
// All the predecessors have been visited, pick the preferred one.
|
|
TBI.Pred = pickTracePred(I);
|
|
LLVM_DEBUG({
|
|
if (TBI.Pred)
|
|
dbgs() << printMBBReference(*TBI.Pred) << '\n';
|
|
else
|
|
dbgs() << "null\n";
|
|
});
|
|
// The trace leading to I is now known, compute the depth resources.
|
|
computeDepthResources(I);
|
|
}
|
|
|
|
// Run a downwards post-order search for the trace end.
|
|
Bounds.Downward = true;
|
|
Bounds.Visited.clear();
|
|
for (auto I : post_order_ext(MBB, Bounds)) {
|
|
LLVM_DEBUG(dbgs() << " succ for " << printMBBReference(*I) << ": ");
|
|
TraceBlockInfo &TBI = BlockInfo[I->getNumber()];
|
|
// All the successors have been visited, pick the preferred one.
|
|
TBI.Succ = pickTraceSucc(I);
|
|
LLVM_DEBUG({
|
|
if (TBI.Succ)
|
|
dbgs() << printMBBReference(*TBI.Succ) << '\n';
|
|
else
|
|
dbgs() << "null\n";
|
|
});
|
|
// The trace leaving I is now known, compute the height resources.
|
|
computeHeightResources(I);
|
|
}
|
|
}
|
|
|
|
/// Invalidate traces through BadMBB.
|
|
void
|
|
MachineTraceMetrics::Ensemble::invalidate(const MachineBasicBlock *BadMBB) {
|
|
SmallVector<const MachineBasicBlock*, 16> WorkList;
|
|
TraceBlockInfo &BadTBI = BlockInfo[BadMBB->getNumber()];
|
|
|
|
// Invalidate height resources of blocks above MBB.
|
|
if (BadTBI.hasValidHeight()) {
|
|
BadTBI.invalidateHeight();
|
|
WorkList.push_back(BadMBB);
|
|
do {
|
|
const MachineBasicBlock *MBB = WorkList.pop_back_val();
|
|
LLVM_DEBUG(dbgs() << "Invalidate " << printMBBReference(*MBB) << ' '
|
|
<< getName() << " height.\n");
|
|
// Find any MBB predecessors that have MBB as their preferred successor.
|
|
// They are the only ones that need to be invalidated.
|
|
for (const MachineBasicBlock *Pred : MBB->predecessors()) {
|
|
TraceBlockInfo &TBI = BlockInfo[Pred->getNumber()];
|
|
if (!TBI.hasValidHeight())
|
|
continue;
|
|
if (TBI.Succ == MBB) {
|
|
TBI.invalidateHeight();
|
|
WorkList.push_back(Pred);
|
|
continue;
|
|
}
|
|
// Verify that TBI.Succ is actually a *I successor.
|
|
assert((!TBI.Succ || Pred->isSuccessor(TBI.Succ)) && "CFG changed");
|
|
}
|
|
} while (!WorkList.empty());
|
|
}
|
|
|
|
// Invalidate depth resources of blocks below MBB.
|
|
if (BadTBI.hasValidDepth()) {
|
|
BadTBI.invalidateDepth();
|
|
WorkList.push_back(BadMBB);
|
|
do {
|
|
const MachineBasicBlock *MBB = WorkList.pop_back_val();
|
|
LLVM_DEBUG(dbgs() << "Invalidate " << printMBBReference(*MBB) << ' '
|
|
<< getName() << " depth.\n");
|
|
// Find any MBB successors that have MBB as their preferred predecessor.
|
|
// They are the only ones that need to be invalidated.
|
|
for (const MachineBasicBlock *Succ : MBB->successors()) {
|
|
TraceBlockInfo &TBI = BlockInfo[Succ->getNumber()];
|
|
if (!TBI.hasValidDepth())
|
|
continue;
|
|
if (TBI.Pred == MBB) {
|
|
TBI.invalidateDepth();
|
|
WorkList.push_back(Succ);
|
|
continue;
|
|
}
|
|
// Verify that TBI.Pred is actually a *I predecessor.
|
|
assert((!TBI.Pred || Succ->isPredecessor(TBI.Pred)) && "CFG changed");
|
|
}
|
|
} while (!WorkList.empty());
|
|
}
|
|
|
|
// Clear any per-instruction data. We only have to do this for BadMBB itself
|
|
// because the instructions in that block may change. Other blocks may be
|
|
// invalidated, but their instructions will stay the same, so there is no
|
|
// need to erase the Cycle entries. They will be overwritten when we
|
|
// recompute.
|
|
for (const auto &I : *BadMBB)
|
|
Cycles.erase(&I);
|
|
}
|
|
|
|
void MachineTraceMetrics::Ensemble::verify() const {
|
|
#ifndef NDEBUG
|
|
assert(BlockInfo.size() == MTM.MF->getNumBlockIDs() &&
|
|
"Outdated BlockInfo size");
|
|
for (unsigned Num = 0, e = BlockInfo.size(); Num != e; ++Num) {
|
|
const TraceBlockInfo &TBI = BlockInfo[Num];
|
|
if (TBI.hasValidDepth() && TBI.Pred) {
|
|
const MachineBasicBlock *MBB = MTM.MF->getBlockNumbered(Num);
|
|
assert(MBB->isPredecessor(TBI.Pred) && "CFG doesn't match trace");
|
|
assert(BlockInfo[TBI.Pred->getNumber()].hasValidDepth() &&
|
|
"Trace is broken, depth should have been invalidated.");
|
|
const MachineLoop *Loop = getLoopFor(MBB);
|
|
assert(!(Loop && MBB == Loop->getHeader()) && "Trace contains backedge");
|
|
}
|
|
if (TBI.hasValidHeight() && TBI.Succ) {
|
|
const MachineBasicBlock *MBB = MTM.MF->getBlockNumbered(Num);
|
|
assert(MBB->isSuccessor(TBI.Succ) && "CFG doesn't match trace");
|
|
assert(BlockInfo[TBI.Succ->getNumber()].hasValidHeight() &&
|
|
"Trace is broken, height should have been invalidated.");
|
|
const MachineLoop *Loop = getLoopFor(MBB);
|
|
const MachineLoop *SuccLoop = getLoopFor(TBI.Succ);
|
|
assert(!(Loop && Loop == SuccLoop && TBI.Succ == Loop->getHeader()) &&
|
|
"Trace contains backedge");
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Data Dependencies
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Compute the depth and height of each instruction based on data dependencies
|
|
// and instruction latencies. These cycle numbers assume that the CPU can issue
|
|
// an infinite number of instructions per cycle as long as their dependencies
|
|
// are ready.
|
|
|
|
// A data dependency is represented as a defining MI and operand numbers on the
|
|
// defining and using MI.
|
|
namespace {
|
|
|
|
struct DataDep {
|
|
const MachineInstr *DefMI;
|
|
unsigned DefOp;
|
|
unsigned UseOp;
|
|
|
|
DataDep(const MachineInstr *DefMI, unsigned DefOp, unsigned UseOp)
|
|
: DefMI(DefMI), DefOp(DefOp), UseOp(UseOp) {}
|
|
|
|
/// Create a DataDep from an SSA form virtual register.
|
|
DataDep(const MachineRegisterInfo *MRI, unsigned VirtReg, unsigned UseOp)
|
|
: UseOp(UseOp) {
|
|
assert(Register::isVirtualRegister(VirtReg));
|
|
MachineRegisterInfo::def_iterator DefI = MRI->def_begin(VirtReg);
|
|
assert(!DefI.atEnd() && "Register has no defs");
|
|
DefMI = DefI->getParent();
|
|
DefOp = DefI.getOperandNo();
|
|
assert((++DefI).atEnd() && "Register has multiple defs");
|
|
}
|
|
};
|
|
|
|
} // end anonymous namespace
|
|
|
|
// Get the input data dependencies that must be ready before UseMI can issue.
|
|
// Return true if UseMI has any physreg operands.
|
|
static bool getDataDeps(const MachineInstr &UseMI,
|
|
SmallVectorImpl<DataDep> &Deps,
|
|
const MachineRegisterInfo *MRI) {
|
|
// Debug values should not be included in any calculations.
|
|
if (UseMI.isDebugInstr())
|
|
return false;
|
|
|
|
bool HasPhysRegs = false;
|
|
for (MachineInstr::const_mop_iterator I = UseMI.operands_begin(),
|
|
E = UseMI.operands_end(); I != E; ++I) {
|
|
const MachineOperand &MO = *I;
|
|
if (!MO.isReg())
|
|
continue;
|
|
Register Reg = MO.getReg();
|
|
if (!Reg)
|
|
continue;
|
|
if (Register::isPhysicalRegister(Reg)) {
|
|
HasPhysRegs = true;
|
|
continue;
|
|
}
|
|
// Collect virtual register reads.
|
|
if (MO.readsReg())
|
|
Deps.push_back(DataDep(MRI, Reg, UseMI.getOperandNo(I)));
|
|
}
|
|
return HasPhysRegs;
|
|
}
|
|
|
|
// Get the input data dependencies of a PHI instruction, using Pred as the
|
|
// preferred predecessor.
|
|
// This will add at most one dependency to Deps.
|
|
static void getPHIDeps(const MachineInstr &UseMI,
|
|
SmallVectorImpl<DataDep> &Deps,
|
|
const MachineBasicBlock *Pred,
|
|
const MachineRegisterInfo *MRI) {
|
|
// No predecessor at the beginning of a trace. Ignore dependencies.
|
|
if (!Pred)
|
|
return;
|
|
assert(UseMI.isPHI() && UseMI.getNumOperands() % 2 && "Bad PHI");
|
|
for (unsigned i = 1; i != UseMI.getNumOperands(); i += 2) {
|
|
if (UseMI.getOperand(i + 1).getMBB() == Pred) {
|
|
Register Reg = UseMI.getOperand(i).getReg();
|
|
Deps.push_back(DataDep(MRI, Reg, i));
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Identify physreg dependencies for UseMI, and update the live regunit
|
|
// tracking set when scanning instructions downwards.
|
|
static void updatePhysDepsDownwards(const MachineInstr *UseMI,
|
|
SmallVectorImpl<DataDep> &Deps,
|
|
SparseSet<LiveRegUnit> &RegUnits,
|
|
const TargetRegisterInfo *TRI) {
|
|
SmallVector<unsigned, 8> Kills;
|
|
SmallVector<unsigned, 8> LiveDefOps;
|
|
|
|
for (MachineInstr::const_mop_iterator MI = UseMI->operands_begin(),
|
|
ME = UseMI->operands_end(); MI != ME; ++MI) {
|
|
const MachineOperand &MO = *MI;
|
|
if (!MO.isReg())
|
|
continue;
|
|
Register Reg = MO.getReg();
|
|
if (!Register::isPhysicalRegister(Reg))
|
|
continue;
|
|
// Track live defs and kills for updating RegUnits.
|
|
if (MO.isDef()) {
|
|
if (MO.isDead())
|
|
Kills.push_back(Reg);
|
|
else
|
|
LiveDefOps.push_back(UseMI->getOperandNo(MI));
|
|
} else if (MO.isKill())
|
|
Kills.push_back(Reg);
|
|
// Identify dependencies.
|
|
if (!MO.readsReg())
|
|
continue;
|
|
for (MCRegUnitIterator Units(Reg, TRI); Units.isValid(); ++Units) {
|
|
SparseSet<LiveRegUnit>::iterator I = RegUnits.find(*Units);
|
|
if (I == RegUnits.end())
|
|
continue;
|
|
Deps.push_back(DataDep(I->MI, I->Op, UseMI->getOperandNo(MI)));
|
|
break;
|
|
}
|
|
}
|
|
|
|
// Update RegUnits to reflect live registers after UseMI.
|
|
// First kills.
|
|
for (unsigned Kill : Kills)
|
|
for (MCRegUnitIterator Units(Kill, TRI); Units.isValid(); ++Units)
|
|
RegUnits.erase(*Units);
|
|
|
|
// Second, live defs.
|
|
for (unsigned DefOp : LiveDefOps) {
|
|
for (MCRegUnitIterator Units(UseMI->getOperand(DefOp).getReg(), TRI);
|
|
Units.isValid(); ++Units) {
|
|
LiveRegUnit &LRU = RegUnits[*Units];
|
|
LRU.MI = UseMI;
|
|
LRU.Op = DefOp;
|
|
}
|
|
}
|
|
}
|
|
|
|
/// The length of the critical path through a trace is the maximum of two path
|
|
/// lengths:
|
|
///
|
|
/// 1. The maximum height+depth over all instructions in the trace center block.
|
|
///
|
|
/// 2. The longest cross-block dependency chain. For small blocks, it is
|
|
/// possible that the critical path through the trace doesn't include any
|
|
/// instructions in the block.
|
|
///
|
|
/// This function computes the second number from the live-in list of the
|
|
/// center block.
|
|
unsigned MachineTraceMetrics::Ensemble::
|
|
computeCrossBlockCriticalPath(const TraceBlockInfo &TBI) {
|
|
assert(TBI.HasValidInstrDepths && "Missing depth info");
|
|
assert(TBI.HasValidInstrHeights && "Missing height info");
|
|
unsigned MaxLen = 0;
|
|
for (const LiveInReg &LIR : TBI.LiveIns) {
|
|
if (!Register::isVirtualRegister(LIR.Reg))
|
|
continue;
|
|
const MachineInstr *DefMI = MTM.MRI->getVRegDef(LIR.Reg);
|
|
// Ignore dependencies outside the current trace.
|
|
const TraceBlockInfo &DefTBI = BlockInfo[DefMI->getParent()->getNumber()];
|
|
if (!DefTBI.isUsefulDominator(TBI))
|
|
continue;
|
|
unsigned Len = LIR.Height + Cycles[DefMI].Depth;
|
|
MaxLen = std::max(MaxLen, Len);
|
|
}
|
|
return MaxLen;
|
|
}
|
|
|
|
void MachineTraceMetrics::Ensemble::
|
|
updateDepth(MachineTraceMetrics::TraceBlockInfo &TBI, const MachineInstr &UseMI,
|
|
SparseSet<LiveRegUnit> &RegUnits) {
|
|
SmallVector<DataDep, 8> Deps;
|
|
// Collect all data dependencies.
|
|
if (UseMI.isPHI())
|
|
getPHIDeps(UseMI, Deps, TBI.Pred, MTM.MRI);
|
|
else if (getDataDeps(UseMI, Deps, MTM.MRI))
|
|
updatePhysDepsDownwards(&UseMI, Deps, RegUnits, MTM.TRI);
|
|
|
|
// Filter and process dependencies, computing the earliest issue cycle.
|
|
unsigned Cycle = 0;
|
|
for (const DataDep &Dep : Deps) {
|
|
const TraceBlockInfo&DepTBI =
|
|
BlockInfo[Dep.DefMI->getParent()->getNumber()];
|
|
// Ignore dependencies from outside the current trace.
|
|
if (!DepTBI.isUsefulDominator(TBI))
|
|
continue;
|
|
assert(DepTBI.HasValidInstrDepths && "Inconsistent dependency");
|
|
unsigned DepCycle = Cycles.lookup(Dep.DefMI).Depth;
|
|
// Add latency if DefMI is a real instruction. Transients get latency 0.
|
|
if (!Dep.DefMI->isTransient())
|
|
DepCycle += MTM.SchedModel
|
|
.computeOperandLatency(Dep.DefMI, Dep.DefOp, &UseMI, Dep.UseOp);
|
|
Cycle = std::max(Cycle, DepCycle);
|
|
}
|
|
// Remember the instruction depth.
|
|
InstrCycles &MICycles = Cycles[&UseMI];
|
|
MICycles.Depth = Cycle;
|
|
|
|
if (TBI.HasValidInstrHeights) {
|
|
// Update critical path length.
|
|
TBI.CriticalPath = std::max(TBI.CriticalPath, Cycle + MICycles.Height);
|
|
LLVM_DEBUG(dbgs() << TBI.CriticalPath << '\t' << Cycle << '\t' << UseMI);
|
|
} else {
|
|
LLVM_DEBUG(dbgs() << Cycle << '\t' << UseMI);
|
|
}
|
|
}
|
|
|
|
void MachineTraceMetrics::Ensemble::
|
|
updateDepth(const MachineBasicBlock *MBB, const MachineInstr &UseMI,
|
|
SparseSet<LiveRegUnit> &RegUnits) {
|
|
updateDepth(BlockInfo[MBB->getNumber()], UseMI, RegUnits);
|
|
}
|
|
|
|
void MachineTraceMetrics::Ensemble::
|
|
updateDepths(MachineBasicBlock::iterator Start,
|
|
MachineBasicBlock::iterator End,
|
|
SparseSet<LiveRegUnit> &RegUnits) {
|
|
for (; Start != End; Start++)
|
|
updateDepth(Start->getParent(), *Start, RegUnits);
|
|
}
|
|
|
|
/// Compute instruction depths for all instructions above or in MBB in its
|
|
/// trace. This assumes that the trace through MBB has already been computed.
|
|
void MachineTraceMetrics::Ensemble::
|
|
computeInstrDepths(const MachineBasicBlock *MBB) {
|
|
// The top of the trace may already be computed, and HasValidInstrDepths
|
|
// implies Head->HasValidInstrDepths, so we only need to start from the first
|
|
// block in the trace that needs to be recomputed.
|
|
SmallVector<const MachineBasicBlock*, 8> Stack;
|
|
do {
|
|
TraceBlockInfo &TBI = BlockInfo[MBB->getNumber()];
|
|
assert(TBI.hasValidDepth() && "Incomplete trace");
|
|
if (TBI.HasValidInstrDepths)
|
|
break;
|
|
Stack.push_back(MBB);
|
|
MBB = TBI.Pred;
|
|
} while (MBB);
|
|
|
|
// FIXME: If MBB is non-null at this point, it is the last pre-computed block
|
|
// in the trace. We should track any live-out physregs that were defined in
|
|
// the trace. This is quite rare in SSA form, typically created by CSE
|
|
// hoisting a compare.
|
|
SparseSet<LiveRegUnit> RegUnits;
|
|
RegUnits.setUniverse(MTM.TRI->getNumRegUnits());
|
|
|
|
// Go through trace blocks in top-down order, stopping after the center block.
|
|
while (!Stack.empty()) {
|
|
MBB = Stack.pop_back_val();
|
|
LLVM_DEBUG(dbgs() << "\nDepths for " << printMBBReference(*MBB) << ":\n");
|
|
TraceBlockInfo &TBI = BlockInfo[MBB->getNumber()];
|
|
TBI.HasValidInstrDepths = true;
|
|
TBI.CriticalPath = 0;
|
|
|
|
// Print out resource depths here as well.
|
|
LLVM_DEBUG({
|
|
dbgs() << format("%7u Instructions\n", TBI.InstrDepth);
|
|
ArrayRef<unsigned> PRDepths = getProcResourceDepths(MBB->getNumber());
|
|
for (unsigned K = 0; K != PRDepths.size(); ++K)
|
|
if (PRDepths[K]) {
|
|
unsigned Factor = MTM.SchedModel.getResourceFactor(K);
|
|
dbgs() << format("%6uc @ ", MTM.getCycles(PRDepths[K]))
|
|
<< MTM.SchedModel.getProcResource(K)->Name << " ("
|
|
<< PRDepths[K]/Factor << " ops x" << Factor << ")\n";
|
|
}
|
|
});
|
|
|
|
// Also compute the critical path length through MBB when possible.
|
|
if (TBI.HasValidInstrHeights)
|
|
TBI.CriticalPath = computeCrossBlockCriticalPath(TBI);
|
|
|
|
for (const auto &UseMI : *MBB) {
|
|
updateDepth(TBI, UseMI, RegUnits);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Identify physreg dependencies for MI when scanning instructions upwards.
|
|
// Return the issue height of MI after considering any live regunits.
|
|
// Height is the issue height computed from virtual register dependencies alone.
|
|
static unsigned updatePhysDepsUpwards(const MachineInstr &MI, unsigned Height,
|
|
SparseSet<LiveRegUnit> &RegUnits,
|
|
const TargetSchedModel &SchedModel,
|
|
const TargetInstrInfo *TII,
|
|
const TargetRegisterInfo *TRI) {
|
|
SmallVector<unsigned, 8> ReadOps;
|
|
|
|
for (MachineInstr::const_mop_iterator MOI = MI.operands_begin(),
|
|
MOE = MI.operands_end();
|
|
MOI != MOE; ++MOI) {
|
|
const MachineOperand &MO = *MOI;
|
|
if (!MO.isReg())
|
|
continue;
|
|
Register Reg = MO.getReg();
|
|
if (!Register::isPhysicalRegister(Reg))
|
|
continue;
|
|
if (MO.readsReg())
|
|
ReadOps.push_back(MI.getOperandNo(MOI));
|
|
if (!MO.isDef())
|
|
continue;
|
|
// This is a def of Reg. Remove corresponding entries from RegUnits, and
|
|
// update MI Height to consider the physreg dependencies.
|
|
for (MCRegUnitIterator Units(Reg, TRI); Units.isValid(); ++Units) {
|
|
SparseSet<LiveRegUnit>::iterator I = RegUnits.find(*Units);
|
|
if (I == RegUnits.end())
|
|
continue;
|
|
unsigned DepHeight = I->Cycle;
|
|
if (!MI.isTransient()) {
|
|
// We may not know the UseMI of this dependency, if it came from the
|
|
// live-in list. SchedModel can handle a NULL UseMI.
|
|
DepHeight += SchedModel.computeOperandLatency(&MI, MI.getOperandNo(MOI),
|
|
I->MI, I->Op);
|
|
}
|
|
Height = std::max(Height, DepHeight);
|
|
// This regunit is dead above MI.
|
|
RegUnits.erase(I);
|
|
}
|
|
}
|
|
|
|
// Now we know the height of MI. Update any regunits read.
|
|
for (unsigned i = 0, e = ReadOps.size(); i != e; ++i) {
|
|
Register Reg = MI.getOperand(ReadOps[i]).getReg();
|
|
for (MCRegUnitIterator Units(Reg, TRI); Units.isValid(); ++Units) {
|
|
LiveRegUnit &LRU = RegUnits[*Units];
|
|
// Set the height to the highest reader of the unit.
|
|
if (LRU.Cycle <= Height && LRU.MI != &MI) {
|
|
LRU.Cycle = Height;
|
|
LRU.MI = &MI;
|
|
LRU.Op = ReadOps[i];
|
|
}
|
|
}
|
|
}
|
|
|
|
return Height;
|
|
}
|
|
|
|
using MIHeightMap = DenseMap<const MachineInstr *, unsigned>;
|
|
|
|
// Push the height of DefMI upwards if required to match UseMI.
|
|
// Return true if this is the first time DefMI was seen.
|
|
static bool pushDepHeight(const DataDep &Dep, const MachineInstr &UseMI,
|
|
unsigned UseHeight, MIHeightMap &Heights,
|
|
const TargetSchedModel &SchedModel,
|
|
const TargetInstrInfo *TII) {
|
|
// Adjust height by Dep.DefMI latency.
|
|
if (!Dep.DefMI->isTransient())
|
|
UseHeight += SchedModel.computeOperandLatency(Dep.DefMI, Dep.DefOp, &UseMI,
|
|
Dep.UseOp);
|
|
|
|
// Update Heights[DefMI] to be the maximum height seen.
|
|
MIHeightMap::iterator I;
|
|
bool New;
|
|
std::tie(I, New) = Heights.insert(std::make_pair(Dep.DefMI, UseHeight));
|
|
if (New)
|
|
return true;
|
|
|
|
// DefMI has been pushed before. Give it the max height.
|
|
if (I->second < UseHeight)
|
|
I->second = UseHeight;
|
|
return false;
|
|
}
|
|
|
|
/// Assuming that the virtual register defined by DefMI:DefOp was used by
|
|
/// Trace.back(), add it to the live-in lists of all the blocks in Trace. Stop
|
|
/// when reaching the block that contains DefMI.
|
|
void MachineTraceMetrics::Ensemble::
|
|
addLiveIns(const MachineInstr *DefMI, unsigned DefOp,
|
|
ArrayRef<const MachineBasicBlock*> Trace) {
|
|
assert(!Trace.empty() && "Trace should contain at least one block");
|
|
unsigned Reg = DefMI->getOperand(DefOp).getReg();
|
|
assert(Register::isVirtualRegister(Reg));
|
|
const MachineBasicBlock *DefMBB = DefMI->getParent();
|
|
|
|
// Reg is live-in to all blocks in Trace that follow DefMBB.
|
|
for (unsigned i = Trace.size(); i; --i) {
|
|
const MachineBasicBlock *MBB = Trace[i-1];
|
|
if (MBB == DefMBB)
|
|
return;
|
|
TraceBlockInfo &TBI = BlockInfo[MBB->getNumber()];
|
|
// Just add the register. The height will be updated later.
|
|
TBI.LiveIns.push_back(Reg);
|
|
}
|
|
}
|
|
|
|
/// Compute instruction heights in the trace through MBB. This updates MBB and
|
|
/// the blocks below it in the trace. It is assumed that the trace has already
|
|
/// been computed.
|
|
void MachineTraceMetrics::Ensemble::
|
|
computeInstrHeights(const MachineBasicBlock *MBB) {
|
|
// The bottom of the trace may already be computed.
|
|
// Find the blocks that need updating.
|
|
SmallVector<const MachineBasicBlock*, 8> Stack;
|
|
do {
|
|
TraceBlockInfo &TBI = BlockInfo[MBB->getNumber()];
|
|
assert(TBI.hasValidHeight() && "Incomplete trace");
|
|
if (TBI.HasValidInstrHeights)
|
|
break;
|
|
Stack.push_back(MBB);
|
|
TBI.LiveIns.clear();
|
|
MBB = TBI.Succ;
|
|
} while (MBB);
|
|
|
|
// As we move upwards in the trace, keep track of instructions that are
|
|
// required by deeper trace instructions. Map MI -> height required so far.
|
|
MIHeightMap Heights;
|
|
|
|
// For physregs, the def isn't known when we see the use.
|
|
// Instead, keep track of the highest use of each regunit.
|
|
SparseSet<LiveRegUnit> RegUnits;
|
|
RegUnits.setUniverse(MTM.TRI->getNumRegUnits());
|
|
|
|
// If the bottom of the trace was already precomputed, initialize heights
|
|
// from its live-in list.
|
|
// MBB is the highest precomputed block in the trace.
|
|
if (MBB) {
|
|
TraceBlockInfo &TBI = BlockInfo[MBB->getNumber()];
|
|
for (LiveInReg &LI : TBI.LiveIns) {
|
|
if (Register::isVirtualRegister(LI.Reg)) {
|
|
// For virtual registers, the def latency is included.
|
|
unsigned &Height = Heights[MTM.MRI->getVRegDef(LI.Reg)];
|
|
if (Height < LI.Height)
|
|
Height = LI.Height;
|
|
} else {
|
|
// For register units, the def latency is not included because we don't
|
|
// know the def yet.
|
|
RegUnits[LI.Reg].Cycle = LI.Height;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Go through the trace blocks in bottom-up order.
|
|
SmallVector<DataDep, 8> Deps;
|
|
for (;!Stack.empty(); Stack.pop_back()) {
|
|
MBB = Stack.back();
|
|
LLVM_DEBUG(dbgs() << "Heights for " << printMBBReference(*MBB) << ":\n");
|
|
TraceBlockInfo &TBI = BlockInfo[MBB->getNumber()];
|
|
TBI.HasValidInstrHeights = true;
|
|
TBI.CriticalPath = 0;
|
|
|
|
LLVM_DEBUG({
|
|
dbgs() << format("%7u Instructions\n", TBI.InstrHeight);
|
|
ArrayRef<unsigned> PRHeights = getProcResourceHeights(MBB->getNumber());
|
|
for (unsigned K = 0; K != PRHeights.size(); ++K)
|
|
if (PRHeights[K]) {
|
|
unsigned Factor = MTM.SchedModel.getResourceFactor(K);
|
|
dbgs() << format("%6uc @ ", MTM.getCycles(PRHeights[K]))
|
|
<< MTM.SchedModel.getProcResource(K)->Name << " ("
|
|
<< PRHeights[K]/Factor << " ops x" << Factor << ")\n";
|
|
}
|
|
});
|
|
|
|
// Get dependencies from PHIs in the trace successor.
|
|
const MachineBasicBlock *Succ = TBI.Succ;
|
|
// If MBB is the last block in the trace, and it has a back-edge to the
|
|
// loop header, get loop-carried dependencies from PHIs in the header. For
|
|
// that purpose, pretend that all the loop header PHIs have height 0.
|
|
if (!Succ)
|
|
if (const MachineLoop *Loop = getLoopFor(MBB))
|
|
if (MBB->isSuccessor(Loop->getHeader()))
|
|
Succ = Loop->getHeader();
|
|
|
|
if (Succ) {
|
|
for (const auto &PHI : *Succ) {
|
|
if (!PHI.isPHI())
|
|
break;
|
|
Deps.clear();
|
|
getPHIDeps(PHI, Deps, MBB, MTM.MRI);
|
|
if (!Deps.empty()) {
|
|
// Loop header PHI heights are all 0.
|
|
unsigned Height = TBI.Succ ? Cycles.lookup(&PHI).Height : 0;
|
|
LLVM_DEBUG(dbgs() << "pred\t" << Height << '\t' << PHI);
|
|
if (pushDepHeight(Deps.front(), PHI, Height, Heights, MTM.SchedModel,
|
|
MTM.TII))
|
|
addLiveIns(Deps.front().DefMI, Deps.front().DefOp, Stack);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Go through the block backwards.
|
|
for (MachineBasicBlock::const_iterator BI = MBB->end(), BB = MBB->begin();
|
|
BI != BB;) {
|
|
const MachineInstr &MI = *--BI;
|
|
|
|
// Find the MI height as determined by virtual register uses in the
|
|
// trace below.
|
|
unsigned Cycle = 0;
|
|
MIHeightMap::iterator HeightI = Heights.find(&MI);
|
|
if (HeightI != Heights.end()) {
|
|
Cycle = HeightI->second;
|
|
// We won't be seeing any more MI uses.
|
|
Heights.erase(HeightI);
|
|
}
|
|
|
|
// Don't process PHI deps. They depend on the specific predecessor, and
|
|
// we'll get them when visiting the predecessor.
|
|
Deps.clear();
|
|
bool HasPhysRegs = !MI.isPHI() && getDataDeps(MI, Deps, MTM.MRI);
|
|
|
|
// There may also be regunit dependencies to include in the height.
|
|
if (HasPhysRegs)
|
|
Cycle = updatePhysDepsUpwards(MI, Cycle, RegUnits, MTM.SchedModel,
|
|
MTM.TII, MTM.TRI);
|
|
|
|
// Update the required height of any virtual registers read by MI.
|
|
for (const DataDep &Dep : Deps)
|
|
if (pushDepHeight(Dep, MI, Cycle, Heights, MTM.SchedModel, MTM.TII))
|
|
addLiveIns(Dep.DefMI, Dep.DefOp, Stack);
|
|
|
|
InstrCycles &MICycles = Cycles[&MI];
|
|
MICycles.Height = Cycle;
|
|
if (!TBI.HasValidInstrDepths) {
|
|
LLVM_DEBUG(dbgs() << Cycle << '\t' << MI);
|
|
continue;
|
|
}
|
|
// Update critical path length.
|
|
TBI.CriticalPath = std::max(TBI.CriticalPath, Cycle + MICycles.Depth);
|
|
LLVM_DEBUG(dbgs() << TBI.CriticalPath << '\t' << Cycle << '\t' << MI);
|
|
}
|
|
|
|
// Update virtual live-in heights. They were added by addLiveIns() with a 0
|
|
// height because the final height isn't known until now.
|
|
LLVM_DEBUG(dbgs() << printMBBReference(*MBB) << " Live-ins:");
|
|
for (LiveInReg &LIR : TBI.LiveIns) {
|
|
const MachineInstr *DefMI = MTM.MRI->getVRegDef(LIR.Reg);
|
|
LIR.Height = Heights.lookup(DefMI);
|
|
LLVM_DEBUG(dbgs() << ' ' << printReg(LIR.Reg) << '@' << LIR.Height);
|
|
}
|
|
|
|
// Transfer the live regunits to the live-in list.
|
|
for (SparseSet<LiveRegUnit>::const_iterator
|
|
RI = RegUnits.begin(), RE = RegUnits.end(); RI != RE; ++RI) {
|
|
TBI.LiveIns.push_back(LiveInReg(RI->RegUnit, RI->Cycle));
|
|
LLVM_DEBUG(dbgs() << ' ' << printRegUnit(RI->RegUnit, MTM.TRI) << '@'
|
|
<< RI->Cycle);
|
|
}
|
|
LLVM_DEBUG(dbgs() << '\n');
|
|
|
|
if (!TBI.HasValidInstrDepths)
|
|
continue;
|
|
// Add live-ins to the critical path length.
|
|
TBI.CriticalPath = std::max(TBI.CriticalPath,
|
|
computeCrossBlockCriticalPath(TBI));
|
|
LLVM_DEBUG(dbgs() << "Critical path: " << TBI.CriticalPath << '\n');
|
|
}
|
|
}
|
|
|
|
MachineTraceMetrics::Trace
|
|
MachineTraceMetrics::Ensemble::getTrace(const MachineBasicBlock *MBB) {
|
|
TraceBlockInfo &TBI = BlockInfo[MBB->getNumber()];
|
|
|
|
if (!TBI.hasValidDepth() || !TBI.hasValidHeight())
|
|
computeTrace(MBB);
|
|
if (!TBI.HasValidInstrDepths)
|
|
computeInstrDepths(MBB);
|
|
if (!TBI.HasValidInstrHeights)
|
|
computeInstrHeights(MBB);
|
|
|
|
return Trace(*this, TBI);
|
|
}
|
|
|
|
unsigned
|
|
MachineTraceMetrics::Trace::getInstrSlack(const MachineInstr &MI) const {
|
|
assert(getBlockNum() == unsigned(MI.getParent()->getNumber()) &&
|
|
"MI must be in the trace center block");
|
|
InstrCycles Cyc = getInstrCycles(MI);
|
|
return getCriticalPath() - (Cyc.Depth + Cyc.Height);
|
|
}
|
|
|
|
unsigned
|
|
MachineTraceMetrics::Trace::getPHIDepth(const MachineInstr &PHI) const {
|
|
const MachineBasicBlock *MBB = TE.MTM.MF->getBlockNumbered(getBlockNum());
|
|
SmallVector<DataDep, 1> Deps;
|
|
getPHIDeps(PHI, Deps, MBB, TE.MTM.MRI);
|
|
assert(Deps.size() == 1 && "PHI doesn't have MBB as a predecessor");
|
|
DataDep &Dep = Deps.front();
|
|
unsigned DepCycle = getInstrCycles(*Dep.DefMI).Depth;
|
|
// Add latency if DefMI is a real instruction. Transients get latency 0.
|
|
if (!Dep.DefMI->isTransient())
|
|
DepCycle += TE.MTM.SchedModel.computeOperandLatency(Dep.DefMI, Dep.DefOp,
|
|
&PHI, Dep.UseOp);
|
|
return DepCycle;
|
|
}
|
|
|
|
/// When bottom is set include instructions in current block in estimate.
|
|
unsigned MachineTraceMetrics::Trace::getResourceDepth(bool Bottom) const {
|
|
// Find the limiting processor resource.
|
|
// Numbers have been pre-scaled to be comparable.
|
|
unsigned PRMax = 0;
|
|
ArrayRef<unsigned> PRDepths = TE.getProcResourceDepths(getBlockNum());
|
|
if (Bottom) {
|
|
ArrayRef<unsigned> PRCycles = TE.MTM.getProcResourceCycles(getBlockNum());
|
|
for (unsigned K = 0; K != PRDepths.size(); ++K)
|
|
PRMax = std::max(PRMax, PRDepths[K] + PRCycles[K]);
|
|
} else {
|
|
for (unsigned K = 0; K != PRDepths.size(); ++K)
|
|
PRMax = std::max(PRMax, PRDepths[K]);
|
|
}
|
|
// Convert to cycle count.
|
|
PRMax = TE.MTM.getCycles(PRMax);
|
|
|
|
/// All instructions before current block
|
|
unsigned Instrs = TBI.InstrDepth;
|
|
// plus instructions in current block
|
|
if (Bottom)
|
|
Instrs += TE.MTM.BlockInfo[getBlockNum()].InstrCount;
|
|
if (unsigned IW = TE.MTM.SchedModel.getIssueWidth())
|
|
Instrs /= IW;
|
|
// Assume issue width 1 without a schedule model.
|
|
return std::max(Instrs, PRMax);
|
|
}
|
|
|
|
unsigned MachineTraceMetrics::Trace::getResourceLength(
|
|
ArrayRef<const MachineBasicBlock *> Extrablocks,
|
|
ArrayRef<const MCSchedClassDesc *> ExtraInstrs,
|
|
ArrayRef<const MCSchedClassDesc *> RemoveInstrs) const {
|
|
// Add up resources above and below the center block.
|
|
ArrayRef<unsigned> PRDepths = TE.getProcResourceDepths(getBlockNum());
|
|
ArrayRef<unsigned> PRHeights = TE.getProcResourceHeights(getBlockNum());
|
|
unsigned PRMax = 0;
|
|
|
|
// Capture computing cycles from extra instructions
|
|
auto extraCycles = [this](ArrayRef<const MCSchedClassDesc *> Instrs,
|
|
unsigned ResourceIdx)
|
|
->unsigned {
|
|
unsigned Cycles = 0;
|
|
for (const MCSchedClassDesc *SC : Instrs) {
|
|
if (!SC->isValid())
|
|
continue;
|
|
for (TargetSchedModel::ProcResIter
|
|
PI = TE.MTM.SchedModel.getWriteProcResBegin(SC),
|
|
PE = TE.MTM.SchedModel.getWriteProcResEnd(SC);
|
|
PI != PE; ++PI) {
|
|
if (PI->ProcResourceIdx != ResourceIdx)
|
|
continue;
|
|
Cycles +=
|
|
(PI->Cycles * TE.MTM.SchedModel.getResourceFactor(ResourceIdx));
|
|
}
|
|
}
|
|
return Cycles;
|
|
};
|
|
|
|
for (unsigned K = 0; K != PRDepths.size(); ++K) {
|
|
unsigned PRCycles = PRDepths[K] + PRHeights[K];
|
|
for (const MachineBasicBlock *MBB : Extrablocks)
|
|
PRCycles += TE.MTM.getProcResourceCycles(MBB->getNumber())[K];
|
|
PRCycles += extraCycles(ExtraInstrs, K);
|
|
PRCycles -= extraCycles(RemoveInstrs, K);
|
|
PRMax = std::max(PRMax, PRCycles);
|
|
}
|
|
// Convert to cycle count.
|
|
PRMax = TE.MTM.getCycles(PRMax);
|
|
|
|
// Instrs: #instructions in current trace outside current block.
|
|
unsigned Instrs = TBI.InstrDepth + TBI.InstrHeight;
|
|
// Add instruction count from the extra blocks.
|
|
for (const MachineBasicBlock *MBB : Extrablocks)
|
|
Instrs += TE.MTM.getResources(MBB)->InstrCount;
|
|
Instrs += ExtraInstrs.size();
|
|
Instrs -= RemoveInstrs.size();
|
|
if (unsigned IW = TE.MTM.SchedModel.getIssueWidth())
|
|
Instrs /= IW;
|
|
// Assume issue width 1 without a schedule model.
|
|
return std::max(Instrs, PRMax);
|
|
}
|
|
|
|
bool MachineTraceMetrics::Trace::isDepInTrace(const MachineInstr &DefMI,
|
|
const MachineInstr &UseMI) const {
|
|
if (DefMI.getParent() == UseMI.getParent())
|
|
return true;
|
|
|
|
const TraceBlockInfo &DepTBI = TE.BlockInfo[DefMI.getParent()->getNumber()];
|
|
const TraceBlockInfo &TBI = TE.BlockInfo[UseMI.getParent()->getNumber()];
|
|
|
|
return DepTBI.isUsefulDominator(TBI);
|
|
}
|
|
|
|
void MachineTraceMetrics::Ensemble::print(raw_ostream &OS) const {
|
|
OS << getName() << " ensemble:\n";
|
|
for (unsigned i = 0, e = BlockInfo.size(); i != e; ++i) {
|
|
OS << " %bb." << i << '\t';
|
|
BlockInfo[i].print(OS);
|
|
OS << '\n';
|
|
}
|
|
}
|
|
|
|
void MachineTraceMetrics::TraceBlockInfo::print(raw_ostream &OS) const {
|
|
if (hasValidDepth()) {
|
|
OS << "depth=" << InstrDepth;
|
|
if (Pred)
|
|
OS << " pred=" << printMBBReference(*Pred);
|
|
else
|
|
OS << " pred=null";
|
|
OS << " head=%bb." << Head;
|
|
if (HasValidInstrDepths)
|
|
OS << " +instrs";
|
|
} else
|
|
OS << "depth invalid";
|
|
OS << ", ";
|
|
if (hasValidHeight()) {
|
|
OS << "height=" << InstrHeight;
|
|
if (Succ)
|
|
OS << " succ=" << printMBBReference(*Succ);
|
|
else
|
|
OS << " succ=null";
|
|
OS << " tail=%bb." << Tail;
|
|
if (HasValidInstrHeights)
|
|
OS << " +instrs";
|
|
} else
|
|
OS << "height invalid";
|
|
if (HasValidInstrDepths && HasValidInstrHeights)
|
|
OS << ", crit=" << CriticalPath;
|
|
}
|
|
|
|
void MachineTraceMetrics::Trace::print(raw_ostream &OS) const {
|
|
unsigned MBBNum = &TBI - &TE.BlockInfo[0];
|
|
|
|
OS << TE.getName() << " trace %bb." << TBI.Head << " --> %bb." << MBBNum
|
|
<< " --> %bb." << TBI.Tail << ':';
|
|
if (TBI.hasValidHeight() && TBI.hasValidDepth())
|
|
OS << ' ' << getInstrCount() << " instrs.";
|
|
if (TBI.HasValidInstrDepths && TBI.HasValidInstrHeights)
|
|
OS << ' ' << TBI.CriticalPath << " cycles.";
|
|
|
|
const MachineTraceMetrics::TraceBlockInfo *Block = &TBI;
|
|
OS << "\n%bb." << MBBNum;
|
|
while (Block->hasValidDepth() && Block->Pred) {
|
|
unsigned Num = Block->Pred->getNumber();
|
|
OS << " <- " << printMBBReference(*Block->Pred);
|
|
Block = &TE.BlockInfo[Num];
|
|
}
|
|
|
|
Block = &TBI;
|
|
OS << "\n ";
|
|
while (Block->hasValidHeight() && Block->Succ) {
|
|
unsigned Num = Block->Succ->getNumber();
|
|
OS << " -> " << printMBBReference(*Block->Succ);
|
|
Block = &TE.BlockInfo[Num];
|
|
}
|
|
OS << '\n';
|
|
}
|