Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 1 | //===----- MIRFSDiscriminator.h: MIR FS Discriminator Support --0-- 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 file contains the supporting functions for adding Machine level IR | ||
| 10 | // Flow Sensitive discriminators to the instruction debug information. With | ||
| 11 | // this, a cloned machine instruction in a different MachineBasicBlock will | ||
| 12 | // have its own discriminator value. This is done in a MIRAddFSDiscriminators | ||
| 13 | // pass. | ||
| 14 | // | ||
| 15 | //===----------------------------------------------------------------------===// | ||
| 16 | |||
| 17 | #ifndef LLVM_CODEGEN_MIRFSDISCRIMINATOR_H | ||
| 18 | #define LLVM_CODEGEN_MIRFSDISCRIMINATOR_H | ||
| 19 | |||
| 20 | #include "llvm/ADT/StringRef.h" | ||
| 21 | #include "llvm/CodeGen/MachineFunction.h" | ||
| 22 | #include "llvm/CodeGen/MachineFunctionPass.h" | ||
| 23 | #include "llvm/Support/Discriminator.h" | ||
| 24 | |||
| 25 | #include <cassert> | ||
| 26 | #include <cstdint> | ||
| 27 | |||
| 28 | namespace llvm { | ||
| 29 | class MachineFunction; | ||
| 30 | |||
| 31 | using namespace sampleprof; | ||
| 32 | class MIRAddFSDiscriminators : public MachineFunctionPass { | ||
| 33 | MachineFunction *MF; | ||
| 34 | unsigned LowBit; | ||
| 35 | unsigned HighBit; | ||
| 36 | |||
| 37 | public: | ||
| 38 | static char ID; | ||
| 39 |   /// PassNum is the sequence number this pass is called, start from 1. | ||
| 40 | MIRAddFSDiscriminators(FSDiscriminatorPass P = FSDiscriminatorPass::Pass1) | ||
| 41 | : MachineFunctionPass(ID) { | ||
| 42 | LowBit = getFSPassBitBegin(P); | ||
| 43 | HighBit = getFSPassBitEnd(P); | ||
| 44 | assert(LowBit < HighBit && "HighBit needs to be greater than Lowbit"); | ||
| 45 |   } | ||
| 46 | |||
| 47 | StringRef getPassName() const override { | ||
| 48 | return "Add FS discriminators in MIR"; | ||
| 49 |   } | ||
| 50 | |||
| 51 |   /// getNumFSBBs() - Return the number of machine BBs that have FS samples. | ||
| 52 | unsigned getNumFSBBs(); | ||
| 53 | |||
| 54 |   /// getNumFSSamples() - Return the number of samples that have flow sensitive | ||
| 55 |   /// values. | ||
| 56 | uint64_t getNumFSSamples(); | ||
| 57 | |||
| 58 |   /// getMachineFunction - Return the current machine function. | ||
| 59 | const MachineFunction *getMachineFunction() const { return MF; } | ||
| 60 | |||
| 61 | private: | ||
| 62 | bool runOnMachineFunction(MachineFunction &) override; | ||
| 63 | }; | ||
| 64 | |||
| 65 | } // namespace llvm | ||
| 66 | |||
| 67 | #endif // LLVM_CODEGEN_MIRFSDISCRIMINATOR_H |