Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //=- FunctionPropertiesAnalysis.h - Function Properties Analysis --*- 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 defines the FunctionPropertiesInfo and FunctionPropertiesAnalysis |
||
| 10 | // classes used to extract function properties. |
||
| 11 | // |
||
| 12 | //===----------------------------------------------------------------------===// |
||
| 13 | |||
| 14 | #ifndef LLVM_ANALYSIS_FUNCTIONPROPERTIESANALYSIS_H |
||
| 15 | #define LLVM_ANALYSIS_FUNCTIONPROPERTIESANALYSIS_H |
||
| 16 | |||
| 17 | #include "llvm/ADT/SmallPtrSet.h" |
||
| 18 | #include "llvm/ADT/iterator_range.h" |
||
| 19 | #include "llvm/IR/InstrTypes.h" |
||
| 20 | #include "llvm/IR/PassManager.h" |
||
| 21 | |||
| 22 | namespace llvm { |
||
| 23 | class Function; |
||
| 24 | class LoopInfo; |
||
| 25 | |||
| 26 | class FunctionPropertiesInfo { |
||
| 27 | friend class FunctionPropertiesUpdater; |
||
| 28 | void updateForBB(const BasicBlock &BB, int64_t Direction); |
||
| 29 | void updateAggregateStats(const Function &F, const LoopInfo &LI); |
||
| 30 | void reIncludeBB(const BasicBlock &BB); |
||
| 31 | |||
| 32 | public: |
||
| 33 | static FunctionPropertiesInfo |
||
| 34 | getFunctionPropertiesInfo(const Function &F, FunctionAnalysisManager &FAM); |
||
| 35 | |||
| 36 | bool operator==(const FunctionPropertiesInfo &FPI) const { |
||
| 37 | return std::memcmp(this, &FPI, sizeof(FunctionPropertiesInfo)) == 0; |
||
| 38 | } |
||
| 39 | |||
| 40 | bool operator!=(const FunctionPropertiesInfo &FPI) const { |
||
| 41 | return !(*this == FPI); |
||
| 42 | } |
||
| 43 | |||
| 44 | void print(raw_ostream &OS) const; |
||
| 45 | |||
| 46 | /// Number of basic blocks |
||
| 47 | int64_t BasicBlockCount = 0; |
||
| 48 | |||
| 49 | /// Number of blocks reached from a conditional instruction, or that are |
||
| 50 | /// 'cases' of a SwitchInstr. |
||
| 51 | // FIXME: We may want to replace this with a more meaningful metric, like |
||
| 52 | // number of conditionally executed blocks: |
||
| 53 | // 'if (a) s();' would be counted here as 2 blocks, just like |
||
| 54 | // 'if (a) s(); else s2(); s3();' would. |
||
| 55 | int64_t BlocksReachedFromConditionalInstruction = 0; |
||
| 56 | |||
| 57 | /// Number of uses of this function, plus 1 if the function is callable |
||
| 58 | /// outside the module. |
||
| 59 | int64_t Uses = 0; |
||
| 60 | |||
| 61 | /// Number of direct calls made from this function to other functions |
||
| 62 | /// defined in this module. |
||
| 63 | int64_t DirectCallsToDefinedFunctions = 0; |
||
| 64 | |||
| 65 | // Load Instruction Count |
||
| 66 | int64_t LoadInstCount = 0; |
||
| 67 | |||
| 68 | // Store Instruction Count |
||
| 69 | int64_t StoreInstCount = 0; |
||
| 70 | |||
| 71 | // Maximum Loop Depth in the Function |
||
| 72 | int64_t MaxLoopDepth = 0; |
||
| 73 | |||
| 74 | // Number of Top Level Loops in the Function |
||
| 75 | int64_t TopLevelLoopCount = 0; |
||
| 76 | |||
| 77 | // All non-debug instructions |
||
| 78 | int64_t TotalInstructionCount = 0; |
||
| 79 | }; |
||
| 80 | |||
| 81 | // Analysis pass |
||
| 82 | class FunctionPropertiesAnalysis |
||
| 83 | : public AnalysisInfoMixin<FunctionPropertiesAnalysis> { |
||
| 84 | |||
| 85 | public: |
||
| 86 | static AnalysisKey Key; |
||
| 87 | |||
| 88 | using Result = const FunctionPropertiesInfo; |
||
| 89 | |||
| 90 | FunctionPropertiesInfo run(Function &F, FunctionAnalysisManager &FAM); |
||
| 91 | }; |
||
| 92 | |||
| 93 | /// Printer pass for the FunctionPropertiesAnalysis results. |
||
| 94 | class FunctionPropertiesPrinterPass |
||
| 95 | : public PassInfoMixin<FunctionPropertiesPrinterPass> { |
||
| 96 | raw_ostream &OS; |
||
| 97 | |||
| 98 | public: |
||
| 99 | explicit FunctionPropertiesPrinterPass(raw_ostream &OS) : OS(OS) {} |
||
| 100 | |||
| 101 | PreservedAnalyses run(Function &F, FunctionAnalysisManager &AM); |
||
| 102 | }; |
||
| 103 | |||
| 104 | /// Correctly update FunctionPropertiesInfo post-inlining. A |
||
| 105 | /// FunctionPropertiesUpdater keeps the state necessary for tracking the changes |
||
| 106 | /// llvm::InlineFunction makes. The idea is that inlining will at most modify |
||
| 107 | /// a few BBs of the Caller (maybe the entry BB and definitely the callsite BB) |
||
| 108 | /// and potentially affect exception handling BBs in the case of invoke |
||
| 109 | /// inlining. |
||
| 110 | class FunctionPropertiesUpdater { |
||
| 111 | public: |
||
| 112 | FunctionPropertiesUpdater(FunctionPropertiesInfo &FPI, const CallBase &CB); |
||
| 113 | |||
| 114 | void finish(FunctionAnalysisManager &FAM) const; |
||
| 115 | |||
| 116 | private: |
||
| 117 | FunctionPropertiesInfo &FPI; |
||
| 118 | const BasicBlock &CallSiteBB; |
||
| 119 | const Function &Caller; |
||
| 120 | |||
| 121 | DenseSet<const BasicBlock *> Successors; |
||
| 122 | }; |
||
| 123 | } // namespace llvm |
||
| 124 | #endif // LLVM_ANALYSIS_FUNCTIONPROPERTIESANALYSIS_H |