Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===-- llvm/Support/Timer.h - Interval Timing Support ----------*- 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 | #ifndef LLVM_SUPPORT_TIMER_H |
||
10 | #define LLVM_SUPPORT_TIMER_H |
||
11 | |||
12 | #include "llvm/ADT/StringMap.h" |
||
13 | #include "llvm/ADT/StringRef.h" |
||
14 | #include "llvm/Support/DataTypes.h" |
||
15 | #include <cassert> |
||
16 | #include <memory> |
||
17 | #include <string> |
||
18 | #include <vector> |
||
19 | |||
20 | namespace llvm { |
||
21 | |||
22 | class TimerGroup; |
||
23 | class raw_ostream; |
||
24 | |||
25 | class TimeRecord { |
||
26 | double WallTime; ///< Wall clock time elapsed in seconds. |
||
27 | double UserTime; ///< User time elapsed. |
||
28 | double SystemTime; ///< System time elapsed. |
||
29 | ssize_t MemUsed; ///< Memory allocated (in bytes). |
||
30 | uint64_t InstructionsExecuted; ///< Number of instructions executed |
||
31 | public: |
||
32 | TimeRecord() |
||
33 | : WallTime(0), UserTime(0), SystemTime(0), MemUsed(0), |
||
34 | InstructionsExecuted(0) {} |
||
35 | |||
36 | /// Get the current time and memory usage. If Start is true we get the memory |
||
37 | /// usage before the time, otherwise we get time before memory usage. This |
||
38 | /// matters if the time to get the memory usage is significant and shouldn't |
||
39 | /// be counted as part of a duration. |
||
40 | static TimeRecord getCurrentTime(bool Start = true); |
||
41 | |||
42 | double getProcessTime() const { return UserTime + SystemTime; } |
||
43 | double getUserTime() const { return UserTime; } |
||
44 | double getSystemTime() const { return SystemTime; } |
||
45 | double getWallTime() const { return WallTime; } |
||
46 | ssize_t getMemUsed() const { return MemUsed; } |
||
47 | uint64_t getInstructionsExecuted() const { return InstructionsExecuted; } |
||
48 | |||
49 | bool operator<(const TimeRecord &T) const { |
||
50 | // Sort by Wall Time elapsed, as it is the only thing really accurate |
||
51 | return WallTime < T.WallTime; |
||
52 | } |
||
53 | |||
54 | void operator+=(const TimeRecord &RHS) { |
||
55 | WallTime += RHS.WallTime; |
||
56 | UserTime += RHS.UserTime; |
||
57 | SystemTime += RHS.SystemTime; |
||
58 | MemUsed += RHS.MemUsed; |
||
59 | InstructionsExecuted += RHS.InstructionsExecuted; |
||
60 | } |
||
61 | void operator-=(const TimeRecord &RHS) { |
||
62 | WallTime -= RHS.WallTime; |
||
63 | UserTime -= RHS.UserTime; |
||
64 | SystemTime -= RHS.SystemTime; |
||
65 | MemUsed -= RHS.MemUsed; |
||
66 | InstructionsExecuted -= RHS.InstructionsExecuted; |
||
67 | } |
||
68 | |||
69 | /// Print the current time record to \p OS, with a breakdown showing |
||
70 | /// contributions to the \p Total time record. |
||
71 | void print(const TimeRecord &Total, raw_ostream &OS) const; |
||
72 | }; |
||
73 | |||
74 | /// This class is used to track the amount of time spent between invocations of |
||
75 | /// its startTimer()/stopTimer() methods. Given appropriate OS support it can |
||
76 | /// also keep track of the RSS of the program at various points. By default, |
||
77 | /// the Timer will print the amount of time it has captured to standard error |
||
78 | /// when the last timer is destroyed, otherwise it is printed when its |
||
79 | /// TimerGroup is destroyed. Timers do not print their information if they are |
||
80 | /// never started. |
||
81 | class Timer { |
||
82 | TimeRecord Time; ///< The total time captured. |
||
83 | TimeRecord StartTime; ///< The time startTimer() was last called. |
||
84 | std::string Name; ///< The name of this time variable. |
||
85 | std::string Description; ///< Description of this time variable. |
||
86 | bool Running = false; ///< Is the timer currently running? |
||
87 | bool Triggered = false; ///< Has the timer ever been triggered? |
||
88 | TimerGroup *TG = nullptr; ///< The TimerGroup this Timer is in. |
||
89 | |||
90 | Timer **Prev = nullptr; ///< Pointer to \p Next of previous timer in group. |
||
91 | Timer *Next = nullptr; ///< Next timer in the group. |
||
92 | public: |
||
93 | explicit Timer(StringRef TimerName, StringRef TimerDescription) { |
||
94 | init(TimerName, TimerDescription); |
||
95 | } |
||
96 | Timer(StringRef TimerName, StringRef TimerDescription, TimerGroup &tg) { |
||
97 | init(TimerName, TimerDescription, tg); |
||
98 | } |
||
99 | Timer(const Timer &RHS) { |
||
100 | assert(!RHS.TG && "Can only copy uninitialized timers"); |
||
101 | } |
||
102 | const Timer &operator=(const Timer &T) { |
||
103 | assert(!TG && !T.TG && "Can only assign uninit timers"); |
||
104 | return *this; |
||
105 | } |
||
106 | ~Timer(); |
||
107 | |||
108 | /// Create an uninitialized timer, client must use 'init'. |
||
109 | explicit Timer() = default; |
||
110 | void init(StringRef TimerName, StringRef TimerDescription); |
||
111 | void init(StringRef TimerName, StringRef TimerDescription, TimerGroup &tg); |
||
112 | |||
113 | const std::string &getName() const { return Name; } |
||
114 | const std::string &getDescription() const { return Description; } |
||
115 | bool isInitialized() const { return TG != nullptr; } |
||
116 | |||
117 | /// Check if the timer is currently running. |
||
118 | bool isRunning() const { return Running; } |
||
119 | |||
120 | /// Check if startTimer() has ever been called on this timer. |
||
121 | bool hasTriggered() const { return Triggered; } |
||
122 | |||
123 | /// Start the timer running. Time between calls to startTimer/stopTimer is |
||
124 | /// counted by the Timer class. Note that these calls must be correctly |
||
125 | /// paired. |
||
126 | void startTimer(); |
||
127 | |||
128 | /// Stop the timer. |
||
129 | void stopTimer(); |
||
130 | |||
131 | /// Clear the timer state. |
||
132 | void clear(); |
||
133 | |||
134 | /// Return the duration for which this timer has been running. |
||
135 | TimeRecord getTotalTime() const { return Time; } |
||
136 | |||
137 | private: |
||
138 | friend class TimerGroup; |
||
139 | }; |
||
140 | |||
141 | /// The TimeRegion class is used as a helper class to call the startTimer() and |
||
142 | /// stopTimer() methods of the Timer class. When the object is constructed, it |
||
143 | /// starts the timer specified as its argument. When it is destroyed, it stops |
||
144 | /// the relevant timer. This makes it easy to time a region of code. |
||
145 | class TimeRegion { |
||
146 | Timer *T; |
||
147 | TimeRegion(const TimeRegion &) = delete; |
||
148 | |||
149 | public: |
||
150 | explicit TimeRegion(Timer &t) : T(&t) { |
||
151 | T->startTimer(); |
||
152 | } |
||
153 | explicit TimeRegion(Timer *t) : T(t) { |
||
154 | if (T) T->startTimer(); |
||
155 | } |
||
156 | ~TimeRegion() { |
||
157 | if (T) T->stopTimer(); |
||
158 | } |
||
159 | }; |
||
160 | |||
161 | /// This class is basically a combination of TimeRegion and Timer. It allows |
||
162 | /// you to declare a new timer, AND specify the region to time, all in one |
||
163 | /// statement. All timers with the same name are merged. This is primarily |
||
164 | /// used for debugging and for hunting performance problems. |
||
165 | struct NamedRegionTimer : public TimeRegion { |
||
166 | explicit NamedRegionTimer(StringRef Name, StringRef Description, |
||
167 | StringRef GroupName, |
||
168 | StringRef GroupDescription, bool Enabled = true); |
||
169 | }; |
||
170 | |||
171 | /// The TimerGroup class is used to group together related timers into a single |
||
172 | /// report that is printed when the TimerGroup is destroyed. It is illegal to |
||
173 | /// destroy a TimerGroup object before all of the Timers in it are gone. A |
||
174 | /// TimerGroup can be specified for a newly created timer in its constructor. |
||
175 | class TimerGroup { |
||
176 | struct PrintRecord { |
||
177 | TimeRecord Time; |
||
178 | std::string Name; |
||
179 | std::string Description; |
||
180 | |||
181 | PrintRecord(const PrintRecord &Other) = default; |
||
182 | PrintRecord &operator=(const PrintRecord &Other) = default; |
||
183 | PrintRecord(const TimeRecord &Time, const std::string &Name, |
||
184 | const std::string &Description) |
||
185 | : Time(Time), Name(Name), Description(Description) {} |
||
186 | |||
187 | bool operator <(const PrintRecord &Other) const { |
||
188 | return Time < Other.Time; |
||
189 | } |
||
190 | }; |
||
191 | std::string Name; |
||
192 | std::string Description; |
||
193 | Timer *FirstTimer = nullptr; ///< First timer in the group. |
||
194 | std::vector<PrintRecord> TimersToPrint; |
||
195 | |||
196 | TimerGroup **Prev; ///< Pointer to Next field of previous timergroup in list. |
||
197 | TimerGroup *Next; ///< Pointer to next timergroup in list. |
||
198 | TimerGroup(const TimerGroup &TG) = delete; |
||
199 | void operator=(const TimerGroup &TG) = delete; |
||
200 | |||
201 | public: |
||
202 | explicit TimerGroup(StringRef Name, StringRef Description); |
||
203 | |||
204 | explicit TimerGroup(StringRef Name, StringRef Description, |
||
205 | const StringMap<TimeRecord> &Records); |
||
206 | |||
207 | ~TimerGroup(); |
||
208 | |||
209 | void setName(StringRef NewName, StringRef NewDescription) { |
||
210 | Name.assign(NewName.begin(), NewName.end()); |
||
211 | Description.assign(NewDescription.begin(), NewDescription.end()); |
||
212 | } |
||
213 | |||
214 | /// Print any started timers in this group, optionally resetting timers after |
||
215 | /// printing them. |
||
216 | void print(raw_ostream &OS, bool ResetAfterPrint = false); |
||
217 | |||
218 | /// Clear all timers in this group. |
||
219 | void clear(); |
||
220 | |||
221 | /// This static method prints all timers. |
||
222 | static void printAll(raw_ostream &OS); |
||
223 | |||
224 | /// Clear out all timers. This is mostly used to disable automatic |
||
225 | /// printing on shutdown, when timers have already been printed explicitly |
||
226 | /// using \c printAll or \c printJSONValues. |
||
227 | static void clearAll(); |
||
228 | |||
229 | const char *printJSONValues(raw_ostream &OS, const char *delim); |
||
230 | |||
231 | /// Prints all timers as JSON key/value pairs. |
||
232 | static const char *printAllJSONValues(raw_ostream &OS, const char *delim); |
||
233 | |||
234 | /// Ensure global objects required for statistics printing are initialized. |
||
235 | /// This function is used by the Statistic code to ensure correct order of |
||
236 | /// global constructors and destructors. |
||
237 | static void constructForStatistics(); |
||
238 | |||
239 | /// This makes the default group unmanaged, and lets the user manage the |
||
240 | /// group's lifetime. |
||
241 | static std::unique_ptr<TimerGroup> aquireDefaultGroup(); |
||
242 | |||
243 | private: |
||
244 | friend class Timer; |
||
245 | friend void PrintStatisticsJSON(raw_ostream &OS); |
||
246 | void addTimer(Timer &T); |
||
247 | void removeTimer(Timer &T); |
||
248 | void prepareToPrintList(bool reset_time = false); |
||
249 | void PrintQueuedTimers(raw_ostream &OS); |
||
250 | void printJSONValue(raw_ostream &OS, const PrintRecord &R, |
||
251 | const char *suffix, double Value); |
||
252 | }; |
||
253 | |||
254 | } // end namespace llvm |
||
255 | |||
256 | #endif |