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 |