blob: dc04dc2c49aced5fb3304f3f2589273cac264e63 [file] [log] [blame]
Aart Bikd14c5952015-09-08 15:25:15 -07001/*
2 * Copyright (C) 2015 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
Aart Bikd14c5952015-09-08 15:25:15 -070017#include "base/arena_allocator.h"
18#include "builder.h"
Aart Bikd14c5952015-09-08 15:25:15 -070019#include "induction_var_analysis.h"
20#include "induction_var_range.h"
21#include "nodes.h"
22#include "optimizing_unit_test.h"
23
24namespace art {
25
26using Value = InductionVarRange::Value;
27
28/**
29 * Fixture class for the InductionVarRange tests.
30 */
David Brazdil4833f5a2015-12-16 10:37:39 +000031class InductionVarRangeTest : public CommonCompilerTest {
Aart Bikd14c5952015-09-08 15:25:15 -070032 public:
Aart Bik7d57d7f2015-12-09 14:39:48 -080033 InductionVarRangeTest()
34 : pool_(),
35 allocator_(&pool_),
36 graph_(CreateGraph(&allocator_)),
37 iva_(new (&allocator_) HInductionVarAnalysis(graph_)),
38 range_(iva_) {
Aart Bikd14c5952015-09-08 15:25:15 -070039 BuildGraph();
40 }
41
42 ~InductionVarRangeTest() { }
43
44 void ExpectEqual(Value v1, Value v2) {
45 EXPECT_EQ(v1.instruction, v2.instruction);
46 EXPECT_EQ(v1.a_constant, v2.a_constant);
47 EXPECT_EQ(v1.b_constant, v2.b_constant);
Aart Bikb3365e02015-09-21 14:45:05 -070048 EXPECT_EQ(v1.is_known, v2.is_known);
Aart Bikd14c5952015-09-08 15:25:15 -070049 }
50
Aart Bik389b3db2015-10-28 14:23:40 -070051 //
52 // Construction methods.
53 //
54
Aart Bikd14c5952015-09-08 15:25:15 -070055 /** Constructs bare minimum graph. */
56 void BuildGraph() {
57 graph_->SetNumberOfVRegs(1);
Aart Bikaec3cce2015-10-14 17:44:55 -070058 entry_block_ = new (&allocator_) HBasicBlock(graph_);
59 exit_block_ = new (&allocator_) HBasicBlock(graph_);
60 graph_->AddBlock(entry_block_);
61 graph_->AddBlock(exit_block_);
62 graph_->SetEntryBlock(entry_block_);
63 graph_->SetExitBlock(exit_block_);
Aart Bik7d57d7f2015-12-09 14:39:48 -080064 // Two parameters.
65 x_ = new (&allocator_) HParameterValue(graph_->GetDexFile(), 0, 0, Primitive::kPrimInt);
66 entry_block_->AddInstruction(x_);
67 y_ = new (&allocator_) HParameterValue(graph_->GetDexFile(), 0, 0, Primitive::kPrimInt);
68 entry_block_->AddInstruction(y_);
Aart Bikaec3cce2015-10-14 17:44:55 -070069 }
70
71 /** Constructs loop with given upper bound. */
Aart Bik389b3db2015-10-28 14:23:40 -070072 void BuildLoop(int32_t lower, HInstruction* upper, int32_t stride) {
Aart Bikaec3cce2015-10-14 17:44:55 -070073 // Control flow.
74 loop_preheader_ = new (&allocator_) HBasicBlock(graph_);
75 graph_->AddBlock(loop_preheader_);
76 HBasicBlock* loop_header = new (&allocator_) HBasicBlock(graph_);
77 graph_->AddBlock(loop_header);
78 HBasicBlock* loop_body = new (&allocator_) HBasicBlock(graph_);
79 graph_->AddBlock(loop_body);
David Brazdildb51efb2015-11-06 01:36:20 +000080 HBasicBlock* return_block = new (&allocator_) HBasicBlock(graph_);
81 graph_->AddBlock(return_block);
Aart Bikaec3cce2015-10-14 17:44:55 -070082 entry_block_->AddSuccessor(loop_preheader_);
83 loop_preheader_->AddSuccessor(loop_header);
84 loop_header->AddSuccessor(loop_body);
David Brazdildb51efb2015-11-06 01:36:20 +000085 loop_header->AddSuccessor(return_block);
Aart Bikaec3cce2015-10-14 17:44:55 -070086 loop_body->AddSuccessor(loop_header);
David Brazdildb51efb2015-11-06 01:36:20 +000087 return_block->AddSuccessor(exit_block_);
Aart Bikaec3cce2015-10-14 17:44:55 -070088 // Instructions.
Aart Bikaec3cce2015-10-14 17:44:55 -070089 loop_preheader_->AddInstruction(new (&allocator_) HGoto());
David Brazdilbadd8262016-02-02 16:28:56 +000090 HPhi* phi = new (&allocator_) HPhi(&allocator_, 0, 0, Primitive::kPrimInt);
91 loop_header->AddPhi(phi);
92 phi->AddInput(graph_->GetIntConstant(lower)); // i = l
Aart Bik389b3db2015-10-28 14:23:40 -070093 if (stride > 0) {
David Brazdilbadd8262016-02-02 16:28:56 +000094 condition_ = new (&allocator_) HLessThan(phi, upper); // i < u
Aart Bik389b3db2015-10-28 14:23:40 -070095 } else {
David Brazdilbadd8262016-02-02 16:28:56 +000096 condition_ = new (&allocator_) HGreaterThan(phi, upper); // i > u
Aart Bik389b3db2015-10-28 14:23:40 -070097 }
Aart Bikaec3cce2015-10-14 17:44:55 -070098 loop_header->AddInstruction(condition_);
Aart Bik389b3db2015-10-28 14:23:40 -070099 loop_header->AddInstruction(new (&allocator_) HIf(condition_));
David Brazdilbadd8262016-02-02 16:28:56 +0000100 increment_ = new (&allocator_) HAdd(Primitive::kPrimInt, phi, graph_->GetIntConstant(stride));
101 loop_body->AddInstruction(increment_); // i += s
102 phi->AddInput(increment_);
Aart Bikaec3cce2015-10-14 17:44:55 -0700103 loop_body->AddInstruction(new (&allocator_) HGoto());
David Brazdildb51efb2015-11-06 01:36:20 +0000104 return_block->AddInstruction(new (&allocator_) HReturnVoid());
105 exit_block_->AddInstruction(new (&allocator_) HExit());
Aart Bikaec3cce2015-10-14 17:44:55 -0700106 }
107
Aart Bik7d57d7f2015-12-09 14:39:48 -0800108 /** Constructs SSA and performs induction variable analysis. */
Aart Bikaec3cce2015-10-14 17:44:55 -0700109 void PerformInductionVarAnalysis() {
David Brazdilbadd8262016-02-02 16:28:56 +0000110 graph_->BuildDominatorTree();
Aart Bikaec3cce2015-10-14 17:44:55 -0700111 iva_->Run();
Aart Bikd14c5952015-09-08 15:25:15 -0700112 }
113
114 /** Constructs an invariant. */
115 HInductionVarAnalysis::InductionInfo* CreateInvariant(char opc,
116 HInductionVarAnalysis::InductionInfo* a,
117 HInductionVarAnalysis::InductionInfo* b) {
118 HInductionVarAnalysis::InductionOp op;
119 switch (opc) {
120 case '+': op = HInductionVarAnalysis::kAdd; break;
121 case '-': op = HInductionVarAnalysis::kSub; break;
122 case 'n': op = HInductionVarAnalysis::kNeg; break;
123 case '*': op = HInductionVarAnalysis::kMul; break;
124 case '/': op = HInductionVarAnalysis::kDiv; break;
125 default: op = HInductionVarAnalysis::kNop; break;
126 }
127 return iva_->CreateInvariantOp(op, a, b);
128 }
129
130 /** Constructs a fetch. */
131 HInductionVarAnalysis::InductionInfo* CreateFetch(HInstruction* fetch) {
132 return iva_->CreateInvariantFetch(fetch);
133 }
134
135 /** Constructs a constant. */
136 HInductionVarAnalysis::InductionInfo* CreateConst(int32_t c) {
137 return CreateFetch(graph_->GetIntConstant(c));
138 }
139
140 /** Constructs a trip-count. */
Aart Bik389b3db2015-10-28 14:23:40 -0700141 HInductionVarAnalysis::InductionInfo* CreateTripCount(int32_t tc, bool in_loop, bool safe) {
Aart Bik0d345cf2016-03-16 10:49:38 -0700142 Primitive::Type type = Primitive::kPrimInt;
Aart Bik389b3db2015-10-28 14:23:40 -0700143 if (in_loop && safe) {
144 return iva_->CreateTripCount(
Aart Bik0d345cf2016-03-16 10:49:38 -0700145 HInductionVarAnalysis::kTripCountInLoop, CreateConst(tc), nullptr, type);
Aart Bik389b3db2015-10-28 14:23:40 -0700146 } else if (in_loop) {
147 return iva_->CreateTripCount(
Aart Bik0d345cf2016-03-16 10:49:38 -0700148 HInductionVarAnalysis::kTripCountInLoopUnsafe, CreateConst(tc), nullptr, type);
Aart Bik389b3db2015-10-28 14:23:40 -0700149 } else if (safe) {
150 return iva_->CreateTripCount(
Aart Bik0d345cf2016-03-16 10:49:38 -0700151 HInductionVarAnalysis::kTripCountInBody, CreateConst(tc), nullptr, type);
Aart Bik389b3db2015-10-28 14:23:40 -0700152 } else {
153 return iva_->CreateTripCount(
Aart Bik0d345cf2016-03-16 10:49:38 -0700154 HInductionVarAnalysis::kTripCountInBodyUnsafe, CreateConst(tc), nullptr, type);
Aart Bik389b3db2015-10-28 14:23:40 -0700155 }
Aart Bikd14c5952015-09-08 15:25:15 -0700156 }
157
158 /** Constructs a linear a * i + b induction. */
159 HInductionVarAnalysis::InductionInfo* CreateLinear(int32_t a, int32_t b) {
Aart Bik0d345cf2016-03-16 10:49:38 -0700160 return iva_->CreateInduction(
161 HInductionVarAnalysis::kLinear, CreateConst(a), CreateConst(b), Primitive::kPrimInt);
Aart Bikd14c5952015-09-08 15:25:15 -0700162 }
163
164 /** Constructs a range [lo, hi] using a periodic induction. */
165 HInductionVarAnalysis::InductionInfo* CreateRange(int32_t lo, int32_t hi) {
166 return iva_->CreateInduction(
Aart Bik0d345cf2016-03-16 10:49:38 -0700167 HInductionVarAnalysis::kPeriodic, CreateConst(lo), CreateConst(hi), Primitive::kPrimInt);
Aart Bikd14c5952015-09-08 15:25:15 -0700168 }
169
Aart Bik389b3db2015-10-28 14:23:40 -0700170 /** Constructs a wrap-around induction consisting of a constant, followed info */
171 HInductionVarAnalysis::InductionInfo* CreateWrapAround(
172 int32_t initial,
173 HInductionVarAnalysis::InductionInfo* info) {
Aart Bik0d345cf2016-03-16 10:49:38 -0700174 return iva_->CreateInduction(
175 HInductionVarAnalysis::kWrapAround, CreateConst(initial), info, Primitive::kPrimInt);
Aart Bik389b3db2015-10-28 14:23:40 -0700176 }
177
Aart Bikd14c5952015-09-08 15:25:15 -0700178 /** Constructs a wrap-around induction consisting of a constant, followed by a range. */
179 HInductionVarAnalysis::InductionInfo* CreateWrapAround(int32_t initial, int32_t lo, int32_t hi) {
Aart Bik389b3db2015-10-28 14:23:40 -0700180 return CreateWrapAround(initial, CreateRange(lo, hi));
Aart Bikd14c5952015-09-08 15:25:15 -0700181 }
182
183 //
184 // Relay methods.
185 //
186
Aart Bik389b3db2015-10-28 14:23:40 -0700187 bool NeedsTripCount(HInductionVarAnalysis::InductionInfo* info) {
Aart Bik7d57d7f2015-12-09 14:39:48 -0800188 return range_.NeedsTripCount(info);
Aart Bik389b3db2015-10-28 14:23:40 -0700189 }
190
191 bool IsBodyTripCount(HInductionVarAnalysis::InductionInfo* trip) {
Aart Bik7d57d7f2015-12-09 14:39:48 -0800192 return range_.IsBodyTripCount(trip);
Aart Bik389b3db2015-10-28 14:23:40 -0700193 }
194
195 bool IsUnsafeTripCount(HInductionVarAnalysis::InductionInfo* trip) {
Aart Bik7d57d7f2015-12-09 14:39:48 -0800196 return range_.IsUnsafeTripCount(trip);
Aart Bik389b3db2015-10-28 14:23:40 -0700197 }
198
Aart Bikd14c5952015-09-08 15:25:15 -0700199 Value GetMin(HInductionVarAnalysis::InductionInfo* info,
200 HInductionVarAnalysis::InductionInfo* induc) {
Aart Bik7d57d7f2015-12-09 14:39:48 -0800201 return range_.GetVal(info, induc, /* in_body */ true, /* is_min */ true);
Aart Bikd14c5952015-09-08 15:25:15 -0700202 }
203
204 Value GetMax(HInductionVarAnalysis::InductionInfo* info,
205 HInductionVarAnalysis::InductionInfo* induc) {
Aart Bik7d57d7f2015-12-09 14:39:48 -0800206 return range_.GetVal(info, induc, /* in_body */ true, /* is_min */ false);
Aart Bikd14c5952015-09-08 15:25:15 -0700207 }
208
209 Value GetMul(HInductionVarAnalysis::InductionInfo* info1,
Aart Bikb3365e02015-09-21 14:45:05 -0700210 HInductionVarAnalysis::InductionInfo* info2,
211 bool is_min) {
Aart Bik7d57d7f2015-12-09 14:39:48 -0800212 return range_.GetMul(info1, info2, nullptr, /* in_body */ true, is_min);
Aart Bikd14c5952015-09-08 15:25:15 -0700213 }
214
215 Value GetDiv(HInductionVarAnalysis::InductionInfo* info1,
Aart Bikb3365e02015-09-21 14:45:05 -0700216 HInductionVarAnalysis::InductionInfo* info2,
217 bool is_min) {
Aart Bik7d57d7f2015-12-09 14:39:48 -0800218 return range_.GetDiv(info1, info2, nullptr, /* in_body */ true, is_min);
Aart Bik9401f532015-09-28 16:25:56 -0700219 }
220
Aart Bik97412c922016-02-19 20:14:38 -0800221 bool IsExact(HInductionVarAnalysis::InductionInfo* info, int64_t* value) {
222 return range_.IsConstant(info, InductionVarRange::kExact, value);
223 }
224
225 bool IsAtMost(HInductionVarAnalysis::InductionInfo* info, int64_t* value) {
226 return range_.IsConstant(info, InductionVarRange::kAtMost, value);
227 }
228
229 bool IsAtLeast(HInductionVarAnalysis::InductionInfo* info, int64_t* value) {
230 return range_.IsConstant(info, InductionVarRange::kAtLeast, value);
Aart Bikd14c5952015-09-08 15:25:15 -0700231 }
232
Aart Bik7d57d7f2015-12-09 14:39:48 -0800233 Value AddValue(Value v1, Value v2) { return range_.AddValue(v1, v2); }
234 Value SubValue(Value v1, Value v2) { return range_.SubValue(v1, v2); }
235 Value MulValue(Value v1, Value v2) { return range_.MulValue(v1, v2); }
236 Value DivValue(Value v1, Value v2) { return range_.DivValue(v1, v2); }
237 Value MinValue(Value v1, Value v2) { return range_.MergeVal(v1, v2, true); }
238 Value MaxValue(Value v1, Value v2) { return range_.MergeVal(v1, v2, false); }
Aart Bikd14c5952015-09-08 15:25:15 -0700239
240 // General building fields.
241 ArenaPool pool_;
242 ArenaAllocator allocator_;
243 HGraph* graph_;
Aart Bikaec3cce2015-10-14 17:44:55 -0700244 HBasicBlock* entry_block_;
245 HBasicBlock* exit_block_;
246 HBasicBlock* loop_preheader_;
Aart Bikd14c5952015-09-08 15:25:15 -0700247 HInductionVarAnalysis* iva_;
Aart Bik7d57d7f2015-12-09 14:39:48 -0800248 InductionVarRange range_;
Aart Bikd14c5952015-09-08 15:25:15 -0700249
Aart Bikaec3cce2015-10-14 17:44:55 -0700250 // Instructions.
251 HInstruction* condition_;
252 HInstruction* increment_;
Aart Bik7d57d7f2015-12-09 14:39:48 -0800253 HInstruction* x_;
254 HInstruction* y_;
Aart Bikd14c5952015-09-08 15:25:15 -0700255};
256
257//
Aart Bik7d57d7f2015-12-09 14:39:48 -0800258// Tests on private methods.
Aart Bikd14c5952015-09-08 15:25:15 -0700259//
260
Aart Bik97412c922016-02-19 20:14:38 -0800261TEST_F(InductionVarRangeTest, IsConstant) {
262 int64_t value;
263 // Constant.
264 EXPECT_TRUE(IsExact(CreateConst(12345), &value));
265 EXPECT_EQ(12345, value);
266 EXPECT_TRUE(IsAtMost(CreateConst(12345), &value));
267 EXPECT_EQ(12345, value);
268 EXPECT_TRUE(IsAtLeast(CreateConst(12345), &value));
269 EXPECT_EQ(12345, value);
270 // Constant trivial range.
271 EXPECT_TRUE(IsExact(CreateRange(111, 111), &value));
272 EXPECT_EQ(111, value);
273 EXPECT_TRUE(IsAtMost(CreateRange(111, 111), &value));
274 EXPECT_EQ(111, value);
275 EXPECT_TRUE(IsAtLeast(CreateRange(111, 111), &value));
276 EXPECT_EQ(111, value);
277 // Constant non-trivial range.
278 EXPECT_FALSE(IsExact(CreateRange(11, 22), &value));
279 EXPECT_TRUE(IsAtMost(CreateRange(11, 22), &value));
280 EXPECT_EQ(22, value);
281 EXPECT_TRUE(IsAtLeast(CreateRange(11, 22), &value));
282 EXPECT_EQ(11, value);
283 // Symbolic.
284 EXPECT_FALSE(IsExact(CreateFetch(x_), &value));
285 EXPECT_FALSE(IsAtMost(CreateFetch(x_), &value));
286 EXPECT_FALSE(IsAtLeast(CreateFetch(x_), &value));
287}
288
Aart Bik389b3db2015-10-28 14:23:40 -0700289TEST_F(InductionVarRangeTest, TripCountProperties) {
290 EXPECT_FALSE(NeedsTripCount(nullptr));
291 EXPECT_FALSE(NeedsTripCount(CreateConst(1)));
292 EXPECT_TRUE(NeedsTripCount(CreateLinear(1, 1)));
293 EXPECT_FALSE(NeedsTripCount(CreateWrapAround(1, 2, 3)));
294 EXPECT_TRUE(NeedsTripCount(CreateWrapAround(1, CreateLinear(1, 1))));
295
296 EXPECT_FALSE(IsBodyTripCount(nullptr));
297 EXPECT_FALSE(IsBodyTripCount(CreateTripCount(100, true, true)));
298 EXPECT_FALSE(IsBodyTripCount(CreateTripCount(100, true, false)));
299 EXPECT_TRUE(IsBodyTripCount(CreateTripCount(100, false, true)));
300 EXPECT_TRUE(IsBodyTripCount(CreateTripCount(100, false, false)));
301
302 EXPECT_FALSE(IsUnsafeTripCount(nullptr));
303 EXPECT_FALSE(IsUnsafeTripCount(CreateTripCount(100, true, true)));
304 EXPECT_TRUE(IsUnsafeTripCount(CreateTripCount(100, true, false)));
305 EXPECT_FALSE(IsUnsafeTripCount(CreateTripCount(100, false, true)));
306 EXPECT_TRUE(IsUnsafeTripCount(CreateTripCount(100, false, false)));
307}
308
Aart Bikd14c5952015-09-08 15:25:15 -0700309TEST_F(InductionVarRangeTest, GetMinMaxNull) {
Aart Bikb3365e02015-09-21 14:45:05 -0700310 ExpectEqual(Value(), GetMin(nullptr, nullptr));
311 ExpectEqual(Value(), GetMax(nullptr, nullptr));
Aart Bikd14c5952015-09-08 15:25:15 -0700312}
313
314TEST_F(InductionVarRangeTest, GetMinMaxAdd) {
315 ExpectEqual(Value(12),
316 GetMin(CreateInvariant('+', CreateConst(2), CreateRange(10, 20)), nullptr));
317 ExpectEqual(Value(22),
318 GetMax(CreateInvariant('+', CreateConst(2), CreateRange(10, 20)), nullptr));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800319 ExpectEqual(Value(x_, 1, -20),
320 GetMin(CreateInvariant('+', CreateFetch(x_), CreateRange(-20, -10)), nullptr));
321 ExpectEqual(Value(x_, 1, -10),
322 GetMax(CreateInvariant('+', CreateFetch(x_), CreateRange(-20, -10)), nullptr));
323 ExpectEqual(Value(x_, 1, 10),
324 GetMin(CreateInvariant('+', CreateRange(10, 20), CreateFetch(x_)), nullptr));
325 ExpectEqual(Value(x_, 1, 20),
326 GetMax(CreateInvariant('+', CreateRange(10, 20), CreateFetch(x_)), nullptr));
Aart Bikd14c5952015-09-08 15:25:15 -0700327 ExpectEqual(Value(5),
328 GetMin(CreateInvariant('+', CreateRange(-5, -1), CreateRange(10, 20)), nullptr));
329 ExpectEqual(Value(19),
330 GetMax(CreateInvariant('+', CreateRange(-5, -1), CreateRange(10, 20)), nullptr));
331}
332
333TEST_F(InductionVarRangeTest, GetMinMaxSub) {
334 ExpectEqual(Value(-18),
335 GetMin(CreateInvariant('-', CreateConst(2), CreateRange(10, 20)), nullptr));
336 ExpectEqual(Value(-8),
337 GetMax(CreateInvariant('-', CreateConst(2), CreateRange(10, 20)), nullptr));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800338 ExpectEqual(Value(x_, 1, 10),
339 GetMin(CreateInvariant('-', CreateFetch(x_), CreateRange(-20, -10)), nullptr));
340 ExpectEqual(Value(x_, 1, 20),
341 GetMax(CreateInvariant('-', CreateFetch(x_), CreateRange(-20, -10)), nullptr));
342 ExpectEqual(Value(x_, -1, 10),
343 GetMin(CreateInvariant('-', CreateRange(10, 20), CreateFetch(x_)), nullptr));
344 ExpectEqual(Value(x_, -1, 20),
345 GetMax(CreateInvariant('-', CreateRange(10, 20), CreateFetch(x_)), nullptr));
Aart Bikd14c5952015-09-08 15:25:15 -0700346 ExpectEqual(Value(-25),
347 GetMin(CreateInvariant('-', CreateRange(-5, -1), CreateRange(10, 20)), nullptr));
348 ExpectEqual(Value(-11),
349 GetMax(CreateInvariant('-', CreateRange(-5, -1), CreateRange(10, 20)), nullptr));
350}
351
352TEST_F(InductionVarRangeTest, GetMinMaxNeg) {
353 ExpectEqual(Value(-20), GetMin(CreateInvariant('n', nullptr, CreateRange(10, 20)), nullptr));
354 ExpectEqual(Value(-10), GetMax(CreateInvariant('n', nullptr, CreateRange(10, 20)), nullptr));
355 ExpectEqual(Value(10), GetMin(CreateInvariant('n', nullptr, CreateRange(-20, -10)), nullptr));
356 ExpectEqual(Value(20), GetMax(CreateInvariant('n', nullptr, CreateRange(-20, -10)), nullptr));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800357 ExpectEqual(Value(x_, -1, 0), GetMin(CreateInvariant('n', nullptr, CreateFetch(x_)), nullptr));
358 ExpectEqual(Value(x_, -1, 0), GetMax(CreateInvariant('n', nullptr, CreateFetch(x_)), nullptr));
Aart Bikd14c5952015-09-08 15:25:15 -0700359}
360
361TEST_F(InductionVarRangeTest, GetMinMaxMul) {
362 ExpectEqual(Value(20),
363 GetMin(CreateInvariant('*', CreateConst(2), CreateRange(10, 20)), nullptr));
364 ExpectEqual(Value(40),
365 GetMax(CreateInvariant('*', CreateConst(2), CreateRange(10, 20)), nullptr));
366}
367
368TEST_F(InductionVarRangeTest, GetMinMaxDiv) {
369 ExpectEqual(Value(3),
370 GetMin(CreateInvariant('/', CreateRange(12, 20), CreateConst(4)), nullptr));
371 ExpectEqual(Value(5),
372 GetMax(CreateInvariant('/', CreateRange(12, 20), CreateConst(4)), nullptr));
373}
374
375TEST_F(InductionVarRangeTest, GetMinMaxConstant) {
376 ExpectEqual(Value(12345), GetMin(CreateConst(12345), nullptr));
377 ExpectEqual(Value(12345), GetMax(CreateConst(12345), nullptr));
378}
379
380TEST_F(InductionVarRangeTest, GetMinMaxFetch) {
Aart Bik7d57d7f2015-12-09 14:39:48 -0800381 ExpectEqual(Value(x_, 1, 0), GetMin(CreateFetch(x_), nullptr));
382 ExpectEqual(Value(x_, 1, 0), GetMax(CreateFetch(x_), nullptr));
Aart Bikd14c5952015-09-08 15:25:15 -0700383}
384
385TEST_F(InductionVarRangeTest, GetMinMaxLinear) {
Aart Bik389b3db2015-10-28 14:23:40 -0700386 ExpectEqual(Value(20), GetMin(CreateLinear(10, 20), CreateTripCount(100, true, true)));
387 ExpectEqual(Value(1010), GetMax(CreateLinear(10, 20), CreateTripCount(100, true, true)));
388 ExpectEqual(Value(-970), GetMin(CreateLinear(-10, 20), CreateTripCount(100, true, true)));
389 ExpectEqual(Value(20), GetMax(CreateLinear(-10, 20), CreateTripCount(100, true, true)));
Aart Bikd14c5952015-09-08 15:25:15 -0700390}
391
392TEST_F(InductionVarRangeTest, GetMinMaxWrapAround) {
393 ExpectEqual(Value(-5), GetMin(CreateWrapAround(-5, -1, 10), nullptr));
394 ExpectEqual(Value(10), GetMax(CreateWrapAround(-5, -1, 10), nullptr));
395 ExpectEqual(Value(-1), GetMin(CreateWrapAround(2, -1, 10), nullptr));
396 ExpectEqual(Value(10), GetMax(CreateWrapAround(2, -1, 10), nullptr));
397 ExpectEqual(Value(-1), GetMin(CreateWrapAround(20, -1, 10), nullptr));
398 ExpectEqual(Value(20), GetMax(CreateWrapAround(20, -1, 10), nullptr));
399}
400
401TEST_F(InductionVarRangeTest, GetMinMaxPeriodic) {
402 ExpectEqual(Value(-2), GetMin(CreateRange(-2, 99), nullptr));
403 ExpectEqual(Value(99), GetMax(CreateRange(-2, 99), nullptr));
404}
405
406TEST_F(InductionVarRangeTest, GetMulMin) {
Aart Bik97412c922016-02-19 20:14:38 -0800407 ExpectEqual(Value(-14), GetMul(CreateConst(2), CreateRange(-7, 8), true));
408 ExpectEqual(Value(-16), GetMul(CreateConst(-2), CreateRange(-7, 8), true));
409 ExpectEqual(Value(-14), GetMul(CreateRange(-7, 8), CreateConst(2), true));
410 ExpectEqual(Value(-16), GetMul(CreateRange(-7, 8), CreateConst(-2), true));
Aart Bikb3365e02015-09-21 14:45:05 -0700411 ExpectEqual(Value(6), GetMul(CreateRange(2, 10), CreateRange(3, 5), true));
412 ExpectEqual(Value(-50), GetMul(CreateRange(2, 10), CreateRange(-5, -3), true));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800413 ExpectEqual(Value(), GetMul(CreateRange(2, 10), CreateRange(-1, 1), true));
Aart Bikb3365e02015-09-21 14:45:05 -0700414 ExpectEqual(Value(-50), GetMul(CreateRange(-10, -2), CreateRange(3, 5), true));
415 ExpectEqual(Value(6), GetMul(CreateRange(-10, -2), CreateRange(-5, -3), true));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800416 ExpectEqual(Value(), GetMul(CreateRange(-10, -2), CreateRange(-1, 1), true));
417 ExpectEqual(Value(), GetMul(CreateRange(-1, 1), CreateRange(2, 10), true));
418 ExpectEqual(Value(), GetMul(CreateRange(-1, 1), CreateRange(-10, -2), true));
419 ExpectEqual(Value(), GetMul(CreateRange(-1, 1), CreateRange(-1, 1), true));
Aart Bikd14c5952015-09-08 15:25:15 -0700420}
421
422TEST_F(InductionVarRangeTest, GetMulMax) {
Aart Bik97412c922016-02-19 20:14:38 -0800423 ExpectEqual(Value(16), GetMul(CreateConst(2), CreateRange(-7, 8), false));
424 ExpectEqual(Value(14), GetMul(CreateConst(-2), CreateRange(-7, 8), false));
425 ExpectEqual(Value(16), GetMul(CreateRange(-7, 8), CreateConst(2), false));
426 ExpectEqual(Value(14), GetMul(CreateRange(-7, 8), CreateConst(-2), false));
Aart Bikb3365e02015-09-21 14:45:05 -0700427 ExpectEqual(Value(50), GetMul(CreateRange(2, 10), CreateRange(3, 5), false));
428 ExpectEqual(Value(-6), GetMul(CreateRange(2, 10), CreateRange(-5, -3), false));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800429 ExpectEqual(Value(), GetMul(CreateRange(2, 10), CreateRange(-1, 1), false));
Aart Bikb3365e02015-09-21 14:45:05 -0700430 ExpectEqual(Value(-6), GetMul(CreateRange(-10, -2), CreateRange(3, 5), false));
431 ExpectEqual(Value(50), GetMul(CreateRange(-10, -2), CreateRange(-5, -3), false));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800432 ExpectEqual(Value(), GetMul(CreateRange(-10, -2), CreateRange(-1, 1), false));
433 ExpectEqual(Value(), GetMul(CreateRange(-1, 1), CreateRange(2, 10), false));
434 ExpectEqual(Value(), GetMul(CreateRange(-1, 1), CreateRange(-10, -2), false));
435 ExpectEqual(Value(), GetMul(CreateRange(-1, 1), CreateRange(-1, 1), false));
Aart Bikd14c5952015-09-08 15:25:15 -0700436}
437
438TEST_F(InductionVarRangeTest, GetDivMin) {
Aart Bik97412c922016-02-19 20:14:38 -0800439 ExpectEqual(Value(-5), GetDiv(CreateRange(-10, 20), CreateConst(2), true));
440 ExpectEqual(Value(-10), GetDiv(CreateRange(-10, 20), CreateConst(-2), true));
Aart Bikb3365e02015-09-21 14:45:05 -0700441 ExpectEqual(Value(10), GetDiv(CreateRange(40, 1000), CreateRange(2, 4), true));
442 ExpectEqual(Value(-500), GetDiv(CreateRange(40, 1000), CreateRange(-4, -2), true));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800443 ExpectEqual(Value(), GetDiv(CreateRange(40, 1000), CreateRange(-1, 1), true));
Aart Bikb3365e02015-09-21 14:45:05 -0700444 ExpectEqual(Value(-500), GetDiv(CreateRange(-1000, -40), CreateRange(2, 4), true));
445 ExpectEqual(Value(10), GetDiv(CreateRange(-1000, -40), CreateRange(-4, -2), true));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800446 ExpectEqual(Value(), GetDiv(CreateRange(-1000, -40), CreateRange(-1, 1), true));
447 ExpectEqual(Value(), GetDiv(CreateRange(-1, 1), CreateRange(40, 1000), true));
448 ExpectEqual(Value(), GetDiv(CreateRange(-1, 1), CreateRange(-1000, -40), true));
449 ExpectEqual(Value(), GetDiv(CreateRange(-1, 1), CreateRange(-1, 1), true));
Aart Bikd14c5952015-09-08 15:25:15 -0700450}
451
452TEST_F(InductionVarRangeTest, GetDivMax) {
Aart Bik97412c922016-02-19 20:14:38 -0800453 ExpectEqual(Value(10), GetDiv(CreateRange(-10, 20), CreateConst(2), false));
454 ExpectEqual(Value(5), GetDiv(CreateRange(-10, 20), CreateConst(-2), false));
Aart Bikb3365e02015-09-21 14:45:05 -0700455 ExpectEqual(Value(500), GetDiv(CreateRange(40, 1000), CreateRange(2, 4), false));
456 ExpectEqual(Value(-10), GetDiv(CreateRange(40, 1000), CreateRange(-4, -2), false));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800457 ExpectEqual(Value(), GetDiv(CreateRange(40, 1000), CreateRange(-1, 1), false));
Aart Bikb3365e02015-09-21 14:45:05 -0700458 ExpectEqual(Value(-10), GetDiv(CreateRange(-1000, -40), CreateRange(2, 4), false));
459 ExpectEqual(Value(500), GetDiv(CreateRange(-1000, -40), CreateRange(-4, -2), false));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800460 ExpectEqual(Value(), GetDiv(CreateRange(-1000, -40), CreateRange(-1, 1), false));
461 ExpectEqual(Value(), GetDiv(CreateRange(-1, 1), CreateRange(40, 1000), false));
462 ExpectEqual(Value(), GetDiv(CreateRange(-1, 1), CreateRange(-1000, 40), false));
463 ExpectEqual(Value(), GetDiv(CreateRange(-1, 1), CreateRange(-1, 1), false));
Aart Bikd14c5952015-09-08 15:25:15 -0700464}
465
466TEST_F(InductionVarRangeTest, AddValue) {
467 ExpectEqual(Value(110), AddValue(Value(10), Value(100)));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800468 ExpectEqual(Value(-5), AddValue(Value(x_, 1, -4), Value(x_, -1, -1)));
469 ExpectEqual(Value(x_, 3, -5), AddValue(Value(x_, 2, -4), Value(x_, 1, -1)));
470 ExpectEqual(Value(), AddValue(Value(x_, 1, 5), Value(y_, 1, -7)));
471 ExpectEqual(Value(x_, 1, 23), AddValue(Value(x_, 1, 20), Value(3)));
472 ExpectEqual(Value(y_, 1, 5), AddValue(Value(55), Value(y_, 1, -50)));
Aart Bikcd26feb2015-09-23 17:50:50 -0700473 const int32_t max_value = std::numeric_limits<int32_t>::max();
474 ExpectEqual(Value(max_value), AddValue(Value(max_value - 5), Value(5)));
475 ExpectEqual(Value(), AddValue(Value(max_value - 5), Value(6))); // unsafe
Aart Bikd14c5952015-09-08 15:25:15 -0700476}
477
478TEST_F(InductionVarRangeTest, SubValue) {
479 ExpectEqual(Value(-90), SubValue(Value(10), Value(100)));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800480 ExpectEqual(Value(-3), SubValue(Value(x_, 1, -4), Value(x_, 1, -1)));
481 ExpectEqual(Value(x_, 2, -3), SubValue(Value(x_, 3, -4), Value(x_, 1, -1)));
482 ExpectEqual(Value(), SubValue(Value(x_, 1, 5), Value(y_, 1, -7)));
483 ExpectEqual(Value(x_, 1, 17), SubValue(Value(x_, 1, 20), Value(3)));
484 ExpectEqual(Value(y_, -4, 105), SubValue(Value(55), Value(y_, 4, -50)));
Aart Bikcd26feb2015-09-23 17:50:50 -0700485 const int32_t min_value = std::numeric_limits<int32_t>::min();
486 ExpectEqual(Value(min_value), SubValue(Value(min_value + 5), Value(5)));
487 ExpectEqual(Value(), SubValue(Value(min_value + 5), Value(6))); // unsafe
Aart Bikd14c5952015-09-08 15:25:15 -0700488}
489
490TEST_F(InductionVarRangeTest, MulValue) {
491 ExpectEqual(Value(1000), MulValue(Value(10), Value(100)));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800492 ExpectEqual(Value(), MulValue(Value(x_, 1, -4), Value(x_, 1, -1)));
493 ExpectEqual(Value(), MulValue(Value(x_, 1, 5), Value(y_, 1, -7)));
494 ExpectEqual(Value(x_, 9, 60), MulValue(Value(x_, 3, 20), Value(3)));
495 ExpectEqual(Value(y_, 55, -110), MulValue(Value(55), Value(y_, 1, -2)));
Aart Bikb3365e02015-09-21 14:45:05 -0700496 ExpectEqual(Value(), MulValue(Value(90000), Value(-90000))); // unsafe
Aart Bikd14c5952015-09-08 15:25:15 -0700497}
498
Aart Bik97412c922016-02-19 20:14:38 -0800499TEST_F(InductionVarRangeTest, MulValueSpecial) {
500 const int32_t min_value = std::numeric_limits<int32_t>::min();
501 const int32_t max_value = std::numeric_limits<int32_t>::max();
502
503 // Unsafe.
504 ExpectEqual(Value(), MulValue(Value(min_value), Value(min_value)));
505 ExpectEqual(Value(), MulValue(Value(min_value), Value(-1)));
506 ExpectEqual(Value(), MulValue(Value(min_value), Value(max_value)));
507 ExpectEqual(Value(), MulValue(Value(max_value), Value(max_value)));
508
509 // Safe.
510 ExpectEqual(Value(min_value), MulValue(Value(min_value), Value(1)));
511 ExpectEqual(Value(max_value), MulValue(Value(max_value), Value(1)));
512 ExpectEqual(Value(-max_value), MulValue(Value(max_value), Value(-1)));
513 ExpectEqual(Value(-1), MulValue(Value(1), Value(-1)));
514 ExpectEqual(Value(1), MulValue(Value(-1), Value(-1)));
515}
516
Aart Bikd14c5952015-09-08 15:25:15 -0700517TEST_F(InductionVarRangeTest, DivValue) {
518 ExpectEqual(Value(25), DivValue(Value(100), Value(4)));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800519 ExpectEqual(Value(), DivValue(Value(x_, 1, -4), Value(x_, 1, -1)));
520 ExpectEqual(Value(), DivValue(Value(x_, 1, 5), Value(y_, 1, -7)));
521 ExpectEqual(Value(), DivValue(Value(x_, 12, 24), Value(3)));
522 ExpectEqual(Value(), DivValue(Value(55), Value(y_, 1, -50)));
Aart Bikb3365e02015-09-21 14:45:05 -0700523 ExpectEqual(Value(), DivValue(Value(1), Value(0))); // unsafe
Aart Bikd14c5952015-09-08 15:25:15 -0700524}
525
Aart Bik97412c922016-02-19 20:14:38 -0800526TEST_F(InductionVarRangeTest, DivValueSpecial) {
527 const int32_t min_value = std::numeric_limits<int32_t>::min();
528 const int32_t max_value = std::numeric_limits<int32_t>::max();
529
530 // Unsafe.
531 ExpectEqual(Value(), DivValue(Value(min_value), Value(-1)));
532
533 // Safe.
534 ExpectEqual(Value(1), DivValue(Value(min_value), Value(min_value)));
535 ExpectEqual(Value(1), DivValue(Value(max_value), Value(max_value)));
536 ExpectEqual(Value(min_value), DivValue(Value(min_value), Value(1)));
537 ExpectEqual(Value(max_value), DivValue(Value(max_value), Value(1)));
538 ExpectEqual(Value(-max_value), DivValue(Value(max_value), Value(-1)));
539 ExpectEqual(Value(-1), DivValue(Value(1), Value(-1)));
540 ExpectEqual(Value(1), DivValue(Value(-1), Value(-1)));
541}
542
Aart Bikd14c5952015-09-08 15:25:15 -0700543TEST_F(InductionVarRangeTest, MinValue) {
544 ExpectEqual(Value(10), MinValue(Value(10), Value(100)));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800545 ExpectEqual(Value(x_, 1, -4), MinValue(Value(x_, 1, -4), Value(x_, 1, -1)));
546 ExpectEqual(Value(x_, 4, -4), MinValue(Value(x_, 4, -4), Value(x_, 4, -1)));
547 ExpectEqual(Value(), MinValue(Value(x_, 1, 5), Value(y_, 1, -7)));
548 ExpectEqual(Value(), MinValue(Value(x_, 1, 20), Value(3)));
549 ExpectEqual(Value(), MinValue(Value(55), Value(y_, 1, -50)));
Aart Bikd14c5952015-09-08 15:25:15 -0700550}
551
552TEST_F(InductionVarRangeTest, MaxValue) {
553 ExpectEqual(Value(100), MaxValue(Value(10), Value(100)));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800554 ExpectEqual(Value(x_, 1, -1), MaxValue(Value(x_, 1, -4), Value(x_, 1, -1)));
555 ExpectEqual(Value(x_, 4, -1), MaxValue(Value(x_, 4, -4), Value(x_, 4, -1)));
556 ExpectEqual(Value(), MaxValue(Value(x_, 1, 5), Value(y_, 1, -7)));
557 ExpectEqual(Value(), MaxValue(Value(x_, 1, 20), Value(3)));
558 ExpectEqual(Value(), MaxValue(Value(55), Value(y_, 1, -50)));
Aart Bikd14c5952015-09-08 15:25:15 -0700559}
560
Aart Bikaec3cce2015-10-14 17:44:55 -0700561//
Aart Bik7d57d7f2015-12-09 14:39:48 -0800562// Tests on public methods.
Aart Bikaec3cce2015-10-14 17:44:55 -0700563//
564
Aart Bik389b3db2015-10-28 14:23:40 -0700565TEST_F(InductionVarRangeTest, ConstantTripCountUp) {
566 BuildLoop(0, graph_->GetIntConstant(1000), 1);
Aart Bikaec3cce2015-10-14 17:44:55 -0700567 PerformInductionVarAnalysis();
Aart Bikaec3cce2015-10-14 17:44:55 -0700568
Aart Bik389b3db2015-10-28 14:23:40 -0700569 Value v1, v2;
570 bool needs_finite_test = true;
571
Aart Bikaec3cce2015-10-14 17:44:55 -0700572 // In context of header: known.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800573 range_.GetInductionRange(condition_, condition_->InputAt(0), &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700574 EXPECT_FALSE(needs_finite_test);
575 ExpectEqual(Value(0), v1);
576 ExpectEqual(Value(1000), v2);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800577 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
Aart Bikaec3cce2015-10-14 17:44:55 -0700578
579 // In context of loop-body: known.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800580 range_.GetInductionRange(increment_, condition_->InputAt(0), &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700581 EXPECT_FALSE(needs_finite_test);
582 ExpectEqual(Value(0), v1);
583 ExpectEqual(Value(999), v2);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800584 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
585 range_.GetInductionRange(increment_, increment_, &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700586 EXPECT_FALSE(needs_finite_test);
587 ExpectEqual(Value(1), v1);
588 ExpectEqual(Value(1000), v2);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800589 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
Aart Bikaec3cce2015-10-14 17:44:55 -0700590}
591
Aart Bik389b3db2015-10-28 14:23:40 -0700592TEST_F(InductionVarRangeTest, ConstantTripCountDown) {
593 BuildLoop(1000, graph_->GetIntConstant(0), -1);
Aart Bikaec3cce2015-10-14 17:44:55 -0700594 PerformInductionVarAnalysis();
Aart Bikaec3cce2015-10-14 17:44:55 -0700595
Aart Bik389b3db2015-10-28 14:23:40 -0700596 Value v1, v2;
597 bool needs_finite_test = true;
598
599 // In context of header: known.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800600 range_.GetInductionRange(condition_, condition_->InputAt(0), &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700601 EXPECT_FALSE(needs_finite_test);
602 ExpectEqual(Value(0), v1);
603 ExpectEqual(Value(1000), v2);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800604 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
Aart Bikaec3cce2015-10-14 17:44:55 -0700605
606 // In context of loop-body: known.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800607 range_.GetInductionRange(increment_, condition_->InputAt(0), &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700608 EXPECT_FALSE(needs_finite_test);
609 ExpectEqual(Value(1), v1);
610 ExpectEqual(Value(1000), v2);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800611 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
612 range_.GetInductionRange(increment_, increment_, &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700613 EXPECT_FALSE(needs_finite_test);
614 ExpectEqual(Value(0), v1);
615 ExpectEqual(Value(999), v2);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800616 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
Aart Bikaec3cce2015-10-14 17:44:55 -0700617}
618
Aart Bik389b3db2015-10-28 14:23:40 -0700619TEST_F(InductionVarRangeTest, SymbolicTripCountUp) {
Aart Bik7d57d7f2015-12-09 14:39:48 -0800620 BuildLoop(0, x_, 1);
Aart Bikaec3cce2015-10-14 17:44:55 -0700621 PerformInductionVarAnalysis();
Aart Bikaec3cce2015-10-14 17:44:55 -0700622
Aart Bik389b3db2015-10-28 14:23:40 -0700623 Value v1, v2;
624 bool needs_finite_test = true;
625 bool needs_taken_test = true;
626
627 // In context of header: upper unknown.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800628 range_.GetInductionRange(condition_, condition_->InputAt(0), &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700629 EXPECT_FALSE(needs_finite_test);
630 ExpectEqual(Value(0), v1);
631 ExpectEqual(Value(), v2);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800632 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
Aart Bik389b3db2015-10-28 14:23:40 -0700633
634 // In context of loop-body: known.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800635 range_.GetInductionRange(increment_, condition_->InputAt(0), &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700636 EXPECT_FALSE(needs_finite_test);
637 ExpectEqual(Value(0), v1);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800638 ExpectEqual(Value(x_, 1, -1), v2);
639 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
640 range_.GetInductionRange(increment_, increment_, &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700641 EXPECT_FALSE(needs_finite_test);
642 ExpectEqual(Value(1), v1);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800643 ExpectEqual(Value(x_, 1, 0), v2);
644 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
Aart Bik389b3db2015-10-28 14:23:40 -0700645
Aart Bikaec3cce2015-10-14 17:44:55 -0700646 HInstruction* lower = nullptr;
647 HInstruction* upper = nullptr;
Aart Bik389b3db2015-10-28 14:23:40 -0700648 HInstruction* taken = nullptr;
Aart Bikaec3cce2015-10-14 17:44:55 -0700649
650 // Can generate code in context of loop-body only.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800651 EXPECT_FALSE(range_.CanGenerateCode(
Aart Bik389b3db2015-10-28 14:23:40 -0700652 condition_, condition_->InputAt(0), &needs_finite_test, &needs_taken_test));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800653 ASSERT_TRUE(range_.CanGenerateCode(
Aart Bik389b3db2015-10-28 14:23:40 -0700654 increment_, condition_->InputAt(0), &needs_finite_test, &needs_taken_test));
655 EXPECT_FALSE(needs_finite_test);
656 EXPECT_TRUE(needs_taken_test);
Aart Bikaec3cce2015-10-14 17:44:55 -0700657
658 // Generates code.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800659 range_.GenerateRangeCode(
660 increment_, condition_->InputAt(0), graph_, loop_preheader_, &lower, &upper);
Aart Bikaec3cce2015-10-14 17:44:55 -0700661
662 // Verify lower is 0+0.
663 ASSERT_TRUE(lower != nullptr);
664 ASSERT_TRUE(lower->IsAdd());
665 ASSERT_TRUE(lower->InputAt(0)->IsIntConstant());
666 EXPECT_EQ(0, lower->InputAt(0)->AsIntConstant()->GetValue());
667 ASSERT_TRUE(lower->InputAt(1)->IsIntConstant());
668 EXPECT_EQ(0, lower->InputAt(1)->AsIntConstant()->GetValue());
669
Aart Bik389b3db2015-10-28 14:23:40 -0700670 // Verify upper is (V-1)+0.
Aart Bikaec3cce2015-10-14 17:44:55 -0700671 ASSERT_TRUE(upper != nullptr);
672 ASSERT_TRUE(upper->IsAdd());
673 ASSERT_TRUE(upper->InputAt(0)->IsSub());
674 EXPECT_TRUE(upper->InputAt(0)->InputAt(0)->IsParameterValue());
675 ASSERT_TRUE(upper->InputAt(0)->InputAt(1)->IsIntConstant());
676 EXPECT_EQ(1, upper->InputAt(0)->InputAt(1)->AsIntConstant()->GetValue());
677 ASSERT_TRUE(upper->InputAt(1)->IsIntConstant());
678 EXPECT_EQ(0, upper->InputAt(1)->AsIntConstant()->GetValue());
Aart Bik389b3db2015-10-28 14:23:40 -0700679
680 // Verify taken-test is 0<V.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800681 range_.GenerateTakenTest(increment_, graph_, loop_preheader_, &taken);
Aart Bik389b3db2015-10-28 14:23:40 -0700682 ASSERT_TRUE(taken != nullptr);
683 ASSERT_TRUE(taken->IsLessThan());
684 ASSERT_TRUE(taken->InputAt(0)->IsIntConstant());
685 EXPECT_EQ(0, taken->InputAt(0)->AsIntConstant()->GetValue());
686 EXPECT_TRUE(taken->InputAt(1)->IsParameterValue());
687}
688
689TEST_F(InductionVarRangeTest, SymbolicTripCountDown) {
Aart Bik7d57d7f2015-12-09 14:39:48 -0800690 BuildLoop(1000, x_, -1);
Aart Bik389b3db2015-10-28 14:23:40 -0700691 PerformInductionVarAnalysis();
Aart Bik389b3db2015-10-28 14:23:40 -0700692
693 Value v1, v2;
694 bool needs_finite_test = true;
695 bool needs_taken_test = true;
696
697 // In context of header: lower unknown.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800698 range_.GetInductionRange(condition_, condition_->InputAt(0), &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700699 EXPECT_FALSE(needs_finite_test);
700 ExpectEqual(Value(), v1);
701 ExpectEqual(Value(1000), v2);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800702 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
Aart Bik389b3db2015-10-28 14:23:40 -0700703
704 // In context of loop-body: known.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800705 range_.GetInductionRange(increment_, condition_->InputAt(0), &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700706 EXPECT_FALSE(needs_finite_test);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800707 ExpectEqual(Value(x_, 1, 1), v1);
Aart Bik389b3db2015-10-28 14:23:40 -0700708 ExpectEqual(Value(1000), v2);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800709 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
710 range_.GetInductionRange(increment_, increment_, &v1, &v2, &needs_finite_test);
Aart Bik389b3db2015-10-28 14:23:40 -0700711 EXPECT_FALSE(needs_finite_test);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800712 ExpectEqual(Value(x_, 1, 0), v1);
Aart Bik389b3db2015-10-28 14:23:40 -0700713 ExpectEqual(Value(999), v2);
Aart Bik7d57d7f2015-12-09 14:39:48 -0800714 EXPECT_FALSE(range_.RefineOuter(&v1, &v2));
Aart Bik389b3db2015-10-28 14:23:40 -0700715
716 HInstruction* lower = nullptr;
717 HInstruction* upper = nullptr;
718 HInstruction* taken = nullptr;
719
720 // Can generate code in context of loop-body only.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800721 EXPECT_FALSE(range_.CanGenerateCode(
Aart Bik389b3db2015-10-28 14:23:40 -0700722 condition_, condition_->InputAt(0), &needs_finite_test, &needs_taken_test));
Aart Bik7d57d7f2015-12-09 14:39:48 -0800723 ASSERT_TRUE(range_.CanGenerateCode(
Aart Bik389b3db2015-10-28 14:23:40 -0700724 increment_, condition_->InputAt(0), &needs_finite_test, &needs_taken_test));
725 EXPECT_FALSE(needs_finite_test);
726 EXPECT_TRUE(needs_taken_test);
727
728 // Generates code.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800729 range_.GenerateRangeCode(
730 increment_, condition_->InputAt(0), graph_, loop_preheader_, &lower, &upper);
Aart Bik389b3db2015-10-28 14:23:40 -0700731
Aart Bik7d57d7f2015-12-09 14:39:48 -0800732 // Verify lower is 1000-((1000-V)-1).
Aart Bik389b3db2015-10-28 14:23:40 -0700733 ASSERT_TRUE(lower != nullptr);
734 ASSERT_TRUE(lower->IsSub());
735 ASSERT_TRUE(lower->InputAt(0)->IsIntConstant());
736 EXPECT_EQ(1000, lower->InputAt(0)->AsIntConstant()->GetValue());
737 lower = lower->InputAt(1);
738 ASSERT_TRUE(lower->IsSub());
739 ASSERT_TRUE(lower->InputAt(1)->IsIntConstant());
740 EXPECT_EQ(1, lower->InputAt(1)->AsIntConstant()->GetValue());
741 lower = lower->InputAt(0);
Aart Bik389b3db2015-10-28 14:23:40 -0700742 ASSERT_TRUE(lower->IsSub());
Aart Bik7d57d7f2015-12-09 14:39:48 -0800743 ASSERT_TRUE(lower->InputAt(0)->IsIntConstant());
744 EXPECT_EQ(1000, lower->InputAt(0)->AsIntConstant()->GetValue());
745 EXPECT_TRUE(lower->InputAt(1)->IsParameterValue());
Aart Bik389b3db2015-10-28 14:23:40 -0700746
747 // Verify upper is 1000-0.
748 ASSERT_TRUE(upper != nullptr);
749 ASSERT_TRUE(upper->IsSub());
750 ASSERT_TRUE(upper->InputAt(0)->IsIntConstant());
751 EXPECT_EQ(1000, upper->InputAt(0)->AsIntConstant()->GetValue());
752 ASSERT_TRUE(upper->InputAt(1)->IsIntConstant());
753 EXPECT_EQ(0, upper->InputAt(1)->AsIntConstant()->GetValue());
754
755 // Verify taken-test is 1000>V.
Aart Bik7d57d7f2015-12-09 14:39:48 -0800756 range_.GenerateTakenTest(increment_, graph_, loop_preheader_, &taken);
Aart Bik389b3db2015-10-28 14:23:40 -0700757 ASSERT_TRUE(taken != nullptr);
758 ASSERT_TRUE(taken->IsGreaterThan());
759 ASSERT_TRUE(taken->InputAt(0)->IsIntConstant());
760 EXPECT_EQ(1000, taken->InputAt(0)->AsIntConstant()->GetValue());
761 EXPECT_TRUE(taken->InputAt(1)->IsParameterValue());
Aart Bikaec3cce2015-10-14 17:44:55 -0700762}
763
Aart Bikd14c5952015-09-08 15:25:15 -0700764} // namespace art