aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/libs/llvm12/lib/CodeGen/MBFIWrapper.cpp
blob: a3de59e64f45c089447a63439e7cb20ee61358b5 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
//===- MBFIWrapper.cpp - MachineBlockFrequencyInfo wrapper ----------------===// 
// 
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 
// See https://llvm.org/LICENSE.txt for license information. 
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 
// 
//===----------------------------------------------------------------------===// 
// 
// This class keeps track of branch frequencies of newly created blocks and 
// tail-merged blocks. Used by the TailDuplication and MachineBlockPlacement. 
// 
//===----------------------------------------------------------------------===// 
 
#include "llvm/CodeGen/MBFIWrapper.h" 
#include "llvm/CodeGen/MachineBlockFrequencyInfo.h" 
 
using namespace llvm; 
 
BlockFrequency MBFIWrapper::getBlockFreq(const MachineBasicBlock *MBB) const { 
  auto I = MergedBBFreq.find(MBB); 
 
  if (I != MergedBBFreq.end()) 
    return I->second; 
 
  return MBFI.getBlockFreq(MBB); 
} 
 
void MBFIWrapper::setBlockFreq(const MachineBasicBlock *MBB, 
                               BlockFrequency F) { 
  MergedBBFreq[MBB] = F; 
} 
 
Optional<uint64_t>
MBFIWrapper::getBlockProfileCount(const MachineBasicBlock *MBB) const {
  auto I = MergedBBFreq.find(MBB);

  // Modified block frequency also impacts profile count. So we should compute
  // profile count from new block frequency if it has been changed.
  if (I != MergedBBFreq.end())
    return MBFI.getProfileCountFromFreq(I->second.getFrequency());

  return MBFI.getBlockProfileCount(MBB);
}

raw_ostream & MBFIWrapper::printBlockFreq(raw_ostream &OS, 
                                          const MachineBasicBlock *MBB) const { 
  return MBFI.printBlockFreq(OS, getBlockFreq(MBB)); 
} 
 
raw_ostream & MBFIWrapper::printBlockFreq(raw_ostream &OS, 
                                          const BlockFrequency Freq) const { 
  return MBFI.printBlockFreq(OS, Freq); 
} 
 
void MBFIWrapper::view(const Twine &Name, bool isSimple) { 
  MBFI.view(Name, isSimple); 
} 
 
uint64_t MBFIWrapper::getEntryFreq() const { 
  return MBFI.getEntryFreq(); 
}