SampleProfile.h 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. #pragma once
  2. #ifdef __GNUC__
  3. #pragma GCC diagnostic push
  4. #pragma GCC diagnostic ignored "-Wunused-parameter"
  5. #endif
  6. //===- SampleProfile.h - SamplePGO pass ---------- --------------*- C++ -*-===//
  7. //
  8. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
  9. // See https://llvm.org/LICENSE.txt for license information.
  10. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  11. //
  12. //===----------------------------------------------------------------------===//
  13. //
  14. /// \file
  15. /// This file provides the interface for the sampled PGO loader pass.
  16. //
  17. //===----------------------------------------------------------------------===//
  18. #ifndef LLVM_TRANSFORMS_IPO_SAMPLEPROFILE_H
  19. #define LLVM_TRANSFORMS_IPO_SAMPLEPROFILE_H
  20. #include "llvm/IR/PassManager.h"
  21. #include "llvm/Pass.h"
  22. #include <string>
  23. namespace llvm {
  24. class Module;
  25. /// The sample profiler data loader pass.
  26. class SampleProfileLoaderPass : public PassInfoMixin<SampleProfileLoaderPass> {
  27. public:
  28. SampleProfileLoaderPass(
  29. std::string File = "", std::string RemappingFile = "",
  30. ThinOrFullLTOPhase LTOPhase = ThinOrFullLTOPhase::None)
  31. : ProfileFileName(File), ProfileRemappingFileName(RemappingFile),
  32. LTOPhase(LTOPhase) {}
  33. PreservedAnalyses run(Module &M, ModuleAnalysisManager &AM);
  34. private:
  35. std::string ProfileFileName;
  36. std::string ProfileRemappingFileName;
  37. const ThinOrFullLTOPhase LTOPhase;
  38. };
  39. } // end namespace llvm
  40. #endif // LLVM_TRANSFORMS_IPO_SAMPLEPROFILE_H
  41. #ifdef __GNUC__
  42. #pragma GCC diagnostic pop
  43. #endif