1 //===- MBFIWrapper.cpp - MachineBlockFrequencyInfo wrapper ----------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This class keeps track of branch frequencies of newly created blocks and
10 // tail-merged blocks. Used by the TailDuplication and MachineBlockPlacement.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "llvm/CodeGen/MBFIWrapper.h"
15 #include "llvm/CodeGen/MachineBlockFrequencyInfo.h"
16 
17 using namespace llvm;
18 
19 BlockFrequency MBFIWrapper::getBlockFreq(const MachineBasicBlock *MBB) const {
20   auto I = MergedBBFreq.find(MBB);
21 
22   if (I != MergedBBFreq.end())
23     return I->second;
24 
25   return MBFI.getBlockFreq(MBB);
26 }
27 
28 void MBFIWrapper::setBlockFreq(const MachineBasicBlock *MBB,
29                                BlockFrequency F) {
30   MergedBBFreq[MBB] = F;
31 }
32 
33 raw_ostream & MBFIWrapper::printBlockFreq(raw_ostream &OS,
34                                           const MachineBasicBlock *MBB) const {
35   return MBFI.printBlockFreq(OS, getBlockFreq(MBB));
36 }
37 
38 raw_ostream & MBFIWrapper::printBlockFreq(raw_ostream &OS,
39                                           const BlockFrequency Freq) const {
40   return MBFI.printBlockFreq(OS, Freq);
41 }
42 
43 void MBFIWrapper::view(const Twine &Name, bool isSimple) {
44   MBFI.view(Name, isSimple);
45 }
46 
47 uint64_t MBFIWrapper::getEntryFreq() const {
48   return MBFI.getEntryFreq();
49 }
50