1 //===- Transforms/PGOInstrumentation.h - PGO gen/use passes  ---*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 /// \file
10 /// This file provides the interface for IR based instrumentation passes (
11 /// (profile-gen, and profile-use).
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef LLVM_TRANSFORMS_PGOINSTRUMENTATION_H
15 #define LLVM_TRANSFORMS_PGOINSTRUMENTATION_H
16 
17 #include "llvm/IR/PassManager.h"
18 #include "llvm/Transforms/Instrumentation.h"
19 
20 namespace llvm {
21 
22 /// The instrumentation (profile-instr-gen) pass for IR based PGO.
23 class PGOInstrumentationGen : public PassInfoMixin<PGOInstrumentationGen> {
24 public:
25   PreservedAnalyses run(Module &M, AnalysisManager<Module> &AM);
26 };
27 
28 /// The profile annotation (profile-instr-use) pass for IR based PGO.
29 class PGOInstrumentationUse : public PassInfoMixin<PGOInstrumentationUse> {
30 public:
31   PreservedAnalyses run(Module &M, AnalysisManager<Module> &AM);
32   PGOInstrumentationUse(std::string Filename = "");
33 
34 private:
35   std::string ProfileFileName;
36 };
37 
38 /// The indirect function call promotion pass.
39 class PGOIndirectCallPromotion : public PassInfoMixin<PGOIndirectCallPromotion> {
40 public:
InLTO(IsInLTO)41   PGOIndirectCallPromotion(bool IsInLTO = false) : InLTO(IsInLTO) {}
42   PreservedAnalyses run(Module &M, AnalysisManager<Module> &AM);
43 private:
44   bool InLTO;
45 };
46 
47 } // End llvm namespace
48 #endif
49