1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
| //===- llvm/unittests/Transforms/Vectorize/VPlanDominatorTreeTest.cpp -----===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "../lib/Transforms/Vectorize/VPlanHCFGBuilder.h"
#include "VPlanTestBase.h"
#include "gtest/gtest.h"
namespace llvm {
namespace {
class VPlanDominatorTreeTest : public VPlanTestBase {};
TEST_F(VPlanDominatorTreeTest, BasicVPBBDomination) {
const char *ModuleString =
"define void @f(i32* %a, i32* %b, i32* %c, i32 %N, i32 %M, i32 %K) {\n"
"entry:\n"
" br label %for.body\n"
"for.body:\n"
" %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.inc ]\n"
" br i1 true, label %if.then, label %if.else\n"
"if.then:\n"
" br label %for.inc\n"
"if.else:\n"
" br label %for.inc\n"
"for.inc:\n"
" %iv.next = add nuw nsw i64 %iv, 1\n"
" %exitcond = icmp eq i64 %iv.next, 300\n"
" br i1 %exitcond, label %for.end, label %for.body\n"
"for.end:\n"
" ret void\n"
"}\n";
Module &M = parseModule(ModuleString);
Function *F = M.getFunction("f");
BasicBlock *LoopHeader = F->getEntryBlock().getSingleSuccessor();
auto Plan = buildPlainCFG(LoopHeader);
// Build VPlan domination tree analysis.
VPRegionBlock *TopRegion = cast<VPRegionBlock>(Plan->getEntry());
VPDominatorTree VPDT;
VPDT.recalculate(*TopRegion);
VPBlockBase *PH = TopRegion->getEntry();
VPBlockBase *H = PH->getSingleSuccessor();
VPBlockBase *IfThen = H->getSuccessors()[0];
VPBlockBase *IfElse = H->getSuccessors()[1];
VPBlockBase *Latch = IfThen->getSingleSuccessor();
VPBlockBase *Exit = Latch->getSuccessors()[0] != H
? Latch->getSuccessors()[0]
: Latch->getSuccessors()[1];
// Reachability.
EXPECT_TRUE(VPDT.isReachableFromEntry(PH));
EXPECT_TRUE(VPDT.isReachableFromEntry(H));
EXPECT_TRUE(VPDT.isReachableFromEntry(IfThen));
EXPECT_TRUE(VPDT.isReachableFromEntry(IfElse));
EXPECT_TRUE(VPDT.isReachableFromEntry(Latch));
EXPECT_TRUE(VPDT.isReachableFromEntry(Exit));
// VPBB dominance.
EXPECT_TRUE(VPDT.dominates(PH, PH));
EXPECT_TRUE(VPDT.dominates(PH, H));
EXPECT_TRUE(VPDT.dominates(PH, IfThen));
EXPECT_TRUE(VPDT.dominates(PH, IfElse));
EXPECT_TRUE(VPDT.dominates(PH, Latch));
EXPECT_TRUE(VPDT.dominates(PH, Exit));
EXPECT_FALSE(VPDT.dominates(H, PH));
EXPECT_TRUE(VPDT.dominates(H, H));
EXPECT_TRUE(VPDT.dominates(H, IfThen));
EXPECT_TRUE(VPDT.dominates(H, IfElse));
EXPECT_TRUE(VPDT.dominates(H, Latch));
EXPECT_TRUE(VPDT.dominates(H, Exit));
EXPECT_FALSE(VPDT.dominates(IfThen, PH));
EXPECT_FALSE(VPDT.dominates(IfThen, H));
EXPECT_TRUE(VPDT.dominates(IfThen, IfThen));
EXPECT_FALSE(VPDT.dominates(IfThen, IfElse));
EXPECT_FALSE(VPDT.dominates(IfThen, Latch));
EXPECT_FALSE(VPDT.dominates(IfThen, Exit));
EXPECT_FALSE(VPDT.dominates(IfElse, PH));
EXPECT_FALSE(VPDT.dominates(IfElse, H));
EXPECT_FALSE(VPDT.dominates(IfElse, IfThen));
EXPECT_TRUE(VPDT.dominates(IfElse, IfElse));
EXPECT_FALSE(VPDT.dominates(IfElse, Latch));
EXPECT_FALSE(VPDT.dominates(IfElse, Exit));
EXPECT_FALSE(VPDT.dominates(Latch, PH));
EXPECT_FALSE(VPDT.dominates(Latch, H));
EXPECT_FALSE(VPDT.dominates(Latch, IfThen));
EXPECT_FALSE(VPDT.dominates(Latch, IfElse));
EXPECT_TRUE(VPDT.dominates(Latch, Latch));
EXPECT_TRUE(VPDT.dominates(Latch, Exit));
EXPECT_FALSE(VPDT.dominates(Exit, PH));
EXPECT_FALSE(VPDT.dominates(Exit, H));
EXPECT_FALSE(VPDT.dominates(Exit, IfThen));
EXPECT_FALSE(VPDT.dominates(Exit, IfElse));
EXPECT_FALSE(VPDT.dominates(Exit, Latch));
EXPECT_TRUE(VPDT.dominates(Exit, Exit));
// VPBB proper dominance.
EXPECT_FALSE(VPDT.properlyDominates(PH, PH));
EXPECT_TRUE(VPDT.properlyDominates(PH, H));
EXPECT_TRUE(VPDT.properlyDominates(PH, IfThen));
EXPECT_TRUE(VPDT.properlyDominates(PH, IfElse));
EXPECT_TRUE(VPDT.properlyDominates(PH, Latch));
EXPECT_TRUE(VPDT.properlyDominates(PH, Exit));
EXPECT_FALSE(VPDT.properlyDominates(H, PH));
EXPECT_FALSE(VPDT.properlyDominates(H, H));
EXPECT_TRUE(VPDT.properlyDominates(H, IfThen));
EXPECT_TRUE(VPDT.properlyDominates(H, IfElse));
EXPECT_TRUE(VPDT.properlyDominates(H, Latch));
EXPECT_TRUE(VPDT.properlyDominates(H, Exit));
EXPECT_FALSE(VPDT.properlyDominates(IfThen, PH));
EXPECT_FALSE(VPDT.properlyDominates(IfThen, H));
EXPECT_FALSE(VPDT.properlyDominates(IfThen, IfThen));
EXPECT_FALSE(VPDT.properlyDominates(IfThen, IfElse));
EXPECT_FALSE(VPDT.properlyDominates(IfThen, Latch));
EXPECT_FALSE(VPDT.properlyDominates(IfThen, Exit));
EXPECT_FALSE(VPDT.properlyDominates(IfElse, PH));
EXPECT_FALSE(VPDT.properlyDominates(IfElse, H));
EXPECT_FALSE(VPDT.properlyDominates(IfElse, IfThen));
EXPECT_FALSE(VPDT.properlyDominates(IfElse, IfElse));
EXPECT_FALSE(VPDT.properlyDominates(IfElse, Latch));
EXPECT_FALSE(VPDT.properlyDominates(IfElse, Exit));
EXPECT_FALSE(VPDT.properlyDominates(Latch, PH));
EXPECT_FALSE(VPDT.properlyDominates(Latch, H));
EXPECT_FALSE(VPDT.properlyDominates(Latch, IfThen));
EXPECT_FALSE(VPDT.properlyDominates(Latch, IfElse));
EXPECT_FALSE(VPDT.properlyDominates(Latch, Latch));
EXPECT_TRUE(VPDT.properlyDominates(Latch, Exit));
EXPECT_FALSE(VPDT.properlyDominates(Exit, PH));
EXPECT_FALSE(VPDT.properlyDominates(Exit, H));
EXPECT_FALSE(VPDT.properlyDominates(Exit, IfThen));
EXPECT_FALSE(VPDT.properlyDominates(Exit, IfElse));
EXPECT_FALSE(VPDT.properlyDominates(Exit, Latch));
EXPECT_FALSE(VPDT.properlyDominates(Exit, Exit));
// VPBB nearest common dominator.
EXPECT_EQ(PH, VPDT.findNearestCommonDominator(PH, PH));
EXPECT_EQ(PH, VPDT.findNearestCommonDominator(PH, H));
EXPECT_EQ(PH, VPDT.findNearestCommonDominator(PH, IfThen));
EXPECT_EQ(PH, VPDT.findNearestCommonDominator(PH, IfElse));
EXPECT_EQ(PH, VPDT.findNearestCommonDominator(PH, Latch));
EXPECT_EQ(PH, VPDT.findNearestCommonDominator(PH, Exit));
EXPECT_EQ(PH, VPDT.findNearestCommonDominator(H, PH));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(H, H));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(H, IfThen));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(H, IfElse));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(H, Latch));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(H, Exit));
EXPECT_EQ(PH, VPDT.findNearestCommonDominator(IfThen, PH));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(IfThen, H));
EXPECT_EQ(IfThen, VPDT.findNearestCommonDominator(IfThen, IfThen));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(IfThen, IfElse));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(IfThen, Latch));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(IfThen, Exit));
EXPECT_EQ(PH, VPDT.findNearestCommonDominator(IfElse, PH));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(IfElse, H));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(IfElse, IfThen));
EXPECT_EQ(IfElse, VPDT.findNearestCommonDominator(IfElse, IfElse));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(IfElse, Latch));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(IfElse, Exit));
EXPECT_EQ(PH, VPDT.findNearestCommonDominator(Latch, PH));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(Latch, H));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(Latch, IfThen));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(Latch, IfElse));
EXPECT_EQ(Latch, VPDT.findNearestCommonDominator(Latch, Latch));
EXPECT_EQ(Latch, VPDT.findNearestCommonDominator(Latch, Exit));
EXPECT_EQ(PH, VPDT.findNearestCommonDominator(Exit, PH));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(Exit, H));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(Exit, IfThen));
EXPECT_EQ(H, VPDT.findNearestCommonDominator(Exit, IfElse));
EXPECT_EQ(Latch, VPDT.findNearestCommonDominator(Exit, Latch));
EXPECT_EQ(Exit, VPDT.findNearestCommonDominator(Exit, Exit));
}
} // namespace
} // namespace llvm
|