1 //===- llvm/CodeGen/MBFIWrapper.h -------------------------------*- C++ -*-===// 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 #ifndef LLVM_CODEGEN_MBFIWRAPPER_H 15 #define LLVM_CODEGEN_MBFIWRAPPER_H 16 17 #include "llvm/ADT/DenseMap.h" 18 #include "llvm/ADT/Optional.h" 19 #include "llvm/Support/BlockFrequency.h" 20 21 namespace llvm { 22 23 class MachineBasicBlock; 24 class MachineBlockFrequencyInfo; 25 26 class MBFIWrapper { 27 public: 28 MBFIWrapper(const MachineBlockFrequencyInfo &I) : MBFI(I) {} 29 30 BlockFrequency getBlockFreq(const MachineBasicBlock *MBB) const; 31 void setBlockFreq(const MachineBasicBlock *MBB, BlockFrequency F); 32 Optional<uint64_t> getBlockProfileCount(const MachineBasicBlock *MBB) const; 33 34 raw_ostream &printBlockFreq(raw_ostream &OS, 35 const MachineBasicBlock *MBB) const; 36 raw_ostream &printBlockFreq(raw_ostream &OS, 37 const BlockFrequency Freq) const; 38 void view(const Twine &Name, bool isSimple = true); 39 uint64_t getEntryFreq() const; 40 const MachineBlockFrequencyInfo &getMBFI() { return MBFI; } 41 42 private: 43 const MachineBlockFrequencyInfo &MBFI; 44 DenseMap<const MachineBasicBlock *, BlockFrequency> MergedBBFreq; 45 }; 46 47 } // end namespace llvm 48 49 #endif // LLVM_CODEGEN_MBFIWRAPPER_H 50