diff options
-rw-r--r-- | include/llvm/Analysis/BlockFrequencyImpl.h | 17 | ||||
-rw-r--r-- | include/llvm/Analysis/BranchProbabilityInfo.h | 10 | ||||
-rw-r--r-- | lib/Analysis/BranchProbabilityInfo.cpp | 24 |
3 files changed, 13 insertions, 38 deletions
diff --git a/include/llvm/Analysis/BlockFrequencyImpl.h b/include/llvm/Analysis/BlockFrequencyImpl.h index 7447e80a31..cef375f10e 100644 --- a/include/llvm/Analysis/BlockFrequencyImpl.h +++ b/include/llvm/Analysis/BlockFrequencyImpl.h @@ -133,6 +133,15 @@ class BlockFrequencyImpl { } + /// Return a probability of getting to the DST block through SRC->DST edge. + /// + BranchProbability getBackEdgeProbability(BlockT *Src, BlockT *Dst) const { + uint32_t N = getEdgeFreq(Src, Dst); + uint32_t D = getBlockFreq(Dst); + + return BranchProbability(N, D); + } + /// isReachable - Returns if BB block is reachable from the entry. /// bool isReachable(BlockT *BB) { @@ -213,7 +222,9 @@ class BlockFrequencyImpl { return; assert(START_FREQ >= CycleProb[BB]); - divBlockFreq(BB, BranchProbability(START_FREQ - CycleProb[BB], START_FREQ)); + uint32_t CProb = CycleProb[BB]; + uint32_t Numerator = START_FREQ - CProb ? START_FREQ - CProb : 1; + divBlockFreq(BB, BranchProbability(Numerator, START_FREQ)); } /// doLoop - Propagate block frequency down throught the loop. @@ -238,19 +249,17 @@ class BlockFrequencyImpl { BlockT *Pred = *PI; assert(Pred); if (isReachable(Pred) && isBackedge(Pred, Head)) { - BranchProbability Prob = BPI->getBackEdgeProbability(Pred, Head); + BranchProbability Prob = getBackEdgeProbability(Pred, Head); uint64_t N = Prob.getNumerator(); uint64_t D = Prob.getDenominator(); uint64_t Res = (N * START_FREQ) / D; - // CycleProb[Head] += getEdgeFreq(Pred, Head); assert(Res <= UINT32_MAX); CycleProb[Head] += (uint32_t) Res; } } } - friend class BlockFrequency; void doFunction(FunctionT *fn, BlockProbInfoT *bpi) { diff --git a/include/llvm/Analysis/BranchProbabilityInfo.h b/include/llvm/Analysis/BranchProbabilityInfo.h index e40d2044dc..5a17a76f5b 100644 --- a/include/llvm/Analysis/BranchProbabilityInfo.h +++ b/include/llvm/Analysis/BranchProbabilityInfo.h @@ -39,9 +39,6 @@ class BranchProbabilityInfo : public FunctionPass { // Get sum of the block successors' weights. uint32_t getSumForBlock(BasicBlock *BB) const; - // Get sum of the edge weights going to the BB block. - uint32_t getBackSumForBlock(BasicBlock *BB) const; - public: static char ID; @@ -74,13 +71,6 @@ public: // only iff SRC block has only one successor. BranchProbability getEdgeProbability(BasicBlock *Src, BasicBlock *Dst) const; - // Return a probability of getting to the DST block through SRC->DST edge. - // Returned value is a fraction between 0 (0% probability) and - // 1 (100% probability), however the value is never equal to 0, and can be 1 - // only iff DST block has only one predecesor. - BranchProbability getBackEdgeProbability(BasicBlock *Src, - BasicBlock *Dst) const; - // Print value between 0 (0% probability) and 1 (100% probability), // however the value is never equal to 0, and can be 1 only iff SRC block // has only one successor. diff --git a/lib/Analysis/BranchProbabilityInfo.cpp b/lib/Analysis/BranchProbabilityInfo.cpp index 263ea2c26b..15059c733a 100644 --- a/lib/Analysis/BranchProbabilityInfo.cpp +++ b/lib/Analysis/BranchProbabilityInfo.cpp @@ -279,21 +279,6 @@ uint32_t BranchProbabilityInfo::getSumForBlock(BasicBlock *BB) const { return Sum; } -uint32_t BranchProbabilityInfo::getBackSumForBlock(BasicBlock *BB) const { - uint32_t Sum = 0; - - for (pred_iterator I = pred_begin(BB), E = pred_end(BB); I != E; ++I) { - BasicBlock *Pred = *I; - uint32_t Weight = getEdgeWeight(Pred, BB); - uint32_t PrevSum = Sum; - - Sum += Weight; - assert(Sum > PrevSum); (void) PrevSum; - } - - return Sum; -} - bool BranchProbabilityInfo::isEdgeHot(BasicBlock *Src, BasicBlock *Dst) const { // Hot probability is at least 4/5 = 80% uint32_t Weight = getEdgeWeight(Src, Dst); @@ -360,15 +345,6 @@ getEdgeProbability(BasicBlock *Src, BasicBlock *Dst) const { return BranchProbability(N, D); } -BranchProbability BranchProbabilityInfo:: -getBackEdgeProbability(BasicBlock *Src, BasicBlock *Dst) const { - - uint32_t N = getEdgeWeight(Src, Dst); - uint32_t D = getBackSumForBlock(Dst); - - return BranchProbability(N, D); -} - raw_ostream & BranchProbabilityInfo::printEdgeProbability(raw_ostream &OS, BasicBlock *Src, BasicBlock *Dst) const { |