Home | History | Annotate | Line # | Download | only in CodeGen 
      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 
     20 namespace llvm {
     21 
     22 class MachineBasicBlock;
     23 class MachineBlockFrequencyInfo;
     24 
     25 class MBFIWrapper {
     26  public:
     27   MBFIWrapper(const MachineBlockFrequencyInfo &I) : MBFI(I) {}
     28 
     29   BlockFrequency getBlockFreq(const MachineBasicBlock *MBB) const;
     30   void setBlockFreq(const MachineBasicBlock *MBB, BlockFrequency F);
     31   Optional<uint64_t> getBlockProfileCount(const MachineBasicBlock *MBB) const;
     32 
     33   raw_ostream &printBlockFreq(raw_ostream &OS,
     34                               const MachineBasicBlock *MBB) const;
     35   raw_ostream &printBlockFreq(raw_ostream &OS,
     36                               const BlockFrequency Freq) const;
     37   void view(const Twine &Name, bool isSimple = true);
     38   uint64_t getEntryFreq() const;
     39   const MachineBlockFrequencyInfo &getMBFI() { return MBFI; }
     40 
     41  private:
     42   const MachineBlockFrequencyInfo &MBFI;
     43   DenseMap<const MachineBasicBlock *, BlockFrequency> MergedBBFreq;
     44 };
     45 
     46 } // end namespace llvm
     47 
     48 #endif // LLVM_CODEGEN_MBFIWRAPPER_H
     49