Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //===- HotColdSplitting.h ---- Outline Cold Regions -------------*- 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 | // This pass outlines cold regions to a separate function. |
||
| 9 | // |
||
| 10 | //===----------------------------------------------------------------------===// |
||
| 11 | |||
| 12 | #ifndef LLVM_TRANSFORMS_IPO_HOTCOLDSPLITTING_H |
||
| 13 | #define LLVM_TRANSFORMS_IPO_HOTCOLDSPLITTING_H |
||
| 14 | |||
| 15 | #include "llvm/IR/PassManager.h" |
||
| 16 | |||
| 17 | namespace llvm { |
||
| 18 | |||
| 19 | class Module; |
||
| 20 | class ProfileSummaryInfo; |
||
| 21 | class BlockFrequencyInfo; |
||
| 22 | class TargetTransformInfo; |
||
| 23 | class OptimizationRemarkEmitter; |
||
| 24 | class AssumptionCache; |
||
| 25 | class DominatorTree; |
||
| 26 | class CodeExtractorAnalysisCache; |
||
| 27 | |||
| 28 | /// A sequence of basic blocks. |
||
| 29 | /// |
||
| 30 | /// A 0-sized SmallVector is slightly cheaper to move than a std::vector. |
||
| 31 | using BlockSequence = SmallVector<BasicBlock *, 0>; |
||
| 32 | |||
| 33 | class HotColdSplitting { |
||
| 34 | public: |
||
| 35 | HotColdSplitting(ProfileSummaryInfo *ProfSI, |
||
| 36 | function_ref<BlockFrequencyInfo *(Function &)> GBFI, |
||
| 37 | function_ref<TargetTransformInfo &(Function &)> GTTI, |
||
| 38 | std::function<OptimizationRemarkEmitter &(Function &)> *GORE, |
||
| 39 | function_ref<AssumptionCache *(Function &)> LAC) |
||
| 40 | : PSI(ProfSI), GetBFI(GBFI), GetTTI(GTTI), GetORE(GORE), LookupAC(LAC) {} |
||
| 41 | bool run(Module &M); |
||
| 42 | |||
| 43 | private: |
||
| 44 | bool isFunctionCold(const Function &F) const; |
||
| 45 | bool shouldOutlineFrom(const Function &F) const; |
||
| 46 | bool outlineColdRegions(Function &F, bool HasProfileSummary); |
||
| 47 | Function *extractColdRegion(const BlockSequence &Region, |
||
| 48 | const CodeExtractorAnalysisCache &CEAC, |
||
| 49 | DominatorTree &DT, BlockFrequencyInfo *BFI, |
||
| 50 | TargetTransformInfo &TTI, |
||
| 51 | OptimizationRemarkEmitter &ORE, |
||
| 52 | AssumptionCache *AC, unsigned Count); |
||
| 53 | ProfileSummaryInfo *PSI; |
||
| 54 | function_ref<BlockFrequencyInfo *(Function &)> GetBFI; |
||
| 55 | function_ref<TargetTransformInfo &(Function &)> GetTTI; |
||
| 56 | std::function<OptimizationRemarkEmitter &(Function &)> *GetORE; |
||
| 57 | function_ref<AssumptionCache *(Function &)> LookupAC; |
||
| 58 | }; |
||
| 59 | |||
| 60 | /// Pass to outline cold regions. |
||
| 61 | class HotColdSplittingPass : public PassInfoMixin<HotColdSplittingPass> { |
||
| 62 | public: |
||
| 63 | PreservedAnalyses run(Module &M, ModuleAnalysisManager &AM); |
||
| 64 | }; |
||
| 65 | |||
| 66 | } // end namespace llvm |
||
| 67 | |||
| 68 | #endif // LLVM_TRANSFORMS_IPO_HOTCOLDSPLITTING_H |
||
| 69 |