Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //==------ llvm/CodeGen/LoopTraversal.h - Loop Traversal -*- 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 | /// \file Loop Traversal logic. |
||
10 | /// |
||
11 | /// This class provides the basic blocks traversal order used by passes like |
||
12 | /// ReachingDefAnalysis and ExecutionDomainFix. |
||
13 | /// It identifies basic blocks that are part of loops and should to be visited |
||
14 | /// twice and returns efficient traversal order for all the blocks. |
||
15 | // |
||
16 | //===----------------------------------------------------------------------===// |
||
17 | |||
18 | #ifndef LLVM_CODEGEN_LOOPTRAVERSAL_H |
||
19 | #define LLVM_CODEGEN_LOOPTRAVERSAL_H |
||
20 | |||
21 | #include "llvm/ADT/SmallVector.h" |
||
22 | |||
23 | namespace llvm { |
||
24 | |||
25 | class MachineBasicBlock; |
||
26 | class MachineFunction; |
||
27 | |||
28 | /// This class provides the basic blocks traversal order used by passes like |
||
29 | /// ReachingDefAnalysis and ExecutionDomainFix. |
||
30 | /// It identifies basic blocks that are part of loops and should to be visited |
||
31 | /// twice and returns efficient traversal order for all the blocks. |
||
32 | /// |
||
33 | /// We want to visit every instruction in every basic block in order to update |
||
34 | /// it's execution domain or collect clearance information. However, for the |
||
35 | /// clearance calculation, we need to know clearances from all predecessors |
||
36 | /// (including any backedges), therfore we need to visit some blocks twice. |
||
37 | /// As an example, consider the following loop. |
||
38 | /// |
||
39 | /// |
||
40 | /// PH -> A -> B (xmm<Undef> -> xmm<Def>) -> C -> D -> EXIT |
||
41 | /// ^ | |
||
42 | /// +----------------------------------+ |
||
43 | /// |
||
44 | /// The iteration order this pass will return is as follows: |
||
45 | /// Optimized: PH A B C A' B' C' D |
||
46 | /// |
||
47 | /// The basic block order is constructed as follows: |
||
48 | /// Once we finish processing some block, we update the counters in MBBInfos |
||
49 | /// and re-process any successors that are now 'done'. |
||
50 | /// We call a block that is ready for its final round of processing `done` |
||
51 | /// (isBlockDone), e.g. when all predecessor information is known. |
||
52 | /// |
||
53 | /// Note that a naive traversal order would be to do two complete passes over |
||
54 | /// all basic blocks/instructions, the first for recording clearances, the |
||
55 | /// second for updating clearance based on backedges. |
||
56 | /// However, for functions without backedges, or functions with a lot of |
||
57 | /// straight-line code, and a small loop, that would be a lot of unnecessary |
||
58 | /// work (since only the BBs that are part of the loop require two passes). |
||
59 | /// |
||
60 | /// E.g., the naive iteration order for the above exmple is as follows: |
||
61 | /// Naive: PH A B C D A' B' C' D' |
||
62 | /// |
||
63 | /// In the optimized approach we avoid processing D twice, because we |
||
64 | /// can entirely process the predecessors before getting to D. |
||
65 | class LoopTraversal { |
||
66 | private: |
||
67 | struct MBBInfo { |
||
68 | /// Whether we have gotten to this block in primary processing yet. |
||
69 | bool PrimaryCompleted = false; |
||
70 | |||
71 | /// The number of predecessors for which primary processing has completed |
||
72 | unsigned IncomingProcessed = 0; |
||
73 | |||
74 | /// The value of `IncomingProcessed` at the start of primary processing |
||
75 | unsigned PrimaryIncoming = 0; |
||
76 | |||
77 | /// The number of predecessors for which all processing steps are done. |
||
78 | unsigned IncomingCompleted = 0; |
||
79 | |||
80 | MBBInfo() = default; |
||
81 | }; |
||
82 | using MBBInfoMap = SmallVector<MBBInfo, 4>; |
||
83 | /// Helps keep track if we proccessed this block and all its predecessors. |
||
84 | MBBInfoMap MBBInfos; |
||
85 | |||
86 | public: |
||
87 | struct TraversedMBBInfo { |
||
88 | /// The basic block. |
||
89 | MachineBasicBlock *MBB = nullptr; |
||
90 | |||
91 | /// True if this is the first time we process the basic block. |
||
92 | bool PrimaryPass = true; |
||
93 | |||
94 | /// True if the block that is ready for its final round of processing. |
||
95 | bool IsDone = true; |
||
96 | |||
97 | TraversedMBBInfo(MachineBasicBlock *BB = nullptr, bool Primary = true, |
||
98 | bool Done = true) |
||
99 | : MBB(BB), PrimaryPass(Primary), IsDone(Done) {} |
||
100 | }; |
||
101 | LoopTraversal() = default; |
||
102 | |||
103 | /// Identifies basic blocks that are part of loops and should to be |
||
104 | /// visited twice and returns efficient traversal order for all the blocks. |
||
105 | typedef SmallVector<TraversedMBBInfo, 4> TraversalOrder; |
||
106 | TraversalOrder traverse(MachineFunction &MF); |
||
107 | |||
108 | private: |
||
109 | /// Returens true if the block is ready for its final round of processing. |
||
110 | bool isBlockDone(MachineBasicBlock *MBB); |
||
111 | }; |
||
112 | |||
113 | } // namespace llvm |
||
114 | |||
115 | #endif // LLVM_CODEGEN_LOOPTRAVERSAL_H |