1 //===- ConstantRangeTest.cpp - ConstantRange tests ------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #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" 15 16 using namespace llvm; 17 18 namespace { 19 20 class ConstantRangeTest : public ::testing::Test { 21 protected: 22 static ConstantRange Full; 23 static ConstantRange Empty; 24 static ConstantRange One; 25 static ConstantRange Some; 26 static ConstantRange Wrap; 27 }; 28 29 template<typename Fn> 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) 36 continue; 37 38 ConstantRange CR(APInt(Bits, Lo), APInt(Bits, Hi)); 39 TestFn(CR); 40 } 41 } 42 } 43 44 template<typename Fn> 45 static void EnumerateTwoConstantRanges(unsigned Bits, Fn TestFn) { 46 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR1) { 47 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR2) { 48 TestFn(CR1, CR2); 49 }); 50 }); 51 } 52 53 template<typename Fn> 54 static void ForeachNumInConstantRange(const ConstantRange &CR, Fn TestFn) { 55 if (!CR.isEmptySet()) { 56 APInt N = CR.getLower(); 57 do TestFn(N); 58 while (++N != CR.getUpper()); 59 } 60 } 61 62 struct OpRangeGathererBase { 63 void account(const APInt &N); 64 ConstantRange getRange(); 65 }; 66 67 struct UnsignedOpRangeGatherer : public OpRangeGathererBase { 68 APInt Min; 69 APInt Max; 70 71 UnsignedOpRangeGatherer(unsigned Bits) 72 : Min(APInt::getMaxValue(Bits)), Max(APInt::getMinValue(Bits)) {} 73 74 void account(const APInt &N) { 75 if (N.ult(Min)) 76 Min = N; 77 if (N.ugt(Max)) 78 Max = N; 79 } 80 81 ConstantRange getRange() { 82 if (Min.ugt(Max)) 83 return ConstantRange::getEmpty(Min.getBitWidth()); 84 return ConstantRange::getNonEmpty(Min, Max + 1); 85 } 86 }; 87 88 struct SignedOpRangeGatherer : public OpRangeGathererBase { 89 APInt Min; 90 APInt Max; 91 92 SignedOpRangeGatherer(unsigned Bits) 93 : Min(APInt::getSignedMaxValue(Bits)), 94 Max(APInt::getSignedMinValue(Bits)) {} 95 96 void account(const APInt &N) { 97 if (N.slt(Min)) 98 Min = N; 99 if (N.sgt(Max)) 100 Max = N; 101 } 102 103 ConstantRange getRange() { 104 if (Min.sgt(Max)) 105 return ConstantRange::getEmpty(Min.getBitWidth()); 106 return ConstantRange::getNonEmpty(Min, Max + 1); 107 } 108 }; 109 110 template <typename Fn1, typename Fn2> 111 static void TestUnsignedUnaryOpExhaustive(Fn1 RangeFn, Fn2 IntFn, 112 bool SkipSignedIntMin = false) { 113 unsigned Bits = 4; 114 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) { 115 UnsignedOpRangeGatherer R(CR.getBitWidth()); 116 ForeachNumInConstantRange(CR, [&](const APInt &N) { 117 if (SkipSignedIntMin && N.isMinSignedValue()) 118 return; 119 R.account(IntFn(N)); 120 }); 121 122 ConstantRange ExactCR = R.getRange(); 123 ConstantRange ActualCR = RangeFn(CR); 124 125 EXPECT_EQ(ExactCR, ActualCR); 126 }); 127 } 128 129 template <typename Fn1, typename Fn2> 130 static void TestUnsignedBinOpExhaustive(Fn1 RangeFn, Fn2 IntFn, 131 bool SkipZeroRHS = false, 132 bool CorrectnessOnly = false) { 133 unsigned Bits = 4; 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) 140 return; 141 R.account(IntFn(N1, N2)); 142 }); 143 }); 144 145 ConstantRange CR = RangeFn(CR1, CR2); 146 147 ConstantRange Exact = R.getRange(); 148 149 if (!CorrectnessOnly) { 150 EXPECT_EQ(Exact, CR); 151 return; 152 } 153 154 EXPECT_TRUE(CR.contains(Exact)); 155 if (Exact.isEmptySet()) 156 EXPECT_TRUE(CR.isEmptySet()); 157 }); 158 } 159 160 template <typename Fn1, typename Fn2> 161 static void TestSignedBinOpExhaustive(Fn1 RangeFn, Fn2 IntFn, 162 bool SkipZeroRHS = false, 163 bool CorrectnessOnly = false) { 164 unsigned Bits = 4; 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) 171 return; 172 173 R.account(IntFn(N1, N2)); 174 }); 175 }); 176 177 ConstantRange CR = RangeFn(CR1, CR2); 178 179 ConstantRange Exact = R.getRange(); 180 if (CorrectnessOnly) { 181 EXPECT_TRUE(CR.contains(Exact)); 182 } else { 183 EXPECT_EQ(Exact, CR); 184 } 185 }); 186 } 187 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)); 193 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))); 204 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))); 214 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))); 224 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))); 233 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))); 242 } 243 244 TEST_F(ConstantRangeTest, Equality) { 245 EXPECT_EQ(Full, Full); 246 EXPECT_EQ(Empty, Empty); 247 EXPECT_EQ(One, One); 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); 260 } 261 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)); 267 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)); 271 272 EXPECT_EQ(One.getSingleMissingElement(), static_cast<APInt *>(nullptr)); 273 EXPECT_EQ(Some.getSingleMissingElement(), static_cast<APInt *>(nullptr)); 274 275 ConstantRange OneInverse = One.inverse(); 276 EXPECT_EQ(*OneInverse.getSingleMissingElement(), *One.getSingleElement()); 277 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()); 283 } 284 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)); 290 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)); 295 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)); 300 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)); 305 306 // Found by Klee 307 EXPECT_EQ(ConstantRange(APInt(4, 7), APInt(4, 0)).getSignedMax(), 308 APInt(4, 7)); 309 } 310 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()); 317 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()); 325 } 326 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()); 339 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()); 344 345 ConstantRange CR2(APInt(8, 42), APInt::getSignedMinValue(8)); 346 EXPECT_FALSE(CR2.isSignWrappedSet()); 347 EXPECT_TRUE(CR2.isUpperSignWrapped()); 348 } 349 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()); 362 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))); 366 367 // trunc([2, 6), 3->2) = full 368 ConstantRange TwoSix(APInt(3, 2), APInt(3, 6)); 369 EXPECT_TRUE(TwoSix.truncate(2).isFullSet()); 370 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))); 374 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))); 378 } 379 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))); 393 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))); 397 } 398 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))); 414 415 EXPECT_EQ(ConstantRange(APInt(8, 120), APInt(8, 140)).signExtend(16), 416 ConstantRange(APInt(16, -128), APInt(16, 128))); 417 418 EXPECT_EQ(ConstantRange(APInt(16, 0x0200), APInt(16, 0x8000)).signExtend(19), 419 ConstantRange(APInt(19, 0x0200), APInt(19, 0x8000))); 420 } 421 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)); 436 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); 443 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))); 448 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))); 453 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))); 458 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))); 463 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))); 468 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))); 473 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))); 478 } 479 480 template<typename Fn1, typename Fn2> 481 void testBinarySetOperationExhaustive(Fn1 OpFn, Fn2 InResultFn) { 482 unsigned Bits = 4; 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; 494 495 APInt Num(Bits, 0); 496 for (unsigned I = 0, Limit = 1 << Bits; I < Limit; ++I, ++Num) { 497 if (!InResultFn(CR1, CR2, Num)) { 498 if (HaveRange3) 499 HaveInterrupt3 = true; 500 else if (HaveRange2) 501 HaveInterrupt2 = true; 502 else if (HaveRange1) 503 HaveInterrupt1 = true; 504 continue; 505 } 506 507 if (HaveRange3) { 508 Upper3 = Num; 509 } else if (HaveInterrupt2) { 510 HaveRange3 = true; 511 Lower3 = Upper3 = Num; 512 } else if (HaveRange2) { 513 Upper2 = Num; 514 } else if (HaveInterrupt1) { 515 HaveRange2 = true; 516 Lower2 = Upper2 = Num; 517 } else if (HaveRange1) { 518 Upper1 = Num; 519 } else { 520 HaveRange1 = true; 521 Lower1 = Upper1 = Num; 522 } 523 } 524 525 (void)HaveInterrupt3; 526 assert(!HaveInterrupt3 && "Should have at most three ranges"); 527 528 ConstantRange SmallestCR = OpFn(CR1, CR2, ConstantRange::Smallest); 529 ConstantRange UnsignedCR = OpFn(CR1, CR2, ConstantRange::Unsigned); 530 ConstantRange SignedCR = OpFn(CR1, CR2, ConstantRange::Signed); 531 532 if (!HaveRange1) { 533 EXPECT_TRUE(SmallestCR.isEmptySet()); 534 EXPECT_TRUE(UnsignedCR.isEmptySet()); 535 EXPECT_TRUE(SignedCR.isEmptySet()); 536 return; 537 } 538 539 if (!HaveRange2) { 540 if (Lower1 == Upper1 + 1) { 541 EXPECT_TRUE(SmallestCR.isFullSet()); 542 EXPECT_TRUE(UnsignedCR.isFullSet()); 543 EXPECT_TRUE(SignedCR.isFullSet()); 544 } else { 545 ConstantRange Expected(Lower1, Upper1 + 1); 546 EXPECT_EQ(Expected, SmallestCR); 547 EXPECT_EQ(Expected, UnsignedCR); 548 EXPECT_EQ(Expected, SignedCR); 549 } 550 return; 551 } 552 553 ConstantRange Variant1(Bits, /*full*/ true); 554 ConstantRange Variant2(Bits, /*full*/ true); 555 if (!HaveRange3) { 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); 561 } else { 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); 568 } 569 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); 575 else 576 EXPECT_TRUE(Variant1 == SmallestCR || Variant2 == SmallestCR); 577 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); 589 else 590 EXPECT_TRUE(Variant1 == UnsignedCR || Variant2 == UnsignedCR); 591 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); 603 else 604 EXPECT_TRUE(Variant1 == SignedCR || Variant2 == SignedCR); 605 }); 606 } 607 608 TEST_F(ConstantRangeTest, IntersectWithExhaustive) { 609 testBinarySetOperationExhaustive( 610 [](const ConstantRange &CR1, const ConstantRange &CR2, 611 ConstantRange::PreferredRangeType Type) { 612 return CR1.intersectWith(CR2, Type); 613 }, 614 [](const ConstantRange &CR1, const ConstantRange &CR2, const APInt &N) { 615 return CR1.contains(N) && CR2.contains(N); 616 }); 617 } 618 619 TEST_F(ConstantRangeTest, UnionWithExhaustive) { 620 testBinarySetOperationExhaustive( 621 [](const ConstantRange &CR1, const ConstantRange &CR2, 622 ConstantRange::PreferredRangeType Type) { 623 return CR1.unionWith(CR2, Type); 624 }, 625 [](const ConstantRange &CR1, const ConstantRange &CR2, const APInt &N) { 626 return CR1.contains(N) || CR2.contains(N); 627 }); 628 } 629 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); 637 638 // PR4545 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)); 648 } 649 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); 655 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); 665 } 666 667 TEST_F(ConstantRangeTest, getActiveBits) { 668 unsigned Bits = 4; 669 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) { 670 unsigned Exact = 0; 671 ForeachNumInConstantRange(CR, [&](const APInt &N) { 672 Exact = std::max(Exact, N.getActiveBits()); 673 }); 674 675 unsigned ResultCR = CR.getActiveBits(); 676 EXPECT_EQ(Exact, ResultCR); 677 }); 678 } 679 TEST_F(ConstantRangeTest, losslessUnsignedTruncationZeroext) { 680 unsigned Bits = 4; 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())); 686 return; 687 } 688 if (MinBitWidth == Bits) 689 return; 690 EXPECT_EQ(CR, CR.truncate(MinBitWidth).zeroExtend(Bits)); 691 }); 692 } 693 694 TEST_F(ConstantRangeTest, getMinSignedBits) { 695 unsigned Bits = 4; 696 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) { 697 unsigned Exact = 0; 698 ForeachNumInConstantRange(CR, [&](const APInt &N) { 699 Exact = std::max(Exact, N.getMinSignedBits()); 700 }); 701 702 unsigned ResultCR = CR.getMinSignedBits(); 703 EXPECT_EQ(Exact, ResultCR); 704 }); 705 } 706 TEST_F(ConstantRangeTest, losslessSignedTruncationSignext) { 707 unsigned Bits = 4; 708 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) { 709 unsigned MinBitWidth = CR.getMinSignedBits(); 710 if (MinBitWidth == 0) { 711 EXPECT_TRUE(CR.isEmptySet()); 712 return; 713 } 714 if (MinBitWidth == Bits) 715 return; 716 EXPECT_EQ(CR, CR.truncate(MinBitWidth).signExtend(Bits)); 717 }); 718 } 719 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))); 729 } 730 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))); 749 } 750 751 template <typename Fn1, typename Fn2> 752 static void TestAddWithNoSignedWrapExhaustive(Fn1 RangeFn, Fn2 IntFn) { 753 unsigned Bits = 4; 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); 763 if (!IsOverflow) { 764 AllOverflow = false; 765 R.account(N); 766 EXPECT_TRUE(CR.contains(N)); 767 } 768 }); 769 }); 770 771 EXPECT_EQ(CR.isEmptySet(), AllOverflow); 772 773 if (CR1.isSignWrappedSet() || CR2.isSignWrappedSet()) 774 return; 775 776 ConstantRange Exact = R.getRange(); 777 EXPECT_EQ(Exact, CR); 778 }); 779 } 780 781 template <typename Fn1, typename Fn2> 782 static void TestAddWithNoUnsignedWrapExhaustive(Fn1 RangeFn, Fn2 IntFn) { 783 unsigned Bits = 4; 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); 793 if (!IsOverflow) { 794 AllOverflow = false; 795 R.account(N); 796 EXPECT_TRUE(CR.contains(N)); 797 } 798 }); 799 }); 800 801 EXPECT_EQ(CR.isEmptySet(), AllOverflow); 802 803 if (CR1.isWrappedSet() || CR2.isWrappedSet()) 804 return; 805 806 ConstantRange Exact = R.getRange(); 807 EXPECT_EQ(Exact, CR); 808 }); 809 } 810 811 template <typename Fn1, typename Fn2, typename Fn3> 812 static void TestAddWithNoSignedUnsignedWrapExhaustive(Fn1 RangeFn, 813 Fn2 IntFnSigned, 814 Fn3 IntFnUnsigned) { 815 unsigned Bits = 4; 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) { 828 AllOverflow = false; 829 UR.account(N); 830 SR.account(N); 831 EXPECT_TRUE(CR.contains(N)); 832 } 833 }); 834 }); 835 836 EXPECT_EQ(CR.isEmptySet(), AllOverflow); 837 838 if (CR1.isWrappedSet() || CR2.isWrappedSet() || 839 CR1.isSignWrappedSet() || CR2.isSignWrappedSet()) 840 return; 841 842 ConstantRange ExactUnsignedCR = UR.getRange(); 843 ConstantRange ExactSignedCR = SR.getRange(); 844 845 if (ExactUnsignedCR.isEmptySet() || ExactSignedCR.isEmptySet()) { 846 EXPECT_TRUE(CR.isEmptySet()); 847 return; 848 } 849 850 ConstantRange Exact = ExactSignedCR.intersectWith(ExactUnsignedCR); 851 EXPECT_EQ(Exact, CR); 852 }); 853 } 854 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)), 863 OBO::NoSignedWrap), 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)), 869 OBO::NoSignedWrap), 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)), 873 OBO::NoSignedWrap), 874 ConstantRange(8, false)); 875 EXPECT_EQ(ConstantRange(APInt(8, -120), APInt(8, -100)) 876 .addWithNoWrap(ConstantRange(APInt(8, -110), APInt(8, -100)), 877 OBO::NoSignedWrap), 878 ConstantRange(8, false)); 879 EXPECT_EQ(ConstantRange(APInt(8, 0), APInt(8, 101)) 880 .addWithNoWrap(ConstantRange(APInt(8, -128), APInt(8, 28)), 881 OBO::NoSignedWrap), 882 ConstantRange(8, true)); 883 EXPECT_EQ(ConstantRange(APInt(8, 0), APInt(8, 101)) 884 .addWithNoWrap(ConstantRange(APInt(8, -120), APInt(8, 29)), 885 OBO::NoSignedWrap), 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)), 889 OBO::NoSignedWrap), 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)), 893 OBO::NoSignedWrap), 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)), 897 OBO::NoSignedWrap), 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)), 901 OBO::NoSignedWrap), 902 ConstantRange(APInt(8, 125), APInt(8, 9))); 903 904 TestAddWithNoSignedWrapExhaustive( 905 [](const ConstantRange &CR1, const ConstantRange &CR2) { 906 return CR1.addWithNoWrap(CR2, OBO::NoSignedWrap); 907 }, 908 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 909 return N1.sadd_ov(N2, IsOverflow); 910 }); 911 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))); 951 952 TestAddWithNoUnsignedWrapExhaustive( 953 [](const ConstantRange &CR1, const ConstantRange &CR2) { 954 return CR1.addWithNoWrap(CR2, OBO::NoUnsignedWrap); 955 }, 956 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 957 return N1.uadd_ov(N2, IsOverflow); 958 }); 959 960 EXPECT_EQ(ConstantRange(APInt(8, 50), APInt(8, 100)) 961 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 70)), 962 OBO::NoSignedWrap), 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))); 972 973 EXPECT_EQ(ConstantRange(APInt(8, -100), APInt(8, -50)) 974 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 30)), 975 OBO::NoSignedWrap), 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))); 985 986 TestAddWithNoSignedUnsignedWrapExhaustive( 987 [](const ConstantRange &CR1, const ConstantRange &CR2) { 988 return CR1.addWithNoWrap(CR2, OBO::NoUnsignedWrap | OBO::NoSignedWrap); 989 }, 990 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 991 return N1.sadd_ov(N2, IsOverflow); 992 }, 993 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 994 return N1.uadd_ov(N2, IsOverflow); 995 }); 996 } 997 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))); 1018 } 1019 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); 1025 }, 1026 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 1027 return N1.ssub_ov(N2, IsOverflow); 1028 }); 1029 TestAddWithNoUnsignedWrapExhaustive( 1030 [](const ConstantRange &CR1, const ConstantRange &CR2) { 1031 return CR1.subWithNoWrap(CR2, OBO::NoUnsignedWrap); 1032 }, 1033 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 1034 return N1.usub_ov(N2, IsOverflow); 1035 }); 1036 TestAddWithNoSignedUnsignedWrapExhaustive( 1037 [](const ConstantRange &CR1, const ConstantRange &CR2) { 1038 return CR1.subWithNoWrap(CR2, OBO::NoUnsignedWrap | OBO::NoSignedWrap); 1039 }, 1040 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 1041 return N1.ssub_ov(N2, IsOverflow); 1042 }, 1043 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 1044 return N1.usub_ov(N2, IsOverflow); 1045 }); 1046 } 1047 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); 1066 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); 1074 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)); 1079 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))); 1086 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))); 1091 } 1092 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); 1110 } 1111 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); 1131 } 1132 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); 1149 } 1150 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), 1155 APInt(16, 0xaaa))); 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), 1163 APInt(16, 0xaaa))); 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), 1168 APInt(16, 0xb))); 1169 EXPECT_EQ(One.smin(One), One); 1170 } 1171 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); 1190 1191 1192 ConstantRange Zero(APInt(16, 0)); 1193 EXPECT_EQ(Zero.udiv(One), Zero); 1194 EXPECT_EQ(Zero.udiv(Full), Zero); 1195 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))); 1200 } 1201 1202 TEST_F(ConstantRangeTest, SDiv) { 1203 unsigned Bits = 4; 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. 1213 if (N2 == 0) 1214 return; 1215 1216 // SignedMin / -1 is UB. 1217 if (N1.isMinSignedValue() && N2.isAllOnesValue()) 1218 return; 1219 1220 APInt N = N1.sdiv(N2); 1221 Results.set(N.getSExtValue() + Bias); 1222 }); 1223 }); 1224 1225 ConstantRange CR = CR1.sdiv(CR2); 1226 if (Results.none()) { 1227 EXPECT_TRUE(CR.isEmptySet()); 1228 return; 1229 } 1230 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); 1237 return; 1238 } 1239 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]) { 1246 if (LastNeg == -1) 1247 ++LastNeg; 1248 else if (LastPos == 1) 1249 --LastPos; 1250 } 1251 1252 APInt WMax(Bits, LastNeg); 1253 APInt WMin(Bits, LastPos); 1254 ConstantRange Wrapped = ConstantRange::getNonEmpty(WMin, WMax + 1); 1255 EXPECT_EQ(Wrapped, CR); 1256 }); 1257 } 1258 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))); 1284 1285 TestUnsignedBinOpExhaustive( 1286 [](const ConstantRange &CR1, const ConstantRange &CR2) { 1287 return CR1.urem(CR2); 1288 }, 1289 [](const APInt &N1, const APInt &N2) { 1290 return N1.urem(N2); 1291 }, 1292 /* SkipZeroRHS */ true, /* CorrectnessOnly */ true); 1293 } 1294 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))); 1303 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)); 1307 1308 ConstantRange Expected(16, true); 1309 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); 1323 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); 1337 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); 1351 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))); 1357 1358 TestSignedBinOpExhaustive( 1359 [](const ConstantRange &CR1, const ConstantRange &CR2) { 1360 return CR1.srem(CR2); 1361 }, 1362 [](const APInt &N1, const APInt &N2) { 1363 return N1.srem(N2); 1364 }, 1365 /* SkipZeroRHS */ true, /* CorrectnessOnly */ true); 1366 } 1367 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); 1387 EXPECT_EQ( 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); 1391 } 1392 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); 1412 } 1413 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))); 1439 } 1440 1441 TEST(ConstantRange, MakeAllowedICmpRegion) { 1442 // PR8250 1443 ConstantRange SMax = ConstantRange(APInt::getSignedMaxValue(32)); 1444 EXPECT_TRUE(ConstantRange::makeAllowedICmpRegion(ICmpInst::ICMP_SGT, SMax) 1445 .isEmptySet()); 1446 } 1447 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); 1452 1453 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_NE, LowHalf), 1454 HighHalf); 1455 1456 EXPECT_EQ( 1457 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_NE, HighHalf), 1458 LowHalf); 1459 1460 EXPECT_TRUE(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_EQ, 1461 HighHalf).isEmptySet()); 1462 1463 ConstantRange UnsignedSample(APInt(8, 5), APInt(8, 200)); 1464 1465 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_ULT, 1466 UnsignedSample), 1467 ConstantRange(APInt(8, 0), APInt(8, 5))); 1468 1469 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_ULE, 1470 UnsignedSample), 1471 ConstantRange(APInt(8, 0), APInt(8, 6))); 1472 1473 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_UGT, 1474 UnsignedSample), 1475 ConstantRange(APInt(8, 200), APInt(8, 0))); 1476 1477 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_UGE, 1478 UnsignedSample), 1479 ConstantRange(APInt(8, 199), APInt(8, 0))); 1480 1481 ConstantRange SignedSample(APInt(8, -5), APInt(8, 5)); 1482 1483 EXPECT_EQ( 1484 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SLT, SignedSample), 1485 ConstantRange(APInt(8, -128), APInt(8, -5))); 1486 1487 EXPECT_EQ( 1488 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SLE, SignedSample), 1489 ConstantRange(APInt(8, -128), APInt(8, -4))); 1490 1491 EXPECT_EQ( 1492 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SGT, SignedSample), 1493 ConstantRange(APInt(8, 5), APInt(8, -128))); 1494 1495 EXPECT_EQ( 1496 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SGE, SignedSample), 1497 ConstantRange(APInt(8, 4), APInt(8, -128))); 1498 } 1499 1500 TEST(ConstantRange, MakeGuaranteedNoWrapRegion) { 1501 const int IntMin4Bits = 8; 1502 const int IntMax4Bits = 7; 1503 typedef OverflowingBinaryOperator OBO; 1504 1505 for (int Const : {0, -1, -2, 1, 2, IntMin4Bits, IntMax4Bits}) { 1506 APInt C(4, Const, true /* = isSigned */); 1507 1508 auto NUWRegion = ConstantRange::makeGuaranteedNoWrapRegion( 1509 Instruction::Add, C, OBO::NoUnsignedWrap); 1510 1511 EXPECT_FALSE(NUWRegion.isEmptySet()); 1512 1513 auto NSWRegion = ConstantRange::makeGuaranteedNoWrapRegion( 1514 Instruction::Add, C, OBO::NoSignedWrap); 1515 1516 EXPECT_FALSE(NSWRegion.isEmptySet()); 1517 1518 for (APInt I = NUWRegion.getLower(), E = NUWRegion.getUpper(); I != E; 1519 ++I) { 1520 bool Overflow = false; 1521 (void)I.uadd_ov(C, Overflow); 1522 EXPECT_FALSE(Overflow); 1523 } 1524 1525 for (APInt I = NSWRegion.getLower(), E = NSWRegion.getUpper(); I != E; 1526 ++I) { 1527 bool Overflow = false; 1528 (void)I.sadd_ov(C, Overflow); 1529 EXPECT_FALSE(Overflow); 1530 } 1531 } 1532 1533 for (int Const : {0, -1, -2, 1, 2, IntMin4Bits, IntMax4Bits}) { 1534 APInt C(4, Const, true /* = isSigned */); 1535 1536 auto NUWRegion = ConstantRange::makeGuaranteedNoWrapRegion( 1537 Instruction::Sub, C, OBO::NoUnsignedWrap); 1538 1539 EXPECT_FALSE(NUWRegion.isEmptySet()); 1540 1541 auto NSWRegion = ConstantRange::makeGuaranteedNoWrapRegion( 1542 Instruction::Sub, C, OBO::NoSignedWrap); 1543 1544 EXPECT_FALSE(NSWRegion.isEmptySet()); 1545 1546 for (APInt I = NUWRegion.getLower(), E = NUWRegion.getUpper(); I != E; 1547 ++I) { 1548 bool Overflow = false; 1549 (void)I.usub_ov(C, Overflow); 1550 EXPECT_FALSE(Overflow); 1551 } 1552 1553 for (APInt I = NSWRegion.getLower(), E = NSWRegion.getUpper(); I != E; 1554 ++I) { 1555 bool Overflow = false; 1556 (void)I.ssub_ov(C, Overflow); 1557 EXPECT_FALSE(Overflow); 1558 } 1559 } 1560 1561 auto NSWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion( 1562 Instruction::Add, ConstantRange(32, /* isFullSet = */ true), 1563 OBO::NoSignedWrap); 1564 EXPECT_TRUE(NSWForAllValues.isSingleElement() && 1565 NSWForAllValues.getSingleElement()->isMinValue()); 1566 1567 NSWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion( 1568 Instruction::Sub, ConstantRange(32, /* isFullSet = */ true), 1569 OBO::NoSignedWrap); 1570 EXPECT_TRUE(NSWForAllValues.isSingleElement() && 1571 NSWForAllValues.getSingleElement()->isMaxValue()); 1572 1573 auto NUWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion( 1574 Instruction::Add, ConstantRange(32, /* isFullSet = */ true), 1575 OBO::NoUnsignedWrap); 1576 EXPECT_TRUE(NUWForAllValues.isSingleElement() && 1577 NUWForAllValues.getSingleElement()->isMinValue()); 1578 1579 NUWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion( 1580 Instruction::Sub, ConstantRange(32, /* isFullSet = */ true), 1581 OBO::NoUnsignedWrap); 1582 EXPECT_TRUE(NUWForAllValues.isSingleElement() && 1583 NUWForAllValues.getSingleElement()->isMaxValue()); 1584 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()); 1593 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))); 1609 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))); 1626 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))); 1643 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))); 1659 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)); 1676 1677 EXPECT_EQ( 1678 ConstantRange::makeGuaranteedNoWrapRegion( 1679 Instruction::Shl, ConstantRange::getFull(32), OBO::NoUnsignedWrap), 1680 ConstantRange::makeGuaranteedNoWrapRegion( 1681 Instruction::Shl, OneLessThanBitWidth, OBO::NoUnsignedWrap)); 1682 EXPECT_EQ( 1683 ConstantRange::makeGuaranteedNoWrapRegion( 1684 Instruction::Shl, ConstantRange::getFull(32), OBO::NoSignedWrap), 1685 ConstantRange::makeGuaranteedNoWrapRegion( 1686 Instruction::Shl, OneLessThanBitWidth, OBO::NoSignedWrap)); 1687 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)); 1695 1696 EXPECT_EQ( 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)); 1703 EXPECT_EQ( 1704 ConstantRange::makeGuaranteedNoWrapRegion( 1705 Instruction::Shl, ConstantRange(APInt(32, -32), APInt(32, 16) + 1), 1706 OBO::NoSignedWrap), 1707 ConstantRange::makeGuaranteedNoWrapRegion( 1708 Instruction::Shl, ConstantRange(APInt(32, 0), APInt(32, 16) + 1), 1709 OBO::NoSignedWrap)); 1710 1711 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1712 Instruction::Shl, 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( 1717 Instruction::Shl, 1718 ConstantRange(APInt(32, -32), APInt(32, 16) + 1), 1719 OBO::NoSignedWrap), 1720 ConstantRange(APInt(32, -32768), APInt(32, 32767) + 1)); 1721 } 1722 1723 template<typename Fn> 1724 void TestNoWrapRegionExhaustive(Instruction::BinaryOps BinOp, 1725 unsigned NoWrapKind, Fn OverflowFn) { 1726 unsigned Bits = 5; 1727 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) { 1728 if (CR.isEmptySet()) 1729 return; 1730 if (Instruction::isShift(BinOp) && CR.getUnsignedMax().uge(Bits)) 1731 return; 1732 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)) 1741 NoOverflow = false; 1742 else 1743 Overflow = false; 1744 }); 1745 EXPECT_EQ(NoOverflow, NoWrap.contains(N1)); 1746 1747 // The no-wrap range is exact for single-element ranges. 1748 if (CR.isSingleElement()) { 1749 EXPECT_EQ(Overflow, !NoWrap.contains(N1)); 1750 } 1751 }); 1752 }); 1753 } 1754 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) { 1761 bool Overflow; 1762 (void) N1.uadd_ov(N2, Overflow); 1763 return Overflow; 1764 }); 1765 TestNoWrapRegionExhaustive( 1766 Instruction::Add, OverflowingBinaryOperator::NoSignedWrap, 1767 [](const APInt &N1, const APInt &N2) { 1768 bool Overflow; 1769 (void) N1.sadd_ov(N2, Overflow); 1770 return Overflow; 1771 }); 1772 TestNoWrapRegionExhaustive( 1773 Instruction::Sub, OverflowingBinaryOperator::NoUnsignedWrap, 1774 [](const APInt &N1, const APInt &N2) { 1775 bool Overflow; 1776 (void) N1.usub_ov(N2, Overflow); 1777 return Overflow; 1778 }); 1779 TestNoWrapRegionExhaustive( 1780 Instruction::Sub, OverflowingBinaryOperator::NoSignedWrap, 1781 [](const APInt &N1, const APInt &N2) { 1782 bool Overflow; 1783 (void) N1.ssub_ov(N2, Overflow); 1784 return Overflow; 1785 }); 1786 TestNoWrapRegionExhaustive( 1787 Instruction::Mul, OverflowingBinaryOperator::NoUnsignedWrap, 1788 [](const APInt &N1, const APInt &N2) { 1789 bool Overflow; 1790 (void) N1.umul_ov(N2, Overflow); 1791 return Overflow; 1792 }); 1793 TestNoWrapRegionExhaustive( 1794 Instruction::Mul, OverflowingBinaryOperator::NoSignedWrap, 1795 [](const APInt &N1, const APInt &N2) { 1796 bool Overflow; 1797 (void) N1.smul_ov(N2, Overflow); 1798 return Overflow; 1799 }); 1800 TestNoWrapRegionExhaustive(Instruction::Shl, 1801 OverflowingBinaryOperator::NoUnsignedWrap, 1802 [](const APInt &N1, const APInt &N2) { 1803 bool Overflow; 1804 (void)N1.ushl_ov(N2, Overflow); 1805 return Overflow; 1806 }); 1807 TestNoWrapRegionExhaustive(Instruction::Shl, 1808 OverflowingBinaryOperator::NoSignedWrap, 1809 [](const APInt &N1, const APInt &N2) { 1810 bool Overflow; 1811 (void)N1.sshl_ov(N2, Overflow); 1812 return Overflow; 1813 }); 1814 } 1815 1816 TEST(ConstantRange, GetEquivalentICmp) { 1817 APInt RHS; 1818 CmpInst::Predicate Pred; 1819 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)); 1824 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)); 1829 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)); 1834 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)); 1839 1840 EXPECT_TRUE( 1841 ConstantRange(32, /*isFullSet=*/true).getEquivalentICmp(Pred, RHS)); 1842 EXPECT_EQ(Pred, CmpInst::ICMP_UGE); 1843 EXPECT_EQ(RHS, APInt(32, 0)); 1844 1845 EXPECT_TRUE( 1846 ConstantRange(32, /*isFullSet=*/false).getEquivalentICmp(Pred, RHS)); 1847 EXPECT_EQ(Pred, CmpInst::ICMP_ULT); 1848 EXPECT_EQ(RHS, APInt(32, 0)); 1849 1850 EXPECT_FALSE(ConstantRange(APInt(32, 100), APInt(32, 200)) 1851 .getEquivalentICmp(Pred, RHS)); 1852 1853 EXPECT_FALSE(ConstantRange(APInt::getSignedMinValue(32) - APInt(32, 100), 1854 APInt::getSignedMinValue(32) + APInt(32, 100)) 1855 .getEquivalentICmp(Pred, RHS)); 1856 1857 EXPECT_FALSE(ConstantRange(APInt::getMinValue(32) - APInt(32, 100), 1858 APInt::getMinValue(32) + APInt(32, 100)) 1859 .getEquivalentICmp(Pred, RHS)); 1860 1861 EXPECT_TRUE(ConstantRange(APInt(32, 100)).getEquivalentICmp(Pred, RHS)); 1862 EXPECT_EQ(Pred, CmpInst::ICMP_EQ); 1863 EXPECT_EQ(RHS, APInt(32, 100)); 1864 1865 EXPECT_TRUE( 1866 ConstantRange(APInt(32, 100)).inverse().getEquivalentICmp(Pred, RHS)); 1867 EXPECT_EQ(Pred, CmpInst::ICMP_NE); 1868 EXPECT_EQ(RHS, APInt(32, 100)); 1869 1870 EXPECT_TRUE( 1871 ConstantRange(APInt(512, 100)).inverse().getEquivalentICmp(Pred, RHS)); 1872 EXPECT_EQ(Pred, CmpInst::ICMP_NE); 1873 EXPECT_EQ(RHS, APInt(512, 100)); 1874 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. 1878 1879 EXPECT_TRUE(ConstantRange(APInt(32, 0)).getEquivalentICmp(Pred, RHS)); 1880 EXPECT_EQ(Pred, CmpInst::ICMP_EQ); 1881 EXPECT_EQ(RHS, APInt(32, 0)); 1882 1883 EXPECT_TRUE( 1884 ConstantRange(APInt(32, 0)).inverse().getEquivalentICmp(Pred, RHS)); 1885 EXPECT_EQ(Pred, CmpInst::ICMP_NE); 1886 EXPECT_EQ(RHS, APInt(32, 0)); 1887 1888 EXPECT_TRUE(ConstantRange(APInt(32, -1)).getEquivalentICmp(Pred, RHS)); 1889 EXPECT_EQ(Pred, CmpInst::ICMP_EQ); 1890 EXPECT_EQ(RHS, APInt(32, -1)); 1891 1892 EXPECT_TRUE( 1893 ConstantRange(APInt(32, -1)).inverse().getEquivalentICmp(Pred, RHS)); 1894 EXPECT_EQ(Pred, CmpInst::ICMP_NE); 1895 EXPECT_EQ(RHS, APInt(32, -1)); 1896 } 1897 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)) 1906 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)); 1911 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)); 1918 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)); 1924 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)); 1932 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)); 1939 } 1940 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)); 1945 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)); 1953 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)); 1959 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)); 1964 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)); 1969 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)); 1974 } 1975 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)); 1980 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)); 1987 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)); 1993 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)); 2007 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)); 2021 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)); 2026 } 2027 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)); 2032 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)); 2037 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)); 2043 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)); 2053 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)); 2063 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)); 2068 } 2069 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. 2073 unsigned Bits = 4; 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; 2086 return; 2087 } 2088 2089 if (IsOverflowHigh) 2090 RangeHasOverflowHigh = true; 2091 else 2092 RangeHasOverflowLow = true; 2093 }); 2094 }); 2095 2096 ConstantRange::OverflowResult OR = MayOverflowFn(CR1, CR2); 2097 switch (OR) { 2098 case ConstantRange::OverflowResult::AlwaysOverflowsLow: 2099 EXPECT_TRUE(RangeHasOverflowLow); 2100 EXPECT_FALSE(RangeHasOverflowHigh); 2101 EXPECT_FALSE(RangeHasNoOverflow); 2102 break; 2103 case ConstantRange::OverflowResult::AlwaysOverflowsHigh: 2104 EXPECT_TRUE(RangeHasOverflowHigh); 2105 EXPECT_FALSE(RangeHasOverflowLow); 2106 EXPECT_FALSE(RangeHasNoOverflow); 2107 break; 2108 case ConstantRange::OverflowResult::NeverOverflows: 2109 EXPECT_FALSE(RangeHasOverflowLow); 2110 EXPECT_FALSE(RangeHasOverflowHigh); 2111 EXPECT_TRUE(RangeHasNoOverflow); 2112 break; 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()) 2118 break; 2119 2120 EXPECT_TRUE(RangeHasOverflowLow || RangeHasOverflowHigh); 2121 EXPECT_TRUE(RangeHasNoOverflow); 2122 break; 2123 } 2124 }); 2125 } 2126 2127 TEST_F(ConstantRangeTest, UnsignedAddOverflowExhaustive) { 2128 TestOverflowExhaustive( 2129 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) { 2130 bool Overflow; 2131 (void) N1.uadd_ov(N2, Overflow); 2132 IsOverflowHigh = true; 2133 return Overflow; 2134 }, 2135 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2136 return CR1.unsignedAddMayOverflow(CR2); 2137 }); 2138 } 2139 2140 TEST_F(ConstantRangeTest, UnsignedSubOverflowExhaustive) { 2141 TestOverflowExhaustive( 2142 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) { 2143 bool Overflow; 2144 (void) N1.usub_ov(N2, Overflow); 2145 IsOverflowHigh = false; 2146 return Overflow; 2147 }, 2148 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2149 return CR1.unsignedSubMayOverflow(CR2); 2150 }); 2151 } 2152 2153 TEST_F(ConstantRangeTest, UnsignedMulOverflowExhaustive) { 2154 TestOverflowExhaustive( 2155 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) { 2156 bool Overflow; 2157 (void) N1.umul_ov(N2, Overflow); 2158 IsOverflowHigh = true; 2159 return Overflow; 2160 }, 2161 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2162 return CR1.unsignedMulMayOverflow(CR2); 2163 }); 2164 } 2165 2166 TEST_F(ConstantRangeTest, SignedAddOverflowExhaustive) { 2167 TestOverflowExhaustive( 2168 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) { 2169 bool Overflow; 2170 (void) N1.sadd_ov(N2, Overflow); 2171 IsOverflowHigh = N1.isNonNegative(); 2172 return Overflow; 2173 }, 2174 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2175 return CR1.signedAddMayOverflow(CR2); 2176 }); 2177 } 2178 2179 TEST_F(ConstantRangeTest, SignedSubOverflowExhaustive) { 2180 TestOverflowExhaustive( 2181 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) { 2182 bool Overflow; 2183 (void) N1.ssub_ov(N2, Overflow); 2184 IsOverflowHigh = N1.isNonNegative(); 2185 return Overflow; 2186 }, 2187 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2188 return CR1.signedSubMayOverflow(CR2); 2189 }); 2190 } 2191 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)); 2196 2197 // .10..01. -> unsigned 01000010 (66) to 11011011 (219) 2198 // -> signed 11000010 (194) to 01011011 (91) 2199 KnownBits Known(8); 2200 Known.Zero = 36; 2201 Known.One = 66; 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)); 2206 2207 // 1.10.10. -> 10100100 (164) to 11101101 (237) 2208 Known.Zero = 18; 2209 Known.One = 164; 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)); 2213 2214 // 01.0.1.0 -> 01000100 (68) to 01101110 (110) 2215 Known.Zero = 145; 2216 Known.One = 68; 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)); 2220 } 2221 2222 TEST_F(ConstantRangeTest, FromKnownBitsExhaustive) { 2223 unsigned Bits = 4; 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) { 2228 Known.Zero = Zero; 2229 Known.One = One; 2230 if (Known.hasConflict() || Known.isUnknown()) 2231 continue; 2232 2233 UnsignedOpRangeGatherer UR(Bits); 2234 SignedOpRangeGatherer SR(Bits); 2235 for (unsigned N = 0; N < Max; ++N) { 2236 APInt Num(Bits, N); 2237 if ((Num & Known.Zero) != 0 || (~Num & Known.One) != 0) 2238 continue; 2239 2240 UR.account(Num); 2241 SR.account(Num); 2242 } 2243 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)); 2248 } 2249 } 2250 } 2251 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()); 2260 2261 unsigned Bits = 4; 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; 2270 }); 2271 assert((CR.isEmptySet() || !AllNegative || !AllNonNegative) && 2272 "Only empty set can be both all negative and all non-negative"); 2273 2274 EXPECT_EQ(AllNegative, CR.isAllNegative()); 2275 EXPECT_EQ(AllNonNegative, CR.isAllNonNegative()); 2276 }); 2277 } 2278 2279 TEST_F(ConstantRangeTest, UAddSat) { 2280 TestUnsignedBinOpExhaustive( 2281 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2282 return CR1.uadd_sat(CR2); 2283 }, 2284 [](const APInt &N1, const APInt &N2) { 2285 return N1.uadd_sat(N2); 2286 }); 2287 } 2288 2289 TEST_F(ConstantRangeTest, USubSat) { 2290 TestUnsignedBinOpExhaustive( 2291 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2292 return CR1.usub_sat(CR2); 2293 }, 2294 [](const APInt &N1, const APInt &N2) { 2295 return N1.usub_sat(N2); 2296 }); 2297 } 2298 2299 TEST_F(ConstantRangeTest, UMulSat) { 2300 TestUnsignedBinOpExhaustive( 2301 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2302 return CR1.umul_sat(CR2); 2303 }, 2304 [](const APInt &N1, const APInt &N2) { return N1.umul_sat(N2); }); 2305 } 2306 2307 TEST_F(ConstantRangeTest, UShlSat) { 2308 TestUnsignedBinOpExhaustive( 2309 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2310 return CR1.ushl_sat(CR2); 2311 }, 2312 [](const APInt &N1, const APInt &N2) { return N1.ushl_sat(N2); }); 2313 } 2314 2315 TEST_F(ConstantRangeTest, SAddSat) { 2316 TestSignedBinOpExhaustive( 2317 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2318 return CR1.sadd_sat(CR2); 2319 }, 2320 [](const APInt &N1, const APInt &N2) { 2321 return N1.sadd_sat(N2); 2322 }); 2323 } 2324 2325 TEST_F(ConstantRangeTest, SSubSat) { 2326 TestSignedBinOpExhaustive( 2327 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2328 return CR1.ssub_sat(CR2); 2329 }, 2330 [](const APInt &N1, const APInt &N2) { 2331 return N1.ssub_sat(N2); 2332 }); 2333 } 2334 2335 TEST_F(ConstantRangeTest, SMulSat) { 2336 TestSignedBinOpExhaustive( 2337 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2338 return CR1.smul_sat(CR2); 2339 }, 2340 [](const APInt &N1, const APInt &N2) { return N1.smul_sat(N2); }); 2341 } 2342 2343 TEST_F(ConstantRangeTest, SShlSat) { 2344 TestSignedBinOpExhaustive( 2345 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2346 return CR1.sshl_sat(CR2); 2347 }, 2348 [](const APInt &N1, const APInt &N2) { return N1.sshl_sat(N2); }); 2349 } 2350 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(); }); 2357 2358 TestUnsignedUnaryOpExhaustive( 2359 [](const ConstantRange &CR) { return CR.abs(/*IntMinIsPoison=*/true); }, 2360 [](const APInt &N) { return N.abs(); }, 2361 /*SkipSignedIntMin=*/true); 2362 } 2363 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()); 2369 2370 ConstantRange FpToI16 = A.castOp(Instruction::FPToSI, 16); 2371 EXPECT_EQ(16u, FpToI16.getBitWidth()); 2372 EXPECT_EQ(A, FpToI16); 2373 2374 ConstantRange FPExtToDouble = A.castOp(Instruction::FPExt, 64); 2375 EXPECT_EQ(64u, FPExtToDouble.getBitWidth()); 2376 EXPECT_TRUE(FPExtToDouble.isFullSet()); 2377 2378 ConstantRange PtrToInt = A.castOp(Instruction::PtrToInt, 64); 2379 EXPECT_EQ(64u, PtrToInt.getBitWidth()); 2380 EXPECT_TRUE(PtrToInt.isFullSet()); 2381 2382 ConstantRange IntToPtr = A.castOp(Instruction::IntToPtr, 64); 2383 EXPECT_EQ(64u, IntToPtr.getBitWidth()); 2384 EXPECT_TRUE(IntToPtr.isFullSet()); 2385 } 2386 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)); 2393 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()); 2400 } 2401 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()))); 2410 }, 2411 [](const APInt &N) { return ~N; }); 2412 TestUnsignedUnaryOpExhaustive( 2413 [](const ConstantRange &CR) { 2414 return ConstantRange(APInt::getAllOnesValue(CR.getBitWidth())) 2415 .binaryXor(CR); 2416 }, 2417 [](const APInt &N) { return ~N; }); 2418 } 2419 2420 } // anonymous namespace 2421