1 //===- ConstantRangeTest.cpp - ConstantRange tests ------------------------===//
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
7 //===----------------------------------------------------------------------===//
9 #include "llvm/ADT/BitVector.h"
10 #include "llvm/IR/ConstantRange.h"
11 #include "llvm/IR/Instructions.h"
12 #include "llvm/IR/Operator.h"
13 #include "llvm/Support/KnownBits.h"
14 #include "gtest/gtest.h"
20 class ConstantRangeTest : public ::testing::Test {
22 static ConstantRange Full;
23 static ConstantRange Empty;
24 static ConstantRange One;
25 static ConstantRange Some;
26 static ConstantRange Wrap;
30 static void EnumerateConstantRanges(unsigned Bits, Fn TestFn) {
31 unsigned Max = 1 << Bits;
32 for (unsigned Lo = 0; Lo < Max; Lo++) {
33 for (unsigned Hi = 0; Hi < Max; Hi++) {
34 // Enforce ConstantRange invariant.
35 if (Lo == Hi && Lo != 0 && Lo != Max - 1)
38 ConstantRange CR(APInt(Bits, Lo), APInt(Bits, Hi));
45 static void EnumerateTwoConstantRanges(unsigned Bits, Fn TestFn) {
46 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR1) {
47 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR2) {
54 static void ForeachNumInConstantRange(const ConstantRange &CR, Fn TestFn) {
55 if (!CR.isEmptySet()) {
56 APInt N = CR.getLower();
58 while (++N != CR.getUpper());
62 struct OpRangeGathererBase {
63 void account(const APInt &N);
64 ConstantRange getRange();
67 struct UnsignedOpRangeGatherer : public OpRangeGathererBase {
71 UnsignedOpRangeGatherer(unsigned Bits)
72 : Min(APInt::getMaxValue(Bits)), Max(APInt::getMinValue(Bits)) {}
74 void account(const APInt &N) {
81 ConstantRange getRange() {
83 return ConstantRange::getEmpty(Min.getBitWidth());
84 return ConstantRange::getNonEmpty(Min, Max + 1);
88 struct SignedOpRangeGatherer : public OpRangeGathererBase {
92 SignedOpRangeGatherer(unsigned Bits)
93 : Min(APInt::getSignedMaxValue(Bits)),
94 Max(APInt::getSignedMinValue(Bits)) {}
96 void account(const APInt &N) {
103 ConstantRange getRange() {
105 return ConstantRange::getEmpty(Min.getBitWidth());
106 return ConstantRange::getNonEmpty(Min, Max + 1);
110 template <typename Fn1, typename Fn2>
111 static void TestUnsignedUnaryOpExhaustive(Fn1 RangeFn, Fn2 IntFn,
112 bool SkipSignedIntMin = false) {
114 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) {
115 UnsignedOpRangeGatherer R(CR.getBitWidth());
116 ForeachNumInConstantRange(CR, [&](const APInt &N) {
117 if (SkipSignedIntMin && N.isMinSignedValue())
122 ConstantRange ExactCR = R.getRange();
123 ConstantRange ActualCR = RangeFn(CR);
125 EXPECT_EQ(ExactCR, ActualCR);
129 template <typename Fn1, typename Fn2>
130 static void TestUnsignedBinOpExhaustive(Fn1 RangeFn, Fn2 IntFn,
131 bool SkipZeroRHS = false,
132 bool CorrectnessOnly = false) {
134 EnumerateTwoConstantRanges(
135 Bits, [&](const ConstantRange &CR1, const ConstantRange &CR2) {
136 UnsignedOpRangeGatherer R(CR1.getBitWidth());
137 ForeachNumInConstantRange(CR1, [&](const APInt &N1) {
138 ForeachNumInConstantRange(CR2, [&](const APInt &N2) {
139 if (SkipZeroRHS && N2 == 0)
141 R.account(IntFn(N1, N2));
145 ConstantRange CR = RangeFn(CR1, CR2);
147 ConstantRange Exact = R.getRange();
149 if (!CorrectnessOnly) {
150 EXPECT_EQ(Exact, CR);
154 EXPECT_TRUE(CR.contains(Exact));
155 if (Exact.isEmptySet())
156 EXPECT_TRUE(CR.isEmptySet());
160 template <typename Fn1, typename Fn2>
161 static void TestSignedBinOpExhaustive(Fn1 RangeFn, Fn2 IntFn,
162 bool SkipZeroRHS = false,
163 bool CorrectnessOnly = false) {
165 EnumerateTwoConstantRanges(
166 Bits, [&](const ConstantRange &CR1, const ConstantRange &CR2) {
167 SignedOpRangeGatherer R(CR1.getBitWidth());
168 ForeachNumInConstantRange(CR1, [&](const APInt &N1) {
169 ForeachNumInConstantRange(CR2, [&](const APInt &N2) {
170 if (SkipZeroRHS && N2 == 0)
173 R.account(IntFn(N1, N2));
177 ConstantRange CR = RangeFn(CR1, CR2);
179 ConstantRange Exact = R.getRange();
180 if (CorrectnessOnly) {
181 EXPECT_TRUE(CR.contains(Exact));
183 EXPECT_EQ(Exact, CR);
188 ConstantRange ConstantRangeTest::Full(16, true);
189 ConstantRange ConstantRangeTest::Empty(16, false);
190 ConstantRange ConstantRangeTest::One(APInt(16, 0xa));
191 ConstantRange ConstantRangeTest::Some(APInt(16, 0xa), APInt(16, 0xaaa));
192 ConstantRange ConstantRangeTest::Wrap(APInt(16, 0xaaa), APInt(16, 0xa));
194 TEST_F(ConstantRangeTest, Basics) {
195 EXPECT_TRUE(Full.isFullSet());
196 EXPECT_FALSE(Full.isEmptySet());
197 EXPECT_TRUE(Full.inverse().isEmptySet());
198 EXPECT_FALSE(Full.isWrappedSet());
199 EXPECT_TRUE(Full.contains(APInt(16, 0x0)));
200 EXPECT_TRUE(Full.contains(APInt(16, 0x9)));
201 EXPECT_TRUE(Full.contains(APInt(16, 0xa)));
202 EXPECT_TRUE(Full.contains(APInt(16, 0xaa9)));
203 EXPECT_TRUE(Full.contains(APInt(16, 0xaaa)));
205 EXPECT_FALSE(Empty.isFullSet());
206 EXPECT_TRUE(Empty.isEmptySet());
207 EXPECT_TRUE(Empty.inverse().isFullSet());
208 EXPECT_FALSE(Empty.isWrappedSet());
209 EXPECT_FALSE(Empty.contains(APInt(16, 0x0)));
210 EXPECT_FALSE(Empty.contains(APInt(16, 0x9)));
211 EXPECT_FALSE(Empty.contains(APInt(16, 0xa)));
212 EXPECT_FALSE(Empty.contains(APInt(16, 0xaa9)));
213 EXPECT_FALSE(Empty.contains(APInt(16, 0xaaa)));
215 EXPECT_FALSE(One.isFullSet());
216 EXPECT_FALSE(One.isEmptySet());
217 EXPECT_FALSE(One.isWrappedSet());
218 EXPECT_FALSE(One.contains(APInt(16, 0x0)));
219 EXPECT_FALSE(One.contains(APInt(16, 0x9)));
220 EXPECT_TRUE(One.contains(APInt(16, 0xa)));
221 EXPECT_FALSE(One.contains(APInt(16, 0xaa9)));
222 EXPECT_FALSE(One.contains(APInt(16, 0xaaa)));
223 EXPECT_FALSE(One.inverse().contains(APInt(16, 0xa)));
225 EXPECT_FALSE(Some.isFullSet());
226 EXPECT_FALSE(Some.isEmptySet());
227 EXPECT_FALSE(Some.isWrappedSet());
228 EXPECT_FALSE(Some.contains(APInt(16, 0x0)));
229 EXPECT_FALSE(Some.contains(APInt(16, 0x9)));
230 EXPECT_TRUE(Some.contains(APInt(16, 0xa)));
231 EXPECT_TRUE(Some.contains(APInt(16, 0xaa9)));
232 EXPECT_FALSE(Some.contains(APInt(16, 0xaaa)));
234 EXPECT_FALSE(Wrap.isFullSet());
235 EXPECT_FALSE(Wrap.isEmptySet());
236 EXPECT_TRUE(Wrap.isWrappedSet());
237 EXPECT_TRUE(Wrap.contains(APInt(16, 0x0)));
238 EXPECT_TRUE(Wrap.contains(APInt(16, 0x9)));
239 EXPECT_FALSE(Wrap.contains(APInt(16, 0xa)));
240 EXPECT_FALSE(Wrap.contains(APInt(16, 0xaa9)));
241 EXPECT_TRUE(Wrap.contains(APInt(16, 0xaaa)));
244 TEST_F(ConstantRangeTest, Equality) {
245 EXPECT_EQ(Full, Full);
246 EXPECT_EQ(Empty, Empty);
248 EXPECT_EQ(Some, Some);
249 EXPECT_EQ(Wrap, Wrap);
250 EXPECT_NE(Full, Empty);
251 EXPECT_NE(Full, One);
252 EXPECT_NE(Full, Some);
253 EXPECT_NE(Full, Wrap);
254 EXPECT_NE(Empty, One);
255 EXPECT_NE(Empty, Some);
256 EXPECT_NE(Empty, Wrap);
257 EXPECT_NE(One, Some);
258 EXPECT_NE(One, Wrap);
259 EXPECT_NE(Some, Wrap);
262 TEST_F(ConstantRangeTest, SingleElement) {
263 EXPECT_EQ(Full.getSingleElement(), static_cast<APInt *>(nullptr));
264 EXPECT_EQ(Empty.getSingleElement(), static_cast<APInt *>(nullptr));
265 EXPECT_EQ(Full.getSingleMissingElement(), static_cast<APInt *>(nullptr));
266 EXPECT_EQ(Empty.getSingleMissingElement(), static_cast<APInt *>(nullptr));
268 EXPECT_EQ(*One.getSingleElement(), APInt(16, 0xa));
269 EXPECT_EQ(Some.getSingleElement(), static_cast<APInt *>(nullptr));
270 EXPECT_EQ(Wrap.getSingleElement(), static_cast<APInt *>(nullptr));
272 EXPECT_EQ(One.getSingleMissingElement(), static_cast<APInt *>(nullptr));
273 EXPECT_EQ(Some.getSingleMissingElement(), static_cast<APInt *>(nullptr));
275 ConstantRange OneInverse = One.inverse();
276 EXPECT_EQ(*OneInverse.getSingleMissingElement(), *One.getSingleElement());
278 EXPECT_FALSE(Full.isSingleElement());
279 EXPECT_FALSE(Empty.isSingleElement());
280 EXPECT_TRUE(One.isSingleElement());
281 EXPECT_FALSE(Some.isSingleElement());
282 EXPECT_FALSE(Wrap.isSingleElement());
285 TEST_F(ConstantRangeTest, GetMinsAndMaxes) {
286 EXPECT_EQ(Full.getUnsignedMax(), APInt(16, UINT16_MAX));
287 EXPECT_EQ(One.getUnsignedMax(), APInt(16, 0xa));
288 EXPECT_EQ(Some.getUnsignedMax(), APInt(16, 0xaa9));
289 EXPECT_EQ(Wrap.getUnsignedMax(), APInt(16, UINT16_MAX));
291 EXPECT_EQ(Full.getUnsignedMin(), APInt(16, 0));
292 EXPECT_EQ(One.getUnsignedMin(), APInt(16, 0xa));
293 EXPECT_EQ(Some.getUnsignedMin(), APInt(16, 0xa));
294 EXPECT_EQ(Wrap.getUnsignedMin(), APInt(16, 0));
296 EXPECT_EQ(Full.getSignedMax(), APInt(16, INT16_MAX));
297 EXPECT_EQ(One.getSignedMax(), APInt(16, 0xa));
298 EXPECT_EQ(Some.getSignedMax(), APInt(16, 0xaa9));
299 EXPECT_EQ(Wrap.getSignedMax(), APInt(16, INT16_MAX));
301 EXPECT_EQ(Full.getSignedMin(), APInt(16, (uint64_t)INT16_MIN));
302 EXPECT_EQ(One.getSignedMin(), APInt(16, 0xa));
303 EXPECT_EQ(Some.getSignedMin(), APInt(16, 0xa));
304 EXPECT_EQ(Wrap.getSignedMin(), APInt(16, (uint64_t)INT16_MIN));
307 EXPECT_EQ(ConstantRange(APInt(4, 7), APInt(4, 0)).getSignedMax(),
311 TEST_F(ConstantRangeTest, SignWrapped) {
312 EXPECT_FALSE(Full.isSignWrappedSet());
313 EXPECT_FALSE(Empty.isSignWrappedSet());
314 EXPECT_FALSE(One.isSignWrappedSet());
315 EXPECT_FALSE(Some.isSignWrappedSet());
316 EXPECT_TRUE(Wrap.isSignWrappedSet());
318 EXPECT_FALSE(ConstantRange(APInt(8, 127), APInt(8, 128)).isSignWrappedSet());
319 EXPECT_TRUE(ConstantRange(APInt(8, 127), APInt(8, 129)).isSignWrappedSet());
320 EXPECT_FALSE(ConstantRange(APInt(8, 128), APInt(8, 129)).isSignWrappedSet());
321 EXPECT_TRUE(ConstantRange(APInt(8, 10), APInt(8, 9)).isSignWrappedSet());
322 EXPECT_TRUE(ConstantRange(APInt(8, 10), APInt(8, 250)).isSignWrappedSet());
323 EXPECT_FALSE(ConstantRange(APInt(8, 250), APInt(8, 10)).isSignWrappedSet());
324 EXPECT_FALSE(ConstantRange(APInt(8, 250), APInt(8, 251)).isSignWrappedSet());
327 TEST_F(ConstantRangeTest, UpperWrapped) {
328 // The behavior here is the same as for isWrappedSet() / isSignWrappedSet().
329 EXPECT_FALSE(Full.isUpperWrapped());
330 EXPECT_FALSE(Empty.isUpperWrapped());
331 EXPECT_FALSE(One.isUpperWrapped());
332 EXPECT_FALSE(Some.isUpperWrapped());
333 EXPECT_TRUE(Wrap.isUpperWrapped());
334 EXPECT_FALSE(Full.isUpperSignWrapped());
335 EXPECT_FALSE(Empty.isUpperSignWrapped());
336 EXPECT_FALSE(One.isUpperSignWrapped());
337 EXPECT_FALSE(Some.isUpperSignWrapped());
338 EXPECT_TRUE(Wrap.isUpperSignWrapped());
340 // The behavior differs if Upper is the Min/SignedMin value.
341 ConstantRange CR1(APInt(8, 42), APInt::getMinValue(8));
342 EXPECT_FALSE(CR1.isWrappedSet());
343 EXPECT_TRUE(CR1.isUpperWrapped());
345 ConstantRange CR2(APInt(8, 42), APInt::getSignedMinValue(8));
346 EXPECT_FALSE(CR2.isSignWrappedSet());
347 EXPECT_TRUE(CR2.isUpperSignWrapped());
350 TEST_F(ConstantRangeTest, Trunc) {
351 ConstantRange TFull = Full.truncate(10);
352 ConstantRange TEmpty = Empty.truncate(10);
353 ConstantRange TOne = One.truncate(10);
354 ConstantRange TSome = Some.truncate(10);
355 ConstantRange TWrap = Wrap.truncate(10);
356 EXPECT_TRUE(TFull.isFullSet());
357 EXPECT_TRUE(TEmpty.isEmptySet());
358 EXPECT_EQ(TOne, ConstantRange(One.getLower().trunc(10),
359 One.getUpper().trunc(10)));
360 EXPECT_TRUE(TSome.isFullSet());
361 EXPECT_TRUE(TWrap.isFullSet());
363 // trunc([2, 5), 3->2) = [2, 1)
364 ConstantRange TwoFive(APInt(3, 2), APInt(3, 5));
365 EXPECT_EQ(TwoFive.truncate(2), ConstantRange(APInt(2, 2), APInt(2, 1)));
367 // trunc([2, 6), 3->2) = full
368 ConstantRange TwoSix(APInt(3, 2), APInt(3, 6));
369 EXPECT_TRUE(TwoSix.truncate(2).isFullSet());
371 // trunc([5, 7), 3->2) = [1, 3)
372 ConstantRange FiveSeven(APInt(3, 5), APInt(3, 7));
373 EXPECT_EQ(FiveSeven.truncate(2), ConstantRange(APInt(2, 1), APInt(2, 3)));
375 // trunc([7, 1), 3->2) = [3, 1)
376 ConstantRange SevenOne(APInt(3, 7), APInt(3, 1));
377 EXPECT_EQ(SevenOne.truncate(2), ConstantRange(APInt(2, 3), APInt(2, 1)));
380 TEST_F(ConstantRangeTest, ZExt) {
381 ConstantRange ZFull = Full.zeroExtend(20);
382 ConstantRange ZEmpty = Empty.zeroExtend(20);
383 ConstantRange ZOne = One.zeroExtend(20);
384 ConstantRange ZSome = Some.zeroExtend(20);
385 ConstantRange ZWrap = Wrap.zeroExtend(20);
386 EXPECT_EQ(ZFull, ConstantRange(APInt(20, 0), APInt(20, 0x10000)));
387 EXPECT_TRUE(ZEmpty.isEmptySet());
388 EXPECT_EQ(ZOne, ConstantRange(One.getLower().zext(20),
389 One.getUpper().zext(20)));
390 EXPECT_EQ(ZSome, ConstantRange(Some.getLower().zext(20),
391 Some.getUpper().zext(20)));
392 EXPECT_EQ(ZWrap, ConstantRange(APInt(20, 0), APInt(20, 0x10000)));
394 // zext([5, 0), 3->7) = [5, 8)
395 ConstantRange FiveZero(APInt(3, 5), APInt(3, 0));
396 EXPECT_EQ(FiveZero.zeroExtend(7), ConstantRange(APInt(7, 5), APInt(7, 8)));
399 TEST_F(ConstantRangeTest, SExt) {
400 ConstantRange SFull = Full.signExtend(20);
401 ConstantRange SEmpty = Empty.signExtend(20);
402 ConstantRange SOne = One.signExtend(20);
403 ConstantRange SSome = Some.signExtend(20);
404 ConstantRange SWrap = Wrap.signExtend(20);
405 EXPECT_EQ(SFull, ConstantRange(APInt(20, (uint64_t)INT16_MIN, true),
406 APInt(20, INT16_MAX + 1, true)));
407 EXPECT_TRUE(SEmpty.isEmptySet());
408 EXPECT_EQ(SOne, ConstantRange(One.getLower().sext(20),
409 One.getUpper().sext(20)));
410 EXPECT_EQ(SSome, ConstantRange(Some.getLower().sext(20),
411 Some.getUpper().sext(20)));
412 EXPECT_EQ(SWrap, ConstantRange(APInt(20, (uint64_t)INT16_MIN, true),
413 APInt(20, INT16_MAX + 1, true)));
415 EXPECT_EQ(ConstantRange(APInt(8, 120), APInt(8, 140)).signExtend(16),
416 ConstantRange(APInt(16, -128), APInt(16, 128)));
418 EXPECT_EQ(ConstantRange(APInt(16, 0x0200), APInt(16, 0x8000)).signExtend(19),
419 ConstantRange(APInt(19, 0x0200), APInt(19, 0x8000)));
422 TEST_F(ConstantRangeTest, IntersectWith) {
423 EXPECT_EQ(Empty.intersectWith(Full), Empty);
424 EXPECT_EQ(Empty.intersectWith(Empty), Empty);
425 EXPECT_EQ(Empty.intersectWith(One), Empty);
426 EXPECT_EQ(Empty.intersectWith(Some), Empty);
427 EXPECT_EQ(Empty.intersectWith(Wrap), Empty);
428 EXPECT_EQ(Full.intersectWith(Full), Full);
429 EXPECT_EQ(Some.intersectWith(Some), Some);
430 EXPECT_EQ(Some.intersectWith(One), One);
431 EXPECT_EQ(Full.intersectWith(One), One);
432 EXPECT_EQ(Full.intersectWith(Some), Some);
433 EXPECT_EQ(Some.intersectWith(Wrap), Empty);
434 EXPECT_EQ(One.intersectWith(Wrap), Empty);
435 EXPECT_EQ(One.intersectWith(Wrap), Wrap.intersectWith(One));
437 // Klee generated testcase from PR4545.
438 // The intersection of i16 [4, 2) and [6, 5) is disjoint, looking like
439 // 01..4.6789ABCDEF where the dots represent values not in the intersection.
440 ConstantRange LHS(APInt(16, 4), APInt(16, 2));
441 ConstantRange RHS(APInt(16, 6), APInt(16, 5));
442 EXPECT_TRUE(LHS.intersectWith(RHS) == LHS);
444 // previous bug: intersection of [min, 3) and [2, max) should be 2
445 LHS = ConstantRange(APInt(32, -2147483646), APInt(32, 3));
446 RHS = ConstantRange(APInt(32, 2), APInt(32, 2147483646));
447 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 2)));
449 // [2, 0) /\ [4, 3) = [2, 0)
450 LHS = ConstantRange(APInt(32, 2), APInt(32, 0));
451 RHS = ConstantRange(APInt(32, 4), APInt(32, 3));
452 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 2), APInt(32, 0)));
454 // [2, 0) /\ [4, 2) = [4, 0)
455 LHS = ConstantRange(APInt(32, 2), APInt(32, 0));
456 RHS = ConstantRange(APInt(32, 4), APInt(32, 2));
457 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 4), APInt(32, 0)));
459 // [4, 2) /\ [5, 1) = [5, 1)
460 LHS = ConstantRange(APInt(32, 4), APInt(32, 2));
461 RHS = ConstantRange(APInt(32, 5), APInt(32, 1));
462 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 5), APInt(32, 1)));
464 // [2, 0) /\ [7, 4) = [7, 4)
465 LHS = ConstantRange(APInt(32, 2), APInt(32, 0));
466 RHS = ConstantRange(APInt(32, 7), APInt(32, 4));
467 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 7), APInt(32, 4)));
469 // [4, 2) /\ [1, 0) = [1, 0)
470 LHS = ConstantRange(APInt(32, 4), APInt(32, 2));
471 RHS = ConstantRange(APInt(32, 1), APInt(32, 0));
472 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 4), APInt(32, 2)));
474 // [15, 0) /\ [7, 6) = [15, 0)
475 LHS = ConstantRange(APInt(32, 15), APInt(32, 0));
476 RHS = ConstantRange(APInt(32, 7), APInt(32, 6));
477 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 15), APInt(32, 0)));
480 template<typename Fn1, typename Fn2>
481 void testBinarySetOperationExhaustive(Fn1 OpFn, Fn2 InResultFn) {
483 EnumerateTwoConstantRanges(Bits,
484 [=](const ConstantRange &CR1, const ConstantRange &CR2) {
485 // Collect up to three contiguous unsigned ranges. The HaveInterrupt
486 // variables are used determine when we have to switch to the next
487 // range because the previous one ended.
488 APInt Lower1(Bits, 0), Upper1(Bits, 0);
489 APInt Lower2(Bits, 0), Upper2(Bits, 0);
490 APInt Lower3(Bits, 0), Upper3(Bits, 0);
491 bool HaveRange1 = false, HaveInterrupt1 = false;
492 bool HaveRange2 = false, HaveInterrupt2 = false;
493 bool HaveRange3 = false, HaveInterrupt3 = false;
496 for (unsigned I = 0, Limit = 1 << Bits; I < Limit; ++I, ++Num) {
497 if (!InResultFn(CR1, CR2, Num)) {
499 HaveInterrupt3 = true;
501 HaveInterrupt2 = true;
503 HaveInterrupt1 = true;
509 } else if (HaveInterrupt2) {
511 Lower3 = Upper3 = Num;
512 } else if (HaveRange2) {
514 } else if (HaveInterrupt1) {
516 Lower2 = Upper2 = Num;
517 } else if (HaveRange1) {
521 Lower1 = Upper1 = Num;
525 (void)HaveInterrupt3;
526 assert(!HaveInterrupt3 && "Should have at most three ranges");
528 ConstantRange SmallestCR = OpFn(CR1, CR2, ConstantRange::Smallest);
529 ConstantRange UnsignedCR = OpFn(CR1, CR2, ConstantRange::Unsigned);
530 ConstantRange SignedCR = OpFn(CR1, CR2, ConstantRange::Signed);
533 EXPECT_TRUE(SmallestCR.isEmptySet());
534 EXPECT_TRUE(UnsignedCR.isEmptySet());
535 EXPECT_TRUE(SignedCR.isEmptySet());
540 if (Lower1 == Upper1 + 1) {
541 EXPECT_TRUE(SmallestCR.isFullSet());
542 EXPECT_TRUE(UnsignedCR.isFullSet());
543 EXPECT_TRUE(SignedCR.isFullSet());
545 ConstantRange Expected(Lower1, Upper1 + 1);
546 EXPECT_EQ(Expected, SmallestCR);
547 EXPECT_EQ(Expected, UnsignedCR);
548 EXPECT_EQ(Expected, SignedCR);
553 ConstantRange Variant1(Bits, /*full*/ true);
554 ConstantRange Variant2(Bits, /*full*/ true);
556 // Compute the two possible ways to cover two disjoint ranges.
557 if (Lower1 != Upper2 + 1)
558 Variant1 = ConstantRange(Lower1, Upper2 + 1);
559 if (Lower2 != Upper1 + 1)
560 Variant2 = ConstantRange(Lower2, Upper1 + 1);
562 // If we have three ranges, the first and last one have to be adjacent
563 // to the unsigned domain. It's better to think of this as having two
564 // holes, and we can construct one range using each hole.
565 assert(Lower1.isNullValue() && Upper3.isMaxValue());
566 Variant1 = ConstantRange(Lower2, Upper1 + 1);
567 Variant2 = ConstantRange(Lower3, Upper2 + 1);
570 // Smallest: Smaller set, then any set.
571 if (Variant1.isSizeStrictlySmallerThan(Variant2))
572 EXPECT_EQ(Variant1, SmallestCR);
573 else if (Variant2.isSizeStrictlySmallerThan(Variant1))
574 EXPECT_EQ(Variant2, SmallestCR);
576 EXPECT_TRUE(Variant1 == SmallestCR || Variant2 == SmallestCR);
578 // Unsigned: Non-wrapped set, then smaller set, then any set.
579 bool Variant1Full = Variant1.isFullSet() || Variant1.isWrappedSet();
580 bool Variant2Full = Variant2.isFullSet() || Variant2.isWrappedSet();
581 if (!Variant1Full && Variant2Full)
582 EXPECT_EQ(Variant1, UnsignedCR);
583 else if (Variant1Full && !Variant2Full)
584 EXPECT_EQ(Variant2, UnsignedCR);
585 else if (Variant1.isSizeStrictlySmallerThan(Variant2))
586 EXPECT_EQ(Variant1, UnsignedCR);
587 else if (Variant2.isSizeStrictlySmallerThan(Variant1))
588 EXPECT_EQ(Variant2, UnsignedCR);
590 EXPECT_TRUE(Variant1 == UnsignedCR || Variant2 == UnsignedCR);
592 // Signed: Signed non-wrapped set, then smaller set, then any set.
593 Variant1Full = Variant1.isFullSet() || Variant1.isSignWrappedSet();
594 Variant2Full = Variant2.isFullSet() || Variant2.isSignWrappedSet();
595 if (!Variant1Full && Variant2Full)
596 EXPECT_EQ(Variant1, SignedCR);
597 else if (Variant1Full && !Variant2Full)
598 EXPECT_EQ(Variant2, SignedCR);
599 else if (Variant1.isSizeStrictlySmallerThan(Variant2))
600 EXPECT_EQ(Variant1, SignedCR);
601 else if (Variant2.isSizeStrictlySmallerThan(Variant1))
602 EXPECT_EQ(Variant2, SignedCR);
604 EXPECT_TRUE(Variant1 == SignedCR || Variant2 == SignedCR);
608 TEST_F(ConstantRangeTest, IntersectWithExhaustive) {
609 testBinarySetOperationExhaustive(
610 [](const ConstantRange &CR1, const ConstantRange &CR2,
611 ConstantRange::PreferredRangeType Type) {
612 return CR1.intersectWith(CR2, Type);
614 [](const ConstantRange &CR1, const ConstantRange &CR2, const APInt &N) {
615 return CR1.contains(N) && CR2.contains(N);
619 TEST_F(ConstantRangeTest, UnionWithExhaustive) {
620 testBinarySetOperationExhaustive(
621 [](const ConstantRange &CR1, const ConstantRange &CR2,
622 ConstantRange::PreferredRangeType Type) {
623 return CR1.unionWith(CR2, Type);
625 [](const ConstantRange &CR1, const ConstantRange &CR2, const APInt &N) {
626 return CR1.contains(N) || CR2.contains(N);
630 TEST_F(ConstantRangeTest, UnionWith) {
631 EXPECT_EQ(Wrap.unionWith(One),
632 ConstantRange(APInt(16, 0xaaa), APInt(16, 0xb)));
633 EXPECT_EQ(One.unionWith(Wrap), Wrap.unionWith(One));
634 EXPECT_EQ(Empty.unionWith(Empty), Empty);
635 EXPECT_EQ(Full.unionWith(Full), Full);
636 EXPECT_EQ(Some.unionWith(Wrap), Full);
639 EXPECT_EQ(ConstantRange(APInt(16, 14), APInt(16, 1)).unionWith(
640 ConstantRange(APInt(16, 0), APInt(16, 8))),
641 ConstantRange(APInt(16, 14), APInt(16, 8)));
642 EXPECT_EQ(ConstantRange(APInt(16, 6), APInt(16, 4)).unionWith(
643 ConstantRange(APInt(16, 4), APInt(16, 0))),
644 ConstantRange::getFull(16));
645 EXPECT_EQ(ConstantRange(APInt(16, 1), APInt(16, 0)).unionWith(
646 ConstantRange(APInt(16, 2), APInt(16, 1))),
647 ConstantRange::getFull(16));
650 TEST_F(ConstantRangeTest, SetDifference) {
651 EXPECT_EQ(Full.difference(Empty), Full);
652 EXPECT_EQ(Full.difference(Full), Empty);
653 EXPECT_EQ(Empty.difference(Empty), Empty);
654 EXPECT_EQ(Empty.difference(Full), Empty);
656 ConstantRange A(APInt(16, 3), APInt(16, 7));
657 ConstantRange B(APInt(16, 5), APInt(16, 9));
658 ConstantRange C(APInt(16, 3), APInt(16, 5));
659 ConstantRange D(APInt(16, 7), APInt(16, 9));
660 ConstantRange E(APInt(16, 5), APInt(16, 4));
661 ConstantRange F(APInt(16, 7), APInt(16, 3));
662 EXPECT_EQ(A.difference(B), C);
663 EXPECT_EQ(B.difference(A), D);
664 EXPECT_EQ(E.difference(A), F);
667 TEST_F(ConstantRangeTest, getActiveBits) {
669 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) {
671 ForeachNumInConstantRange(CR, [&](const APInt &N) {
672 Exact = std::max(Exact, N.getActiveBits());
675 unsigned ResultCR = CR.getActiveBits();
676 EXPECT_EQ(Exact, ResultCR);
679 TEST_F(ConstantRangeTest, losslessUnsignedTruncationZeroext) {
681 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) {
682 unsigned MinBitWidth = CR.getActiveBits();
683 if (MinBitWidth == 0) {
684 EXPECT_TRUE(CR.isEmptySet() || (CR.isSingleElement() &&
685 CR.getSingleElement()->isNullValue()));
688 if (MinBitWidth == Bits)
690 EXPECT_EQ(CR, CR.truncate(MinBitWidth).zeroExtend(Bits));
694 TEST_F(ConstantRangeTest, getMinSignedBits) {
696 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) {
698 ForeachNumInConstantRange(CR, [&](const APInt &N) {
699 Exact = std::max(Exact, N.getMinSignedBits());
702 unsigned ResultCR = CR.getMinSignedBits();
703 EXPECT_EQ(Exact, ResultCR);
706 TEST_F(ConstantRangeTest, losslessSignedTruncationSignext) {
708 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) {
709 unsigned MinBitWidth = CR.getMinSignedBits();
710 if (MinBitWidth == 0) {
711 EXPECT_TRUE(CR.isEmptySet());
714 if (MinBitWidth == Bits)
716 EXPECT_EQ(CR, CR.truncate(MinBitWidth).signExtend(Bits));
720 TEST_F(ConstantRangeTest, SubtractAPInt) {
721 EXPECT_EQ(Full.subtract(APInt(16, 4)), Full);
722 EXPECT_EQ(Empty.subtract(APInt(16, 4)), Empty);
723 EXPECT_EQ(Some.subtract(APInt(16, 4)),
724 ConstantRange(APInt(16, 0x6), APInt(16, 0xaa6)));
725 EXPECT_EQ(Wrap.subtract(APInt(16, 4)),
726 ConstantRange(APInt(16, 0xaa6), APInt(16, 0x6)));
727 EXPECT_EQ(One.subtract(APInt(16, 4)),
728 ConstantRange(APInt(16, 0x6)));
731 TEST_F(ConstantRangeTest, Add) {
732 EXPECT_EQ(Full.add(APInt(16, 4)), Full);
733 EXPECT_EQ(Full.add(Full), Full);
734 EXPECT_EQ(Full.add(Empty), Empty);
735 EXPECT_EQ(Full.add(One), Full);
736 EXPECT_EQ(Full.add(Some), Full);
737 EXPECT_EQ(Full.add(Wrap), Full);
738 EXPECT_EQ(Empty.add(Empty), Empty);
739 EXPECT_EQ(Empty.add(One), Empty);
740 EXPECT_EQ(Empty.add(Some), Empty);
741 EXPECT_EQ(Empty.add(Wrap), Empty);
742 EXPECT_EQ(Empty.add(APInt(16, 4)), Empty);
743 EXPECT_EQ(Some.add(APInt(16, 4)),
744 ConstantRange(APInt(16, 0xe), APInt(16, 0xaae)));
745 EXPECT_EQ(Wrap.add(APInt(16, 4)),
746 ConstantRange(APInt(16, 0xaae), APInt(16, 0xe)));
747 EXPECT_EQ(One.add(APInt(16, 4)),
748 ConstantRange(APInt(16, 0xe)));
751 template <typename Fn1, typename Fn2>
752 static void TestAddWithNoSignedWrapExhaustive(Fn1 RangeFn, Fn2 IntFn) {
754 EnumerateTwoConstantRanges(Bits, [&](const ConstantRange &CR1,
755 const ConstantRange &CR2) {
756 ConstantRange CR = RangeFn(CR1, CR2);
757 SignedOpRangeGatherer R(CR.getBitWidth());
758 bool AllOverflow = true;
759 ForeachNumInConstantRange(CR1, [&](const APInt &N1) {
760 ForeachNumInConstantRange(CR2, [&](const APInt &N2) {
761 bool IsOverflow = false;
762 APInt N = IntFn(IsOverflow, N1, N2);
766 EXPECT_TRUE(CR.contains(N));
771 EXPECT_EQ(CR.isEmptySet(), AllOverflow);
773 if (CR1.isSignWrappedSet() || CR2.isSignWrappedSet())
776 ConstantRange Exact = R.getRange();
777 EXPECT_EQ(Exact, CR);
781 template <typename Fn1, typename Fn2>
782 static void TestAddWithNoUnsignedWrapExhaustive(Fn1 RangeFn, Fn2 IntFn) {
784 EnumerateTwoConstantRanges(Bits, [&](const ConstantRange &CR1,
785 const ConstantRange &CR2) {
786 ConstantRange CR = RangeFn(CR1, CR2);
787 UnsignedOpRangeGatherer R(CR.getBitWidth());
788 bool AllOverflow = true;
789 ForeachNumInConstantRange(CR1, [&](const APInt &N1) {
790 ForeachNumInConstantRange(CR2, [&](const APInt &N2) {
791 bool IsOverflow = false;
792 APInt N = IntFn(IsOverflow, N1, N2);
796 EXPECT_TRUE(CR.contains(N));
801 EXPECT_EQ(CR.isEmptySet(), AllOverflow);
803 if (CR1.isWrappedSet() || CR2.isWrappedSet())
806 ConstantRange Exact = R.getRange();
807 EXPECT_EQ(Exact, CR);
811 template <typename Fn1, typename Fn2, typename Fn3>
812 static void TestAddWithNoSignedUnsignedWrapExhaustive(Fn1 RangeFn,
816 EnumerateTwoConstantRanges(
817 Bits, [&](const ConstantRange &CR1, const ConstantRange &CR2) {
818 ConstantRange CR = RangeFn(CR1, CR2);
819 UnsignedOpRangeGatherer UR(CR.getBitWidth());
820 SignedOpRangeGatherer SR(CR.getBitWidth());
821 bool AllOverflow = true;
822 ForeachNumInConstantRange(CR1, [&](const APInt &N1) {
823 ForeachNumInConstantRange(CR2, [&](const APInt &N2) {
824 bool IsOverflow = false, IsSignedOverflow = false;
825 APInt N = IntFnSigned(IsSignedOverflow, N1, N2);
826 (void) IntFnUnsigned(IsOverflow, N1, N2);
827 if (!IsSignedOverflow && !IsOverflow) {
831 EXPECT_TRUE(CR.contains(N));
836 EXPECT_EQ(CR.isEmptySet(), AllOverflow);
838 if (CR1.isWrappedSet() || CR2.isWrappedSet() ||
839 CR1.isSignWrappedSet() || CR2.isSignWrappedSet())
842 ConstantRange ExactUnsignedCR = UR.getRange();
843 ConstantRange ExactSignedCR = SR.getRange();
845 if (ExactUnsignedCR.isEmptySet() || ExactSignedCR.isEmptySet()) {
846 EXPECT_TRUE(CR.isEmptySet());
850 ConstantRange Exact = ExactSignedCR.intersectWith(ExactUnsignedCR);
851 EXPECT_EQ(Exact, CR);
855 TEST_F(ConstantRangeTest, AddWithNoWrap) {
856 typedef OverflowingBinaryOperator OBO;
857 EXPECT_EQ(Empty.addWithNoWrap(Some, OBO::NoSignedWrap), Empty);
858 EXPECT_EQ(Some.addWithNoWrap(Empty, OBO::NoSignedWrap), Empty);
859 EXPECT_EQ(Full.addWithNoWrap(Full, OBO::NoSignedWrap), Full);
860 EXPECT_NE(Full.addWithNoWrap(Some, OBO::NoSignedWrap), Full);
861 EXPECT_NE(Some.addWithNoWrap(Full, OBO::NoSignedWrap), Full);
862 EXPECT_EQ(Full.addWithNoWrap(ConstantRange(APInt(16, 1), APInt(16, 2)),
864 ConstantRange(APInt(16, INT16_MIN + 1), APInt(16, INT16_MIN)));
865 EXPECT_EQ(ConstantRange(APInt(16, 1), APInt(16, 2))
866 .addWithNoWrap(Full, OBO::NoSignedWrap),
867 ConstantRange(APInt(16, INT16_MIN + 1), APInt(16, INT16_MIN)));
868 EXPECT_EQ(Full.addWithNoWrap(ConstantRange(APInt(16, -1), APInt(16, 0)),
870 ConstantRange(APInt(16, INT16_MIN), APInt(16, INT16_MAX)));
871 EXPECT_EQ(ConstantRange(APInt(8, 100), APInt(8, 120))
872 .addWithNoWrap(ConstantRange(APInt(8, 120), APInt(8, 123)),
874 ConstantRange(8, false));
875 EXPECT_EQ(ConstantRange(APInt(8, -120), APInt(8, -100))
876 .addWithNoWrap(ConstantRange(APInt(8, -110), APInt(8, -100)),
878 ConstantRange(8, false));
879 EXPECT_EQ(ConstantRange(APInt(8, 0), APInt(8, 101))
880 .addWithNoWrap(ConstantRange(APInt(8, -128), APInt(8, 28)),
882 ConstantRange(8, true));
883 EXPECT_EQ(ConstantRange(APInt(8, 0), APInt(8, 101))
884 .addWithNoWrap(ConstantRange(APInt(8, -120), APInt(8, 29)),
886 ConstantRange(APInt(8, -120), APInt(8, -128)));
887 EXPECT_EQ(ConstantRange(APInt(8, -50), APInt(8, 50))
888 .addWithNoWrap(ConstantRange(APInt(8, 10), APInt(8, 20)),
890 ConstantRange(APInt(8, -40), APInt(8, 69)));
891 EXPECT_EQ(ConstantRange(APInt(8, 10), APInt(8, 20))
892 .addWithNoWrap(ConstantRange(APInt(8, -50), APInt(8, 50)),
894 ConstantRange(APInt(8, -40), APInt(8, 69)));
895 EXPECT_EQ(ConstantRange(APInt(8, 120), APInt(8, -10))
896 .addWithNoWrap(ConstantRange(APInt(8, 5), APInt(8, 20)),
898 ConstantRange(APInt(8, 125), APInt(8, 9)));
899 EXPECT_EQ(ConstantRange(APInt(8, 5), APInt(8, 20))
900 .addWithNoWrap(ConstantRange(APInt(8, 120), APInt(8, -10)),
902 ConstantRange(APInt(8, 125), APInt(8, 9)));
904 TestAddWithNoSignedWrapExhaustive(
905 [](const ConstantRange &CR1, const ConstantRange &CR2) {
906 return CR1.addWithNoWrap(CR2, OBO::NoSignedWrap);
908 [](bool &IsOverflow, const APInt &N1, const APInt &N2) {
909 return N1.sadd_ov(N2, IsOverflow);
912 EXPECT_EQ(Empty.addWithNoWrap(Some, OBO::NoUnsignedWrap), Empty);
913 EXPECT_EQ(Some.addWithNoWrap(Empty, OBO::NoUnsignedWrap), Empty);
914 EXPECT_EQ(Full.addWithNoWrap(Full, OBO::NoUnsignedWrap), Full);
915 EXPECT_NE(Full.addWithNoWrap(Some, OBO::NoUnsignedWrap), Full);
916 EXPECT_NE(Some.addWithNoWrap(Full, OBO::NoUnsignedWrap), Full);
917 EXPECT_EQ(Full.addWithNoWrap(ConstantRange(APInt(16, 1), APInt(16, 2)),
918 OBO::NoUnsignedWrap),
919 ConstantRange(APInt(16, 1), APInt(16, 0)));
920 EXPECT_EQ(ConstantRange(APInt(16, 1), APInt(16, 2))
921 .addWithNoWrap(Full, OBO::NoUnsignedWrap),
922 ConstantRange(APInt(16, 1), APInt(16, 0)));
923 EXPECT_EQ(ConstantRange(APInt(8, 200), APInt(8, 220))
924 .addWithNoWrap(ConstantRange(APInt(8, 100), APInt(8, 123)),
925 OBO::NoUnsignedWrap),
926 ConstantRange(8, false));
927 EXPECT_EQ(ConstantRange(APInt(8, 0), APInt(8, 101))
928 .addWithNoWrap(ConstantRange(APInt(8, 0), APInt(8, 156)),
929 OBO::NoUnsignedWrap),
930 ConstantRange(8, true));
931 EXPECT_EQ(ConstantRange(APInt(8, 0), APInt(8, 101))
932 .addWithNoWrap(ConstantRange(APInt(8, 10), APInt(8, 29)),
933 OBO::NoUnsignedWrap),
934 ConstantRange(APInt(8, 10), APInt(8, 129)));
935 EXPECT_EQ(ConstantRange(APInt(8, 20), APInt(8, 10))
936 .addWithNoWrap(ConstantRange(APInt(8, 50), APInt(8, 200)),
937 OBO::NoUnsignedWrap),
938 ConstantRange(APInt(8, 50), APInt(8, 0)));
939 EXPECT_EQ(ConstantRange(APInt(8, 10), APInt(8, 20))
940 .addWithNoWrap(ConstantRange(APInt(8, 50), APInt(8, 200)),
941 OBO::NoUnsignedWrap),
942 ConstantRange(APInt(8, 60), APInt(8, -37)));
943 EXPECT_EQ(ConstantRange(APInt(8, 20), APInt(8, -30))
944 .addWithNoWrap(ConstantRange(APInt(8, 5), APInt(8, 20)),
945 OBO::NoUnsignedWrap),
946 ConstantRange(APInt(8, 25), APInt(8, -11)));
947 EXPECT_EQ(ConstantRange(APInt(8, 5), APInt(8, 20))
948 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, -30)),
949 OBO::NoUnsignedWrap),
950 ConstantRange(APInt(8, 25), APInt(8, -11)));
952 TestAddWithNoUnsignedWrapExhaustive(
953 [](const ConstantRange &CR1, const ConstantRange &CR2) {
954 return CR1.addWithNoWrap(CR2, OBO::NoUnsignedWrap);
956 [](bool &IsOverflow, const APInt &N1, const APInt &N2) {
957 return N1.uadd_ov(N2, IsOverflow);
960 EXPECT_EQ(ConstantRange(APInt(8, 50), APInt(8, 100))
961 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 70)),
963 ConstantRange(APInt(8, 70), APInt(8, -128)));
964 EXPECT_EQ(ConstantRange(APInt(8, 50), APInt(8, 100))
965 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 70)),
966 OBO::NoUnsignedWrap),
967 ConstantRange(APInt(8, 70), APInt(8, 169)));
968 EXPECT_EQ(ConstantRange(APInt(8, 50), APInt(8, 100))
969 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 70)),
970 OBO::NoUnsignedWrap | OBO::NoSignedWrap),
971 ConstantRange(APInt(8, 70), APInt(8, -128)));
973 EXPECT_EQ(ConstantRange(APInt(8, -100), APInt(8, -50))
974 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 30)),
976 ConstantRange(APInt(8, -80), APInt(8, -21)));
977 EXPECT_EQ(ConstantRange(APInt(8, -100), APInt(8, -50))
978 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 30)),
979 OBO::NoUnsignedWrap),
980 ConstantRange(APInt(8, 176), APInt(8, 235)));
981 EXPECT_EQ(ConstantRange(APInt(8, -100), APInt(8, -50))
982 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 30)),
983 OBO::NoUnsignedWrap | OBO::NoSignedWrap),
984 ConstantRange(APInt(8, 176), APInt(8, 235)));
986 TestAddWithNoSignedUnsignedWrapExhaustive(
987 [](const ConstantRange &CR1, const ConstantRange &CR2) {
988 return CR1.addWithNoWrap(CR2, OBO::NoUnsignedWrap | OBO::NoSignedWrap);
990 [](bool &IsOverflow, const APInt &N1, const APInt &N2) {
991 return N1.sadd_ov(N2, IsOverflow);
993 [](bool &IsOverflow, const APInt &N1, const APInt &N2) {
994 return N1.uadd_ov(N2, IsOverflow);
998 TEST_F(ConstantRangeTest, Sub) {
999 EXPECT_EQ(Full.sub(APInt(16, 4)), Full);
1000 EXPECT_EQ(Full.sub(Full), Full);
1001 EXPECT_EQ(Full.sub(Empty), Empty);
1002 EXPECT_EQ(Full.sub(One), Full);
1003 EXPECT_EQ(Full.sub(Some), Full);
1004 EXPECT_EQ(Full.sub(Wrap), Full);
1005 EXPECT_EQ(Empty.sub(Empty), Empty);
1006 EXPECT_EQ(Empty.sub(One), Empty);
1007 EXPECT_EQ(Empty.sub(Some), Empty);
1008 EXPECT_EQ(Empty.sub(Wrap), Empty);
1009 EXPECT_EQ(Empty.sub(APInt(16, 4)), Empty);
1010 EXPECT_EQ(Some.sub(APInt(16, 4)),
1011 ConstantRange(APInt(16, 0x6), APInt(16, 0xaa6)));
1012 EXPECT_EQ(Some.sub(Some),
1013 ConstantRange(APInt(16, 0xf561), APInt(16, 0xaa0)));
1014 EXPECT_EQ(Wrap.sub(APInt(16, 4)),
1015 ConstantRange(APInt(16, 0xaa6), APInt(16, 0x6)));
1016 EXPECT_EQ(One.sub(APInt(16, 4)),
1017 ConstantRange(APInt(16, 0x6)));
1020 TEST_F(ConstantRangeTest, SubWithNoWrap) {
1021 typedef OverflowingBinaryOperator OBO;
1022 TestAddWithNoSignedWrapExhaustive(
1023 [](const ConstantRange &CR1, const ConstantRange &CR2) {
1024 return CR1.subWithNoWrap(CR2, OBO::NoSignedWrap);
1026 [](bool &IsOverflow, const APInt &N1, const APInt &N2) {
1027 return N1.ssub_ov(N2, IsOverflow);
1029 TestAddWithNoUnsignedWrapExhaustive(
1030 [](const ConstantRange &CR1, const ConstantRange &CR2) {
1031 return CR1.subWithNoWrap(CR2, OBO::NoUnsignedWrap);
1033 [](bool &IsOverflow, const APInt &N1, const APInt &N2) {
1034 return N1.usub_ov(N2, IsOverflow);
1036 TestAddWithNoSignedUnsignedWrapExhaustive(
1037 [](const ConstantRange &CR1, const ConstantRange &CR2) {
1038 return CR1.subWithNoWrap(CR2, OBO::NoUnsignedWrap | OBO::NoSignedWrap);
1040 [](bool &IsOverflow, const APInt &N1, const APInt &N2) {
1041 return N1.ssub_ov(N2, IsOverflow);
1043 [](bool &IsOverflow, const APInt &N1, const APInt &N2) {
1044 return N1.usub_ov(N2, IsOverflow);
1048 TEST_F(ConstantRangeTest, Multiply) {
1049 EXPECT_EQ(Full.multiply(Full), Full);
1050 EXPECT_EQ(Full.multiply(Empty), Empty);
1051 EXPECT_EQ(Full.multiply(One), Full);
1052 EXPECT_EQ(Full.multiply(Some), Full);
1053 EXPECT_EQ(Full.multiply(Wrap), Full);
1054 EXPECT_EQ(Empty.multiply(Empty), Empty);
1055 EXPECT_EQ(Empty.multiply(One), Empty);
1056 EXPECT_EQ(Empty.multiply(Some), Empty);
1057 EXPECT_EQ(Empty.multiply(Wrap), Empty);
1058 EXPECT_EQ(One.multiply(One), ConstantRange(APInt(16, 0xa*0xa),
1059 APInt(16, 0xa*0xa + 1)));
1060 EXPECT_EQ(One.multiply(Some), ConstantRange(APInt(16, 0xa*0xa),
1061 APInt(16, 0xa*0xaa9 + 1)));
1062 EXPECT_EQ(One.multiply(Wrap), Full);
1063 EXPECT_EQ(Some.multiply(Some), Full);
1064 EXPECT_EQ(Some.multiply(Wrap), Full);
1065 EXPECT_EQ(Wrap.multiply(Wrap), Full);
1067 ConstantRange Zero(APInt(16, 0));
1068 EXPECT_EQ(Zero.multiply(Full), Zero);
1069 EXPECT_EQ(Zero.multiply(Some), Zero);
1070 EXPECT_EQ(Zero.multiply(Wrap), Zero);
1071 EXPECT_EQ(Full.multiply(Zero), Zero);
1072 EXPECT_EQ(Some.multiply(Zero), Zero);
1073 EXPECT_EQ(Wrap.multiply(Zero), Zero);
1075 // http://llvm.org/PR4545
1076 EXPECT_EQ(ConstantRange(APInt(4, 1), APInt(4, 6)).multiply(
1077 ConstantRange(APInt(4, 6), APInt(4, 2))),
1078 ConstantRange(4, /*isFullSet=*/true));
1080 EXPECT_EQ(ConstantRange(APInt(8, 254), APInt(8, 0)).multiply(
1081 ConstantRange(APInt(8, 252), APInt(8, 4))),
1082 ConstantRange(APInt(8, 250), APInt(8, 9)));
1083 EXPECT_EQ(ConstantRange(APInt(8, 254), APInt(8, 255)).multiply(
1084 ConstantRange(APInt(8, 2), APInt(8, 4))),
1085 ConstantRange(APInt(8, 250), APInt(8, 253)));
1087 // TODO: This should be return [-2, 0]
1088 EXPECT_EQ(ConstantRange(APInt(8, -2)).multiply(
1089 ConstantRange(APInt(8, 0), APInt(8, 2))),
1090 ConstantRange(APInt(8, -2), APInt(8, 1)));
1093 TEST_F(ConstantRangeTest, UMax) {
1094 EXPECT_EQ(Full.umax(Full), Full);
1095 EXPECT_EQ(Full.umax(Empty), Empty);
1096 EXPECT_EQ(Full.umax(Some), ConstantRange(APInt(16, 0xa), APInt(16, 0)));
1097 EXPECT_EQ(Full.umax(Wrap), Full);
1098 EXPECT_EQ(Full.umax(Some), ConstantRange(APInt(16, 0xa), APInt(16, 0)));
1099 EXPECT_EQ(Empty.umax(Empty), Empty);
1100 EXPECT_EQ(Empty.umax(Some), Empty);
1101 EXPECT_EQ(Empty.umax(Wrap), Empty);
1102 EXPECT_EQ(Empty.umax(One), Empty);
1103 EXPECT_EQ(Some.umax(Some), Some);
1104 EXPECT_EQ(Some.umax(Wrap), ConstantRange(APInt(16, 0xa), APInt(16, 0)));
1105 EXPECT_EQ(Some.umax(One), Some);
1106 // TODO: ConstantRange is currently over-conservative here.
1107 EXPECT_EQ(Wrap.umax(Wrap), Full);
1108 EXPECT_EQ(Wrap.umax(One), ConstantRange(APInt(16, 0xa), APInt(16, 0)));
1109 EXPECT_EQ(One.umax(One), One);
1112 TEST_F(ConstantRangeTest, SMax) {
1113 EXPECT_EQ(Full.smax(Full), Full);
1114 EXPECT_EQ(Full.smax(Empty), Empty);
1115 EXPECT_EQ(Full.smax(Some), ConstantRange(APInt(16, 0xa),
1116 APInt::getSignedMinValue(16)));
1117 EXPECT_EQ(Full.smax(Wrap), Full);
1118 EXPECT_EQ(Full.smax(One), ConstantRange(APInt(16, 0xa),
1119 APInt::getSignedMinValue(16)));
1120 EXPECT_EQ(Empty.smax(Empty), Empty);
1121 EXPECT_EQ(Empty.smax(Some), Empty);
1122 EXPECT_EQ(Empty.smax(Wrap), Empty);
1123 EXPECT_EQ(Empty.smax(One), Empty);
1124 EXPECT_EQ(Some.smax(Some), Some);
1125 EXPECT_EQ(Some.smax(Wrap), ConstantRange(APInt(16, 0xa),
1126 APInt(16, (uint64_t)INT16_MIN)));
1127 EXPECT_EQ(Some.smax(One), Some);
1128 EXPECT_EQ(Wrap.smax(One), ConstantRange(APInt(16, 0xa),
1129 APInt(16, (uint64_t)INT16_MIN)));
1130 EXPECT_EQ(One.smax(One), One);
1133 TEST_F(ConstantRangeTest, UMin) {
1134 EXPECT_EQ(Full.umin(Full), Full);
1135 EXPECT_EQ(Full.umin(Empty), Empty);
1136 EXPECT_EQ(Full.umin(Some), ConstantRange(APInt(16, 0), APInt(16, 0xaaa)));
1137 EXPECT_EQ(Full.umin(Wrap), Full);
1138 EXPECT_EQ(Empty.umin(Empty), Empty);
1139 EXPECT_EQ(Empty.umin(Some), Empty);
1140 EXPECT_EQ(Empty.umin(Wrap), Empty);
1141 EXPECT_EQ(Empty.umin(One), Empty);
1142 EXPECT_EQ(Some.umin(Some), Some);
1143 EXPECT_EQ(Some.umin(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xaaa)));
1144 EXPECT_EQ(Some.umin(One), One);
1145 // TODO: ConstantRange is currently over-conservative here.
1146 EXPECT_EQ(Wrap.umin(Wrap), Full);
1147 EXPECT_EQ(Wrap.umin(One), ConstantRange(APInt(16, 0), APInt(16, 0xb)));
1148 EXPECT_EQ(One.umin(One), One);
1151 TEST_F(ConstantRangeTest, SMin) {
1152 EXPECT_EQ(Full.smin(Full), Full);
1153 EXPECT_EQ(Full.smin(Empty), Empty);
1154 EXPECT_EQ(Full.smin(Some), ConstantRange(APInt(16, (uint64_t)INT16_MIN),
1156 EXPECT_EQ(Full.smin(Wrap), Full);
1157 EXPECT_EQ(Empty.smin(Empty), Empty);
1158 EXPECT_EQ(Empty.smin(Some), Empty);
1159 EXPECT_EQ(Empty.smin(Wrap), Empty);
1160 EXPECT_EQ(Empty.smin(One), Empty);
1161 EXPECT_EQ(Some.smin(Some), Some);
1162 EXPECT_EQ(Some.smin(Wrap), ConstantRange(APInt(16, (uint64_t)INT16_MIN),
1164 EXPECT_EQ(Some.smin(One), One);
1165 // TODO: ConstantRange is currently over-conservative here.
1166 EXPECT_EQ(Wrap.smin(Wrap), Full);
1167 EXPECT_EQ(Wrap.smin(One), ConstantRange(APInt(16, (uint64_t)INT16_MIN),
1169 EXPECT_EQ(One.smin(One), One);
1172 TEST_F(ConstantRangeTest, UDiv) {
1173 EXPECT_EQ(Full.udiv(Full), Full);
1174 EXPECT_EQ(Full.udiv(Empty), Empty);
1175 EXPECT_EQ(Full.udiv(One), ConstantRange(APInt(16, 0),
1176 APInt(16, 0xffff / 0xa + 1)));
1177 EXPECT_EQ(Full.udiv(Some), ConstantRange(APInt(16, 0),
1178 APInt(16, 0xffff / 0xa + 1)));
1179 EXPECT_EQ(Full.udiv(Wrap), Full);
1180 EXPECT_EQ(Empty.udiv(Empty), Empty);
1181 EXPECT_EQ(Empty.udiv(One), Empty);
1182 EXPECT_EQ(Empty.udiv(Some), Empty);
1183 EXPECT_EQ(Empty.udiv(Wrap), Empty);
1184 EXPECT_EQ(One.udiv(One), ConstantRange(APInt(16, 1)));
1185 EXPECT_EQ(One.udiv(Some), ConstantRange(APInt(16, 0), APInt(16, 2)));
1186 EXPECT_EQ(One.udiv(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xb)));
1187 EXPECT_EQ(Some.udiv(Some), ConstantRange(APInt(16, 0), APInt(16, 0x111)));
1188 EXPECT_EQ(Some.udiv(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xaaa)));
1189 EXPECT_EQ(Wrap.udiv(Wrap), Full);
1192 ConstantRange Zero(APInt(16, 0));
1193 EXPECT_EQ(Zero.udiv(One), Zero);
1194 EXPECT_EQ(Zero.udiv(Full), Zero);
1196 EXPECT_EQ(ConstantRange(APInt(16, 0), APInt(16, 99)).udiv(Full),
1197 ConstantRange(APInt(16, 0), APInt(16, 99)));
1198 EXPECT_EQ(ConstantRange(APInt(16, 10), APInt(16, 99)).udiv(Full),
1199 ConstantRange(APInt(16, 0), APInt(16, 99)));
1202 TEST_F(ConstantRangeTest, SDiv) {
1204 EnumerateTwoConstantRanges(Bits, [&](const ConstantRange &CR1,
1205 const ConstantRange &CR2) {
1206 // Collect possible results in a bit vector. We store the signed value plus
1207 // a bias to make it unsigned.
1208 int Bias = 1 << (Bits - 1);
1209 BitVector Results(1 << Bits);
1210 ForeachNumInConstantRange(CR1, [&](const APInt &N1) {
1211 ForeachNumInConstantRange(CR2, [&](const APInt &N2) {
1212 // Division by zero is UB.
1216 // SignedMin / -1 is UB.
1217 if (N1.isMinSignedValue() && N2.isAllOnesValue())
1220 APInt N = N1.sdiv(N2);
1221 Results.set(N.getSExtValue() + Bias);
1225 ConstantRange CR = CR1.sdiv(CR2);
1226 if (Results.none()) {
1227 EXPECT_TRUE(CR.isEmptySet());
1231 // If there is a non-full signed envelope, that should be the result.
1232 APInt SMin(Bits, Results.find_first() - Bias);
1233 APInt SMax(Bits, Results.find_last() - Bias);
1234 ConstantRange Envelope = ConstantRange::getNonEmpty(SMin, SMax + 1);
1235 if (!Envelope.isFullSet()) {
1236 EXPECT_EQ(Envelope, CR);
1240 // If the signed envelope is a full set, try to find a smaller sign wrapped
1241 // set that is separated in negative and positive components (or one which
1242 // can also additionally contain zero).
1243 int LastNeg = Results.find_last_in(0, Bias) - Bias;
1244 int LastPos = Results.find_next(Bias) - Bias;
1245 if (Results[Bias]) {
1248 else if (LastPos == 1)
1252 APInt WMax(Bits, LastNeg);
1253 APInt WMin(Bits, LastPos);
1254 ConstantRange Wrapped = ConstantRange::getNonEmpty(WMin, WMax + 1);
1255 EXPECT_EQ(Wrapped, CR);
1259 TEST_F(ConstantRangeTest, URem) {
1260 EXPECT_EQ(Full.urem(Empty), Empty);
1261 EXPECT_EQ(Empty.urem(Full), Empty);
1262 // urem by zero is poison.
1263 EXPECT_EQ(Full.urem(ConstantRange(APInt(16, 0))), Empty);
1264 // urem by full range doesn't contain MaxValue.
1265 EXPECT_EQ(Full.urem(Full), ConstantRange(APInt(16, 0), APInt(16, 0xffff)));
1266 // urem is upper bounded by maximum RHS minus one.
1267 EXPECT_EQ(Full.urem(ConstantRange(APInt(16, 0), APInt(16, 123))),
1268 ConstantRange(APInt(16, 0), APInt(16, 122)));
1269 // urem is upper bounded by maximum LHS.
1270 EXPECT_EQ(ConstantRange(APInt(16, 0), APInt(16, 123)).urem(Full),
1271 ConstantRange(APInt(16, 0), APInt(16, 123)));
1272 // If the LHS is always lower than the RHS, the result is the LHS.
1273 EXPECT_EQ(ConstantRange(APInt(16, 10), APInt(16, 20))
1274 .urem(ConstantRange(APInt(16, 20), APInt(16, 30))),
1275 ConstantRange(APInt(16, 10), APInt(16, 20)));
1276 // It has to be strictly lower, otherwise the top value may wrap to zero.
1277 EXPECT_EQ(ConstantRange(APInt(16, 10), APInt(16, 20))
1278 .urem(ConstantRange(APInt(16, 19), APInt(16, 30))),
1279 ConstantRange(APInt(16, 0), APInt(16, 20)));
1280 // [12, 14] % 10 is [2, 4], but we conservatively compute [0, 9].
1281 EXPECT_EQ(ConstantRange(APInt(16, 12), APInt(16, 15))
1282 .urem(ConstantRange(APInt(16, 10))),
1283 ConstantRange(APInt(16, 0), APInt(16, 10)));
1285 TestUnsignedBinOpExhaustive(
1286 [](const ConstantRange &CR1, const ConstantRange &CR2) {
1287 return CR1.urem(CR2);
1289 [](const APInt &N1, const APInt &N2) {
1292 /* SkipZeroRHS */ true, /* CorrectnessOnly */ true);
1295 TEST_F(ConstantRangeTest, SRem) {
1296 EXPECT_EQ(Full.srem(Empty), Empty);
1297 EXPECT_EQ(Empty.srem(Full), Empty);
1298 // srem by zero is UB.
1299 EXPECT_EQ(Full.srem(ConstantRange(APInt(16, 0))), Empty);
1300 // srem by full range doesn't contain SignedMinValue.
1301 EXPECT_EQ(Full.srem(Full), ConstantRange(APInt::getSignedMinValue(16) + 1,
1302 APInt::getSignedMinValue(16)));
1304 ConstantRange PosMod(APInt(16, 10), APInt(16, 21)); // [10, 20]
1305 ConstantRange NegMod(APInt(16, -20), APInt(16, -9)); // [-20, -10]
1306 ConstantRange IntMinMod(APInt::getSignedMinValue(16));
1308 ConstantRange Expected(16, true);
1310 // srem is bounded by abs(RHS) minus one.
1311 ConstantRange PosLargeLHS(APInt(16, 0), APInt(16, 41));
1312 Expected = ConstantRange(APInt(16, 0), APInt(16, 20));
1313 EXPECT_EQ(PosLargeLHS.srem(PosMod), Expected);
1314 EXPECT_EQ(PosLargeLHS.srem(NegMod), Expected);
1315 ConstantRange NegLargeLHS(APInt(16, -40), APInt(16, 1));
1316 Expected = ConstantRange(APInt(16, -19), APInt(16, 1));
1317 EXPECT_EQ(NegLargeLHS.srem(PosMod), Expected);
1318 EXPECT_EQ(NegLargeLHS.srem(NegMod), Expected);
1319 ConstantRange PosNegLargeLHS(APInt(16, -32), APInt(16, 38));
1320 Expected = ConstantRange(APInt(16, -19), APInt(16, 20));
1321 EXPECT_EQ(PosNegLargeLHS.srem(PosMod), Expected);
1322 EXPECT_EQ(PosNegLargeLHS.srem(NegMod), Expected);
1324 // srem is bounded by LHS.
1325 ConstantRange PosLHS(APInt(16, 0), APInt(16, 16));
1326 EXPECT_EQ(PosLHS.srem(PosMod), PosLHS);
1327 EXPECT_EQ(PosLHS.srem(NegMod), PosLHS);
1328 EXPECT_EQ(PosLHS.srem(IntMinMod), PosLHS);
1329 ConstantRange NegLHS(APInt(16, -15), APInt(16, 1));
1330 EXPECT_EQ(NegLHS.srem(PosMod), NegLHS);
1331 EXPECT_EQ(NegLHS.srem(NegMod), NegLHS);
1332 EXPECT_EQ(NegLHS.srem(IntMinMod), NegLHS);
1333 ConstantRange PosNegLHS(APInt(16, -12), APInt(16, 18));
1334 EXPECT_EQ(PosNegLHS.srem(PosMod), PosNegLHS);
1335 EXPECT_EQ(PosNegLHS.srem(NegMod), PosNegLHS);
1336 EXPECT_EQ(PosNegLHS.srem(IntMinMod), PosNegLHS);
1338 // srem is LHS if it is smaller than RHS.
1339 ConstantRange PosSmallLHS(APInt(16, 3), APInt(16, 8));
1340 EXPECT_EQ(PosSmallLHS.srem(PosMod), PosSmallLHS);
1341 EXPECT_EQ(PosSmallLHS.srem(NegMod), PosSmallLHS);
1342 EXPECT_EQ(PosSmallLHS.srem(IntMinMod), PosSmallLHS);
1343 ConstantRange NegSmallLHS(APInt(16, -7), APInt(16, -2));
1344 EXPECT_EQ(NegSmallLHS.srem(PosMod), NegSmallLHS);
1345 EXPECT_EQ(NegSmallLHS.srem(NegMod), NegSmallLHS);
1346 EXPECT_EQ(NegSmallLHS.srem(IntMinMod), NegSmallLHS);
1347 ConstantRange PosNegSmallLHS(APInt(16, -3), APInt(16, 8));
1348 EXPECT_EQ(PosNegSmallLHS.srem(PosMod), PosNegSmallLHS);
1349 EXPECT_EQ(PosNegSmallLHS.srem(NegMod), PosNegSmallLHS);
1350 EXPECT_EQ(PosNegSmallLHS.srem(IntMinMod), PosNegSmallLHS);
1352 // Example of a suboptimal result:
1353 // [12, 14] srem 10 is [2, 4], but we conservatively compute [0, 9].
1354 EXPECT_EQ(ConstantRange(APInt(16, 12), APInt(16, 15))
1355 .srem(ConstantRange(APInt(16, 10))),
1356 ConstantRange(APInt(16, 0), APInt(16, 10)));
1358 TestSignedBinOpExhaustive(
1359 [](const ConstantRange &CR1, const ConstantRange &CR2) {
1360 return CR1.srem(CR2);
1362 [](const APInt &N1, const APInt &N2) {
1365 /* SkipZeroRHS */ true, /* CorrectnessOnly */ true);
1368 TEST_F(ConstantRangeTest, Shl) {
1369 ConstantRange Some2(APInt(16, 0xfff), APInt(16, 0x8000));
1370 ConstantRange WrapNullMax(APInt(16, 0x1), APInt(16, 0x0));
1371 EXPECT_EQ(Full.shl(Full), Full);
1372 EXPECT_EQ(Full.shl(Empty), Empty);
1373 EXPECT_EQ(Full.shl(One), Full); // TODO: [0, (-1 << 0xa) + 1)
1374 EXPECT_EQ(Full.shl(Some), Full); // TODO: [0, (-1 << 0xa) + 1)
1375 EXPECT_EQ(Full.shl(Wrap), Full);
1376 EXPECT_EQ(Empty.shl(Empty), Empty);
1377 EXPECT_EQ(Empty.shl(One), Empty);
1378 EXPECT_EQ(Empty.shl(Some), Empty);
1379 EXPECT_EQ(Empty.shl(Wrap), Empty);
1380 EXPECT_EQ(One.shl(One), ConstantRange(APInt(16, 0xa << 0xa),
1381 APInt(16, (0xa << 0xa) + 1)));
1382 EXPECT_EQ(One.shl(Some), Full); // TODO: [0xa << 0xa, 0)
1383 EXPECT_EQ(One.shl(Wrap), Full); // TODO: [0xa, 0xa << 14 + 1)
1384 EXPECT_EQ(Some.shl(Some), Full); // TODO: [0xa << 0xa, 0xfc01)
1385 EXPECT_EQ(Some.shl(Wrap), Full); // TODO: [0xa, 0x7ff << 0x5 + 1)
1386 EXPECT_EQ(Wrap.shl(Wrap), Full);
1388 Some2.shl(ConstantRange(APInt(16, 0x1))),
1389 ConstantRange(APInt(16, 0xfff << 0x1), APInt(16, 0x7fff << 0x1) + 1));
1390 EXPECT_EQ(One.shl(WrapNullMax), Full);
1393 TEST_F(ConstantRangeTest, Lshr) {
1394 EXPECT_EQ(Full.lshr(Full), Full);
1395 EXPECT_EQ(Full.lshr(Empty), Empty);
1396 EXPECT_EQ(Full.lshr(One), ConstantRange(APInt(16, 0),
1397 APInt(16, (0xffff >> 0xa) + 1)));
1398 EXPECT_EQ(Full.lshr(Some), ConstantRange(APInt(16, 0),
1399 APInt(16, (0xffff >> 0xa) + 1)));
1400 EXPECT_EQ(Full.lshr(Wrap), Full);
1401 EXPECT_EQ(Empty.lshr(Empty), Empty);
1402 EXPECT_EQ(Empty.lshr(One), Empty);
1403 EXPECT_EQ(Empty.lshr(Some), Empty);
1404 EXPECT_EQ(Empty.lshr(Wrap), Empty);
1405 EXPECT_EQ(One.lshr(One), ConstantRange(APInt(16, 0)));
1406 EXPECT_EQ(One.lshr(Some), ConstantRange(APInt(16, 0)));
1407 EXPECT_EQ(One.lshr(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xb)));
1408 EXPECT_EQ(Some.lshr(Some), ConstantRange(APInt(16, 0),
1409 APInt(16, (0xaaa >> 0xa) + 1)));
1410 EXPECT_EQ(Some.lshr(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xaaa)));
1411 EXPECT_EQ(Wrap.lshr(Wrap), Full);
1414 TEST_F(ConstantRangeTest, Ashr) {
1415 EXPECT_EQ(Full.ashr(Full), Full);
1416 EXPECT_EQ(Full.ashr(Empty), Empty);
1417 EXPECT_EQ(Full.ashr(One), ConstantRange(APInt(16, 0xffe0),
1418 APInt(16, (0x7fff >> 0xa) + 1 )));
1419 ConstantRange Small(APInt(16, 0xa), APInt(16, 0xb));
1420 EXPECT_EQ(Full.ashr(Small), ConstantRange(APInt(16, 0xffe0),
1421 APInt(16, (0x7fff >> 0xa) + 1 )));
1422 EXPECT_EQ(Full.ashr(Some), ConstantRange(APInt(16, 0xffe0),
1423 APInt(16, (0x7fff >> 0xa) + 1 )));
1424 EXPECT_EQ(Full.ashr(Wrap), Full);
1425 EXPECT_EQ(Empty.ashr(Empty), Empty);
1426 EXPECT_EQ(Empty.ashr(One), Empty);
1427 EXPECT_EQ(Empty.ashr(Some), Empty);
1428 EXPECT_EQ(Empty.ashr(Wrap), Empty);
1429 EXPECT_EQ(One.ashr(One), ConstantRange(APInt(16, 0)));
1430 EXPECT_EQ(One.ashr(Some), ConstantRange(APInt(16, 0)));
1431 EXPECT_EQ(One.ashr(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xb)));
1432 EXPECT_EQ(Some.ashr(Some), ConstantRange(APInt(16, 0),
1433 APInt(16, (0xaaa >> 0xa) + 1)));
1434 EXPECT_EQ(Some.ashr(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xaaa)));
1435 EXPECT_EQ(Wrap.ashr(Wrap), Full);
1436 ConstantRange Neg(APInt(16, 0xf3f0, true), APInt(16, 0xf7f8, true));
1437 EXPECT_EQ(Neg.ashr(Small), ConstantRange(APInt(16, 0xfffc, true),
1438 APInt(16, 0xfffe, true)));
1441 TEST(ConstantRange, MakeAllowedICmpRegion) {
1443 ConstantRange SMax = ConstantRange(APInt::getSignedMaxValue(32));
1444 EXPECT_TRUE(ConstantRange::makeAllowedICmpRegion(ICmpInst::ICMP_SGT, SMax)
1448 TEST(ConstantRange, MakeSatisfyingICmpRegion) {
1449 ConstantRange LowHalf(APInt(8, 0), APInt(8, 128));
1450 ConstantRange HighHalf(APInt(8, 128), APInt(8, 0));
1451 ConstantRange EmptySet(8, /* isFullSet = */ false);
1453 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_NE, LowHalf),
1457 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_NE, HighHalf),
1460 EXPECT_TRUE(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_EQ,
1461 HighHalf).isEmptySet());
1463 ConstantRange UnsignedSample(APInt(8, 5), APInt(8, 200));
1465 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_ULT,
1467 ConstantRange(APInt(8, 0), APInt(8, 5)));
1469 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_ULE,
1471 ConstantRange(APInt(8, 0), APInt(8, 6)));
1473 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_UGT,
1475 ConstantRange(APInt(8, 200), APInt(8, 0)));
1477 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_UGE,
1479 ConstantRange(APInt(8, 199), APInt(8, 0)));
1481 ConstantRange SignedSample(APInt(8, -5), APInt(8, 5));
1484 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SLT, SignedSample),
1485 ConstantRange(APInt(8, -128), APInt(8, -5)));
1488 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SLE, SignedSample),
1489 ConstantRange(APInt(8, -128), APInt(8, -4)));
1492 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SGT, SignedSample),
1493 ConstantRange(APInt(8, 5), APInt(8, -128)));
1496 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SGE, SignedSample),
1497 ConstantRange(APInt(8, 4), APInt(8, -128)));
1500 TEST(ConstantRange, MakeGuaranteedNoWrapRegion) {
1501 const int IntMin4Bits = 8;
1502 const int IntMax4Bits = 7;
1503 typedef OverflowingBinaryOperator OBO;
1505 for (int Const : {0, -1, -2, 1, 2, IntMin4Bits, IntMax4Bits}) {
1506 APInt C(4, Const, true /* = isSigned */);
1508 auto NUWRegion = ConstantRange::makeGuaranteedNoWrapRegion(
1509 Instruction::Add, C, OBO::NoUnsignedWrap);
1511 EXPECT_FALSE(NUWRegion.isEmptySet());
1513 auto NSWRegion = ConstantRange::makeGuaranteedNoWrapRegion(
1514 Instruction::Add, C, OBO::NoSignedWrap);
1516 EXPECT_FALSE(NSWRegion.isEmptySet());
1518 for (APInt I = NUWRegion.getLower(), E = NUWRegion.getUpper(); I != E;
1520 bool Overflow = false;
1521 (void)I.uadd_ov(C, Overflow);
1522 EXPECT_FALSE(Overflow);
1525 for (APInt I = NSWRegion.getLower(), E = NSWRegion.getUpper(); I != E;
1527 bool Overflow = false;
1528 (void)I.sadd_ov(C, Overflow);
1529 EXPECT_FALSE(Overflow);
1533 for (int Const : {0, -1, -2, 1, 2, IntMin4Bits, IntMax4Bits}) {
1534 APInt C(4, Const, true /* = isSigned */);
1536 auto NUWRegion = ConstantRange::makeGuaranteedNoWrapRegion(
1537 Instruction::Sub, C, OBO::NoUnsignedWrap);
1539 EXPECT_FALSE(NUWRegion.isEmptySet());
1541 auto NSWRegion = ConstantRange::makeGuaranteedNoWrapRegion(
1542 Instruction::Sub, C, OBO::NoSignedWrap);
1544 EXPECT_FALSE(NSWRegion.isEmptySet());
1546 for (APInt I = NUWRegion.getLower(), E = NUWRegion.getUpper(); I != E;
1548 bool Overflow = false;
1549 (void)I.usub_ov(C, Overflow);
1550 EXPECT_FALSE(Overflow);
1553 for (APInt I = NSWRegion.getLower(), E = NSWRegion.getUpper(); I != E;
1555 bool Overflow = false;
1556 (void)I.ssub_ov(C, Overflow);
1557 EXPECT_FALSE(Overflow);
1561 auto NSWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion(
1562 Instruction::Add, ConstantRange(32, /* isFullSet = */ true),
1564 EXPECT_TRUE(NSWForAllValues.isSingleElement() &&
1565 NSWForAllValues.getSingleElement()->isMinValue());
1567 NSWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion(
1568 Instruction::Sub, ConstantRange(32, /* isFullSet = */ true),
1570 EXPECT_TRUE(NSWForAllValues.isSingleElement() &&
1571 NSWForAllValues.getSingleElement()->isMaxValue());
1573 auto NUWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion(
1574 Instruction::Add, ConstantRange(32, /* isFullSet = */ true),
1575 OBO::NoUnsignedWrap);
1576 EXPECT_TRUE(NUWForAllValues.isSingleElement() &&
1577 NUWForAllValues.getSingleElement()->isMinValue());
1579 NUWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion(
1580 Instruction::Sub, ConstantRange(32, /* isFullSet = */ true),
1581 OBO::NoUnsignedWrap);
1582 EXPECT_TRUE(NUWForAllValues.isSingleElement() &&
1583 NUWForAllValues.getSingleElement()->isMaxValue());
1585 EXPECT_TRUE(ConstantRange::makeGuaranteedNoWrapRegion(
1586 Instruction::Add, APInt(32, 0), OBO::NoUnsignedWrap).isFullSet());
1587 EXPECT_TRUE(ConstantRange::makeGuaranteedNoWrapRegion(
1588 Instruction::Add, APInt(32, 0), OBO::NoSignedWrap).isFullSet());
1589 EXPECT_TRUE(ConstantRange::makeGuaranteedNoWrapRegion(
1590 Instruction::Sub, APInt(32, 0), OBO::NoUnsignedWrap).isFullSet());
1591 EXPECT_TRUE(ConstantRange::makeGuaranteedNoWrapRegion(
1592 Instruction::Sub, APInt(32, 0), OBO::NoSignedWrap).isFullSet());
1594 ConstantRange OneToFive(APInt(32, 1), APInt(32, 6));
1595 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1596 Instruction::Add, OneToFive, OBO::NoSignedWrap),
1597 ConstantRange(APInt::getSignedMinValue(32),
1598 APInt::getSignedMaxValue(32) - 4));
1599 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1600 Instruction::Add, OneToFive, OBO::NoUnsignedWrap),
1601 ConstantRange(APInt::getMinValue(32), APInt::getMinValue(32) - 5));
1602 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1603 Instruction::Sub, OneToFive, OBO::NoSignedWrap),
1604 ConstantRange(APInt::getSignedMinValue(32) + 5,
1605 APInt::getSignedMinValue(32)));
1606 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1607 Instruction::Sub, OneToFive, OBO::NoUnsignedWrap),
1608 ConstantRange(APInt::getMinValue(32) + 5, APInt::getMinValue(32)));
1610 ConstantRange MinusFiveToMinusTwo(APInt(32, -5), APInt(32, -1));
1611 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1612 Instruction::Add, MinusFiveToMinusTwo, OBO::NoSignedWrap),
1613 ConstantRange(APInt::getSignedMinValue(32) + 5,
1614 APInt::getSignedMinValue(32)));
1615 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1616 Instruction::Add, MinusFiveToMinusTwo, OBO::NoUnsignedWrap),
1617 ConstantRange(APInt(32, 0), APInt(32, 2)));
1618 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1619 Instruction::Sub, MinusFiveToMinusTwo, OBO::NoSignedWrap),
1620 ConstantRange(APInt::getSignedMinValue(32),
1621 APInt::getSignedMaxValue(32) - 4));
1622 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1623 Instruction::Sub, MinusFiveToMinusTwo, OBO::NoUnsignedWrap),
1624 ConstantRange(APInt::getMaxValue(32) - 1,
1625 APInt::getMinValue(32)));
1627 ConstantRange MinusOneToOne(APInt(32, -1), APInt(32, 2));
1628 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1629 Instruction::Add, MinusOneToOne, OBO::NoSignedWrap),
1630 ConstantRange(APInt::getSignedMinValue(32) + 1,
1631 APInt::getSignedMinValue(32) - 1));
1632 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1633 Instruction::Add, MinusOneToOne, OBO::NoUnsignedWrap),
1634 ConstantRange(APInt(32, 0), APInt(32, 1)));
1635 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1636 Instruction::Sub, MinusOneToOne, OBO::NoSignedWrap),
1637 ConstantRange(APInt::getSignedMinValue(32) + 1,
1638 APInt::getSignedMinValue(32) - 1));
1639 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1640 Instruction::Sub, MinusOneToOne, OBO::NoUnsignedWrap),
1641 ConstantRange(APInt::getMaxValue(32),
1642 APInt::getMinValue(32)));
1644 ConstantRange One(APInt(32, 1), APInt(32, 2));
1645 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1646 Instruction::Add, One, OBO::NoSignedWrap),
1647 ConstantRange(APInt::getSignedMinValue(32),
1648 APInt::getSignedMaxValue(32)));
1649 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1650 Instruction::Add, One, OBO::NoUnsignedWrap),
1651 ConstantRange(APInt::getMinValue(32), APInt::getMaxValue(32)));
1652 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1653 Instruction::Sub, One, OBO::NoSignedWrap),
1654 ConstantRange(APInt::getSignedMinValue(32) + 1,
1655 APInt::getSignedMinValue(32)));
1656 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1657 Instruction::Sub, One, OBO::NoUnsignedWrap),
1658 ConstantRange(APInt::getMinValue(32) + 1, APInt::getMinValue(32)));
1660 ConstantRange OneLessThanBitWidth(APInt(32, 0), APInt(32, 31) + 1);
1661 ConstantRange UpToBitWidth(APInt(32, 0), APInt(32, 32) + 1);
1662 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1663 Instruction::Shl, UpToBitWidth, OBO::NoUnsignedWrap),
1664 ConstantRange::makeGuaranteedNoWrapRegion(
1665 Instruction::Shl, OneLessThanBitWidth, OBO::NoUnsignedWrap));
1666 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1667 Instruction::Shl, UpToBitWidth, OBO::NoSignedWrap),
1668 ConstantRange::makeGuaranteedNoWrapRegion(
1669 Instruction::Shl, OneLessThanBitWidth, OBO::NoSignedWrap));
1670 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1671 Instruction::Shl, UpToBitWidth, OBO::NoUnsignedWrap),
1672 ConstantRange(APInt(32, 0), APInt(32, 1) + 1));
1673 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1674 Instruction::Shl, UpToBitWidth, OBO::NoSignedWrap),
1675 ConstantRange(APInt(32, -1), APInt(32, 0) + 1));
1678 ConstantRange::makeGuaranteedNoWrapRegion(
1679 Instruction::Shl, ConstantRange::getFull(32), OBO::NoUnsignedWrap),
1680 ConstantRange::makeGuaranteedNoWrapRegion(
1681 Instruction::Shl, OneLessThanBitWidth, OBO::NoUnsignedWrap));
1683 ConstantRange::makeGuaranteedNoWrapRegion(
1684 Instruction::Shl, ConstantRange::getFull(32), OBO::NoSignedWrap),
1685 ConstantRange::makeGuaranteedNoWrapRegion(
1686 Instruction::Shl, OneLessThanBitWidth, OBO::NoSignedWrap));
1688 ConstantRange IllegalShAmt(APInt(32, 32), APInt(32, 0) + 1);
1689 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1690 Instruction::Shl, IllegalShAmt, OBO::NoUnsignedWrap),
1691 ConstantRange::getFull(32));
1692 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1693 Instruction::Shl, IllegalShAmt, OBO::NoSignedWrap),
1694 ConstantRange::getFull(32));
1697 ConstantRange::makeGuaranteedNoWrapRegion(
1698 Instruction::Shl, ConstantRange(APInt(32, -32), APInt(32, 16) + 1),
1699 OBO::NoUnsignedWrap),
1700 ConstantRange::makeGuaranteedNoWrapRegion(
1701 Instruction::Shl, ConstantRange(APInt(32, 0), APInt(32, 16) + 1),
1702 OBO::NoUnsignedWrap));
1704 ConstantRange::makeGuaranteedNoWrapRegion(
1705 Instruction::Shl, ConstantRange(APInt(32, -32), APInt(32, 16) + 1),
1707 ConstantRange::makeGuaranteedNoWrapRegion(
1708 Instruction::Shl, ConstantRange(APInt(32, 0), APInt(32, 16) + 1),
1709 OBO::NoSignedWrap));
1711 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1713 ConstantRange(APInt(32, -32), APInt(32, 16) + 1),
1714 OBO::NoUnsignedWrap),
1715 ConstantRange(APInt(32, 0), APInt(32, 65535) + 1));
1716 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion(
1718 ConstantRange(APInt(32, -32), APInt(32, 16) + 1),
1720 ConstantRange(APInt(32, -32768), APInt(32, 32767) + 1));
1723 template<typename Fn>
1724 void TestNoWrapRegionExhaustive(Instruction::BinaryOps BinOp,
1725 unsigned NoWrapKind, Fn OverflowFn) {
1727 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) {
1728 if (CR.isEmptySet())
1730 if (Instruction::isShift(BinOp) && CR.getUnsignedMax().uge(Bits))
1733 ConstantRange NoWrap =
1734 ConstantRange::makeGuaranteedNoWrapRegion(BinOp, CR, NoWrapKind);
1735 ConstantRange Full = ConstantRange::getFull(Bits);
1736 ForeachNumInConstantRange(Full, [&](const APInt &N1) {
1737 bool NoOverflow = true;
1738 bool Overflow = true;
1739 ForeachNumInConstantRange(CR, [&](const APInt &N2) {
1740 if (OverflowFn(N1, N2))
1745 EXPECT_EQ(NoOverflow, NoWrap.contains(N1));
1747 // The no-wrap range is exact for single-element ranges.
1748 if (CR.isSingleElement()) {
1749 EXPECT_EQ(Overflow, !NoWrap.contains(N1));
1755 // Show that makeGuaranteedNoWrapRegion() is maximal, and for single-element
1756 // ranges also exact.
1757 TEST(ConstantRange, NoWrapRegionExhaustive) {
1758 TestNoWrapRegionExhaustive(
1759 Instruction::Add, OverflowingBinaryOperator::NoUnsignedWrap,
1760 [](const APInt &N1, const APInt &N2) {
1762 (void) N1.uadd_ov(N2, Overflow);
1765 TestNoWrapRegionExhaustive(
1766 Instruction::Add, OverflowingBinaryOperator::NoSignedWrap,
1767 [](const APInt &N1, const APInt &N2) {
1769 (void) N1.sadd_ov(N2, Overflow);
1772 TestNoWrapRegionExhaustive(
1773 Instruction::Sub, OverflowingBinaryOperator::NoUnsignedWrap,
1774 [](const APInt &N1, const APInt &N2) {
1776 (void) N1.usub_ov(N2, Overflow);
1779 TestNoWrapRegionExhaustive(
1780 Instruction::Sub, OverflowingBinaryOperator::NoSignedWrap,
1781 [](const APInt &N1, const APInt &N2) {
1783 (void) N1.ssub_ov(N2, Overflow);
1786 TestNoWrapRegionExhaustive(
1787 Instruction::Mul, OverflowingBinaryOperator::NoUnsignedWrap,
1788 [](const APInt &N1, const APInt &N2) {
1790 (void) N1.umul_ov(N2, Overflow);
1793 TestNoWrapRegionExhaustive(
1794 Instruction::Mul, OverflowingBinaryOperator::NoSignedWrap,
1795 [](const APInt &N1, const APInt &N2) {
1797 (void) N1.smul_ov(N2, Overflow);
1800 TestNoWrapRegionExhaustive(Instruction::Shl,
1801 OverflowingBinaryOperator::NoUnsignedWrap,
1802 [](const APInt &N1, const APInt &N2) {
1804 (void)N1.ushl_ov(N2, Overflow);
1807 TestNoWrapRegionExhaustive(Instruction::Shl,
1808 OverflowingBinaryOperator::NoSignedWrap,
1809 [](const APInt &N1, const APInt &N2) {
1811 (void)N1.sshl_ov(N2, Overflow);
1816 TEST(ConstantRange, GetEquivalentICmp) {
1818 CmpInst::Predicate Pred;
1820 EXPECT_TRUE(ConstantRange(APInt::getMinValue(32), APInt(32, 100))
1821 .getEquivalentICmp(Pred, RHS));
1822 EXPECT_EQ(Pred, CmpInst::ICMP_ULT);
1823 EXPECT_EQ(RHS, APInt(32, 100));
1825 EXPECT_TRUE(ConstantRange(APInt::getSignedMinValue(32), APInt(32, 100))
1826 .getEquivalentICmp(Pred, RHS));
1827 EXPECT_EQ(Pred, CmpInst::ICMP_SLT);
1828 EXPECT_EQ(RHS, APInt(32, 100));
1830 EXPECT_TRUE(ConstantRange(APInt(32, 100), APInt::getMinValue(32))
1831 .getEquivalentICmp(Pred, RHS));
1832 EXPECT_EQ(Pred, CmpInst::ICMP_UGE);
1833 EXPECT_EQ(RHS, APInt(32, 100));
1835 EXPECT_TRUE(ConstantRange(APInt(32, 100), APInt::getSignedMinValue(32))
1836 .getEquivalentICmp(Pred, RHS));
1837 EXPECT_EQ(Pred, CmpInst::ICMP_SGE);
1838 EXPECT_EQ(RHS, APInt(32, 100));
1841 ConstantRange(32, /*isFullSet=*/true).getEquivalentICmp(Pred, RHS));
1842 EXPECT_EQ(Pred, CmpInst::ICMP_UGE);
1843 EXPECT_EQ(RHS, APInt(32, 0));
1846 ConstantRange(32, /*isFullSet=*/false).getEquivalentICmp(Pred, RHS));
1847 EXPECT_EQ(Pred, CmpInst::ICMP_ULT);
1848 EXPECT_EQ(RHS, APInt(32, 0));
1850 EXPECT_FALSE(ConstantRange(APInt(32, 100), APInt(32, 200))
1851 .getEquivalentICmp(Pred, RHS));
1853 EXPECT_FALSE(ConstantRange(APInt::getSignedMinValue(32) - APInt(32, 100),
1854 APInt::getSignedMinValue(32) + APInt(32, 100))
1855 .getEquivalentICmp(Pred, RHS));
1857 EXPECT_FALSE(ConstantRange(APInt::getMinValue(32) - APInt(32, 100),
1858 APInt::getMinValue(32) + APInt(32, 100))
1859 .getEquivalentICmp(Pred, RHS));
1861 EXPECT_TRUE(ConstantRange(APInt(32, 100)).getEquivalentICmp(Pred, RHS));
1862 EXPECT_EQ(Pred, CmpInst::ICMP_EQ);
1863 EXPECT_EQ(RHS, APInt(32, 100));
1866 ConstantRange(APInt(32, 100)).inverse().getEquivalentICmp(Pred, RHS));
1867 EXPECT_EQ(Pred, CmpInst::ICMP_NE);
1868 EXPECT_EQ(RHS, APInt(32, 100));
1871 ConstantRange(APInt(512, 100)).inverse().getEquivalentICmp(Pred, RHS));
1872 EXPECT_EQ(Pred, CmpInst::ICMP_NE);
1873 EXPECT_EQ(RHS, APInt(512, 100));
1875 // NB! It would be correct for the following four calls to getEquivalentICmp
1876 // to return ordered predicates like CmpInst::ICMP_ULT or CmpInst::ICMP_UGT.
1877 // However, that's not the case today.
1879 EXPECT_TRUE(ConstantRange(APInt(32, 0)).getEquivalentICmp(Pred, RHS));
1880 EXPECT_EQ(Pred, CmpInst::ICMP_EQ);
1881 EXPECT_EQ(RHS, APInt(32, 0));
1884 ConstantRange(APInt(32, 0)).inverse().getEquivalentICmp(Pred, RHS));
1885 EXPECT_EQ(Pred, CmpInst::ICMP_NE);
1886 EXPECT_EQ(RHS, APInt(32, 0));
1888 EXPECT_TRUE(ConstantRange(APInt(32, -1)).getEquivalentICmp(Pred, RHS));
1889 EXPECT_EQ(Pred, CmpInst::ICMP_EQ);
1890 EXPECT_EQ(RHS, APInt(32, -1));
1893 ConstantRange(APInt(32, -1)).inverse().getEquivalentICmp(Pred, RHS));
1894 EXPECT_EQ(Pred, CmpInst::ICMP_NE);
1895 EXPECT_EQ(RHS, APInt(32, -1));
1898 #define EXPECT_MAY_OVERFLOW(op) \
1899 EXPECT_EQ(ConstantRange::OverflowResult::MayOverflow, (op))
1900 #define EXPECT_ALWAYS_OVERFLOWS_LOW(op) \
1901 EXPECT_EQ(ConstantRange::OverflowResult::AlwaysOverflowsLow, (op))
1902 #define EXPECT_ALWAYS_OVERFLOWS_HIGH(op) \
1903 EXPECT_EQ(ConstantRange::OverflowResult::AlwaysOverflowsHigh, (op))
1904 #define EXPECT_NEVER_OVERFLOWS(op) \
1905 EXPECT_EQ(ConstantRange::OverflowResult::NeverOverflows, (op))
1907 TEST_F(ConstantRangeTest, UnsignedAddOverflow) {
1908 // Ill-defined - may overflow is a conservative result.
1909 EXPECT_MAY_OVERFLOW(Some.unsignedAddMayOverflow(Empty));
1910 EXPECT_MAY_OVERFLOW(Empty.unsignedAddMayOverflow(Some));
1912 // Never overflow despite one full/wrap set.
1913 ConstantRange Zero(APInt::getNullValue(16));
1914 EXPECT_NEVER_OVERFLOWS(Full.unsignedAddMayOverflow(Zero));
1915 EXPECT_NEVER_OVERFLOWS(Wrap.unsignedAddMayOverflow(Zero));
1916 EXPECT_NEVER_OVERFLOWS(Zero.unsignedAddMayOverflow(Full));
1917 EXPECT_NEVER_OVERFLOWS(Zero.unsignedAddMayOverflow(Wrap));
1919 // But usually full/wrap always may overflow.
1920 EXPECT_MAY_OVERFLOW(Full.unsignedAddMayOverflow(One));
1921 EXPECT_MAY_OVERFLOW(Wrap.unsignedAddMayOverflow(One));
1922 EXPECT_MAY_OVERFLOW(One.unsignedAddMayOverflow(Full));
1923 EXPECT_MAY_OVERFLOW(One.unsignedAddMayOverflow(Wrap));
1925 ConstantRange A(APInt(16, 0xfd00), APInt(16, 0xfe00));
1926 ConstantRange B1(APInt(16, 0x0100), APInt(16, 0x0201));
1927 ConstantRange B2(APInt(16, 0x0100), APInt(16, 0x0202));
1928 EXPECT_NEVER_OVERFLOWS(A.unsignedAddMayOverflow(B1));
1929 EXPECT_MAY_OVERFLOW(A.unsignedAddMayOverflow(B2));
1930 EXPECT_NEVER_OVERFLOWS(B1.unsignedAddMayOverflow(A));
1931 EXPECT_MAY_OVERFLOW(B2.unsignedAddMayOverflow(A));
1933 ConstantRange C1(APInt(16, 0x0299), APInt(16, 0x0400));
1934 ConstantRange C2(APInt(16, 0x0300), APInt(16, 0x0400));
1935 EXPECT_MAY_OVERFLOW(A.unsignedAddMayOverflow(C1));
1936 EXPECT_ALWAYS_OVERFLOWS_HIGH(A.unsignedAddMayOverflow(C2));
1937 EXPECT_MAY_OVERFLOW(C1.unsignedAddMayOverflow(A));
1938 EXPECT_ALWAYS_OVERFLOWS_HIGH(C2.unsignedAddMayOverflow(A));
1941 TEST_F(ConstantRangeTest, UnsignedSubOverflow) {
1942 // Ill-defined - may overflow is a conservative result.
1943 EXPECT_MAY_OVERFLOW(Some.unsignedSubMayOverflow(Empty));
1944 EXPECT_MAY_OVERFLOW(Empty.unsignedSubMayOverflow(Some));
1946 // Never overflow despite one full/wrap set.
1947 ConstantRange Zero(APInt::getNullValue(16));
1948 ConstantRange Max(APInt::getAllOnesValue(16));
1949 EXPECT_NEVER_OVERFLOWS(Full.unsignedSubMayOverflow(Zero));
1950 EXPECT_NEVER_OVERFLOWS(Wrap.unsignedSubMayOverflow(Zero));
1951 EXPECT_NEVER_OVERFLOWS(Max.unsignedSubMayOverflow(Full));
1952 EXPECT_NEVER_OVERFLOWS(Max.unsignedSubMayOverflow(Wrap));
1954 // But usually full/wrap always may overflow.
1955 EXPECT_MAY_OVERFLOW(Full.unsignedSubMayOverflow(One));
1956 EXPECT_MAY_OVERFLOW(Wrap.unsignedSubMayOverflow(One));
1957 EXPECT_MAY_OVERFLOW(One.unsignedSubMayOverflow(Full));
1958 EXPECT_MAY_OVERFLOW(One.unsignedSubMayOverflow(Wrap));
1960 ConstantRange A(APInt(16, 0x0000), APInt(16, 0x0100));
1961 ConstantRange B(APInt(16, 0x0100), APInt(16, 0x0200));
1962 EXPECT_NEVER_OVERFLOWS(B.unsignedSubMayOverflow(A));
1963 EXPECT_ALWAYS_OVERFLOWS_LOW(A.unsignedSubMayOverflow(B));
1965 ConstantRange A1(APInt(16, 0x0000), APInt(16, 0x0101));
1966 ConstantRange B1(APInt(16, 0x0100), APInt(16, 0x0201));
1967 EXPECT_NEVER_OVERFLOWS(B1.unsignedSubMayOverflow(A1));
1968 EXPECT_MAY_OVERFLOW(A1.unsignedSubMayOverflow(B1));
1970 ConstantRange A2(APInt(16, 0x0000), APInt(16, 0x0102));
1971 ConstantRange B2(APInt(16, 0x0100), APInt(16, 0x0202));
1972 EXPECT_MAY_OVERFLOW(B2.unsignedSubMayOverflow(A2));
1973 EXPECT_MAY_OVERFLOW(A2.unsignedSubMayOverflow(B2));
1976 TEST_F(ConstantRangeTest, SignedAddOverflow) {
1977 // Ill-defined - may overflow is a conservative result.
1978 EXPECT_MAY_OVERFLOW(Some.signedAddMayOverflow(Empty));
1979 EXPECT_MAY_OVERFLOW(Empty.signedAddMayOverflow(Some));
1981 // Never overflow despite one full/wrap set.
1982 ConstantRange Zero(APInt::getNullValue(16));
1983 EXPECT_NEVER_OVERFLOWS(Full.signedAddMayOverflow(Zero));
1984 EXPECT_NEVER_OVERFLOWS(Wrap.signedAddMayOverflow(Zero));
1985 EXPECT_NEVER_OVERFLOWS(Zero.signedAddMayOverflow(Full));
1986 EXPECT_NEVER_OVERFLOWS(Zero.signedAddMayOverflow(Wrap));
1988 // But usually full/wrap always may overflow.
1989 EXPECT_MAY_OVERFLOW(Full.signedAddMayOverflow(One));
1990 EXPECT_MAY_OVERFLOW(Wrap.signedAddMayOverflow(One));
1991 EXPECT_MAY_OVERFLOW(One.signedAddMayOverflow(Full));
1992 EXPECT_MAY_OVERFLOW(One.signedAddMayOverflow(Wrap));
1994 ConstantRange A(APInt(16, 0x7d00), APInt(16, 0x7e00));
1995 ConstantRange B1(APInt(16, 0x0100), APInt(16, 0x0201));
1996 ConstantRange B2(APInt(16, 0x0100), APInt(16, 0x0202));
1997 EXPECT_NEVER_OVERFLOWS(A.signedAddMayOverflow(B1));
1998 EXPECT_MAY_OVERFLOW(A.signedAddMayOverflow(B2));
1999 ConstantRange B3(APInt(16, 0x8000), APInt(16, 0x0201));
2000 ConstantRange B4(APInt(16, 0x8000), APInt(16, 0x0202));
2001 EXPECT_NEVER_OVERFLOWS(A.signedAddMayOverflow(B3));
2002 EXPECT_MAY_OVERFLOW(A.signedAddMayOverflow(B4));
2003 ConstantRange B5(APInt(16, 0x0299), APInt(16, 0x0400));
2004 ConstantRange B6(APInt(16, 0x0300), APInt(16, 0x0400));
2005 EXPECT_MAY_OVERFLOW(A.signedAddMayOverflow(B5));
2006 EXPECT_ALWAYS_OVERFLOWS_HIGH(A.signedAddMayOverflow(B6));
2008 ConstantRange C(APInt(16, 0x8200), APInt(16, 0x8300));
2009 ConstantRange D1(APInt(16, 0xfe00), APInt(16, 0xff00));
2010 ConstantRange D2(APInt(16, 0xfd99), APInt(16, 0xff00));
2011 EXPECT_NEVER_OVERFLOWS(C.signedAddMayOverflow(D1));
2012 EXPECT_MAY_OVERFLOW(C.signedAddMayOverflow(D2));
2013 ConstantRange D3(APInt(16, 0xfe00), APInt(16, 0x8000));
2014 ConstantRange D4(APInt(16, 0xfd99), APInt(16, 0x8000));
2015 EXPECT_NEVER_OVERFLOWS(C.signedAddMayOverflow(D3));
2016 EXPECT_MAY_OVERFLOW(C.signedAddMayOverflow(D4));
2017 ConstantRange D5(APInt(16, 0xfc00), APInt(16, 0xfd02));
2018 ConstantRange D6(APInt(16, 0xfc00), APInt(16, 0xfd01));
2019 EXPECT_MAY_OVERFLOW(C.signedAddMayOverflow(D5));
2020 EXPECT_ALWAYS_OVERFLOWS_LOW(C.signedAddMayOverflow(D6));
2022 ConstantRange E(APInt(16, 0xff00), APInt(16, 0x0100));
2023 EXPECT_NEVER_OVERFLOWS(E.signedAddMayOverflow(E));
2024 ConstantRange F(APInt(16, 0xf000), APInt(16, 0x7000));
2025 EXPECT_MAY_OVERFLOW(F.signedAddMayOverflow(F));
2028 TEST_F(ConstantRangeTest, SignedSubOverflow) {
2029 // Ill-defined - may overflow is a conservative result.
2030 EXPECT_MAY_OVERFLOW(Some.signedSubMayOverflow(Empty));
2031 EXPECT_MAY_OVERFLOW(Empty.signedSubMayOverflow(Some));
2033 // Never overflow despite one full/wrap set.
2034 ConstantRange Zero(APInt::getNullValue(16));
2035 EXPECT_NEVER_OVERFLOWS(Full.signedSubMayOverflow(Zero));
2036 EXPECT_NEVER_OVERFLOWS(Wrap.signedSubMayOverflow(Zero));
2038 // But usually full/wrap always may overflow.
2039 EXPECT_MAY_OVERFLOW(Full.signedSubMayOverflow(One));
2040 EXPECT_MAY_OVERFLOW(Wrap.signedSubMayOverflow(One));
2041 EXPECT_MAY_OVERFLOW(One.signedSubMayOverflow(Full));
2042 EXPECT_MAY_OVERFLOW(One.signedSubMayOverflow(Wrap));
2044 ConstantRange A(APInt(16, 0x7d00), APInt(16, 0x7e00));
2045 ConstantRange B1(APInt(16, 0xfe00), APInt(16, 0xff00));
2046 ConstantRange B2(APInt(16, 0xfd99), APInt(16, 0xff00));
2047 EXPECT_NEVER_OVERFLOWS(A.signedSubMayOverflow(B1));
2048 EXPECT_MAY_OVERFLOW(A.signedSubMayOverflow(B2));
2049 ConstantRange B3(APInt(16, 0xfc00), APInt(16, 0xfd02));
2050 ConstantRange B4(APInt(16, 0xfc00), APInt(16, 0xfd01));
2051 EXPECT_MAY_OVERFLOW(A.signedSubMayOverflow(B3));
2052 EXPECT_ALWAYS_OVERFLOWS_HIGH(A.signedSubMayOverflow(B4));
2054 ConstantRange C(APInt(16, 0x8200), APInt(16, 0x8300));
2055 ConstantRange D1(APInt(16, 0x0100), APInt(16, 0x0201));
2056 ConstantRange D2(APInt(16, 0x0100), APInt(16, 0x0202));
2057 EXPECT_NEVER_OVERFLOWS(C.signedSubMayOverflow(D1));
2058 EXPECT_MAY_OVERFLOW(C.signedSubMayOverflow(D2));
2059 ConstantRange D3(APInt(16, 0x0299), APInt(16, 0x0400));
2060 ConstantRange D4(APInt(16, 0x0300), APInt(16, 0x0400));
2061 EXPECT_MAY_OVERFLOW(C.signedSubMayOverflow(D3));
2062 EXPECT_ALWAYS_OVERFLOWS_LOW(C.signedSubMayOverflow(D4));
2064 ConstantRange E(APInt(16, 0xff00), APInt(16, 0x0100));
2065 EXPECT_NEVER_OVERFLOWS(E.signedSubMayOverflow(E));
2066 ConstantRange F(APInt(16, 0xf000), APInt(16, 0x7001));
2067 EXPECT_MAY_OVERFLOW(F.signedSubMayOverflow(F));
2070 template<typename Fn1, typename Fn2>
2071 static void TestOverflowExhaustive(Fn1 OverflowFn, Fn2 MayOverflowFn) {
2072 // Constant range overflow checks are tested exhaustively on 4-bit numbers.
2074 EnumerateTwoConstantRanges(Bits, [=](const ConstantRange &CR1,
2075 const ConstantRange &CR2) {
2076 // Loop over all N1 in CR1 and N2 in CR2 and check whether any of the
2077 // operations have overflow / have no overflow.
2078 bool RangeHasOverflowLow = false;
2079 bool RangeHasOverflowHigh = false;
2080 bool RangeHasNoOverflow = false;
2081 ForeachNumInConstantRange(CR1, [&](const APInt &N1) {
2082 ForeachNumInConstantRange(CR2, [&](const APInt &N2) {
2083 bool IsOverflowHigh;
2084 if (!OverflowFn(IsOverflowHigh, N1, N2)) {
2085 RangeHasNoOverflow = true;
2090 RangeHasOverflowHigh = true;
2092 RangeHasOverflowLow = true;
2096 ConstantRange::OverflowResult OR = MayOverflowFn(CR1, CR2);
2098 case ConstantRange::OverflowResult::AlwaysOverflowsLow:
2099 EXPECT_TRUE(RangeHasOverflowLow);
2100 EXPECT_FALSE(RangeHasOverflowHigh);
2101 EXPECT_FALSE(RangeHasNoOverflow);
2103 case ConstantRange::OverflowResult::AlwaysOverflowsHigh:
2104 EXPECT_TRUE(RangeHasOverflowHigh);
2105 EXPECT_FALSE(RangeHasOverflowLow);
2106 EXPECT_FALSE(RangeHasNoOverflow);
2108 case ConstantRange::OverflowResult::NeverOverflows:
2109 EXPECT_FALSE(RangeHasOverflowLow);
2110 EXPECT_FALSE(RangeHasOverflowHigh);
2111 EXPECT_TRUE(RangeHasNoOverflow);
2113 case ConstantRange::OverflowResult::MayOverflow:
2114 // We return MayOverflow for empty sets as a conservative result,
2115 // but of course neither the RangeHasOverflow nor the
2116 // RangeHasNoOverflow flags will be set.
2117 if (CR1.isEmptySet() || CR2.isEmptySet())
2120 EXPECT_TRUE(RangeHasOverflowLow || RangeHasOverflowHigh);
2121 EXPECT_TRUE(RangeHasNoOverflow);
2127 TEST_F(ConstantRangeTest, UnsignedAddOverflowExhaustive) {
2128 TestOverflowExhaustive(
2129 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) {
2131 (void) N1.uadd_ov(N2, Overflow);
2132 IsOverflowHigh = true;
2135 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2136 return CR1.unsignedAddMayOverflow(CR2);
2140 TEST_F(ConstantRangeTest, UnsignedSubOverflowExhaustive) {
2141 TestOverflowExhaustive(
2142 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) {
2144 (void) N1.usub_ov(N2, Overflow);
2145 IsOverflowHigh = false;
2148 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2149 return CR1.unsignedSubMayOverflow(CR2);
2153 TEST_F(ConstantRangeTest, UnsignedMulOverflowExhaustive) {
2154 TestOverflowExhaustive(
2155 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) {
2157 (void) N1.umul_ov(N2, Overflow);
2158 IsOverflowHigh = true;
2161 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2162 return CR1.unsignedMulMayOverflow(CR2);
2166 TEST_F(ConstantRangeTest, SignedAddOverflowExhaustive) {
2167 TestOverflowExhaustive(
2168 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) {
2170 (void) N1.sadd_ov(N2, Overflow);
2171 IsOverflowHigh = N1.isNonNegative();
2174 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2175 return CR1.signedAddMayOverflow(CR2);
2179 TEST_F(ConstantRangeTest, SignedSubOverflowExhaustive) {
2180 TestOverflowExhaustive(
2181 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) {
2183 (void) N1.ssub_ov(N2, Overflow);
2184 IsOverflowHigh = N1.isNonNegative();
2187 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2188 return CR1.signedSubMayOverflow(CR2);
2192 TEST_F(ConstantRangeTest, FromKnownBits) {
2193 KnownBits Unknown(16);
2194 EXPECT_EQ(Full, ConstantRange::fromKnownBits(Unknown, /*signed*/false));
2195 EXPECT_EQ(Full, ConstantRange::fromKnownBits(Unknown, /*signed*/true));
2197 // .10..01. -> unsigned 01000010 (66) to 11011011 (219)
2198 // -> signed 11000010 (194) to 01011011 (91)
2202 ConstantRange Unsigned(APInt(8, 66), APInt(8, 219 + 1));
2203 ConstantRange Signed(APInt(8, 194), APInt(8, 91 + 1));
2204 EXPECT_EQ(Unsigned, ConstantRange::fromKnownBits(Known, /*signed*/false));
2205 EXPECT_EQ(Signed, ConstantRange::fromKnownBits(Known, /*signed*/true));
2207 // 1.10.10. -> 10100100 (164) to 11101101 (237)
2210 ConstantRange CR1(APInt(8, 164), APInt(8, 237 + 1));
2211 EXPECT_EQ(CR1, ConstantRange::fromKnownBits(Known, /*signed*/false));
2212 EXPECT_EQ(CR1, ConstantRange::fromKnownBits(Known, /*signed*/true));
2214 // 01.0.1.0 -> 01000100 (68) to 01101110 (110)
2217 ConstantRange CR2(APInt(8, 68), APInt(8, 110 + 1));
2218 EXPECT_EQ(CR2, ConstantRange::fromKnownBits(Known, /*signed*/false));
2219 EXPECT_EQ(CR2, ConstantRange::fromKnownBits(Known, /*signed*/true));
2222 TEST_F(ConstantRangeTest, FromKnownBitsExhaustive) {
2224 unsigned Max = 1 << Bits;
2225 KnownBits Known(Bits);
2226 for (unsigned Zero = 0; Zero < Max; ++Zero) {
2227 for (unsigned One = 0; One < Max; ++One) {
2230 if (Known.hasConflict() || Known.isUnknown())
2233 UnsignedOpRangeGatherer UR(Bits);
2234 SignedOpRangeGatherer SR(Bits);
2235 for (unsigned N = 0; N < Max; ++N) {
2237 if ((Num & Known.Zero) != 0 || (~Num & Known.One) != 0)
2244 ConstantRange UnsignedCR = UR.getRange();
2245 ConstantRange SignedCR = SR.getRange();
2246 EXPECT_EQ(UnsignedCR, ConstantRange::fromKnownBits(Known, false));
2247 EXPECT_EQ(SignedCR, ConstantRange::fromKnownBits(Known, true));
2252 TEST_F(ConstantRangeTest, Negative) {
2253 // All elements in an empty set (of which there are none) are both negative
2254 // and non-negative. Empty & full sets checked explicitly for clarity, but
2255 // they are also covered by the exhaustive test below.
2256 EXPECT_TRUE(Empty.isAllNegative());
2257 EXPECT_TRUE(Empty.isAllNonNegative());
2258 EXPECT_FALSE(Full.isAllNegative());
2259 EXPECT_FALSE(Full.isAllNonNegative());
2262 EnumerateConstantRanges(Bits, [](const ConstantRange &CR) {
2263 bool AllNegative = true;
2264 bool AllNonNegative = true;
2265 ForeachNumInConstantRange(CR, [&](const APInt &N) {
2266 if (!N.isNegative())
2267 AllNegative = false;
2268 if (!N.isNonNegative())
2269 AllNonNegative = false;
2271 assert((CR.isEmptySet() || !AllNegative || !AllNonNegative) &&
2272 "Only empty set can be both all negative and all non-negative");
2274 EXPECT_EQ(AllNegative, CR.isAllNegative());
2275 EXPECT_EQ(AllNonNegative, CR.isAllNonNegative());
2279 TEST_F(ConstantRangeTest, UAddSat) {
2280 TestUnsignedBinOpExhaustive(
2281 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2282 return CR1.uadd_sat(CR2);
2284 [](const APInt &N1, const APInt &N2) {
2285 return N1.uadd_sat(N2);
2289 TEST_F(ConstantRangeTest, USubSat) {
2290 TestUnsignedBinOpExhaustive(
2291 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2292 return CR1.usub_sat(CR2);
2294 [](const APInt &N1, const APInt &N2) {
2295 return N1.usub_sat(N2);
2299 TEST_F(ConstantRangeTest, UMulSat) {
2300 TestUnsignedBinOpExhaustive(
2301 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2302 return CR1.umul_sat(CR2);
2304 [](const APInt &N1, const APInt &N2) { return N1.umul_sat(N2); });
2307 TEST_F(ConstantRangeTest, UShlSat) {
2308 TestUnsignedBinOpExhaustive(
2309 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2310 return CR1.ushl_sat(CR2);
2312 [](const APInt &N1, const APInt &N2) { return N1.ushl_sat(N2); });
2315 TEST_F(ConstantRangeTest, SAddSat) {
2316 TestSignedBinOpExhaustive(
2317 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2318 return CR1.sadd_sat(CR2);
2320 [](const APInt &N1, const APInt &N2) {
2321 return N1.sadd_sat(N2);
2325 TEST_F(ConstantRangeTest, SSubSat) {
2326 TestSignedBinOpExhaustive(
2327 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2328 return CR1.ssub_sat(CR2);
2330 [](const APInt &N1, const APInt &N2) {
2331 return N1.ssub_sat(N2);
2335 TEST_F(ConstantRangeTest, SMulSat) {
2336 TestSignedBinOpExhaustive(
2337 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2338 return CR1.smul_sat(CR2);
2340 [](const APInt &N1, const APInt &N2) { return N1.smul_sat(N2); });
2343 TEST_F(ConstantRangeTest, SShlSat) {
2344 TestSignedBinOpExhaustive(
2345 [](const ConstantRange &CR1, const ConstantRange &CR2) {
2346 return CR1.sshl_sat(CR2);
2348 [](const APInt &N1, const APInt &N2) { return N1.sshl_sat(N2); });
2351 TEST_F(ConstantRangeTest, Abs) {
2352 // We're working with unsigned integers here, because it makes the signed
2353 // min case non-wrapping.
2354 TestUnsignedUnaryOpExhaustive(
2355 [](const ConstantRange &CR) { return CR.abs(); },
2356 [](const APInt &N) { return N.abs(); });
2358 TestUnsignedUnaryOpExhaustive(
2359 [](const ConstantRange &CR) { return CR.abs(/*IntMinIsPoison=*/true); },
2360 [](const APInt &N) { return N.abs(); },
2361 /*SkipSignedIntMin=*/true);
2364 TEST_F(ConstantRangeTest, castOps) {
2365 ConstantRange A(APInt(16, 66), APInt(16, 128));
2366 ConstantRange FpToI8 = A.castOp(Instruction::FPToSI, 8);
2367 EXPECT_EQ(8u, FpToI8.getBitWidth());
2368 EXPECT_TRUE(FpToI8.isFullSet());
2370 ConstantRange FpToI16 = A.castOp(Instruction::FPToSI, 16);
2371 EXPECT_EQ(16u, FpToI16.getBitWidth());
2372 EXPECT_EQ(A, FpToI16);
2374 ConstantRange FPExtToDouble = A.castOp(Instruction::FPExt, 64);
2375 EXPECT_EQ(64u, FPExtToDouble.getBitWidth());
2376 EXPECT_TRUE(FPExtToDouble.isFullSet());
2378 ConstantRange PtrToInt = A.castOp(Instruction::PtrToInt, 64);
2379 EXPECT_EQ(64u, PtrToInt.getBitWidth());
2380 EXPECT_TRUE(PtrToInt.isFullSet());
2382 ConstantRange IntToPtr = A.castOp(Instruction::IntToPtr, 64);
2383 EXPECT_EQ(64u, IntToPtr.getBitWidth());
2384 EXPECT_TRUE(IntToPtr.isFullSet());
2387 TEST_F(ConstantRangeTest, binaryXor) {
2388 // Single element ranges.
2389 ConstantRange R16(APInt(8, 16));
2390 ConstantRange R20(APInt(8, 20));
2391 EXPECT_EQ(*R16.binaryXor(R16).getSingleElement(), APInt(8, 0));
2392 EXPECT_EQ(*R16.binaryXor(R20).getSingleElement(), APInt(8, 16 ^ 20));
2394 // Ranges with more than a single element. Handled conservatively for now.
2395 ConstantRange R16_35(APInt(8, 16), APInt(8, 35));
2396 ConstantRange R0_99(APInt(8, 0), APInt(8, 99));
2397 EXPECT_TRUE(R16_35.binaryXor(R16_35).isFullSet());
2398 EXPECT_TRUE(R16_35.binaryXor(R0_99).isFullSet());
2399 EXPECT_TRUE(R0_99.binaryXor(R16_35).isFullSet());
2402 TEST_F(ConstantRangeTest, binaryNot) {
2403 TestUnsignedUnaryOpExhaustive(
2404 [](const ConstantRange &CR) { return CR.binaryNot(); },
2405 [](const APInt &N) { return ~N; });
2406 TestUnsignedUnaryOpExhaustive(
2407 [](const ConstantRange &CR) {
2408 return CR.binaryXor(
2409 ConstantRange(APInt::getAllOnesValue(CR.getBitWidth())));
2411 [](const APInt &N) { return ~N; });
2412 TestUnsignedUnaryOpExhaustive(
2413 [](const ConstantRange &CR) {
2414 return ConstantRange(APInt::getAllOnesValue(CR.getBitWidth()))
2417 [](const APInt &N) { return ~N; });
2420 } // anonymous namespace