Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 1 | //===- PassTimingInfo.h - pass execution timing -----------------*- 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 | /// \file | ||
| 9 | /// | ||
| 10 | /// This header defines classes/functions to handle pass execution timing | ||
| 11 | /// information with interfaces for both pass managers. | ||
| 12 | /// | ||
| 13 | //===----------------------------------------------------------------------===// | ||
| 14 | |||
| 15 | #ifndef LLVM_IR_PASSTIMINGINFO_H | ||
| 16 | #define LLVM_IR_PASSTIMINGINFO_H | ||
| 17 | |||
| 18 | #include "llvm/ADT/SmallVector.h" | ||
| 19 | #include "llvm/ADT/StringMap.h" | ||
| 20 | #include "llvm/ADT/StringRef.h" | ||
| 21 | #include "llvm/Support/Timer.h" | ||
| 22 | #include <memory> | ||
| 23 | #include <utility> | ||
| 24 | |||
| 25 | namespace llvm { | ||
| 26 | |||
| 27 | class Pass; | ||
| 28 | class PassInstrumentationCallbacks; | ||
| 29 | class raw_ostream; | ||
| 30 | |||
| 31 | /// If -time-passes has been specified, report the timings immediately and then | ||
| 32 | /// reset the timers to zero. By default it uses the stream created by | ||
| 33 | /// CreateInfoOutputFile(). | ||
| 34 | void reportAndResetTimings(raw_ostream *OutStream = nullptr); | ||
| 35 | |||
| 36 | /// Request the timer for this legacy-pass-manager's pass instance. | ||
| 37 | Timer *getPassTimer(Pass *); | ||
| 38 | |||
| 39 | /// This class implements -time-passes functionality for new pass manager. | ||
| 40 | /// It provides the pass-instrumentation callbacks that measure the pass | ||
| 41 | /// execution time. They collect timing info into individual timers as | ||
| 42 | /// passes are being run. At the end of its life-time it prints the resulting | ||
| 43 | /// timing report. | ||
| 44 | class TimePassesHandler { | ||
| 45 |   /// Value of this type is capable of uniquely identifying pass invocations. | ||
| 46 |   /// It is a pair of string Pass-Identifier (which for now is common | ||
| 47 |   /// to all the instance of a given pass) + sequential invocation counter. | ||
| 48 | using PassInvocationID = std::pair<StringRef, unsigned>; | ||
| 49 | |||
| 50 |   /// Groups of timers for passes and analyses. | ||
| 51 |   TimerGroup PassTG; | ||
| 52 |   TimerGroup AnalysisTG; | ||
| 53 | |||
| 54 | using TimerVector = llvm::SmallVector<std::unique_ptr<Timer>, 4>; | ||
| 55 |   /// Map of timers for pass invocations | ||
| 56 | StringMap<TimerVector> TimingData; | ||
| 57 | |||
| 58 |   /// Currently active pass timer. | ||
| 59 | Timer *ActivePassTimer = nullptr; | ||
| 60 |   /// Stack of currently active analysis timers. Analyses can request other | ||
| 61 |   /// analyses. | ||
| 62 | SmallVector<Timer *, 8> AnalysisActiveTimerStack; | ||
| 63 | |||
| 64 |   /// Custom output stream to print timing information into. | ||
| 65 |   /// By default (== nullptr) we emit time report into the stream created by | ||
| 66 |   /// CreateInfoOutputFile(). | ||
| 67 | raw_ostream *OutStream = nullptr; | ||
| 68 | |||
| 69 | bool Enabled; | ||
| 70 | bool PerRun; | ||
| 71 | |||
| 72 | public: | ||
| 73 | TimePassesHandler(); | ||
| 74 | TimePassesHandler(bool Enabled, bool PerRun = false); | ||
| 75 | |||
| 76 |   /// Destructor handles the print action if it has not been handled before. | ||
| 77 | ~TimePassesHandler() { print(); } | ||
| 78 | |||
| 79 |   /// Prints out timing information and then resets the timers. | ||
| 80 | void print(); | ||
| 81 | |||
| 82 |   // We intend this to be unique per-compilation, thus no copies. | ||
| 83 | TimePassesHandler(const TimePassesHandler &) = delete; | ||
| 84 | void operator=(const TimePassesHandler &) = delete; | ||
| 85 | |||
| 86 | void registerCallbacks(PassInstrumentationCallbacks &PIC); | ||
| 87 | |||
| 88 |   /// Set a custom output stream for subsequent reporting. | ||
| 89 | void setOutStream(raw_ostream &OutStream); | ||
| 90 | |||
| 91 | private: | ||
| 92 |   /// Dumps information for running/triggered timers, useful for debugging | ||
| 93 | LLVM_DUMP_METHOD void dump() const; | ||
| 94 | |||
| 95 |   /// Returns the new timer for each new run of the pass. | ||
| 96 | Timer &getPassTimer(StringRef PassID, bool IsPass); | ||
| 97 | |||
| 98 | void startAnalysisTimer(StringRef PassID); | ||
| 99 | void stopAnalysisTimer(StringRef PassID); | ||
| 100 | void startPassTimer(StringRef PassID); | ||
| 101 | void stopPassTimer(StringRef PassID); | ||
| 102 | }; | ||
| 103 | |||
| 104 | } // namespace llvm | ||
| 105 | |||
| 106 | #endif |