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/MachineBlockFrequencyInfo.h" 15 #include "llvm/CodeGen/MBFIWrapper.h" 16 #include <optional> 17 18 using namespace llvm; 19 20 BlockFrequency MBFIWrapper::getBlockFreq(const MachineBasicBlock *MBB) const { 21 auto I = MergedBBFreq.find(MBB); 22 23 if (I != MergedBBFreq.end()) 24 return I->second; 25 26 return MBFI.getBlockFreq(MBB); 27 } 28 29 void MBFIWrapper::setBlockFreq(const MachineBasicBlock *MBB, 30 BlockFrequency F) { 31 MergedBBFreq[MBB] = F; 32 } 33 34 std::optional<uint64_t> 35 MBFIWrapper::getBlockProfileCount(const MachineBasicBlock *MBB) const { 36 auto I = MergedBBFreq.find(MBB); 37 38 // Modified block frequency also impacts profile count. So we should compute 39 // profile count from new block frequency if it has been changed. 40 if (I != MergedBBFreq.end()) 41 return MBFI.getProfileCountFromFreq(I->second.getFrequency()); 42 43 return MBFI.getBlockProfileCount(MBB); 44 } 45 46 raw_ostream & MBFIWrapper::printBlockFreq(raw_ostream &OS, 47 const MachineBasicBlock *MBB) const { 48 return MBFI.printBlockFreq(OS, getBlockFreq(MBB)); 49 } 50 51 raw_ostream & MBFIWrapper::printBlockFreq(raw_ostream &OS, 52 const BlockFrequency Freq) const { 53 return MBFI.printBlockFreq(OS, Freq); 54 } 55 56 void MBFIWrapper::view(const Twine &Name, bool isSimple) { 57 MBFI.view(Name, isSimple); 58 } 59 60 uint64_t MBFIWrapper::getEntryFreq() const { 61 return MBFI.getEntryFreq(); 62 } 63