Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- lib/CodeGen/MachineTraceMetrics.h - Super-scalar metrics -*- 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 interface for the MachineTraceMetrics analysis pass |
||
10 | // that estimates CPU resource usage and critical data dependency paths through |
||
11 | // preferred traces. This is useful for super-scalar CPUs where execution speed |
||
12 | // can be limited both by data dependencies and by limited execution resources. |
||
13 | // |
||
14 | // Out-of-order CPUs will often be executing instructions from multiple basic |
||
15 | // blocks at the same time. This makes it difficult to estimate the resource |
||
16 | // usage accurately in a single basic block. Resources can be estimated better |
||
17 | // by looking at a trace through the current basic block. |
||
18 | // |
||
19 | // For every block, the MachineTraceMetrics pass will pick a preferred trace |
||
20 | // that passes through the block. The trace is chosen based on loop structure, |
||
21 | // branch probabilities, and resource usage. The intention is to pick likely |
||
22 | // traces that would be the most affected by code transformations. |
||
23 | // |
||
24 | // It is expensive to compute a full arbitrary trace for every block, so to |
||
25 | // save some computations, traces are chosen to be convergent. This means that |
||
26 | // if the traces through basic blocks A and B ever cross when moving away from |
||
27 | // A and B, they never diverge again. This applies in both directions - If the |
||
28 | // traces meet above A and B, they won't diverge when going further back. |
||
29 | // |
||
30 | // Traces tend to align with loops. The trace through a block in an inner loop |
||
31 | // will begin at the loop entry block and end at a back edge. If there are |
||
32 | // nested loops, the trace may begin and end at those instead. |
||
33 | // |
||
34 | // For each trace, we compute the critical path length, which is the number of |
||
35 | // cycles required to execute the trace when execution is limited by data |
||
36 | // dependencies only. We also compute the resource height, which is the number |
||
37 | // of cycles required to execute all instructions in the trace when ignoring |
||
38 | // data dependencies. |
||
39 | // |
||
40 | // Every instruction in the current block has a slack - the number of cycles |
||
41 | // execution of the instruction can be delayed without extending the critical |
||
42 | // path. |
||
43 | // |
||
44 | //===----------------------------------------------------------------------===// |
||
45 | |||
46 | #ifndef LLVM_CODEGEN_MACHINETRACEMETRICS_H |
||
47 | #define LLVM_CODEGEN_MACHINETRACEMETRICS_H |
||
48 | |||
49 | #include "llvm/ADT/SparseSet.h" |
||
50 | #include "llvm/ADT/ArrayRef.h" |
||
51 | #include "llvm/ADT/DenseMap.h" |
||
52 | #include "llvm/ADT/SmallVector.h" |
||
53 | #include "llvm/CodeGen/MachineBasicBlock.h" |
||
54 | #include "llvm/CodeGen/MachineFunctionPass.h" |
||
55 | #include "llvm/CodeGen/TargetSchedule.h" |
||
56 | |||
57 | namespace llvm { |
||
58 | |||
59 | class AnalysisUsage; |
||
60 | class MachineFunction; |
||
61 | class MachineInstr; |
||
62 | class MachineLoop; |
||
63 | class MachineLoopInfo; |
||
64 | class MachineRegisterInfo; |
||
65 | struct MCSchedClassDesc; |
||
66 | class raw_ostream; |
||
67 | class TargetInstrInfo; |
||
68 | class TargetRegisterInfo; |
||
69 | |||
70 | // Keep track of physreg data dependencies by recording each live register unit. |
||
71 | // Associate each regunit with an instruction operand. Depending on the |
||
72 | // direction instructions are scanned, it could be the operand that defined the |
||
73 | // regunit, or the highest operand to read the regunit. |
||
74 | struct LiveRegUnit { |
||
75 | unsigned RegUnit; |
||
76 | unsigned Cycle = 0; |
||
77 | const MachineInstr *MI = nullptr; |
||
78 | unsigned Op = 0; |
||
79 | |||
80 | unsigned getSparseSetIndex() const { return RegUnit; } |
||
81 | |||
82 | LiveRegUnit(unsigned RU) : RegUnit(RU) {} |
||
83 | }; |
||
84 | |||
85 | |||
86 | class MachineTraceMetrics : public MachineFunctionPass { |
||
87 | const MachineFunction *MF = nullptr; |
||
88 | const TargetInstrInfo *TII = nullptr; |
||
89 | const TargetRegisterInfo *TRI = nullptr; |
||
90 | const MachineRegisterInfo *MRI = nullptr; |
||
91 | const MachineLoopInfo *Loops = nullptr; |
||
92 | TargetSchedModel SchedModel; |
||
93 | |||
94 | public: |
||
95 | friend class Ensemble; |
||
96 | friend class Trace; |
||
97 | |||
98 | class Ensemble; |
||
99 | |||
100 | static char ID; |
||
101 | |||
102 | MachineTraceMetrics(); |
||
103 | |||
104 | void getAnalysisUsage(AnalysisUsage&) const override; |
||
105 | bool runOnMachineFunction(MachineFunction&) override; |
||
106 | void releaseMemory() override; |
||
107 | void verifyAnalysis() const override; |
||
108 | |||
109 | /// Per-basic block information that doesn't depend on the trace through the |
||
110 | /// block. |
||
111 | struct FixedBlockInfo { |
||
112 | /// The number of non-trivial instructions in the block. |
||
113 | /// Doesn't count PHI and COPY instructions that are likely to be removed. |
||
114 | unsigned InstrCount = ~0u; |
||
115 | |||
116 | /// True when the block contains calls. |
||
117 | bool HasCalls = false; |
||
118 | |||
119 | FixedBlockInfo() = default; |
||
120 | |||
121 | /// Returns true when resource information for this block has been computed. |
||
122 | bool hasResources() const { return InstrCount != ~0u; } |
||
123 | |||
124 | /// Invalidate resource information. |
||
125 | void invalidate() { InstrCount = ~0u; } |
||
126 | }; |
||
127 | |||
128 | /// Get the fixed resource information about MBB. Compute it on demand. |
||
129 | const FixedBlockInfo *getResources(const MachineBasicBlock*); |
||
130 | |||
131 | /// Get the scaled number of cycles used per processor resource in MBB. |
||
132 | /// This is an array with SchedModel.getNumProcResourceKinds() entries. |
||
133 | /// The getResources() function above must have been called first. |
||
134 | /// |
||
135 | /// These numbers have already been scaled by SchedModel.getResourceFactor(). |
||
136 | ArrayRef<unsigned> getProcResourceCycles(unsigned MBBNum) const; |
||
137 | |||
138 | /// A virtual register or regunit required by a basic block or its trace |
||
139 | /// successors. |
||
140 | struct LiveInReg { |
||
141 | /// The virtual register required, or a register unit. |
||
142 | Register Reg; |
||
143 | |||
144 | /// For virtual registers: Minimum height of the defining instruction. |
||
145 | /// For regunits: Height of the highest user in the trace. |
||
146 | unsigned Height; |
||
147 | |||
148 | LiveInReg(Register Reg, unsigned Height = 0) : Reg(Reg), Height(Height) {} |
||
149 | }; |
||
150 | |||
151 | /// Per-basic block information that relates to a specific trace through the |
||
152 | /// block. Convergent traces means that only one of these is required per |
||
153 | /// block in a trace ensemble. |
||
154 | struct TraceBlockInfo { |
||
155 | /// Trace predecessor, or NULL for the first block in the trace. |
||
156 | /// Valid when hasValidDepth(). |
||
157 | const MachineBasicBlock *Pred = nullptr; |
||
158 | |||
159 | /// Trace successor, or NULL for the last block in the trace. |
||
160 | /// Valid when hasValidHeight(). |
||
161 | const MachineBasicBlock *Succ = nullptr; |
||
162 | |||
163 | /// The block number of the head of the trace. (When hasValidDepth()). |
||
164 | unsigned Head; |
||
165 | |||
166 | /// The block number of the tail of the trace. (When hasValidHeight()). |
||
167 | unsigned Tail; |
||
168 | |||
169 | /// Accumulated number of instructions in the trace above this block. |
||
170 | /// Does not include instructions in this block. |
||
171 | unsigned InstrDepth = ~0u; |
||
172 | |||
173 | /// Accumulated number of instructions in the trace below this block. |
||
174 | /// Includes instructions in this block. |
||
175 | unsigned InstrHeight = ~0u; |
||
176 | |||
177 | TraceBlockInfo() = default; |
||
178 | |||
179 | /// Returns true if the depth resources have been computed from the trace |
||
180 | /// above this block. |
||
181 | bool hasValidDepth() const { return InstrDepth != ~0u; } |
||
182 | |||
183 | /// Returns true if the height resources have been computed from the trace |
||
184 | /// below this block. |
||
185 | bool hasValidHeight() const { return InstrHeight != ~0u; } |
||
186 | |||
187 | /// Invalidate depth resources when some block above this one has changed. |
||
188 | void invalidateDepth() { InstrDepth = ~0u; HasValidInstrDepths = false; } |
||
189 | |||
190 | /// Invalidate height resources when a block below this one has changed. |
||
191 | void invalidateHeight() { InstrHeight = ~0u; HasValidInstrHeights = false; } |
||
192 | |||
193 | /// Assuming that this is a dominator of TBI, determine if it contains |
||
194 | /// useful instruction depths. A dominating block can be above the current |
||
195 | /// trace head, and any dependencies from such a far away dominator are not |
||
196 | /// expected to affect the critical path. |
||
197 | /// |
||
198 | /// Also returns true when TBI == this. |
||
199 | bool isUsefulDominator(const TraceBlockInfo &TBI) const { |
||
200 | // The trace for TBI may not even be calculated yet. |
||
201 | if (!hasValidDepth() || !TBI.hasValidDepth()) |
||
202 | return false; |
||
203 | // Instruction depths are only comparable if the traces share a head. |
||
204 | if (Head != TBI.Head) |
||
205 | return false; |
||
206 | // It is almost always the case that TBI belongs to the same trace as |
||
207 | // this block, but rare convoluted cases involving irreducible control |
||
208 | // flow, a dominator may share a trace head without actually being on the |
||
209 | // same trace as TBI. This is not a big problem as long as it doesn't |
||
210 | // increase the instruction depth. |
||
211 | return HasValidInstrDepths && InstrDepth <= TBI.InstrDepth; |
||
212 | } |
||
213 | |||
214 | // Data-dependency-related information. Per-instruction depth and height |
||
215 | // are computed from data dependencies in the current trace, using |
||
216 | // itinerary data. |
||
217 | |||
218 | /// Instruction depths have been computed. This implies hasValidDepth(). |
||
219 | bool HasValidInstrDepths = false; |
||
220 | |||
221 | /// Instruction heights have been computed. This implies hasValidHeight(). |
||
222 | bool HasValidInstrHeights = false; |
||
223 | |||
224 | /// Critical path length. This is the number of cycles in the longest data |
||
225 | /// dependency chain through the trace. This is only valid when both |
||
226 | /// HasValidInstrDepths and HasValidInstrHeights are set. |
||
227 | unsigned CriticalPath; |
||
228 | |||
229 | /// Live-in registers. These registers are defined above the current block |
||
230 | /// and used by this block or a block below it. |
||
231 | /// This does not include PHI uses in the current block, but it does |
||
232 | /// include PHI uses in deeper blocks. |
||
233 | SmallVector<LiveInReg, 4> LiveIns; |
||
234 | |||
235 | void print(raw_ostream&) const; |
||
236 | }; |
||
237 | |||
238 | /// InstrCycles represents the cycle height and depth of an instruction in a |
||
239 | /// trace. |
||
240 | struct InstrCycles { |
||
241 | /// Earliest issue cycle as determined by data dependencies and instruction |
||
242 | /// latencies from the beginning of the trace. Data dependencies from |
||
243 | /// before the trace are not included. |
||
244 | unsigned Depth; |
||
245 | |||
246 | /// Minimum number of cycles from this instruction is issued to the of the |
||
247 | /// trace, as determined by data dependencies and instruction latencies. |
||
248 | unsigned Height; |
||
249 | }; |
||
250 | |||
251 | /// A trace represents a plausible sequence of executed basic blocks that |
||
252 | /// passes through the current basic block one. The Trace class serves as a |
||
253 | /// handle to internal cached data structures. |
||
254 | class Trace { |
||
255 | Ensemble &TE; |
||
256 | TraceBlockInfo &TBI; |
||
257 | |||
258 | unsigned getBlockNum() const { return &TBI - &TE.BlockInfo[0]; } |
||
259 | |||
260 | public: |
||
261 | explicit Trace(Ensemble &te, TraceBlockInfo &tbi) : TE(te), TBI(tbi) {} |
||
262 | |||
263 | void print(raw_ostream&) const; |
||
264 | |||
265 | /// Compute the total number of instructions in the trace. |
||
266 | unsigned getInstrCount() const { |
||
267 | return TBI.InstrDepth + TBI.InstrHeight; |
||
268 | } |
||
269 | |||
270 | /// Return the resource depth of the top/bottom of the trace center block. |
||
271 | /// This is the number of cycles required to execute all instructions from |
||
272 | /// the trace head to the trace center block. The resource depth only |
||
273 | /// considers execution resources, it ignores data dependencies. |
||
274 | /// When Bottom is set, instructions in the trace center block are included. |
||
275 | unsigned getResourceDepth(bool Bottom) const; |
||
276 | |||
277 | /// Return the resource length of the trace. This is the number of cycles |
||
278 | /// required to execute the instructions in the trace if they were all |
||
279 | /// independent, exposing the maximum instruction-level parallelism. |
||
280 | /// |
||
281 | /// Any blocks in Extrablocks are included as if they were part of the |
||
282 | /// trace. Likewise, extra resources required by the specified scheduling |
||
283 | /// classes are included. For the caller to account for extra machine |
||
284 | /// instructions, it must first resolve each instruction's scheduling class. |
||
285 | unsigned getResourceLength( |
||
286 | ArrayRef<const MachineBasicBlock *> Extrablocks = std::nullopt, |
||
287 | ArrayRef<const MCSchedClassDesc *> ExtraInstrs = std::nullopt, |
||
288 | ArrayRef<const MCSchedClassDesc *> RemoveInstrs = std::nullopt) const; |
||
289 | |||
290 | /// Return the length of the (data dependency) critical path through the |
||
291 | /// trace. |
||
292 | unsigned getCriticalPath() const { return TBI.CriticalPath; } |
||
293 | |||
294 | /// Return the depth and height of MI. The depth is only valid for |
||
295 | /// instructions in or above the trace center block. The height is only |
||
296 | /// valid for instructions in or below the trace center block. |
||
297 | InstrCycles getInstrCycles(const MachineInstr &MI) const { |
||
298 | return TE.Cycles.lookup(&MI); |
||
299 | } |
||
300 | |||
301 | /// Return the slack of MI. This is the number of cycles MI can be delayed |
||
302 | /// before the critical path becomes longer. |
||
303 | /// MI must be an instruction in the trace center block. |
||
304 | unsigned getInstrSlack(const MachineInstr &MI) const; |
||
305 | |||
306 | /// Return the Depth of a PHI instruction in a trace center block successor. |
||
307 | /// The PHI does not have to be part of the trace. |
||
308 | unsigned getPHIDepth(const MachineInstr &PHI) const; |
||
309 | |||
310 | /// A dependence is useful if the basic block of the defining instruction |
||
311 | /// is part of the trace of the user instruction. It is assumed that DefMI |
||
312 | /// dominates UseMI (see also isUsefulDominator). |
||
313 | bool isDepInTrace(const MachineInstr &DefMI, |
||
314 | const MachineInstr &UseMI) const; |
||
315 | }; |
||
316 | |||
317 | /// A trace ensemble is a collection of traces selected using the same |
||
318 | /// strategy, for example 'minimum resource height'. There is one trace for |
||
319 | /// every block in the function. |
||
320 | class Ensemble { |
||
321 | friend class Trace; |
||
322 | |||
323 | SmallVector<TraceBlockInfo, 4> BlockInfo; |
||
324 | DenseMap<const MachineInstr*, InstrCycles> Cycles; |
||
325 | SmallVector<unsigned, 0> ProcResourceDepths; |
||
326 | SmallVector<unsigned, 0> ProcResourceHeights; |
||
327 | |||
328 | void computeTrace(const MachineBasicBlock*); |
||
329 | void computeDepthResources(const MachineBasicBlock*); |
||
330 | void computeHeightResources(const MachineBasicBlock*); |
||
331 | unsigned computeCrossBlockCriticalPath(const TraceBlockInfo&); |
||
332 | void computeInstrDepths(const MachineBasicBlock*); |
||
333 | void computeInstrHeights(const MachineBasicBlock*); |
||
334 | void addLiveIns(const MachineInstr *DefMI, unsigned DefOp, |
||
335 | ArrayRef<const MachineBasicBlock*> Trace); |
||
336 | |||
337 | protected: |
||
338 | MachineTraceMetrics &MTM; |
||
339 | |||
340 | explicit Ensemble(MachineTraceMetrics*); |
||
341 | |||
342 | virtual const MachineBasicBlock *pickTracePred(const MachineBasicBlock*) =0; |
||
343 | virtual const MachineBasicBlock *pickTraceSucc(const MachineBasicBlock*) =0; |
||
344 | const MachineLoop *getLoopFor(const MachineBasicBlock*) const; |
||
345 | const TraceBlockInfo *getDepthResources(const MachineBasicBlock*) const; |
||
346 | const TraceBlockInfo *getHeightResources(const MachineBasicBlock*) const; |
||
347 | ArrayRef<unsigned> getProcResourceDepths(unsigned MBBNum) const; |
||
348 | ArrayRef<unsigned> getProcResourceHeights(unsigned MBBNum) const; |
||
349 | |||
350 | public: |
||
351 | virtual ~Ensemble(); |
||
352 | |||
353 | virtual const char *getName() const = 0; |
||
354 | void print(raw_ostream&) const; |
||
355 | void invalidate(const MachineBasicBlock *MBB); |
||
356 | void verify() const; |
||
357 | |||
358 | /// Get the trace that passes through MBB. |
||
359 | /// The trace is computed on demand. |
||
360 | Trace getTrace(const MachineBasicBlock *MBB); |
||
361 | |||
362 | /// Updates the depth of an machine instruction, given RegUnits. |
||
363 | void updateDepth(TraceBlockInfo &TBI, const MachineInstr&, |
||
364 | SparseSet<LiveRegUnit> &RegUnits); |
||
365 | void updateDepth(const MachineBasicBlock *, const MachineInstr&, |
||
366 | SparseSet<LiveRegUnit> &RegUnits); |
||
367 | |||
368 | /// Updates the depth of the instructions from Start to End. |
||
369 | void updateDepths(MachineBasicBlock::iterator Start, |
||
370 | MachineBasicBlock::iterator End, |
||
371 | SparseSet<LiveRegUnit> &RegUnits); |
||
372 | |||
373 | }; |
||
374 | |||
375 | /// Strategies for selecting traces. |
||
376 | enum Strategy { |
||
377 | /// Select the trace through a block that has the fewest instructions. |
||
378 | TS_MinInstrCount, |
||
379 | |||
380 | TS_NumStrategies |
||
381 | }; |
||
382 | |||
383 | /// Get the trace ensemble representing the given trace selection strategy. |
||
384 | /// The returned Ensemble object is owned by the MachineTraceMetrics analysis, |
||
385 | /// and valid for the lifetime of the analysis pass. |
||
386 | Ensemble *getEnsemble(Strategy); |
||
387 | |||
388 | /// Invalidate cached information about MBB. This must be called *before* MBB |
||
389 | /// is erased, or the CFG is otherwise changed. |
||
390 | /// |
||
391 | /// This invalidates per-block information about resource usage for MBB only, |
||
392 | /// and it invalidates per-trace information for any trace that passes |
||
393 | /// through MBB. |
||
394 | /// |
||
395 | /// Call Ensemble::getTrace() again to update any trace handles. |
||
396 | void invalidate(const MachineBasicBlock *MBB); |
||
397 | |||
398 | private: |
||
399 | // One entry per basic block, indexed by block number. |
||
400 | SmallVector<FixedBlockInfo, 4> BlockInfo; |
||
401 | |||
402 | // Cycles consumed on each processor resource per block. |
||
403 | // The number of processor resource kinds is constant for a given subtarget, |
||
404 | // but it is not known at compile time. The number of cycles consumed by |
||
405 | // block B on processor resource R is at ProcResourceCycles[B*Kinds + R] |
||
406 | // where Kinds = SchedModel.getNumProcResourceKinds(). |
||
407 | SmallVector<unsigned, 0> ProcResourceCycles; |
||
408 | |||
409 | // One ensemble per strategy. |
||
410 | Ensemble* Ensembles[TS_NumStrategies]; |
||
411 | |||
412 | // Convert scaled resource usage to a cycle count that can be compared with |
||
413 | // latencies. |
||
414 | unsigned getCycles(unsigned Scaled) { |
||
415 | unsigned Factor = SchedModel.getLatencyFactor(); |
||
416 | return (Scaled + Factor - 1) / Factor; |
||
417 | } |
||
418 | }; |
||
419 | |||
420 | inline raw_ostream &operator<<(raw_ostream &OS, |
||
421 | const MachineTraceMetrics::Trace &Tr) { |
||
422 | Tr.print(OS); |
||
423 | return OS; |
||
424 | } |
||
425 | |||
426 | inline raw_ostream &operator<<(raw_ostream &OS, |
||
427 | const MachineTraceMetrics::Ensemble &En) { |
||
428 | En.print(OS); |
||
429 | return OS; |
||
430 | } |
||
431 | |||
432 | } // end namespace llvm |
||
433 | |||
434 | #endif // LLVM_CODEGEN_MACHINETRACEMETRICS_H |