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