2017-07-22 05:37:46 +08:00
|
|
|
//===- BranchProbabilityInfo.cpp - Branch Probability Analysis ------------===//
|
2011-06-04 09:16:30 +08:00
|
|
|
//
|
|
|
|
// The LLVM Compiler Infrastructure
|
|
|
|
//
|
|
|
|
// This file is distributed under the University of Illinois Open Source
|
|
|
|
// License. See LICENSE.TXT for details.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
//
|
|
|
|
// Loops should be simplified before this analysis.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
2012-12-04 00:50:05 +08:00
|
|
|
#include "llvm/Analysis/BranchProbabilityInfo.h"
|
|
|
|
#include "llvm/ADT/PostOrderIterator.h"
|
2017-07-22 05:37:46 +08:00
|
|
|
#include "llvm/ADT/STLExtras.h"
|
|
|
|
#include "llvm/ADT/SmallVector.h"
|
2012-12-04 00:50:05 +08:00
|
|
|
#include "llvm/Analysis/LoopInfo.h"
|
2017-06-08 17:44:40 +08:00
|
|
|
#include "llvm/Analysis/TargetLibraryInfo.h"
|
2017-07-22 05:37:46 +08:00
|
|
|
#include "llvm/IR/Attributes.h"
|
|
|
|
#include "llvm/IR/BasicBlock.h"
|
2014-03-04 19:45:46 +08:00
|
|
|
#include "llvm/IR/CFG.h"
|
2013-01-02 19:36:10 +08:00
|
|
|
#include "llvm/IR/Constants.h"
|
|
|
|
#include "llvm/IR/Function.h"
|
2017-07-22 05:37:46 +08:00
|
|
|
#include "llvm/IR/InstrTypes.h"
|
|
|
|
#include "llvm/IR/Instruction.h"
|
2013-01-02 19:36:10 +08:00
|
|
|
#include "llvm/IR/Instructions.h"
|
|
|
|
#include "llvm/IR/LLVMContext.h"
|
|
|
|
#include "llvm/IR/Metadata.h"
|
2017-07-22 05:37:46 +08:00
|
|
|
#include "llvm/IR/PassManager.h"
|
|
|
|
#include "llvm/IR/Type.h"
|
|
|
|
#include "llvm/IR/Value.h"
|
|
|
|
#include "llvm/Pass.h"
|
|
|
|
#include "llvm/Support/BranchProbability.h"
|
|
|
|
#include "llvm/Support/Casting.h"
|
2011-06-11 09:05:22 +08:00
|
|
|
#include "llvm/Support/Debug.h"
|
2015-03-24 02:07:13 +08:00
|
|
|
#include "llvm/Support/raw_ostream.h"
|
2017-07-22 05:37:46 +08:00
|
|
|
#include <cassert>
|
|
|
|
#include <cstdint>
|
|
|
|
#include <iterator>
|
|
|
|
#include <utility>
|
2011-06-04 09:16:30 +08:00
|
|
|
|
|
|
|
using namespace llvm;
|
|
|
|
|
2014-04-22 10:48:03 +08:00
|
|
|
#define DEBUG_TYPE "branch-prob"
|
|
|
|
|
2017-08-26 08:31:00 +08:00
|
|
|
static cl::opt<bool> PrintBranchProb(
|
|
|
|
"print-bpi", cl::init(false), cl::Hidden,
|
|
|
|
cl::desc("Print the branch probability info."));
|
|
|
|
|
|
|
|
cl::opt<std::string> PrintBranchProbFuncName(
|
|
|
|
"print-bpi-func-name", cl::Hidden,
|
|
|
|
cl::desc("The option to specify the name of the function "
|
|
|
|
"whose branch probability info is printed."));
|
|
|
|
|
2015-07-16 06:48:29 +08:00
|
|
|
INITIALIZE_PASS_BEGIN(BranchProbabilityInfoWrapperPass, "branch-prob",
|
2011-06-04 09:16:30 +08:00
|
|
|
"Branch Probability Analysis", false, true)
|
2015-01-17 22:16:18 +08:00
|
|
|
INITIALIZE_PASS_DEPENDENCY(LoopInfoWrapperPass)
|
2017-06-08 17:44:40 +08:00
|
|
|
INITIALIZE_PASS_DEPENDENCY(TargetLibraryInfoWrapperPass)
|
2015-07-16 06:48:29 +08:00
|
|
|
INITIALIZE_PASS_END(BranchProbabilityInfoWrapperPass, "branch-prob",
|
2011-06-04 09:16:30 +08:00
|
|
|
"Branch Probability Analysis", false, true)
|
|
|
|
|
2015-07-16 06:48:29 +08:00
|
|
|
char BranchProbabilityInfoWrapperPass::ID = 0;
|
2011-06-04 09:16:30 +08:00
|
|
|
|
2011-10-24 09:40:45 +08:00
|
|
|
// Weights are for internal use only. They are used by heuristics to help to
|
|
|
|
// estimate edges' probability. Example:
|
|
|
|
//
|
|
|
|
// Using "Loop Branch Heuristics" we predict weights of edges for the
|
|
|
|
// block BB2.
|
|
|
|
// ...
|
|
|
|
// |
|
|
|
|
// V
|
|
|
|
// BB1<-+
|
|
|
|
// | |
|
|
|
|
// | | (Weight = 124)
|
|
|
|
// V |
|
|
|
|
// BB2--+
|
|
|
|
// |
|
|
|
|
// | (Weight = 4)
|
|
|
|
// V
|
|
|
|
// BB3
|
|
|
|
//
|
|
|
|
// Probability of the edge BB2->BB1 = 124 / (124 + 4) = 0.96875
|
|
|
|
// Probability of the edge BB2->BB3 = 4 / (124 + 4) = 0.03125
|
|
|
|
static const uint32_t LBH_TAKEN_WEIGHT = 124;
|
|
|
|
static const uint32_t LBH_NONTAKEN_WEIGHT = 4;
|
2011-06-04 09:16:30 +08:00
|
|
|
|
2017-05-18 14:11:56 +08:00
|
|
|
/// \brief Unreachable-terminating branch taken probability.
|
2011-10-24 20:01:08 +08:00
|
|
|
///
|
2017-05-18 14:11:56 +08:00
|
|
|
/// This is the probability for a branch being taken to a block that terminates
|
2011-10-24 20:01:08 +08:00
|
|
|
/// (eventually) in unreachable. These are predicted as unlikely as possible.
|
2017-05-18 14:11:56 +08:00
|
|
|
/// All reachable probability will equally share the remaining part.
|
|
|
|
static const BranchProbability UR_TAKEN_PROB = BranchProbability::getRaw(1);
|
2017-04-17 12:33:04 +08:00
|
|
|
|
2013-05-24 20:26:52 +08:00
|
|
|
/// \brief Weight for a branch taken going into a cold block.
|
|
|
|
///
|
|
|
|
/// This is the weight for a branch taken toward a block marked
|
|
|
|
/// cold. A block is marked cold if it's postdominated by a
|
|
|
|
/// block containing a call to a cold function. Cold functions
|
|
|
|
/// are those marked with attribute 'cold'.
|
|
|
|
static const uint32_t CC_TAKEN_WEIGHT = 4;
|
|
|
|
|
|
|
|
/// \brief Weight for a branch not-taken into a cold block.
|
|
|
|
///
|
|
|
|
/// This is the weight for a branch not taken toward a block marked
|
|
|
|
/// cold.
|
|
|
|
static const uint32_t CC_NONTAKEN_WEIGHT = 64;
|
|
|
|
|
2011-10-24 09:40:45 +08:00
|
|
|
static const uint32_t PH_TAKEN_WEIGHT = 20;
|
|
|
|
static const uint32_t PH_NONTAKEN_WEIGHT = 12;
|
2011-06-04 09:16:30 +08:00
|
|
|
|
2011-10-24 09:40:45 +08:00
|
|
|
static const uint32_t ZH_TAKEN_WEIGHT = 20;
|
|
|
|
static const uint32_t ZH_NONTAKEN_WEIGHT = 12;
|
|
|
|
|
|
|
|
static const uint32_t FPH_TAKEN_WEIGHT = 20;
|
|
|
|
static const uint32_t FPH_NONTAKEN_WEIGHT = 12;
|
|
|
|
|
2012-08-15 20:22:35 +08:00
|
|
|
/// \brief Invoke-terminating normal branch taken weight
|
|
|
|
///
|
|
|
|
/// This is the weight for branching to the normal destination of an invoke
|
|
|
|
/// instruction. We expect this to happen most of the time. Set the weight to an
|
|
|
|
/// absurdly high value so that nested loops subsume it.
|
|
|
|
static const uint32_t IH_TAKEN_WEIGHT = 1024 * 1024 - 1;
|
|
|
|
|
|
|
|
/// \brief Invoke-terminating normal branch not-taken weight.
|
|
|
|
///
|
|
|
|
/// This is the weight for branching to the unwind destination of an invoke
|
|
|
|
/// instruction. This is essentially never taken.
|
|
|
|
static const uint32_t IH_NONTAKEN_WEIGHT = 1;
|
|
|
|
|
2017-04-12 13:42:14 +08:00
|
|
|
/// \brief Add \p BB to PostDominatedByUnreachable set if applicable.
|
|
|
|
void
|
|
|
|
BranchProbabilityInfo::updatePostDominatedByUnreachable(const BasicBlock *BB) {
|
2016-04-08 05:59:28 +08:00
|
|
|
const TerminatorInst *TI = BB->getTerminator();
|
2011-10-24 20:01:08 +08:00
|
|
|
if (TI->getNumSuccessors() == 0) {
|
2016-04-19 03:01:28 +08:00
|
|
|
if (isa<UnreachableInst>(TI) ||
|
|
|
|
// If this block is terminated by a call to
|
|
|
|
// @llvm.experimental.deoptimize then treat it like an unreachable since
|
|
|
|
// the @llvm.experimental.deoptimize call is expected to practically
|
|
|
|
// never execute.
|
|
|
|
BB->getTerminatingDeoptimizeCall())
|
2011-10-24 20:01:08 +08:00
|
|
|
PostDominatedByUnreachable.insert(BB);
|
2017-04-12 13:42:14 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
// If the terminator is an InvokeInst, check only the normal destination block
|
|
|
|
// as the unwind edge of InvokeInst is also very unlikely taken.
|
|
|
|
if (auto *II = dyn_cast<InvokeInst>(TI)) {
|
|
|
|
if (PostDominatedByUnreachable.count(II->getNormalDest()))
|
|
|
|
PostDominatedByUnreachable.insert(BB);
|
|
|
|
return;
|
2011-10-24 20:01:08 +08:00
|
|
|
}
|
2011-10-24 09:40:45 +08:00
|
|
|
|
2017-04-12 13:42:14 +08:00
|
|
|
for (auto *I : successors(BB))
|
|
|
|
// If any of successor is not post dominated then BB is also not.
|
|
|
|
if (!PostDominatedByUnreachable.count(I))
|
|
|
|
return;
|
|
|
|
|
|
|
|
PostDominatedByUnreachable.insert(BB);
|
|
|
|
}
|
|
|
|
|
|
|
|
/// \brief Add \p BB to PostDominatedByColdCall set if applicable.
|
|
|
|
void
|
|
|
|
BranchProbabilityInfo::updatePostDominatedByColdCall(const BasicBlock *BB) {
|
|
|
|
assert(!PostDominatedByColdCall.count(BB));
|
|
|
|
const TerminatorInst *TI = BB->getTerminator();
|
|
|
|
if (TI->getNumSuccessors() == 0)
|
|
|
|
return;
|
|
|
|
|
|
|
|
// If all of successor are post dominated then BB is also done.
|
|
|
|
if (llvm::all_of(successors(BB), [&](const BasicBlock *SuccBB) {
|
|
|
|
return PostDominatedByColdCall.count(SuccBB);
|
|
|
|
})) {
|
|
|
|
PostDominatedByColdCall.insert(BB);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
// If the terminator is an InvokeInst, check only the normal destination
|
|
|
|
// block as the unwind edge of InvokeInst is also very unlikely taken.
|
|
|
|
if (auto *II = dyn_cast<InvokeInst>(TI))
|
|
|
|
if (PostDominatedByColdCall.count(II->getNormalDest())) {
|
|
|
|
PostDominatedByColdCall.insert(BB);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Otherwise, if the block itself contains a cold function, add it to the
|
|
|
|
// set of blocks post-dominated by a cold call.
|
|
|
|
for (auto &I : *BB)
|
|
|
|
if (const CallInst *CI = dyn_cast<CallInst>(&I))
|
|
|
|
if (CI->hasFnAttr(Attribute::Cold)) {
|
|
|
|
PostDominatedByColdCall.insert(BB);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// \brief Calculate edge weights for successors lead to unreachable.
|
|
|
|
///
|
|
|
|
/// Predict that a successor which leads necessarily to an
|
|
|
|
/// unreachable-terminated block as extremely unlikely.
|
|
|
|
bool BranchProbabilityInfo::calcUnreachableHeuristics(const BasicBlock *BB) {
|
|
|
|
const TerminatorInst *TI = BB->getTerminator();
|
2017-04-17 14:39:47 +08:00
|
|
|
assert(TI->getNumSuccessors() > 1 && "expected more than one successor!");
|
|
|
|
|
|
|
|
// Return false here so that edge weights for InvokeInst could be decided
|
|
|
|
// in calcInvokeHeuristics().
|
|
|
|
if (isa<InvokeInst>(TI))
|
2017-04-12 13:42:14 +08:00
|
|
|
return false;
|
|
|
|
|
2012-08-25 02:14:27 +08:00
|
|
|
SmallVector<unsigned, 4> UnreachableEdges;
|
|
|
|
SmallVector<unsigned, 4> ReachableEdges;
|
2011-06-04 09:16:30 +08:00
|
|
|
|
2017-04-12 13:42:14 +08:00
|
|
|
for (succ_const_iterator I = succ_begin(BB), E = succ_end(BB); I != E; ++I)
|
2011-10-24 20:01:08 +08:00
|
|
|
if (PostDominatedByUnreachable.count(*I))
|
2012-08-25 02:14:27 +08:00
|
|
|
UnreachableEdges.push_back(I.getSuccessorIndex());
|
2011-10-24 20:01:08 +08:00
|
|
|
else
|
2012-08-25 02:14:27 +08:00
|
|
|
ReachableEdges.push_back(I.getSuccessorIndex());
|
2011-10-24 09:40:45 +08:00
|
|
|
|
2017-04-17 14:39:47 +08:00
|
|
|
// Skip probabilities if all were reachable.
|
|
|
|
if (UnreachableEdges.empty())
|
2017-04-12 13:42:14 +08:00
|
|
|
return false;
|
2015-12-22 06:00:51 +08:00
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
if (ReachableEdges.empty()) {
|
|
|
|
BranchProbability Prob(1, UnreachableEdges.size());
|
|
|
|
for (unsigned SuccIdx : UnreachableEdges)
|
|
|
|
setEdgeProbability(BB, SuccIdx, Prob);
|
2011-10-24 20:01:08 +08:00
|
|
|
return true;
|
2015-12-23 02:56:14 +08:00
|
|
|
}
|
|
|
|
|
2017-05-18 14:11:56 +08:00
|
|
|
auto UnreachableProb = UR_TAKEN_PROB;
|
|
|
|
auto ReachableProb =
|
|
|
|
(BranchProbability::getOne() - UR_TAKEN_PROB * UnreachableEdges.size()) /
|
|
|
|
ReachableEdges.size();
|
2015-12-23 02:56:14 +08:00
|
|
|
|
|
|
|
for (unsigned SuccIdx : UnreachableEdges)
|
|
|
|
setEdgeProbability(BB, SuccIdx, UnreachableProb);
|
|
|
|
for (unsigned SuccIdx : ReachableEdges)
|
|
|
|
setEdgeProbability(BB, SuccIdx, ReachableProb);
|
2011-10-24 20:01:08 +08:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
2011-06-04 09:16:30 +08:00
|
|
|
|
2011-10-19 18:30:30 +08:00
|
|
|
// Propagate existing explicit probabilities from either profile data or
|
2017-04-17 12:33:04 +08:00
|
|
|
// 'expect' intrinsic processing. Examine metadata against unreachable
|
|
|
|
// heuristic. The probability of the edge coming to unreachable block is
|
|
|
|
// set to min of metadata and unreachable heuristic.
|
2016-04-08 05:59:28 +08:00
|
|
|
bool BranchProbabilityInfo::calcMetadataWeights(const BasicBlock *BB) {
|
|
|
|
const TerminatorInst *TI = BB->getTerminator();
|
2017-04-17 14:39:47 +08:00
|
|
|
assert(TI->getNumSuccessors() > 1 && "expected more than one successor!");
|
2017-08-24 05:36:02 +08:00
|
|
|
if (!(isa<BranchInst>(TI) || isa<SwitchInst>(TI) || isa<IndirectBrInst>(TI)))
|
2011-10-19 18:30:30 +08:00
|
|
|
return false;
|
|
|
|
|
2014-11-12 05:30:22 +08:00
|
|
|
MDNode *WeightsNode = TI->getMetadata(LLVMContext::MD_prof);
|
2011-10-19 18:32:19 +08:00
|
|
|
if (!WeightsNode)
|
2011-10-19 18:30:30 +08:00
|
|
|
return false;
|
|
|
|
|
2015-05-08 01:22:06 +08:00
|
|
|
// Check that the number of successors is manageable.
|
|
|
|
assert(TI->getNumSuccessors() < UINT32_MAX && "Too many successors");
|
|
|
|
|
2011-10-19 18:32:19 +08:00
|
|
|
// Ensure there are weights for all of the successors. Note that the first
|
|
|
|
// operand to the metadata node is a name, not a weight.
|
|
|
|
if (WeightsNode->getNumOperands() != TI->getNumSuccessors() + 1)
|
2011-10-19 18:30:30 +08:00
|
|
|
return false;
|
|
|
|
|
2015-05-08 01:22:06 +08:00
|
|
|
// Build up the final weights that will be used in a temporary buffer.
|
|
|
|
// Compute the sum of all weights to later decide whether they need to
|
|
|
|
// be scaled to fit in 32 bits.
|
|
|
|
uint64_t WeightSum = 0;
|
2011-10-19 18:32:19 +08:00
|
|
|
SmallVector<uint32_t, 2> Weights;
|
2017-04-17 12:33:04 +08:00
|
|
|
SmallVector<unsigned, 2> UnreachableIdxs;
|
|
|
|
SmallVector<unsigned, 2> ReachableIdxs;
|
2011-10-19 18:32:19 +08:00
|
|
|
Weights.reserve(TI->getNumSuccessors());
|
|
|
|
for (unsigned i = 1, e = WeightsNode->getNumOperands(); i != e; ++i) {
|
IR: Split Metadata from Value
Split `Metadata` away from the `Value` class hierarchy, as part of
PR21532. Assembly and bitcode changes are in the wings, but this is the
bulk of the change for the IR C++ API.
I have a follow-up patch prepared for `clang`. If this breaks other
sub-projects, I apologize in advance :(. Help me compile it on Darwin
I'll try to fix it. FWIW, the errors should be easy to fix, so it may
be simpler to just fix it yourself.
This breaks the build for all metadata-related code that's out-of-tree.
Rest assured the transition is mechanical and the compiler should catch
almost all of the problems.
Here's a quick guide for updating your code:
- `Metadata` is the root of a class hierarchy with three main classes:
`MDNode`, `MDString`, and `ValueAsMetadata`. It is distinct from
the `Value` class hierarchy. It is typeless -- i.e., instances do
*not* have a `Type`.
- `MDNode`'s operands are all `Metadata *` (instead of `Value *`).
- `TrackingVH<MDNode>` and `WeakVH` referring to metadata can be
replaced with `TrackingMDNodeRef` and `TrackingMDRef`, respectively.
If you're referring solely to resolved `MDNode`s -- post graph
construction -- just use `MDNode*`.
- `MDNode` (and the rest of `Metadata`) have only limited support for
`replaceAllUsesWith()`.
As long as an `MDNode` is pointing at a forward declaration -- the
result of `MDNode::getTemporary()` -- it maintains a side map of its
uses and can RAUW itself. Once the forward declarations are fully
resolved RAUW support is dropped on the ground. This means that
uniquing collisions on changing operands cause nodes to become
"distinct". (This already happened fairly commonly, whenever an
operand went to null.)
If you're constructing complex (non self-reference) `MDNode` cycles,
you need to call `MDNode::resolveCycles()` on each node (or on a
top-level node that somehow references all of the nodes). Also,
don't do that. Metadata cycles (and the RAUW machinery needed to
construct them) are expensive.
- An `MDNode` can only refer to a `Constant` through a bridge called
`ConstantAsMetadata` (one of the subclasses of `ValueAsMetadata`).
As a side effect, accessing an operand of an `MDNode` that is known
to be, e.g., `ConstantInt`, takes three steps: first, cast from
`Metadata` to `ConstantAsMetadata`; second, extract the `Constant`;
third, cast down to `ConstantInt`.
The eventual goal is to introduce `MDInt`/`MDFloat`/etc. and have
metadata schema owners transition away from using `Constant`s when
the type isn't important (and they don't care about referring to
`GlobalValue`s).
In the meantime, I've added transitional API to the `mdconst`
namespace that matches semantics with the old code, in order to
avoid adding the error-prone three-step equivalent to every call
site. If your old code was:
MDNode *N = foo();
bar(isa <ConstantInt>(N->getOperand(0)));
baz(cast <ConstantInt>(N->getOperand(1)));
bak(cast_or_null <ConstantInt>(N->getOperand(2)));
bat(dyn_cast <ConstantInt>(N->getOperand(3)));
bay(dyn_cast_or_null<ConstantInt>(N->getOperand(4)));
you can trivially match its semantics with:
MDNode *N = foo();
bar(mdconst::hasa <ConstantInt>(N->getOperand(0)));
baz(mdconst::extract <ConstantInt>(N->getOperand(1)));
bak(mdconst::extract_or_null <ConstantInt>(N->getOperand(2)));
bat(mdconst::dyn_extract <ConstantInt>(N->getOperand(3)));
bay(mdconst::dyn_extract_or_null<ConstantInt>(N->getOperand(4)));
and when you transition your metadata schema to `MDInt`:
MDNode *N = foo();
bar(isa <MDInt>(N->getOperand(0)));
baz(cast <MDInt>(N->getOperand(1)));
bak(cast_or_null <MDInt>(N->getOperand(2)));
bat(dyn_cast <MDInt>(N->getOperand(3)));
bay(dyn_cast_or_null<MDInt>(N->getOperand(4)));
- A `CallInst` -- specifically, intrinsic instructions -- can refer to
metadata through a bridge called `MetadataAsValue`. This is a
subclass of `Value` where `getType()->isMetadataTy()`.
`MetadataAsValue` is the *only* class that can legally refer to a
`LocalAsMetadata`, which is a bridged form of non-`Constant` values
like `Argument` and `Instruction`. It can also refer to any other
`Metadata` subclass.
(I'll break all your testcases in a follow-up commit, when I propagate
this change to assembly.)
llvm-svn: 223802
2014-12-10 02:38:53 +08:00
|
|
|
ConstantInt *Weight =
|
|
|
|
mdconst::dyn_extract<ConstantInt>(WeightsNode->getOperand(i));
|
2011-10-19 18:32:19 +08:00
|
|
|
if (!Weight)
|
|
|
|
return false;
|
2015-05-08 01:22:06 +08:00
|
|
|
assert(Weight->getValue().getActiveBits() <= 32 &&
|
|
|
|
"Too many bits for uint32_t");
|
|
|
|
Weights.push_back(Weight->getZExtValue());
|
|
|
|
WeightSum += Weights.back();
|
2017-04-17 12:33:04 +08:00
|
|
|
if (PostDominatedByUnreachable.count(TI->getSuccessor(i - 1)))
|
|
|
|
UnreachableIdxs.push_back(i - 1);
|
|
|
|
else
|
|
|
|
ReachableIdxs.push_back(i - 1);
|
2011-10-19 18:32:19 +08:00
|
|
|
}
|
|
|
|
assert(Weights.size() == TI->getNumSuccessors() && "Checked above");
|
2015-05-08 01:22:06 +08:00
|
|
|
|
|
|
|
// If the sum of weights does not fit in 32 bits, scale every weight down
|
|
|
|
// accordingly.
|
|
|
|
uint64_t ScalingFactor =
|
|
|
|
(WeightSum > UINT32_MAX) ? WeightSum / UINT32_MAX + 1 : 1;
|
|
|
|
|
2017-04-17 12:33:04 +08:00
|
|
|
if (ScalingFactor > 1) {
|
|
|
|
WeightSum = 0;
|
|
|
|
for (unsigned i = 0, e = TI->getNumSuccessors(); i != e; ++i) {
|
|
|
|
Weights[i] /= ScalingFactor;
|
|
|
|
WeightSum += Weights[i];
|
|
|
|
}
|
2015-05-08 01:22:06 +08:00
|
|
|
}
|
2017-05-12 15:50:06 +08:00
|
|
|
assert(WeightSum <= UINT32_MAX &&
|
|
|
|
"Expected weights to scale down to 32 bits");
|
2015-12-23 07:45:55 +08:00
|
|
|
|
2017-04-17 12:33:04 +08:00
|
|
|
if (WeightSum == 0 || ReachableIdxs.size() == 0) {
|
2015-12-23 07:45:55 +08:00
|
|
|
for (unsigned i = 0, e = TI->getNumSuccessors(); i != e; ++i)
|
2017-04-17 12:33:04 +08:00
|
|
|
Weights[i] = 1;
|
|
|
|
WeightSum = TI->getNumSuccessors();
|
2015-12-23 07:45:55 +08:00
|
|
|
}
|
2015-12-23 02:56:14 +08:00
|
|
|
|
2017-04-17 12:33:04 +08:00
|
|
|
// Set the probability.
|
|
|
|
SmallVector<BranchProbability, 2> BP;
|
|
|
|
for (unsigned i = 0, e = TI->getNumSuccessors(); i != e; ++i)
|
|
|
|
BP.push_back({ Weights[i], static_cast<uint32_t>(WeightSum) });
|
|
|
|
|
|
|
|
// Examine the metadata against unreachable heuristic.
|
|
|
|
// If the unreachable heuristic is more strong then we use it for this edge.
|
|
|
|
if (UnreachableIdxs.size() > 0 && ReachableIdxs.size() > 0) {
|
|
|
|
auto ToDistribute = BranchProbability::getZero();
|
2017-05-18 14:11:56 +08:00
|
|
|
auto UnreachableProb = UR_TAKEN_PROB;
|
2017-04-17 12:33:04 +08:00
|
|
|
for (auto i : UnreachableIdxs)
|
|
|
|
if (UnreachableProb < BP[i]) {
|
|
|
|
ToDistribute += BP[i] - UnreachableProb;
|
|
|
|
BP[i] = UnreachableProb;
|
|
|
|
}
|
|
|
|
|
|
|
|
// If we modified the probability of some edges then we must distribute
|
|
|
|
// the difference between reachable blocks.
|
|
|
|
if (ToDistribute > BranchProbability::getZero()) {
|
|
|
|
BranchProbability PerEdge = ToDistribute / ReachableIdxs.size();
|
2017-05-12 15:50:06 +08:00
|
|
|
for (auto i : ReachableIdxs)
|
2017-04-17 12:33:04 +08:00
|
|
|
BP[i] += PerEdge;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for (unsigned i = 0, e = TI->getNumSuccessors(); i != e; ++i)
|
|
|
|
setEdgeProbability(BB, i, BP[i]);
|
|
|
|
|
2011-10-19 18:30:30 +08:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2013-05-24 20:26:52 +08:00
|
|
|
/// \brief Calculate edge weights for edges leading to cold blocks.
|
|
|
|
///
|
|
|
|
/// A cold block is one post-dominated by a block with a call to a
|
|
|
|
/// cold function. Those edges are unlikely to be taken, so we give
|
|
|
|
/// them relatively low weight.
|
|
|
|
///
|
|
|
|
/// Return true if we could compute the weights for cold edges.
|
|
|
|
/// Return false, otherwise.
|
2016-04-08 05:59:28 +08:00
|
|
|
bool BranchProbabilityInfo::calcColdCallHeuristics(const BasicBlock *BB) {
|
|
|
|
const TerminatorInst *TI = BB->getTerminator();
|
2017-04-17 14:39:47 +08:00
|
|
|
assert(TI->getNumSuccessors() > 1 && "expected more than one successor!");
|
|
|
|
|
|
|
|
// Return false here so that edge weights for InvokeInst could be decided
|
|
|
|
// in calcInvokeHeuristics().
|
|
|
|
if (isa<InvokeInst>(TI))
|
2013-05-24 20:26:52 +08:00
|
|
|
return false;
|
|
|
|
|
|
|
|
// Determine which successors are post-dominated by a cold block.
|
|
|
|
SmallVector<unsigned, 4> ColdEdges;
|
|
|
|
SmallVector<unsigned, 4> NormalEdges;
|
2016-04-08 05:59:28 +08:00
|
|
|
for (succ_const_iterator I = succ_begin(BB), E = succ_end(BB); I != E; ++I)
|
2013-05-24 20:26:52 +08:00
|
|
|
if (PostDominatedByColdCall.count(*I))
|
|
|
|
ColdEdges.push_back(I.getSuccessorIndex());
|
|
|
|
else
|
|
|
|
NormalEdges.push_back(I.getSuccessorIndex());
|
|
|
|
|
2017-04-17 14:39:47 +08:00
|
|
|
// Skip probabilities if no cold edges.
|
|
|
|
if (ColdEdges.empty())
|
2013-05-24 20:26:52 +08:00
|
|
|
return false;
|
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
if (NormalEdges.empty()) {
|
|
|
|
BranchProbability Prob(1, ColdEdges.size());
|
|
|
|
for (unsigned SuccIdx : ColdEdges)
|
|
|
|
setEdgeProbability(BB, SuccIdx, Prob);
|
2013-05-24 20:26:52 +08:00
|
|
|
return true;
|
2015-12-23 02:56:14 +08:00
|
|
|
}
|
|
|
|
|
2016-12-17 09:02:08 +08:00
|
|
|
auto ColdProb = BranchProbability::getBranchProbability(
|
|
|
|
CC_TAKEN_WEIGHT,
|
|
|
|
(CC_TAKEN_WEIGHT + CC_NONTAKEN_WEIGHT) * uint64_t(ColdEdges.size()));
|
|
|
|
auto NormalProb = BranchProbability::getBranchProbability(
|
|
|
|
CC_NONTAKEN_WEIGHT,
|
|
|
|
(CC_TAKEN_WEIGHT + CC_NONTAKEN_WEIGHT) * uint64_t(NormalEdges.size()));
|
2015-12-23 02:56:14 +08:00
|
|
|
|
|
|
|
for (unsigned SuccIdx : ColdEdges)
|
|
|
|
setEdgeProbability(BB, SuccIdx, ColdProb);
|
|
|
|
for (unsigned SuccIdx : NormalEdges)
|
|
|
|
setEdgeProbability(BB, SuccIdx, NormalProb);
|
2013-05-24 20:26:52 +08:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2011-06-04 09:16:30 +08:00
|
|
|
// Calculate Edge Weights using "Pointer Heuristics". Predict a comparsion
|
|
|
|
// between two pointer or pointer and NULL will fail.
|
2016-04-08 05:59:28 +08:00
|
|
|
bool BranchProbabilityInfo::calcPointerHeuristics(const BasicBlock *BB) {
|
|
|
|
const BranchInst *BI = dyn_cast<BranchInst>(BB->getTerminator());
|
2011-06-04 09:16:30 +08:00
|
|
|
if (!BI || !BI->isConditional())
|
2011-07-29 05:45:07 +08:00
|
|
|
return false;
|
2011-06-04 09:16:30 +08:00
|
|
|
|
|
|
|
Value *Cond = BI->getCondition();
|
|
|
|
ICmpInst *CI = dyn_cast<ICmpInst>(Cond);
|
2011-07-16 04:51:06 +08:00
|
|
|
if (!CI || !CI->isEquality())
|
2011-07-29 05:45:07 +08:00
|
|
|
return false;
|
2011-06-04 09:16:30 +08:00
|
|
|
|
|
|
|
Value *LHS = CI->getOperand(0);
|
|
|
|
|
|
|
|
if (!LHS->getType()->isPointerTy())
|
2011-07-29 05:45:07 +08:00
|
|
|
return false;
|
2011-06-04 09:16:30 +08:00
|
|
|
|
2011-06-04 10:07:10 +08:00
|
|
|
assert(CI->getOperand(1)->getType()->isPointerTy());
|
2011-06-04 09:16:30 +08:00
|
|
|
|
|
|
|
// p != 0 -> isProb = true
|
|
|
|
// p == 0 -> isProb = false
|
|
|
|
// p != q -> isProb = true
|
|
|
|
// p == q -> isProb = false;
|
2012-08-25 02:14:27 +08:00
|
|
|
unsigned TakenIdx = 0, NonTakenIdx = 1;
|
2011-07-16 04:51:06 +08:00
|
|
|
bool isProb = CI->getPredicate() == ICmpInst::ICMP_NE;
|
2011-06-04 09:16:30 +08:00
|
|
|
if (!isProb)
|
2012-08-25 02:14:27 +08:00
|
|
|
std::swap(TakenIdx, NonTakenIdx);
|
2011-06-04 09:16:30 +08:00
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
BranchProbability TakenProb(PH_TAKEN_WEIGHT,
|
|
|
|
PH_TAKEN_WEIGHT + PH_NONTAKEN_WEIGHT);
|
|
|
|
setEdgeProbability(BB, TakenIdx, TakenProb);
|
|
|
|
setEdgeProbability(BB, NonTakenIdx, TakenProb.getCompl());
|
2011-07-29 05:45:07 +08:00
|
|
|
return true;
|
2011-06-04 09:16:30 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Calculate Edge Weights using "Loop Branch Heuristics". Predict backedges
|
|
|
|
// as taken, exiting edges as not-taken.
|
2016-04-08 05:59:28 +08:00
|
|
|
bool BranchProbabilityInfo::calcLoopBranchHeuristics(const BasicBlock *BB,
|
2015-07-16 06:48:29 +08:00
|
|
|
const LoopInfo &LI) {
|
|
|
|
Loop *L = LI.getLoopFor(BB);
|
2011-06-04 09:16:30 +08:00
|
|
|
if (!L)
|
2011-07-29 05:45:07 +08:00
|
|
|
return false;
|
2011-06-04 09:16:30 +08:00
|
|
|
|
2012-08-25 02:14:27 +08:00
|
|
|
SmallVector<unsigned, 8> BackEdges;
|
|
|
|
SmallVector<unsigned, 8> ExitingEdges;
|
|
|
|
SmallVector<unsigned, 8> InEdges; // Edges from header to the loop.
|
2011-07-29 05:33:46 +08:00
|
|
|
|
2016-04-08 05:59:28 +08:00
|
|
|
for (succ_const_iterator I = succ_begin(BB), E = succ_end(BB); I != E; ++I) {
|
2011-10-25 17:47:41 +08:00
|
|
|
if (!L->contains(*I))
|
2012-08-25 02:14:27 +08:00
|
|
|
ExitingEdges.push_back(I.getSuccessorIndex());
|
2011-10-25 17:47:41 +08:00
|
|
|
else if (L->getHeader() == *I)
|
2012-08-25 02:14:27 +08:00
|
|
|
BackEdges.push_back(I.getSuccessorIndex());
|
2011-10-25 17:47:41 +08:00
|
|
|
else
|
2012-08-25 02:14:27 +08:00
|
|
|
InEdges.push_back(I.getSuccessorIndex());
|
2011-06-04 09:16:30 +08:00
|
|
|
}
|
|
|
|
|
2014-04-15 00:56:19 +08:00
|
|
|
if (BackEdges.empty() && ExitingEdges.empty())
|
|
|
|
return false;
|
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
// Collect the sum of probabilities of back-edges/in-edges/exiting-edges, and
|
|
|
|
// normalize them so that they sum up to one.
|
2016-06-17 21:15:10 +08:00
|
|
|
BranchProbability Probs[] = {BranchProbability::getZero(),
|
|
|
|
BranchProbability::getZero(),
|
|
|
|
BranchProbability::getZero()};
|
2015-12-23 02:56:14 +08:00
|
|
|
unsigned Denom = (BackEdges.empty() ? 0 : LBH_TAKEN_WEIGHT) +
|
|
|
|
(InEdges.empty() ? 0 : LBH_TAKEN_WEIGHT) +
|
|
|
|
(ExitingEdges.empty() ? 0 : LBH_NONTAKEN_WEIGHT);
|
|
|
|
if (!BackEdges.empty())
|
|
|
|
Probs[0] = BranchProbability(LBH_TAKEN_WEIGHT, Denom);
|
|
|
|
if (!InEdges.empty())
|
|
|
|
Probs[1] = BranchProbability(LBH_TAKEN_WEIGHT, Denom);
|
|
|
|
if (!ExitingEdges.empty())
|
|
|
|
Probs[2] = BranchProbability(LBH_NONTAKEN_WEIGHT, Denom);
|
2011-06-04 09:16:30 +08:00
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
if (uint32_t numBackEdges = BackEdges.size()) {
|
|
|
|
auto Prob = Probs[0] / numBackEdges;
|
|
|
|
for (unsigned SuccIdx : BackEdges)
|
|
|
|
setEdgeProbability(BB, SuccIdx, Prob);
|
2011-06-04 09:16:30 +08:00
|
|
|
}
|
|
|
|
|
2011-07-29 05:33:46 +08:00
|
|
|
if (uint32_t numInEdges = InEdges.size()) {
|
2015-12-23 02:56:14 +08:00
|
|
|
auto Prob = Probs[1] / numInEdges;
|
|
|
|
for (unsigned SuccIdx : InEdges)
|
|
|
|
setEdgeProbability(BB, SuccIdx, Prob);
|
2011-07-29 05:33:46 +08:00
|
|
|
}
|
|
|
|
|
2011-10-25 17:47:41 +08:00
|
|
|
if (uint32_t numExitingEdges = ExitingEdges.size()) {
|
2015-12-23 02:56:14 +08:00
|
|
|
auto Prob = Probs[2] / numExitingEdges;
|
|
|
|
for (unsigned SuccIdx : ExitingEdges)
|
|
|
|
setEdgeProbability(BB, SuccIdx, Prob);
|
2011-06-04 09:16:30 +08:00
|
|
|
}
|
2011-07-29 05:45:07 +08:00
|
|
|
|
|
|
|
return true;
|
2011-06-04 09:16:30 +08:00
|
|
|
}
|
|
|
|
|
2017-06-08 17:44:40 +08:00
|
|
|
bool BranchProbabilityInfo::calcZeroHeuristics(const BasicBlock *BB,
|
|
|
|
const TargetLibraryInfo *TLI) {
|
2016-04-08 05:59:28 +08:00
|
|
|
const BranchInst *BI = dyn_cast<BranchInst>(BB->getTerminator());
|
2011-07-31 11:27:24 +08:00
|
|
|
if (!BI || !BI->isConditional())
|
|
|
|
return false;
|
|
|
|
|
|
|
|
Value *Cond = BI->getCondition();
|
|
|
|
ICmpInst *CI = dyn_cast<ICmpInst>(Cond);
|
|
|
|
if (!CI)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
Value *RHS = CI->getOperand(1);
|
2011-07-31 12:47:20 +08:00
|
|
|
ConstantInt *CV = dyn_cast<ConstantInt>(RHS);
|
2011-09-05 07:53:04 +08:00
|
|
|
if (!CV)
|
2011-07-31 11:27:24 +08:00
|
|
|
return false;
|
|
|
|
|
2015-04-15 14:24:07 +08:00
|
|
|
// If the LHS is the result of AND'ing a value with a single bit bitmask,
|
|
|
|
// we don't have information about probabilities.
|
|
|
|
if (Instruction *LHS = dyn_cast<Instruction>(CI->getOperand(0)))
|
|
|
|
if (LHS->getOpcode() == Instruction::And)
|
|
|
|
if (ConstantInt *AndRHS = dyn_cast<ConstantInt>(LHS->getOperand(1)))
|
2017-08-05 00:59:29 +08:00
|
|
|
if (AndRHS->getValue().isPowerOf2())
|
2015-04-15 14:24:07 +08:00
|
|
|
return false;
|
|
|
|
|
2017-06-08 17:44:40 +08:00
|
|
|
// Check if the LHS is the return value of a library function
|
|
|
|
LibFunc Func = NumLibFuncs;
|
|
|
|
if (TLI)
|
|
|
|
if (CallInst *Call = dyn_cast<CallInst>(CI->getOperand(0)))
|
|
|
|
if (Function *CalledFn = Call->getCalledFunction())
|
|
|
|
TLI->getLibFunc(*CalledFn, Func);
|
|
|
|
|
2011-07-31 11:27:24 +08:00
|
|
|
bool isProb;
|
2017-06-08 17:44:40 +08:00
|
|
|
if (Func == LibFunc_strcasecmp ||
|
|
|
|
Func == LibFunc_strcmp ||
|
|
|
|
Func == LibFunc_strncasecmp ||
|
|
|
|
Func == LibFunc_strncmp ||
|
|
|
|
Func == LibFunc_memcmp) {
|
|
|
|
// strcmp and similar functions return zero, negative, or positive, if the
|
|
|
|
// first string is equal, less, or greater than the second. We consider it
|
|
|
|
// likely that the strings are not equal, so a comparison with zero is
|
|
|
|
// probably false, but also a comparison with any other number is also
|
|
|
|
// probably false given that what exactly is returned for nonzero values is
|
|
|
|
// not specified. Any kind of comparison other than equality we know
|
|
|
|
// nothing about.
|
|
|
|
switch (CI->getPredicate()) {
|
|
|
|
case CmpInst::ICMP_EQ:
|
|
|
|
isProb = false;
|
|
|
|
break;
|
|
|
|
case CmpInst::ICMP_NE:
|
|
|
|
isProb = true;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
} else if (CV->isZero()) {
|
2011-09-05 07:53:04 +08:00
|
|
|
switch (CI->getPredicate()) {
|
|
|
|
case CmpInst::ICMP_EQ:
|
|
|
|
// X == 0 -> Unlikely
|
|
|
|
isProb = false;
|
|
|
|
break;
|
|
|
|
case CmpInst::ICMP_NE:
|
|
|
|
// X != 0 -> Likely
|
|
|
|
isProb = true;
|
|
|
|
break;
|
|
|
|
case CmpInst::ICMP_SLT:
|
|
|
|
// X < 0 -> Unlikely
|
|
|
|
isProb = false;
|
|
|
|
break;
|
|
|
|
case CmpInst::ICMP_SGT:
|
|
|
|
// X > 0 -> Likely
|
|
|
|
isProb = true;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
} else if (CV->isOne() && CI->getPredicate() == CmpInst::ICMP_SLT) {
|
|
|
|
// InstCombine canonicalizes X <= 0 into X < 1.
|
|
|
|
// X <= 0 -> Unlikely
|
2011-07-31 12:47:20 +08:00
|
|
|
isProb = false;
|
2017-07-07 02:39:47 +08:00
|
|
|
} else if (CV->isMinusOne()) {
|
2013-11-01 18:58:22 +08:00
|
|
|
switch (CI->getPredicate()) {
|
|
|
|
case CmpInst::ICMP_EQ:
|
|
|
|
// X == -1 -> Unlikely
|
|
|
|
isProb = false;
|
|
|
|
break;
|
|
|
|
case CmpInst::ICMP_NE:
|
|
|
|
// X != -1 -> Likely
|
|
|
|
isProb = true;
|
|
|
|
break;
|
|
|
|
case CmpInst::ICMP_SGT:
|
|
|
|
// InstCombine canonicalizes X >= 0 into X > -1.
|
|
|
|
// X >= 0 -> Likely
|
|
|
|
isProb = true;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
return false;
|
|
|
|
}
|
2011-09-05 07:53:04 +08:00
|
|
|
} else {
|
2011-07-31 11:27:24 +08:00
|
|
|
return false;
|
2011-09-05 07:53:04 +08:00
|
|
|
}
|
2011-07-31 11:27:24 +08:00
|
|
|
|
2012-08-25 02:14:27 +08:00
|
|
|
unsigned TakenIdx = 0, NonTakenIdx = 1;
|
2011-07-31 11:27:24 +08:00
|
|
|
|
|
|
|
if (!isProb)
|
2012-08-25 02:14:27 +08:00
|
|
|
std::swap(TakenIdx, NonTakenIdx);
|
2011-07-31 11:27:24 +08:00
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
BranchProbability TakenProb(ZH_TAKEN_WEIGHT,
|
|
|
|
ZH_TAKEN_WEIGHT + ZH_NONTAKEN_WEIGHT);
|
|
|
|
setEdgeProbability(BB, TakenIdx, TakenProb);
|
|
|
|
setEdgeProbability(BB, NonTakenIdx, TakenProb.getCompl());
|
2011-07-31 11:27:24 +08:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2016-04-08 05:59:28 +08:00
|
|
|
bool BranchProbabilityInfo::calcFloatingPointHeuristics(const BasicBlock *BB) {
|
|
|
|
const BranchInst *BI = dyn_cast<BranchInst>(BB->getTerminator());
|
2011-10-22 04:12:47 +08:00
|
|
|
if (!BI || !BI->isConditional())
|
|
|
|
return false;
|
|
|
|
|
|
|
|
Value *Cond = BI->getCondition();
|
|
|
|
FCmpInst *FCmp = dyn_cast<FCmpInst>(Cond);
|
2011-10-22 05:13:47 +08:00
|
|
|
if (!FCmp)
|
2011-10-22 04:12:47 +08:00
|
|
|
return false;
|
|
|
|
|
2011-10-22 05:13:47 +08:00
|
|
|
bool isProb;
|
|
|
|
if (FCmp->isEquality()) {
|
|
|
|
// f1 == f2 -> Unlikely
|
|
|
|
// f1 != f2 -> Likely
|
|
|
|
isProb = !FCmp->isTrueWhenEqual();
|
|
|
|
} else if (FCmp->getPredicate() == FCmpInst::FCMP_ORD) {
|
|
|
|
// !isnan -> Likely
|
|
|
|
isProb = true;
|
|
|
|
} else if (FCmp->getPredicate() == FCmpInst::FCMP_UNO) {
|
|
|
|
// isnan -> Unlikely
|
|
|
|
isProb = false;
|
|
|
|
} else {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2012-08-25 02:14:27 +08:00
|
|
|
unsigned TakenIdx = 0, NonTakenIdx = 1;
|
2011-10-22 04:12:47 +08:00
|
|
|
|
2011-10-22 05:13:47 +08:00
|
|
|
if (!isProb)
|
2012-08-25 02:14:27 +08:00
|
|
|
std::swap(TakenIdx, NonTakenIdx);
|
2011-10-22 04:12:47 +08:00
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
BranchProbability TakenProb(FPH_TAKEN_WEIGHT,
|
|
|
|
FPH_TAKEN_WEIGHT + FPH_NONTAKEN_WEIGHT);
|
|
|
|
setEdgeProbability(BB, TakenIdx, TakenProb);
|
|
|
|
setEdgeProbability(BB, NonTakenIdx, TakenProb.getCompl());
|
2011-10-22 04:12:47 +08:00
|
|
|
return true;
|
|
|
|
}
|
2011-07-31 11:27:24 +08:00
|
|
|
|
2016-04-08 05:59:28 +08:00
|
|
|
bool BranchProbabilityInfo::calcInvokeHeuristics(const BasicBlock *BB) {
|
|
|
|
const InvokeInst *II = dyn_cast<InvokeInst>(BB->getTerminator());
|
2012-08-15 20:22:35 +08:00
|
|
|
if (!II)
|
|
|
|
return false;
|
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
BranchProbability TakenProb(IH_TAKEN_WEIGHT,
|
|
|
|
IH_TAKEN_WEIGHT + IH_NONTAKEN_WEIGHT);
|
|
|
|
setEdgeProbability(BB, 0 /*Index for Normal*/, TakenProb);
|
|
|
|
setEdgeProbability(BB, 1 /*Index for Unwind*/, TakenProb.getCompl());
|
2012-08-15 20:22:35 +08:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2015-05-29 03:43:06 +08:00
|
|
|
void BranchProbabilityInfo::releaseMemory() {
|
2015-12-23 02:56:14 +08:00
|
|
|
Probs.clear();
|
2015-05-29 03:43:06 +08:00
|
|
|
}
|
|
|
|
|
2015-07-16 06:48:29 +08:00
|
|
|
void BranchProbabilityInfo::print(raw_ostream &OS) const {
|
2011-10-24 05:21:50 +08:00
|
|
|
OS << "---- Branch Probabilities ----\n";
|
|
|
|
// We print the probabilities from the last function the analysis ran over,
|
|
|
|
// or the function it is currently running over.
|
|
|
|
assert(LastF && "Cannot print prior to running over a function");
|
Analysis: Remove implicit ilist iterator conversions
Remove implicit ilist iterator conversions from LLVMAnalysis.
I came across something really scary in `llvm::isKnownNotFullPoison()`
which relied on `Instruction::getNextNode()` being completely broken
(not surprising, but scary nevertheless). This function is documented
(and coded to) return `nullptr` when it gets to the sentinel, but with
an `ilist_half_node` as a sentinel, the sentinel check looks into some
other memory and we don't recognize we've hit the end.
Rooting out these scary cases is the reason I'm removing the implicit
conversions before doing anything else with `ilist`; I'm not at all
surprised that clients rely on badness.
I found another scary case -- this time, not relying on badness, just
bad (but I guess getting lucky so far) -- in
`ObjectSizeOffsetEvaluator::compute_()`. Here, we save out the
insertion point, do some things, and then restore it. Previously, we
let the iterator auto-convert to `Instruction*`, and then set it back
using the `Instruction*` version:
Instruction *PrevInsertPoint = Builder.GetInsertPoint();
/* Logic that may change insert point */
if (PrevInsertPoint)
Builder.SetInsertPoint(PrevInsertPoint);
The check for `PrevInsertPoint` doesn't protect correctly against bad
accesses. If the insertion point has been set to the end of a basic
block (i.e., `SetInsertPoint(SomeBB)`), then `GetInsertPoint()` returns
an iterator pointing at the list sentinel. The version of
`SetInsertPoint()` that's getting called will then call
`PrevInsertPoint->getParent()`, which explodes horribly. The only
reason this hasn't blown up is that it's fairly unlikely the builder is
adding to the end of the block; usually, we're adding instructions
somewhere before the terminator.
llvm-svn: 249925
2015-10-10 08:53:03 +08:00
|
|
|
for (const auto &BI : *LastF) {
|
|
|
|
for (succ_const_iterator SI = succ_begin(&BI), SE = succ_end(&BI); SI != SE;
|
|
|
|
++SI) {
|
|
|
|
printEdgeProbability(OS << " ", &BI, *SI);
|
2014-07-22 01:06:51 +08:00
|
|
|
}
|
|
|
|
}
|
2011-10-24 05:21:50 +08:00
|
|
|
}
|
|
|
|
|
2011-07-30 03:30:00 +08:00
|
|
|
bool BranchProbabilityInfo::
|
|
|
|
isEdgeHot(const BasicBlock *Src, const BasicBlock *Dst) const {
|
2011-06-11 09:05:22 +08:00
|
|
|
// Hot probability is at least 4/5 = 80%
|
2011-10-23 19:19:14 +08:00
|
|
|
// FIXME: Compare against a static "hot" BranchProbability.
|
|
|
|
return getEdgeProbability(Src, Dst) > BranchProbability(4, 5);
|
2011-06-04 09:16:30 +08:00
|
|
|
}
|
|
|
|
|
2016-04-08 05:59:28 +08:00
|
|
|
const BasicBlock *
|
|
|
|
BranchProbabilityInfo::getHotSucc(const BasicBlock *BB) const {
|
2015-12-23 02:56:14 +08:00
|
|
|
auto MaxProb = BranchProbability::getZero();
|
2016-04-08 05:59:28 +08:00
|
|
|
const BasicBlock *MaxSucc = nullptr;
|
2011-06-04 09:16:30 +08:00
|
|
|
|
2016-04-08 05:59:28 +08:00
|
|
|
for (succ_const_iterator I = succ_begin(BB), E = succ_end(BB); I != E; ++I) {
|
|
|
|
const BasicBlock *Succ = *I;
|
2015-12-23 02:56:14 +08:00
|
|
|
auto Prob = getEdgeProbability(BB, Succ);
|
|
|
|
if (Prob > MaxProb) {
|
|
|
|
MaxProb = Prob;
|
2011-06-04 09:16:30 +08:00
|
|
|
MaxSucc = Succ;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-10-23 19:19:14 +08:00
|
|
|
// Hot probability is at least 4/5 = 80%
|
2015-12-23 02:56:14 +08:00
|
|
|
if (MaxProb > BranchProbability(4, 5))
|
2011-06-04 09:16:30 +08:00
|
|
|
return MaxSucc;
|
|
|
|
|
2014-04-15 12:59:12 +08:00
|
|
|
return nullptr;
|
2011-06-04 09:16:30 +08:00
|
|
|
}
|
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
/// Get the raw edge probability for the edge. If can't find it, return a
|
|
|
|
/// default probability 1/N where N is the number of successors. Here an edge is
|
|
|
|
/// specified using PredBlock and an
|
|
|
|
/// index to the successors.
|
|
|
|
BranchProbability
|
|
|
|
BranchProbabilityInfo::getEdgeProbability(const BasicBlock *Src,
|
|
|
|
unsigned IndexInSuccessors) const {
|
|
|
|
auto I = Probs.find(std::make_pair(Src, IndexInSuccessors));
|
2011-06-04 09:16:30 +08:00
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
if (I != Probs.end())
|
2011-06-04 09:16:30 +08:00
|
|
|
return I->second;
|
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
return {1,
|
|
|
|
static_cast<uint32_t>(std::distance(succ_begin(Src), succ_end(Src)))};
|
2011-06-04 09:16:30 +08:00
|
|
|
}
|
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
BranchProbability
|
|
|
|
BranchProbabilityInfo::getEdgeProbability(const BasicBlock *Src,
|
|
|
|
succ_const_iterator Dst) const {
|
|
|
|
return getEdgeProbability(Src, Dst.getSuccessorIndex());
|
2013-12-14 10:24:25 +08:00
|
|
|
}
|
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
/// Get the raw edge probability calculated for the block pair. This returns the
|
|
|
|
/// sum of all raw edge probabilities from Src to Dst.
|
|
|
|
BranchProbability
|
|
|
|
BranchProbabilityInfo::getEdgeProbability(const BasicBlock *Src,
|
|
|
|
const BasicBlock *Dst) const {
|
|
|
|
auto Prob = BranchProbability::getZero();
|
|
|
|
bool FoundProb = false;
|
2012-08-25 02:14:27 +08:00
|
|
|
for (succ_const_iterator I = succ_begin(Src), E = succ_end(Src); I != E; ++I)
|
|
|
|
if (*I == Dst) {
|
2015-12-23 02:56:14 +08:00
|
|
|
auto MapI = Probs.find(std::make_pair(Src, I.getSuccessorIndex()));
|
|
|
|
if (MapI != Probs.end()) {
|
|
|
|
FoundProb = true;
|
|
|
|
Prob += MapI->second;
|
2015-05-07 01:55:11 +08:00
|
|
|
}
|
2012-08-25 02:14:27 +08:00
|
|
|
}
|
2015-12-23 02:56:14 +08:00
|
|
|
uint32_t succ_num = std::distance(succ_begin(Src), succ_end(Src));
|
|
|
|
return FoundProb ? Prob : BranchProbability(1, succ_num);
|
2012-08-25 02:14:27 +08:00
|
|
|
}
|
|
|
|
|
2015-12-23 02:56:14 +08:00
|
|
|
/// Set the edge probability for a given edge specified by PredBlock and an
|
|
|
|
/// index to the successors.
|
|
|
|
void BranchProbabilityInfo::setEdgeProbability(const BasicBlock *Src,
|
|
|
|
unsigned IndexInSuccessors,
|
|
|
|
BranchProbability Prob) {
|
|
|
|
Probs[std::make_pair(Src, IndexInSuccessors)] = Prob;
|
2016-07-15 22:31:16 +08:00
|
|
|
Handles.insert(BasicBlockCallbackVH(Src, this));
|
2015-12-23 02:56:14 +08:00
|
|
|
DEBUG(dbgs() << "set edge " << Src->getName() << " -> " << IndexInSuccessors
|
|
|
|
<< " successor probability to " << Prob << "\n");
|
2015-12-01 13:29:22 +08:00
|
|
|
}
|
|
|
|
|
2011-06-11 09:05:22 +08:00
|
|
|
raw_ostream &
|
2011-10-24 05:21:50 +08:00
|
|
|
BranchProbabilityInfo::printEdgeProbability(raw_ostream &OS,
|
|
|
|
const BasicBlock *Src,
|
|
|
|
const BasicBlock *Dst) const {
|
2011-06-17 04:22:37 +08:00
|
|
|
const BranchProbability Prob = getEdgeProbability(Src, Dst);
|
2011-11-16 00:27:03 +08:00
|
|
|
OS << "edge " << Src->getName() << " -> " << Dst->getName()
|
2011-06-11 09:05:22 +08:00
|
|
|
<< " probability is " << Prob
|
|
|
|
<< (isEdgeHot(Src, Dst) ? " [HOT edge]\n" : "\n");
|
2011-06-04 09:16:30 +08:00
|
|
|
|
|
|
|
return OS;
|
|
|
|
}
|
2015-07-16 06:48:29 +08:00
|
|
|
|
2016-07-15 22:31:16 +08:00
|
|
|
void BranchProbabilityInfo::eraseBlock(const BasicBlock *BB) {
|
|
|
|
for (auto I = Probs.begin(), E = Probs.end(); I != E; ++I) {
|
|
|
|
auto Key = I->first;
|
|
|
|
if (Key.first == BB)
|
|
|
|
Probs.erase(Key);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-06-08 17:44:40 +08:00
|
|
|
void BranchProbabilityInfo::calculate(const Function &F, const LoopInfo &LI,
|
|
|
|
const TargetLibraryInfo *TLI) {
|
2015-07-16 06:48:29 +08:00
|
|
|
DEBUG(dbgs() << "---- Branch Probability Info : " << F.getName()
|
|
|
|
<< " ----\n\n");
|
|
|
|
LastF = &F; // Store the last function we ran on for printing.
|
|
|
|
assert(PostDominatedByUnreachable.empty());
|
|
|
|
assert(PostDominatedByColdCall.empty());
|
|
|
|
|
|
|
|
// Walk the basic blocks in post-order so that we can build up state about
|
|
|
|
// the successors of a block iteratively.
|
|
|
|
for (auto BB : post_order(&F.getEntryBlock())) {
|
|
|
|
DEBUG(dbgs() << "Computing probabilities for " << BB->getName() << "\n");
|
2017-04-12 13:42:14 +08:00
|
|
|
updatePostDominatedByUnreachable(BB);
|
|
|
|
updatePostDominatedByColdCall(BB);
|
2017-04-17 14:39:47 +08:00
|
|
|
// If there is no at least two successors, no sense to set probability.
|
|
|
|
if (BB->getTerminator()->getNumSuccessors() < 2)
|
|
|
|
continue;
|
2015-07-16 06:48:29 +08:00
|
|
|
if (calcMetadataWeights(BB))
|
|
|
|
continue;
|
2017-04-17 12:33:04 +08:00
|
|
|
if (calcUnreachableHeuristics(BB))
|
|
|
|
continue;
|
2015-07-16 06:48:29 +08:00
|
|
|
if (calcColdCallHeuristics(BB))
|
|
|
|
continue;
|
|
|
|
if (calcLoopBranchHeuristics(BB, LI))
|
|
|
|
continue;
|
|
|
|
if (calcPointerHeuristics(BB))
|
|
|
|
continue;
|
2017-06-08 17:44:40 +08:00
|
|
|
if (calcZeroHeuristics(BB, TLI))
|
2015-07-16 06:48:29 +08:00
|
|
|
continue;
|
|
|
|
if (calcFloatingPointHeuristics(BB))
|
|
|
|
continue;
|
|
|
|
calcInvokeHeuristics(BB);
|
|
|
|
}
|
|
|
|
|
|
|
|
PostDominatedByUnreachable.clear();
|
|
|
|
PostDominatedByColdCall.clear();
|
2017-08-26 08:31:00 +08:00
|
|
|
|
|
|
|
if (PrintBranchProb &&
|
|
|
|
(PrintBranchProbFuncName.empty() ||
|
|
|
|
F.getName().equals(PrintBranchProbFuncName))) {
|
|
|
|
print(dbgs());
|
|
|
|
}
|
2015-07-16 06:48:29 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void BranchProbabilityInfoWrapperPass::getAnalysisUsage(
|
|
|
|
AnalysisUsage &AU) const {
|
|
|
|
AU.addRequired<LoopInfoWrapperPass>();
|
2017-06-08 17:44:40 +08:00
|
|
|
AU.addRequired<TargetLibraryInfoWrapperPass>();
|
2015-07-16 06:48:29 +08:00
|
|
|
AU.setPreservesAll();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool BranchProbabilityInfoWrapperPass::runOnFunction(Function &F) {
|
|
|
|
const LoopInfo &LI = getAnalysis<LoopInfoWrapperPass>().getLoopInfo();
|
2017-06-08 17:44:40 +08:00
|
|
|
const TargetLibraryInfo &TLI = getAnalysis<TargetLibraryInfoWrapperPass>().getTLI();
|
|
|
|
BPI.calculate(F, LI, &TLI);
|
2015-07-16 06:48:29 +08:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
void BranchProbabilityInfoWrapperPass::releaseMemory() { BPI.releaseMemory(); }
|
|
|
|
|
|
|
|
void BranchProbabilityInfoWrapperPass::print(raw_ostream &OS,
|
|
|
|
const Module *) const {
|
|
|
|
BPI.print(OS);
|
|
|
|
}
|
2016-05-05 10:59:57 +08:00
|
|
|
|
2016-11-24 01:53:26 +08:00
|
|
|
AnalysisKey BranchProbabilityAnalysis::Key;
|
2016-05-05 10:59:57 +08:00
|
|
|
BranchProbabilityInfo
|
2016-08-09 08:28:15 +08:00
|
|
|
BranchProbabilityAnalysis::run(Function &F, FunctionAnalysisManager &AM) {
|
2016-05-05 10:59:57 +08:00
|
|
|
BranchProbabilityInfo BPI;
|
2017-06-08 17:44:40 +08:00
|
|
|
BPI.calculate(F, AM.getResult<LoopAnalysis>(F), &AM.getResult<TargetLibraryAnalysis>(F));
|
2016-05-05 10:59:57 +08:00
|
|
|
return BPI;
|
|
|
|
}
|
|
|
|
|
|
|
|
PreservedAnalyses
|
2016-08-09 08:28:15 +08:00
|
|
|
BranchProbabilityPrinterPass::run(Function &F, FunctionAnalysisManager &AM) {
|
2016-05-05 10:59:57 +08:00
|
|
|
OS << "Printing analysis results of BPI for function "
|
|
|
|
<< "'" << F.getName() << "':"
|
|
|
|
<< "\n";
|
|
|
|
AM.getResult<BranchProbabilityAnalysis>(F).print(OS);
|
|
|
|
return PreservedAnalyses::all();
|
|
|
|
}
|