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/Support/BlockFrequency.h" 19 #include <optional> 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 std::optional<uint64_t> 33 getBlockProfileCount(const MachineBasicBlock *MBB) const; 34 35 void view(const Twine &Name, bool isSimple = true); 36 BlockFrequency getEntryFreq() const; 37 const MachineBlockFrequencyInfo &getMBFI() const { return MBFI; } 38 39 private: 40 const MachineBlockFrequencyInfo &MBFI; 41 DenseMap<const MachineBasicBlock *, BlockFrequency> MergedBBFreq; 42 }; 43 44 } // end namespace llvm 45 46 #endif // LLVM_CODEGEN_MBFIWRAPPER_H 47