1 //===- llvm/CodeGen/MBFIWrapper.h -------------------------------*- C++ -*-===//
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
7 //===----------------------------------------------------------------------===//
9 // This class keeps track of branch frequencies of newly created blocks and
10 // tail-merged blocks. Used by the TailDuplication and MachineBlockPlacement.
12 //===----------------------------------------------------------------------===//
14 #ifndef LLVM_CODEGEN_MBFIWRAPPER_H
15 #define LLVM_CODEGEN_MBFIWRAPPER_H
17 #include "llvm/ADT/DenseMap.h"
18 #include "llvm/Support/BlockFrequency.h"
22 class MachineBasicBlock;
23 class MachineBlockFrequencyInfo;
27 MBFIWrapper(const MachineBlockFrequencyInfo &I) : MBFI(I) {}
29 BlockFrequency getBlockFreq(const MachineBasicBlock *MBB) const;
30 void setBlockFreq(const MachineBasicBlock *MBB, BlockFrequency F);
31 raw_ostream &printBlockFreq(raw_ostream &OS,
32 const MachineBasicBlock *MBB) const;
33 raw_ostream &printBlockFreq(raw_ostream &OS,
34 const BlockFrequency Freq) const;
35 void view(const Twine &Name, bool isSimple = true);
36 uint64_t getEntryFreq() const;
37 const MachineBlockFrequencyInfo &getMBFI() { return MBFI; }
40 const MachineBlockFrequencyInfo &MBFI;
41 DenseMap<const MachineBasicBlock *, BlockFrequency> MergedBBFreq;
44 } // end namespace llvm
46 #endif // LLVM_CODEGEN_MBFIWRAPPER_H