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 template<typename Fn1, typename Fn2> 63 static void TestUnsignedBinOpExhaustive( 64 Fn1 RangeFn, Fn2 IntFn, 65 bool SkipZeroRHS = false, bool CorrectnessOnly = false) { 66 unsigned Bits = 4; 67 EnumerateTwoConstantRanges(Bits, [&](const ConstantRange &CR1, 68 const ConstantRange &CR2) { 69 APInt Min = APInt::getMaxValue(Bits); 70 APInt Max = APInt::getMinValue(Bits); 71 ForeachNumInConstantRange(CR1, [&](const APInt &N1) { 72 ForeachNumInConstantRange(CR2, [&](const APInt &N2) { 73 if (SkipZeroRHS && N2 == 0) 74 return; 75 76 APInt N = IntFn(N1, N2); 77 if (N.ult(Min)) 78 Min = N; 79 if (N.ugt(Max)) 80 Max = N; 81 }); 82 }); 83 84 ConstantRange CR = RangeFn(CR1, CR2); 85 if (Min.ugt(Max)) { 86 EXPECT_TRUE(CR.isEmptySet()); 87 return; 88 } 89 90 ConstantRange Exact = ConstantRange::getNonEmpty(Min, Max + 1); 91 if (CorrectnessOnly) { 92 EXPECT_TRUE(CR.contains(Exact)); 93 } else { 94 EXPECT_EQ(Exact, CR); 95 } 96 }); 97 } 98 99 template<typename Fn1, typename Fn2> 100 static void TestSignedBinOpExhaustive( 101 Fn1 RangeFn, Fn2 IntFn, 102 bool SkipZeroRHS = false, bool CorrectnessOnly = false) { 103 unsigned Bits = 4; 104 EnumerateTwoConstantRanges(Bits, [&](const ConstantRange &CR1, 105 const ConstantRange &CR2) { 106 APInt Min = APInt::getSignedMaxValue(Bits); 107 APInt Max = APInt::getSignedMinValue(Bits); 108 ForeachNumInConstantRange(CR1, [&](const APInt &N1) { 109 ForeachNumInConstantRange(CR2, [&](const APInt &N2) { 110 if (SkipZeroRHS && N2 == 0) 111 return; 112 113 APInt N = IntFn(N1, N2); 114 if (N.slt(Min)) 115 Min = N; 116 if (N.sgt(Max)) 117 Max = N; 118 }); 119 }); 120 121 ConstantRange CR = RangeFn(CR1, CR2); 122 if (Min.sgt(Max)) { 123 EXPECT_TRUE(CR.isEmptySet()); 124 return; 125 } 126 127 ConstantRange Exact = ConstantRange::getNonEmpty(Min, Max + 1); 128 if (CorrectnessOnly) { 129 EXPECT_TRUE(CR.contains(Exact)); 130 } else { 131 EXPECT_EQ(Exact, CR); 132 } 133 }); 134 } 135 136 ConstantRange ConstantRangeTest::Full(16, true); 137 ConstantRange ConstantRangeTest::Empty(16, false); 138 ConstantRange ConstantRangeTest::One(APInt(16, 0xa)); 139 ConstantRange ConstantRangeTest::Some(APInt(16, 0xa), APInt(16, 0xaaa)); 140 ConstantRange ConstantRangeTest::Wrap(APInt(16, 0xaaa), APInt(16, 0xa)); 141 142 TEST_F(ConstantRangeTest, Basics) { 143 EXPECT_TRUE(Full.isFullSet()); 144 EXPECT_FALSE(Full.isEmptySet()); 145 EXPECT_TRUE(Full.inverse().isEmptySet()); 146 EXPECT_FALSE(Full.isWrappedSet()); 147 EXPECT_TRUE(Full.contains(APInt(16, 0x0))); 148 EXPECT_TRUE(Full.contains(APInt(16, 0x9))); 149 EXPECT_TRUE(Full.contains(APInt(16, 0xa))); 150 EXPECT_TRUE(Full.contains(APInt(16, 0xaa9))); 151 EXPECT_TRUE(Full.contains(APInt(16, 0xaaa))); 152 153 EXPECT_FALSE(Empty.isFullSet()); 154 EXPECT_TRUE(Empty.isEmptySet()); 155 EXPECT_TRUE(Empty.inverse().isFullSet()); 156 EXPECT_FALSE(Empty.isWrappedSet()); 157 EXPECT_FALSE(Empty.contains(APInt(16, 0x0))); 158 EXPECT_FALSE(Empty.contains(APInt(16, 0x9))); 159 EXPECT_FALSE(Empty.contains(APInt(16, 0xa))); 160 EXPECT_FALSE(Empty.contains(APInt(16, 0xaa9))); 161 EXPECT_FALSE(Empty.contains(APInt(16, 0xaaa))); 162 163 EXPECT_FALSE(One.isFullSet()); 164 EXPECT_FALSE(One.isEmptySet()); 165 EXPECT_FALSE(One.isWrappedSet()); 166 EXPECT_FALSE(One.contains(APInt(16, 0x0))); 167 EXPECT_FALSE(One.contains(APInt(16, 0x9))); 168 EXPECT_TRUE(One.contains(APInt(16, 0xa))); 169 EXPECT_FALSE(One.contains(APInt(16, 0xaa9))); 170 EXPECT_FALSE(One.contains(APInt(16, 0xaaa))); 171 EXPECT_FALSE(One.inverse().contains(APInt(16, 0xa))); 172 173 EXPECT_FALSE(Some.isFullSet()); 174 EXPECT_FALSE(Some.isEmptySet()); 175 EXPECT_FALSE(Some.isWrappedSet()); 176 EXPECT_FALSE(Some.contains(APInt(16, 0x0))); 177 EXPECT_FALSE(Some.contains(APInt(16, 0x9))); 178 EXPECT_TRUE(Some.contains(APInt(16, 0xa))); 179 EXPECT_TRUE(Some.contains(APInt(16, 0xaa9))); 180 EXPECT_FALSE(Some.contains(APInt(16, 0xaaa))); 181 182 EXPECT_FALSE(Wrap.isFullSet()); 183 EXPECT_FALSE(Wrap.isEmptySet()); 184 EXPECT_TRUE(Wrap.isWrappedSet()); 185 EXPECT_TRUE(Wrap.contains(APInt(16, 0x0))); 186 EXPECT_TRUE(Wrap.contains(APInt(16, 0x9))); 187 EXPECT_FALSE(Wrap.contains(APInt(16, 0xa))); 188 EXPECT_FALSE(Wrap.contains(APInt(16, 0xaa9))); 189 EXPECT_TRUE(Wrap.contains(APInt(16, 0xaaa))); 190 } 191 192 TEST_F(ConstantRangeTest, Equality) { 193 EXPECT_EQ(Full, Full); 194 EXPECT_EQ(Empty, Empty); 195 EXPECT_EQ(One, One); 196 EXPECT_EQ(Some, Some); 197 EXPECT_EQ(Wrap, Wrap); 198 EXPECT_NE(Full, Empty); 199 EXPECT_NE(Full, One); 200 EXPECT_NE(Full, Some); 201 EXPECT_NE(Full, Wrap); 202 EXPECT_NE(Empty, One); 203 EXPECT_NE(Empty, Some); 204 EXPECT_NE(Empty, Wrap); 205 EXPECT_NE(One, Some); 206 EXPECT_NE(One, Wrap); 207 EXPECT_NE(Some, Wrap); 208 } 209 210 TEST_F(ConstantRangeTest, SingleElement) { 211 EXPECT_EQ(Full.getSingleElement(), static_cast<APInt *>(nullptr)); 212 EXPECT_EQ(Empty.getSingleElement(), static_cast<APInt *>(nullptr)); 213 EXPECT_EQ(Full.getSingleMissingElement(), static_cast<APInt *>(nullptr)); 214 EXPECT_EQ(Empty.getSingleMissingElement(), static_cast<APInt *>(nullptr)); 215 216 EXPECT_EQ(*One.getSingleElement(), APInt(16, 0xa)); 217 EXPECT_EQ(Some.getSingleElement(), static_cast<APInt *>(nullptr)); 218 EXPECT_EQ(Wrap.getSingleElement(), static_cast<APInt *>(nullptr)); 219 220 EXPECT_EQ(One.getSingleMissingElement(), static_cast<APInt *>(nullptr)); 221 EXPECT_EQ(Some.getSingleMissingElement(), static_cast<APInt *>(nullptr)); 222 223 ConstantRange OneInverse = One.inverse(); 224 EXPECT_EQ(*OneInverse.getSingleMissingElement(), *One.getSingleElement()); 225 226 EXPECT_FALSE(Full.isSingleElement()); 227 EXPECT_FALSE(Empty.isSingleElement()); 228 EXPECT_TRUE(One.isSingleElement()); 229 EXPECT_FALSE(Some.isSingleElement()); 230 EXPECT_FALSE(Wrap.isSingleElement()); 231 } 232 233 TEST_F(ConstantRangeTest, GetMinsAndMaxes) { 234 EXPECT_EQ(Full.getUnsignedMax(), APInt(16, UINT16_MAX)); 235 EXPECT_EQ(One.getUnsignedMax(), APInt(16, 0xa)); 236 EXPECT_EQ(Some.getUnsignedMax(), APInt(16, 0xaa9)); 237 EXPECT_EQ(Wrap.getUnsignedMax(), APInt(16, UINT16_MAX)); 238 239 EXPECT_EQ(Full.getUnsignedMin(), APInt(16, 0)); 240 EXPECT_EQ(One.getUnsignedMin(), APInt(16, 0xa)); 241 EXPECT_EQ(Some.getUnsignedMin(), APInt(16, 0xa)); 242 EXPECT_EQ(Wrap.getUnsignedMin(), APInt(16, 0)); 243 244 EXPECT_EQ(Full.getSignedMax(), APInt(16, INT16_MAX)); 245 EXPECT_EQ(One.getSignedMax(), APInt(16, 0xa)); 246 EXPECT_EQ(Some.getSignedMax(), APInt(16, 0xaa9)); 247 EXPECT_EQ(Wrap.getSignedMax(), APInt(16, INT16_MAX)); 248 249 EXPECT_EQ(Full.getSignedMin(), APInt(16, (uint64_t)INT16_MIN)); 250 EXPECT_EQ(One.getSignedMin(), APInt(16, 0xa)); 251 EXPECT_EQ(Some.getSignedMin(), APInt(16, 0xa)); 252 EXPECT_EQ(Wrap.getSignedMin(), APInt(16, (uint64_t)INT16_MIN)); 253 254 // Found by Klee 255 EXPECT_EQ(ConstantRange(APInt(4, 7), APInt(4, 0)).getSignedMax(), 256 APInt(4, 7)); 257 } 258 259 TEST_F(ConstantRangeTest, SignWrapped) { 260 EXPECT_FALSE(Full.isSignWrappedSet()); 261 EXPECT_FALSE(Empty.isSignWrappedSet()); 262 EXPECT_FALSE(One.isSignWrappedSet()); 263 EXPECT_FALSE(Some.isSignWrappedSet()); 264 EXPECT_TRUE(Wrap.isSignWrappedSet()); 265 266 EXPECT_FALSE(ConstantRange(APInt(8, 127), APInt(8, 128)).isSignWrappedSet()); 267 EXPECT_TRUE(ConstantRange(APInt(8, 127), APInt(8, 129)).isSignWrappedSet()); 268 EXPECT_FALSE(ConstantRange(APInt(8, 128), APInt(8, 129)).isSignWrappedSet()); 269 EXPECT_TRUE(ConstantRange(APInt(8, 10), APInt(8, 9)).isSignWrappedSet()); 270 EXPECT_TRUE(ConstantRange(APInt(8, 10), APInt(8, 250)).isSignWrappedSet()); 271 EXPECT_FALSE(ConstantRange(APInt(8, 250), APInt(8, 10)).isSignWrappedSet()); 272 EXPECT_FALSE(ConstantRange(APInt(8, 250), APInt(8, 251)).isSignWrappedSet()); 273 } 274 275 TEST_F(ConstantRangeTest, UpperWrapped) { 276 // The behavior here is the same as for isWrappedSet() / isSignWrappedSet(). 277 EXPECT_FALSE(Full.isUpperWrapped()); 278 EXPECT_FALSE(Empty.isUpperWrapped()); 279 EXPECT_FALSE(One.isUpperWrapped()); 280 EXPECT_FALSE(Some.isUpperWrapped()); 281 EXPECT_TRUE(Wrap.isUpperWrapped()); 282 EXPECT_FALSE(Full.isUpperSignWrapped()); 283 EXPECT_FALSE(Empty.isUpperSignWrapped()); 284 EXPECT_FALSE(One.isUpperSignWrapped()); 285 EXPECT_FALSE(Some.isUpperSignWrapped()); 286 EXPECT_TRUE(Wrap.isUpperSignWrapped()); 287 288 // The behavior differs if Upper is the Min/SignedMin value. 289 ConstantRange CR1(APInt(8, 42), APInt::getMinValue(8)); 290 EXPECT_FALSE(CR1.isWrappedSet()); 291 EXPECT_TRUE(CR1.isUpperWrapped()); 292 293 ConstantRange CR2(APInt(8, 42), APInt::getSignedMinValue(8)); 294 EXPECT_FALSE(CR2.isSignWrappedSet()); 295 EXPECT_TRUE(CR2.isUpperSignWrapped()); 296 } 297 298 TEST_F(ConstantRangeTest, Trunc) { 299 ConstantRange TFull = Full.truncate(10); 300 ConstantRange TEmpty = Empty.truncate(10); 301 ConstantRange TOne = One.truncate(10); 302 ConstantRange TSome = Some.truncate(10); 303 ConstantRange TWrap = Wrap.truncate(10); 304 EXPECT_TRUE(TFull.isFullSet()); 305 EXPECT_TRUE(TEmpty.isEmptySet()); 306 EXPECT_EQ(TOne, ConstantRange(One.getLower().trunc(10), 307 One.getUpper().trunc(10))); 308 EXPECT_TRUE(TSome.isFullSet()); 309 EXPECT_TRUE(TWrap.isFullSet()); 310 311 // trunc([2, 5), 3->2) = [2, 1) 312 ConstantRange TwoFive(APInt(3, 2), APInt(3, 5)); 313 EXPECT_EQ(TwoFive.truncate(2), ConstantRange(APInt(2, 2), APInt(2, 1))); 314 315 // trunc([2, 6), 3->2) = full 316 ConstantRange TwoSix(APInt(3, 2), APInt(3, 6)); 317 EXPECT_TRUE(TwoSix.truncate(2).isFullSet()); 318 319 // trunc([5, 7), 3->2) = [1, 3) 320 ConstantRange FiveSeven(APInt(3, 5), APInt(3, 7)); 321 EXPECT_EQ(FiveSeven.truncate(2), ConstantRange(APInt(2, 1), APInt(2, 3))); 322 323 // trunc([7, 1), 3->2) = [3, 1) 324 ConstantRange SevenOne(APInt(3, 7), APInt(3, 1)); 325 EXPECT_EQ(SevenOne.truncate(2), ConstantRange(APInt(2, 3), APInt(2, 1))); 326 } 327 328 TEST_F(ConstantRangeTest, ZExt) { 329 ConstantRange ZFull = Full.zeroExtend(20); 330 ConstantRange ZEmpty = Empty.zeroExtend(20); 331 ConstantRange ZOne = One.zeroExtend(20); 332 ConstantRange ZSome = Some.zeroExtend(20); 333 ConstantRange ZWrap = Wrap.zeroExtend(20); 334 EXPECT_EQ(ZFull, ConstantRange(APInt(20, 0), APInt(20, 0x10000))); 335 EXPECT_TRUE(ZEmpty.isEmptySet()); 336 EXPECT_EQ(ZOne, ConstantRange(One.getLower().zext(20), 337 One.getUpper().zext(20))); 338 EXPECT_EQ(ZSome, ConstantRange(Some.getLower().zext(20), 339 Some.getUpper().zext(20))); 340 EXPECT_EQ(ZWrap, ConstantRange(APInt(20, 0), APInt(20, 0x10000))); 341 342 // zext([5, 0), 3->7) = [5, 8) 343 ConstantRange FiveZero(APInt(3, 5), APInt(3, 0)); 344 EXPECT_EQ(FiveZero.zeroExtend(7), ConstantRange(APInt(7, 5), APInt(7, 8))); 345 } 346 347 TEST_F(ConstantRangeTest, SExt) { 348 ConstantRange SFull = Full.signExtend(20); 349 ConstantRange SEmpty = Empty.signExtend(20); 350 ConstantRange SOne = One.signExtend(20); 351 ConstantRange SSome = Some.signExtend(20); 352 ConstantRange SWrap = Wrap.signExtend(20); 353 EXPECT_EQ(SFull, ConstantRange(APInt(20, (uint64_t)INT16_MIN, true), 354 APInt(20, INT16_MAX + 1, true))); 355 EXPECT_TRUE(SEmpty.isEmptySet()); 356 EXPECT_EQ(SOne, ConstantRange(One.getLower().sext(20), 357 One.getUpper().sext(20))); 358 EXPECT_EQ(SSome, ConstantRange(Some.getLower().sext(20), 359 Some.getUpper().sext(20))); 360 EXPECT_EQ(SWrap, ConstantRange(APInt(20, (uint64_t)INT16_MIN, true), 361 APInt(20, INT16_MAX + 1, true))); 362 363 EXPECT_EQ(ConstantRange(APInt(8, 120), APInt(8, 140)).signExtend(16), 364 ConstantRange(APInt(16, -128), APInt(16, 128))); 365 366 EXPECT_EQ(ConstantRange(APInt(16, 0x0200), APInt(16, 0x8000)).signExtend(19), 367 ConstantRange(APInt(19, 0x0200), APInt(19, 0x8000))); 368 } 369 370 TEST_F(ConstantRangeTest, IntersectWith) { 371 EXPECT_EQ(Empty.intersectWith(Full), Empty); 372 EXPECT_EQ(Empty.intersectWith(Empty), Empty); 373 EXPECT_EQ(Empty.intersectWith(One), Empty); 374 EXPECT_EQ(Empty.intersectWith(Some), Empty); 375 EXPECT_EQ(Empty.intersectWith(Wrap), Empty); 376 EXPECT_EQ(Full.intersectWith(Full), Full); 377 EXPECT_EQ(Some.intersectWith(Some), Some); 378 EXPECT_EQ(Some.intersectWith(One), One); 379 EXPECT_EQ(Full.intersectWith(One), One); 380 EXPECT_EQ(Full.intersectWith(Some), Some); 381 EXPECT_EQ(Some.intersectWith(Wrap), Empty); 382 EXPECT_EQ(One.intersectWith(Wrap), Empty); 383 EXPECT_EQ(One.intersectWith(Wrap), Wrap.intersectWith(One)); 384 385 // Klee generated testcase from PR4545. 386 // The intersection of i16 [4, 2) and [6, 5) is disjoint, looking like 387 // 01..4.6789ABCDEF where the dots represent values not in the intersection. 388 ConstantRange LHS(APInt(16, 4), APInt(16, 2)); 389 ConstantRange RHS(APInt(16, 6), APInt(16, 5)); 390 EXPECT_TRUE(LHS.intersectWith(RHS) == LHS); 391 392 // previous bug: intersection of [min, 3) and [2, max) should be 2 393 LHS = ConstantRange(APInt(32, -2147483646), APInt(32, 3)); 394 RHS = ConstantRange(APInt(32, 2), APInt(32, 2147483646)); 395 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 2))); 396 397 // [2, 0) /\ [4, 3) = [2, 0) 398 LHS = ConstantRange(APInt(32, 2), APInt(32, 0)); 399 RHS = ConstantRange(APInt(32, 4), APInt(32, 3)); 400 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 2), APInt(32, 0))); 401 402 // [2, 0) /\ [4, 2) = [4, 0) 403 LHS = ConstantRange(APInt(32, 2), APInt(32, 0)); 404 RHS = ConstantRange(APInt(32, 4), APInt(32, 2)); 405 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 4), APInt(32, 0))); 406 407 // [4, 2) /\ [5, 1) = [5, 1) 408 LHS = ConstantRange(APInt(32, 4), APInt(32, 2)); 409 RHS = ConstantRange(APInt(32, 5), APInt(32, 1)); 410 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 5), APInt(32, 1))); 411 412 // [2, 0) /\ [7, 4) = [7, 4) 413 LHS = ConstantRange(APInt(32, 2), APInt(32, 0)); 414 RHS = ConstantRange(APInt(32, 7), APInt(32, 4)); 415 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 7), APInt(32, 4))); 416 417 // [4, 2) /\ [1, 0) = [1, 0) 418 LHS = ConstantRange(APInt(32, 4), APInt(32, 2)); 419 RHS = ConstantRange(APInt(32, 1), APInt(32, 0)); 420 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 4), APInt(32, 2))); 421 422 // [15, 0) /\ [7, 6) = [15, 0) 423 LHS = ConstantRange(APInt(32, 15), APInt(32, 0)); 424 RHS = ConstantRange(APInt(32, 7), APInt(32, 6)); 425 EXPECT_EQ(LHS.intersectWith(RHS), ConstantRange(APInt(32, 15), APInt(32, 0))); 426 } 427 428 template<typename Fn1, typename Fn2> 429 void testBinarySetOperationExhaustive(Fn1 OpFn, Fn2 InResultFn) { 430 unsigned Bits = 4; 431 EnumerateTwoConstantRanges(Bits, 432 [=](const ConstantRange &CR1, const ConstantRange &CR2) { 433 // Collect up to three contiguous unsigned ranges. The HaveInterrupt 434 // variables are used determine when we have to switch to the next 435 // range because the previous one ended. 436 APInt Lower1(Bits, 0), Upper1(Bits, 0); 437 APInt Lower2(Bits, 0), Upper2(Bits, 0); 438 APInt Lower3(Bits, 0), Upper3(Bits, 0); 439 bool HaveRange1 = false, HaveInterrupt1 = false; 440 bool HaveRange2 = false, HaveInterrupt2 = false; 441 bool HaveRange3 = false, HaveInterrupt3 = false; 442 443 APInt Num(Bits, 0); 444 for (unsigned I = 0, Limit = 1 << Bits; I < Limit; ++I, ++Num) { 445 if (!InResultFn(CR1, CR2, Num)) { 446 if (HaveRange3) 447 HaveInterrupt3 = true; 448 else if (HaveRange2) 449 HaveInterrupt2 = true; 450 else if (HaveRange1) 451 HaveInterrupt1 = true; 452 continue; 453 } 454 455 if (HaveRange3) { 456 Upper3 = Num; 457 } else if (HaveInterrupt2) { 458 HaveRange3 = true; 459 Lower3 = Upper3 = Num; 460 } else if (HaveRange2) { 461 Upper2 = Num; 462 } else if (HaveInterrupt1) { 463 HaveRange2 = true; 464 Lower2 = Upper2 = Num; 465 } else if (HaveRange1) { 466 Upper1 = Num; 467 } else { 468 HaveRange1 = true; 469 Lower1 = Upper1 = Num; 470 } 471 } 472 473 (void)HaveInterrupt3; 474 assert(!HaveInterrupt3 && "Should have at most three ranges"); 475 476 ConstantRange SmallestCR = OpFn(CR1, CR2, ConstantRange::Smallest); 477 ConstantRange UnsignedCR = OpFn(CR1, CR2, ConstantRange::Unsigned); 478 ConstantRange SignedCR = OpFn(CR1, CR2, ConstantRange::Signed); 479 480 if (!HaveRange1) { 481 EXPECT_TRUE(SmallestCR.isEmptySet()); 482 EXPECT_TRUE(UnsignedCR.isEmptySet()); 483 EXPECT_TRUE(SignedCR.isEmptySet()); 484 return; 485 } 486 487 if (!HaveRange2) { 488 if (Lower1 == Upper1 + 1) { 489 EXPECT_TRUE(SmallestCR.isFullSet()); 490 EXPECT_TRUE(UnsignedCR.isFullSet()); 491 EXPECT_TRUE(SignedCR.isFullSet()); 492 } else { 493 ConstantRange Expected(Lower1, Upper1 + 1); 494 EXPECT_EQ(Expected, SmallestCR); 495 EXPECT_EQ(Expected, UnsignedCR); 496 EXPECT_EQ(Expected, SignedCR); 497 } 498 return; 499 } 500 501 ConstantRange Variant1(Bits, /*full*/ true); 502 ConstantRange Variant2(Bits, /*full*/ true); 503 if (!HaveRange3) { 504 // Compute the two possible ways to cover two disjoint ranges. 505 if (Lower1 != Upper2 + 1) 506 Variant1 = ConstantRange(Lower1, Upper2 + 1); 507 if (Lower2 != Upper1 + 1) 508 Variant2 = ConstantRange(Lower2, Upper1 + 1); 509 } else { 510 // If we have three ranges, the first and last one have to be adjacent 511 // to the unsigned domain. It's better to think of this as having two 512 // holes, and we can construct one range using each hole. 513 assert(Lower1.isNullValue() && Upper3.isMaxValue()); 514 Variant1 = ConstantRange(Lower2, Upper1 + 1); 515 Variant2 = ConstantRange(Lower3, Upper2 + 1); 516 } 517 518 // Smallest: Smaller set, then any set. 519 if (Variant1.isSizeStrictlySmallerThan(Variant2)) 520 EXPECT_EQ(Variant1, SmallestCR); 521 else if (Variant2.isSizeStrictlySmallerThan(Variant1)) 522 EXPECT_EQ(Variant2, SmallestCR); 523 else 524 EXPECT_TRUE(Variant1 == SmallestCR || Variant2 == SmallestCR); 525 526 // Unsigned: Non-wrapped set, then smaller set, then any set. 527 bool Variant1Full = Variant1.isFullSet() || Variant1.isWrappedSet(); 528 bool Variant2Full = Variant2.isFullSet() || Variant2.isWrappedSet(); 529 if (!Variant1Full && Variant2Full) 530 EXPECT_EQ(Variant1, UnsignedCR); 531 else if (Variant1Full && !Variant2Full) 532 EXPECT_EQ(Variant2, UnsignedCR); 533 else if (Variant1.isSizeStrictlySmallerThan(Variant2)) 534 EXPECT_EQ(Variant1, UnsignedCR); 535 else if (Variant2.isSizeStrictlySmallerThan(Variant1)) 536 EXPECT_EQ(Variant2, UnsignedCR); 537 else 538 EXPECT_TRUE(Variant1 == UnsignedCR || Variant2 == UnsignedCR); 539 540 // Signed: Signed non-wrapped set, then smaller set, then any set. 541 Variant1Full = Variant1.isFullSet() || Variant1.isSignWrappedSet(); 542 Variant2Full = Variant2.isFullSet() || Variant2.isSignWrappedSet(); 543 if (!Variant1Full && Variant2Full) 544 EXPECT_EQ(Variant1, SignedCR); 545 else if (Variant1Full && !Variant2Full) 546 EXPECT_EQ(Variant2, SignedCR); 547 else if (Variant1.isSizeStrictlySmallerThan(Variant2)) 548 EXPECT_EQ(Variant1, SignedCR); 549 else if (Variant2.isSizeStrictlySmallerThan(Variant1)) 550 EXPECT_EQ(Variant2, SignedCR); 551 else 552 EXPECT_TRUE(Variant1 == SignedCR || Variant2 == SignedCR); 553 }); 554 } 555 556 TEST_F(ConstantRangeTest, IntersectWithExhaustive) { 557 testBinarySetOperationExhaustive( 558 [](const ConstantRange &CR1, const ConstantRange &CR2, 559 ConstantRange::PreferredRangeType Type) { 560 return CR1.intersectWith(CR2, Type); 561 }, 562 [](const ConstantRange &CR1, const ConstantRange &CR2, const APInt &N) { 563 return CR1.contains(N) && CR2.contains(N); 564 }); 565 } 566 567 TEST_F(ConstantRangeTest, UnionWithExhaustive) { 568 testBinarySetOperationExhaustive( 569 [](const ConstantRange &CR1, const ConstantRange &CR2, 570 ConstantRange::PreferredRangeType Type) { 571 return CR1.unionWith(CR2, Type); 572 }, 573 [](const ConstantRange &CR1, const ConstantRange &CR2, const APInt &N) { 574 return CR1.contains(N) || CR2.contains(N); 575 }); 576 } 577 578 TEST_F(ConstantRangeTest, UnionWith) { 579 EXPECT_EQ(Wrap.unionWith(One), 580 ConstantRange(APInt(16, 0xaaa), APInt(16, 0xb))); 581 EXPECT_EQ(One.unionWith(Wrap), Wrap.unionWith(One)); 582 EXPECT_EQ(Empty.unionWith(Empty), Empty); 583 EXPECT_EQ(Full.unionWith(Full), Full); 584 EXPECT_EQ(Some.unionWith(Wrap), Full); 585 586 // PR4545 587 EXPECT_EQ(ConstantRange(APInt(16, 14), APInt(16, 1)).unionWith( 588 ConstantRange(APInt(16, 0), APInt(16, 8))), 589 ConstantRange(APInt(16, 14), APInt(16, 8))); 590 EXPECT_EQ(ConstantRange(APInt(16, 6), APInt(16, 4)).unionWith( 591 ConstantRange(APInt(16, 4), APInt(16, 0))), 592 ConstantRange::getFull(16)); 593 EXPECT_EQ(ConstantRange(APInt(16, 1), APInt(16, 0)).unionWith( 594 ConstantRange(APInt(16, 2), APInt(16, 1))), 595 ConstantRange::getFull(16)); 596 } 597 598 TEST_F(ConstantRangeTest, SetDifference) { 599 EXPECT_EQ(Full.difference(Empty), Full); 600 EXPECT_EQ(Full.difference(Full), Empty); 601 EXPECT_EQ(Empty.difference(Empty), Empty); 602 EXPECT_EQ(Empty.difference(Full), Empty); 603 604 ConstantRange A(APInt(16, 3), APInt(16, 7)); 605 ConstantRange B(APInt(16, 5), APInt(16, 9)); 606 ConstantRange C(APInt(16, 3), APInt(16, 5)); 607 ConstantRange D(APInt(16, 7), APInt(16, 9)); 608 ConstantRange E(APInt(16, 5), APInt(16, 4)); 609 ConstantRange F(APInt(16, 7), APInt(16, 3)); 610 EXPECT_EQ(A.difference(B), C); 611 EXPECT_EQ(B.difference(A), D); 612 EXPECT_EQ(E.difference(A), F); 613 } 614 615 TEST_F(ConstantRangeTest, SubtractAPInt) { 616 EXPECT_EQ(Full.subtract(APInt(16, 4)), Full); 617 EXPECT_EQ(Empty.subtract(APInt(16, 4)), Empty); 618 EXPECT_EQ(Some.subtract(APInt(16, 4)), 619 ConstantRange(APInt(16, 0x6), APInt(16, 0xaa6))); 620 EXPECT_EQ(Wrap.subtract(APInt(16, 4)), 621 ConstantRange(APInt(16, 0xaa6), APInt(16, 0x6))); 622 EXPECT_EQ(One.subtract(APInt(16, 4)), 623 ConstantRange(APInt(16, 0x6))); 624 } 625 626 TEST_F(ConstantRangeTest, Add) { 627 EXPECT_EQ(Full.add(APInt(16, 4)), Full); 628 EXPECT_EQ(Full.add(Full), Full); 629 EXPECT_EQ(Full.add(Empty), Empty); 630 EXPECT_EQ(Full.add(One), Full); 631 EXPECT_EQ(Full.add(Some), Full); 632 EXPECT_EQ(Full.add(Wrap), Full); 633 EXPECT_EQ(Empty.add(Empty), Empty); 634 EXPECT_EQ(Empty.add(One), Empty); 635 EXPECT_EQ(Empty.add(Some), Empty); 636 EXPECT_EQ(Empty.add(Wrap), Empty); 637 EXPECT_EQ(Empty.add(APInt(16, 4)), Empty); 638 EXPECT_EQ(Some.add(APInt(16, 4)), 639 ConstantRange(APInt(16, 0xe), APInt(16, 0xaae))); 640 EXPECT_EQ(Wrap.add(APInt(16, 4)), 641 ConstantRange(APInt(16, 0xaae), APInt(16, 0xe))); 642 EXPECT_EQ(One.add(APInt(16, 4)), 643 ConstantRange(APInt(16, 0xe))); 644 } 645 646 template <typename Fn1, typename Fn2> 647 static void TestAddWithNoSignedWrapExhaustive(Fn1 RangeFn, Fn2 IntFn) { 648 unsigned Bits = 4; 649 EnumerateTwoConstantRanges(Bits, [&](const ConstantRange &CR1, 650 const ConstantRange &CR2) { 651 ConstantRange CR = RangeFn(CR1, CR2); 652 APInt Min = APInt::getSignedMaxValue(Bits); 653 APInt Max = APInt::getSignedMinValue(Bits); 654 bool AllOverflow = true; 655 ForeachNumInConstantRange(CR1, [&](const APInt &N1) { 656 ForeachNumInConstantRange(CR2, [&](const APInt &N2) { 657 bool IsOverflow = false; 658 APInt N = IntFn(IsOverflow, N1, N2); 659 if (!IsOverflow) { 660 AllOverflow = false; 661 if (N.slt(Min)) 662 Min = N; 663 if (N.sgt(Max)) 664 Max = N; 665 EXPECT_TRUE(CR.contains(N)); 666 } 667 }); 668 }); 669 670 EXPECT_EQ(CR.isEmptySet(), AllOverflow); 671 672 if (!CR1.isSignWrappedSet() && !CR2.isSignWrappedSet()) { 673 if (Min.sgt(Max)) { 674 EXPECT_TRUE(CR.isEmptySet()); 675 return; 676 } 677 678 ConstantRange Exact = ConstantRange::getNonEmpty(Min, Max + 1); 679 EXPECT_EQ(Exact, CR); 680 } 681 }); 682 } 683 684 template <typename Fn1, typename Fn2> 685 static void TestAddWithNoUnsignedWrapExhaustive(Fn1 RangeFn, Fn2 IntFn) { 686 unsigned Bits = 4; 687 EnumerateTwoConstantRanges(Bits, [&](const ConstantRange &CR1, 688 const ConstantRange &CR2) { 689 ConstantRange CR = RangeFn(CR1, CR2); 690 APInt Min = APInt::getMaxValue(Bits); 691 APInt Max = APInt::getMinValue(Bits); 692 bool AllOverflow = true; 693 ForeachNumInConstantRange(CR1, [&](const APInt &N1) { 694 ForeachNumInConstantRange(CR2, [&](const APInt &N2) { 695 bool IsOverflow = false; 696 APInt N = IntFn(IsOverflow, N1, N2); 697 if (!IsOverflow) { 698 AllOverflow = false; 699 if (N.ult(Min)) 700 Min = N; 701 if (N.ugt(Max)) 702 Max = N; 703 EXPECT_TRUE(CR.contains(N)); 704 } 705 }); 706 }); 707 708 EXPECT_EQ(CR.isEmptySet(), AllOverflow); 709 710 if (!CR1.isWrappedSet() && !CR2.isWrappedSet()) { 711 if (Min.ugt(Max)) { 712 EXPECT_TRUE(CR.isEmptySet()); 713 return; 714 } 715 716 ConstantRange Exact = ConstantRange::getNonEmpty(Min, Max + 1); 717 EXPECT_EQ(Exact, CR); 718 } 719 }); 720 } 721 722 template <typename Fn1, typename Fn2, typename Fn3> 723 static void TestAddWithNoSignedUnsignedWrapExhaustive(Fn1 RangeFn, 724 Fn2 IntFnSigned, 725 Fn3 IntFnUnsigned) { 726 unsigned Bits = 4; 727 EnumerateTwoConstantRanges( 728 Bits, [&](const ConstantRange &CR1, const ConstantRange &CR2) { 729 ConstantRange CR = RangeFn(CR1, CR2); 730 APInt UMin = APInt::getMaxValue(Bits); 731 APInt UMax = APInt::getMinValue(Bits); 732 APInt SMin = APInt::getSignedMaxValue(Bits); 733 APInt SMax = APInt::getSignedMinValue(Bits); 734 bool AllOverflow = true; 735 ForeachNumInConstantRange(CR1, [&](const APInt &N1) { 736 ForeachNumInConstantRange(CR2, [&](const APInt &N2) { 737 bool IsOverflow = false, IsSignedOverflow = false; 738 APInt N = IntFnSigned(IsSignedOverflow, N1, N2); 739 (void) IntFnUnsigned(IsOverflow, N1, N2); 740 if (!IsSignedOverflow && !IsOverflow) { 741 AllOverflow = false; 742 if (N.slt(SMin)) 743 SMin = N; 744 if (N.sgt(SMax)) 745 SMax = N; 746 if (N.ult(UMin)) 747 UMin = N; 748 if (N.ugt(UMax)) 749 UMax = N; 750 EXPECT_TRUE(CR.contains(N)); 751 } 752 }); 753 }); 754 755 EXPECT_EQ(CR.isEmptySet(), AllOverflow); 756 757 if (!CR1.isWrappedSet() && !CR2.isWrappedSet() && 758 !CR1.isSignWrappedSet() && !CR2.isSignWrappedSet()) { 759 if (UMin.ugt(UMax) || SMin.sgt(SMax)) { 760 EXPECT_TRUE(CR.isEmptySet()); 761 return; 762 } 763 764 ConstantRange Exact = 765 ConstantRange::getNonEmpty(SMin, SMax + 1) 766 .intersectWith(ConstantRange::getNonEmpty(UMin, UMax + 1)); 767 EXPECT_EQ(Exact, CR); 768 } 769 }); 770 } 771 772 TEST_F(ConstantRangeTest, AddWithNoWrap) { 773 typedef OverflowingBinaryOperator OBO; 774 EXPECT_EQ(Empty.addWithNoWrap(Some, OBO::NoSignedWrap), Empty); 775 EXPECT_EQ(Some.addWithNoWrap(Empty, OBO::NoSignedWrap), Empty); 776 EXPECT_EQ(Full.addWithNoWrap(Full, OBO::NoSignedWrap), Full); 777 EXPECT_NE(Full.addWithNoWrap(Some, OBO::NoSignedWrap), Full); 778 EXPECT_NE(Some.addWithNoWrap(Full, OBO::NoSignedWrap), Full); 779 EXPECT_EQ(Full.addWithNoWrap(ConstantRange(APInt(16, 1), APInt(16, 2)), 780 OBO::NoSignedWrap), 781 ConstantRange(APInt(16, INT16_MIN + 1), APInt(16, INT16_MIN))); 782 EXPECT_EQ(ConstantRange(APInt(16, 1), APInt(16, 2)) 783 .addWithNoWrap(Full, OBO::NoSignedWrap), 784 ConstantRange(APInt(16, INT16_MIN + 1), APInt(16, INT16_MIN))); 785 EXPECT_EQ(Full.addWithNoWrap(ConstantRange(APInt(16, -1), APInt(16, 0)), 786 OBO::NoSignedWrap), 787 ConstantRange(APInt(16, INT16_MIN), APInt(16, INT16_MAX))); 788 EXPECT_EQ(ConstantRange(APInt(8, 100), APInt(8, 120)) 789 .addWithNoWrap(ConstantRange(APInt(8, 120), APInt(8, 123)), 790 OBO::NoSignedWrap), 791 ConstantRange(8, false)); 792 EXPECT_EQ(ConstantRange(APInt(8, -120), APInt(8, -100)) 793 .addWithNoWrap(ConstantRange(APInt(8, -110), APInt(8, -100)), 794 OBO::NoSignedWrap), 795 ConstantRange(8, false)); 796 EXPECT_EQ(ConstantRange(APInt(8, 0), APInt(8, 101)) 797 .addWithNoWrap(ConstantRange(APInt(8, -128), APInt(8, 28)), 798 OBO::NoSignedWrap), 799 ConstantRange(8, true)); 800 EXPECT_EQ(ConstantRange(APInt(8, 0), APInt(8, 101)) 801 .addWithNoWrap(ConstantRange(APInt(8, -120), APInt(8, 29)), 802 OBO::NoSignedWrap), 803 ConstantRange(APInt(8, -120), APInt(8, -128))); 804 EXPECT_EQ(ConstantRange(APInt(8, -50), APInt(8, 50)) 805 .addWithNoWrap(ConstantRange(APInt(8, 10), APInt(8, 20)), 806 OBO::NoSignedWrap), 807 ConstantRange(APInt(8, -40), APInt(8, 69))); 808 EXPECT_EQ(ConstantRange(APInt(8, 10), APInt(8, 20)) 809 .addWithNoWrap(ConstantRange(APInt(8, -50), APInt(8, 50)), 810 OBO::NoSignedWrap), 811 ConstantRange(APInt(8, -40), APInt(8, 69))); 812 EXPECT_EQ(ConstantRange(APInt(8, 120), APInt(8, -10)) 813 .addWithNoWrap(ConstantRange(APInt(8, 5), APInt(8, 20)), 814 OBO::NoSignedWrap), 815 ConstantRange(APInt(8, 125), APInt(8, 9))); 816 EXPECT_EQ(ConstantRange(APInt(8, 5), APInt(8, 20)) 817 .addWithNoWrap(ConstantRange(APInt(8, 120), APInt(8, -10)), 818 OBO::NoSignedWrap), 819 ConstantRange(APInt(8, 125), APInt(8, 9))); 820 821 TestAddWithNoSignedWrapExhaustive( 822 [](const ConstantRange &CR1, const ConstantRange &CR2) { 823 return CR1.addWithNoWrap(CR2, OBO::NoSignedWrap); 824 }, 825 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 826 return N1.sadd_ov(N2, IsOverflow); 827 }); 828 829 EXPECT_EQ(Empty.addWithNoWrap(Some, OBO::NoUnsignedWrap), Empty); 830 EXPECT_EQ(Some.addWithNoWrap(Empty, OBO::NoUnsignedWrap), Empty); 831 EXPECT_EQ(Full.addWithNoWrap(Full, OBO::NoUnsignedWrap), Full); 832 EXPECT_NE(Full.addWithNoWrap(Some, OBO::NoUnsignedWrap), Full); 833 EXPECT_NE(Some.addWithNoWrap(Full, OBO::NoUnsignedWrap), Full); 834 EXPECT_EQ(Full.addWithNoWrap(ConstantRange(APInt(16, 1), APInt(16, 2)), 835 OBO::NoUnsignedWrap), 836 ConstantRange(APInt(16, 1), APInt(16, 0))); 837 EXPECT_EQ(ConstantRange(APInt(16, 1), APInt(16, 2)) 838 .addWithNoWrap(Full, OBO::NoUnsignedWrap), 839 ConstantRange(APInt(16, 1), APInt(16, 0))); 840 EXPECT_EQ(ConstantRange(APInt(8, 200), APInt(8, 220)) 841 .addWithNoWrap(ConstantRange(APInt(8, 100), APInt(8, 123)), 842 OBO::NoUnsignedWrap), 843 ConstantRange(8, false)); 844 EXPECT_EQ(ConstantRange(APInt(8, 0), APInt(8, 101)) 845 .addWithNoWrap(ConstantRange(APInt(8, 0), APInt(8, 156)), 846 OBO::NoUnsignedWrap), 847 ConstantRange(8, true)); 848 EXPECT_EQ(ConstantRange(APInt(8, 0), APInt(8, 101)) 849 .addWithNoWrap(ConstantRange(APInt(8, 10), APInt(8, 29)), 850 OBO::NoUnsignedWrap), 851 ConstantRange(APInt(8, 10), APInt(8, 129))); 852 EXPECT_EQ(ConstantRange(APInt(8, 20), APInt(8, 10)) 853 .addWithNoWrap(ConstantRange(APInt(8, 50), APInt(8, 200)), 854 OBO::NoUnsignedWrap), 855 ConstantRange(APInt(8, 50), APInt(8, 0))); 856 EXPECT_EQ(ConstantRange(APInt(8, 10), APInt(8, 20)) 857 .addWithNoWrap(ConstantRange(APInt(8, 50), APInt(8, 200)), 858 OBO::NoUnsignedWrap), 859 ConstantRange(APInt(8, 60), APInt(8, -37))); 860 EXPECT_EQ(ConstantRange(APInt(8, 20), APInt(8, -30)) 861 .addWithNoWrap(ConstantRange(APInt(8, 5), APInt(8, 20)), 862 OBO::NoUnsignedWrap), 863 ConstantRange(APInt(8, 25), APInt(8, -11))); 864 EXPECT_EQ(ConstantRange(APInt(8, 5), APInt(8, 20)) 865 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, -30)), 866 OBO::NoUnsignedWrap), 867 ConstantRange(APInt(8, 25), APInt(8, -11))); 868 869 TestAddWithNoUnsignedWrapExhaustive( 870 [](const ConstantRange &CR1, const ConstantRange &CR2) { 871 return CR1.addWithNoWrap(CR2, OBO::NoUnsignedWrap); 872 }, 873 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 874 return N1.uadd_ov(N2, IsOverflow); 875 }); 876 877 EXPECT_EQ(ConstantRange(APInt(8, 50), APInt(8, 100)) 878 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 70)), 879 OBO::NoSignedWrap), 880 ConstantRange(APInt(8, 70), APInt(8, -128))); 881 EXPECT_EQ(ConstantRange(APInt(8, 50), APInt(8, 100)) 882 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 70)), 883 OBO::NoUnsignedWrap), 884 ConstantRange(APInt(8, 70), APInt(8, 169))); 885 EXPECT_EQ(ConstantRange(APInt(8, 50), APInt(8, 100)) 886 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 70)), 887 OBO::NoUnsignedWrap | OBO::NoSignedWrap), 888 ConstantRange(APInt(8, 70), APInt(8, -128))); 889 890 EXPECT_EQ(ConstantRange(APInt(8, -100), APInt(8, -50)) 891 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 30)), 892 OBO::NoSignedWrap), 893 ConstantRange(APInt(8, -80), APInt(8, -21))); 894 EXPECT_EQ(ConstantRange(APInt(8, -100), APInt(8, -50)) 895 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 30)), 896 OBO::NoUnsignedWrap), 897 ConstantRange(APInt(8, 176), APInt(8, 235))); 898 EXPECT_EQ(ConstantRange(APInt(8, -100), APInt(8, -50)) 899 .addWithNoWrap(ConstantRange(APInt(8, 20), APInt(8, 30)), 900 OBO::NoUnsignedWrap | OBO::NoSignedWrap), 901 ConstantRange(APInt(8, 176), APInt(8, 235))); 902 903 TestAddWithNoSignedUnsignedWrapExhaustive( 904 [](const ConstantRange &CR1, const ConstantRange &CR2) { 905 return CR1.addWithNoWrap(CR2, OBO::NoUnsignedWrap | OBO::NoSignedWrap); 906 }, 907 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 908 return N1.sadd_ov(N2, IsOverflow); 909 }, 910 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 911 return N1.uadd_ov(N2, IsOverflow); 912 }); 913 } 914 915 TEST_F(ConstantRangeTest, Sub) { 916 EXPECT_EQ(Full.sub(APInt(16, 4)), Full); 917 EXPECT_EQ(Full.sub(Full), Full); 918 EXPECT_EQ(Full.sub(Empty), Empty); 919 EXPECT_EQ(Full.sub(One), Full); 920 EXPECT_EQ(Full.sub(Some), Full); 921 EXPECT_EQ(Full.sub(Wrap), Full); 922 EXPECT_EQ(Empty.sub(Empty), Empty); 923 EXPECT_EQ(Empty.sub(One), Empty); 924 EXPECT_EQ(Empty.sub(Some), Empty); 925 EXPECT_EQ(Empty.sub(Wrap), Empty); 926 EXPECT_EQ(Empty.sub(APInt(16, 4)), Empty); 927 EXPECT_EQ(Some.sub(APInt(16, 4)), 928 ConstantRange(APInt(16, 0x6), APInt(16, 0xaa6))); 929 EXPECT_EQ(Some.sub(Some), 930 ConstantRange(APInt(16, 0xf561), APInt(16, 0xaa0))); 931 EXPECT_EQ(Wrap.sub(APInt(16, 4)), 932 ConstantRange(APInt(16, 0xaa6), APInt(16, 0x6))); 933 EXPECT_EQ(One.sub(APInt(16, 4)), 934 ConstantRange(APInt(16, 0x6))); 935 } 936 937 TEST_F(ConstantRangeTest, SubWithNoWrap) { 938 typedef OverflowingBinaryOperator OBO; 939 TestAddWithNoSignedWrapExhaustive( 940 [](const ConstantRange &CR1, const ConstantRange &CR2) { 941 return CR1.subWithNoWrap(CR2, OBO::NoSignedWrap); 942 }, 943 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 944 return N1.ssub_ov(N2, IsOverflow); 945 }); 946 TestAddWithNoUnsignedWrapExhaustive( 947 [](const ConstantRange &CR1, const ConstantRange &CR2) { 948 return CR1.subWithNoWrap(CR2, OBO::NoUnsignedWrap); 949 }, 950 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 951 return N1.usub_ov(N2, IsOverflow); 952 }); 953 TestAddWithNoSignedUnsignedWrapExhaustive( 954 [](const ConstantRange &CR1, const ConstantRange &CR2) { 955 return CR1.subWithNoWrap(CR2, OBO::NoUnsignedWrap | OBO::NoSignedWrap); 956 }, 957 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 958 return N1.ssub_ov(N2, IsOverflow); 959 }, 960 [](bool &IsOverflow, const APInt &N1, const APInt &N2) { 961 return N1.usub_ov(N2, IsOverflow); 962 }); 963 } 964 965 TEST_F(ConstantRangeTest, Multiply) { 966 EXPECT_EQ(Full.multiply(Full), Full); 967 EXPECT_EQ(Full.multiply(Empty), Empty); 968 EXPECT_EQ(Full.multiply(One), Full); 969 EXPECT_EQ(Full.multiply(Some), Full); 970 EXPECT_EQ(Full.multiply(Wrap), Full); 971 EXPECT_EQ(Empty.multiply(Empty), Empty); 972 EXPECT_EQ(Empty.multiply(One), Empty); 973 EXPECT_EQ(Empty.multiply(Some), Empty); 974 EXPECT_EQ(Empty.multiply(Wrap), Empty); 975 EXPECT_EQ(One.multiply(One), ConstantRange(APInt(16, 0xa*0xa), 976 APInt(16, 0xa*0xa + 1))); 977 EXPECT_EQ(One.multiply(Some), ConstantRange(APInt(16, 0xa*0xa), 978 APInt(16, 0xa*0xaa9 + 1))); 979 EXPECT_EQ(One.multiply(Wrap), Full); 980 EXPECT_EQ(Some.multiply(Some), Full); 981 EXPECT_EQ(Some.multiply(Wrap), Full); 982 EXPECT_EQ(Wrap.multiply(Wrap), Full); 983 984 ConstantRange Zero(APInt(16, 0)); 985 EXPECT_EQ(Zero.multiply(Full), Zero); 986 EXPECT_EQ(Zero.multiply(Some), Zero); 987 EXPECT_EQ(Zero.multiply(Wrap), Zero); 988 EXPECT_EQ(Full.multiply(Zero), Zero); 989 EXPECT_EQ(Some.multiply(Zero), Zero); 990 EXPECT_EQ(Wrap.multiply(Zero), Zero); 991 992 // http://llvm.org/PR4545 993 EXPECT_EQ(ConstantRange(APInt(4, 1), APInt(4, 6)).multiply( 994 ConstantRange(APInt(4, 6), APInt(4, 2))), 995 ConstantRange(4, /*isFullSet=*/true)); 996 997 EXPECT_EQ(ConstantRange(APInt(8, 254), APInt(8, 0)).multiply( 998 ConstantRange(APInt(8, 252), APInt(8, 4))), 999 ConstantRange(APInt(8, 250), APInt(8, 9))); 1000 EXPECT_EQ(ConstantRange(APInt(8, 254), APInt(8, 255)).multiply( 1001 ConstantRange(APInt(8, 2), APInt(8, 4))), 1002 ConstantRange(APInt(8, 250), APInt(8, 253))); 1003 1004 // TODO: This should be return [-2, 0] 1005 EXPECT_EQ(ConstantRange(APInt(8, -2)).multiply( 1006 ConstantRange(APInt(8, 0), APInt(8, 2))), 1007 ConstantRange(APInt(8, -2), APInt(8, 1))); 1008 } 1009 1010 TEST_F(ConstantRangeTest, UMax) { 1011 EXPECT_EQ(Full.umax(Full), Full); 1012 EXPECT_EQ(Full.umax(Empty), Empty); 1013 EXPECT_EQ(Full.umax(Some), ConstantRange(APInt(16, 0xa), APInt(16, 0))); 1014 EXPECT_EQ(Full.umax(Wrap), Full); 1015 EXPECT_EQ(Full.umax(Some), ConstantRange(APInt(16, 0xa), APInt(16, 0))); 1016 EXPECT_EQ(Empty.umax(Empty), Empty); 1017 EXPECT_EQ(Empty.umax(Some), Empty); 1018 EXPECT_EQ(Empty.umax(Wrap), Empty); 1019 EXPECT_EQ(Empty.umax(One), Empty); 1020 EXPECT_EQ(Some.umax(Some), Some); 1021 EXPECT_EQ(Some.umax(Wrap), ConstantRange(APInt(16, 0xa), APInt(16, 0))); 1022 EXPECT_EQ(Some.umax(One), Some); 1023 // TODO: ConstantRange is currently over-conservative here. 1024 EXPECT_EQ(Wrap.umax(Wrap), Full); 1025 EXPECT_EQ(Wrap.umax(One), ConstantRange(APInt(16, 0xa), APInt(16, 0))); 1026 EXPECT_EQ(One.umax(One), One); 1027 } 1028 1029 TEST_F(ConstantRangeTest, SMax) { 1030 EXPECT_EQ(Full.smax(Full), Full); 1031 EXPECT_EQ(Full.smax(Empty), Empty); 1032 EXPECT_EQ(Full.smax(Some), ConstantRange(APInt(16, 0xa), 1033 APInt::getSignedMinValue(16))); 1034 EXPECT_EQ(Full.smax(Wrap), Full); 1035 EXPECT_EQ(Full.smax(One), ConstantRange(APInt(16, 0xa), 1036 APInt::getSignedMinValue(16))); 1037 EXPECT_EQ(Empty.smax(Empty), Empty); 1038 EXPECT_EQ(Empty.smax(Some), Empty); 1039 EXPECT_EQ(Empty.smax(Wrap), Empty); 1040 EXPECT_EQ(Empty.smax(One), Empty); 1041 EXPECT_EQ(Some.smax(Some), Some); 1042 EXPECT_EQ(Some.smax(Wrap), ConstantRange(APInt(16, 0xa), 1043 APInt(16, (uint64_t)INT16_MIN))); 1044 EXPECT_EQ(Some.smax(One), Some); 1045 EXPECT_EQ(Wrap.smax(One), ConstantRange(APInt(16, 0xa), 1046 APInt(16, (uint64_t)INT16_MIN))); 1047 EXPECT_EQ(One.smax(One), One); 1048 } 1049 1050 TEST_F(ConstantRangeTest, UMin) { 1051 EXPECT_EQ(Full.umin(Full), Full); 1052 EXPECT_EQ(Full.umin(Empty), Empty); 1053 EXPECT_EQ(Full.umin(Some), ConstantRange(APInt(16, 0), APInt(16, 0xaaa))); 1054 EXPECT_EQ(Full.umin(Wrap), Full); 1055 EXPECT_EQ(Empty.umin(Empty), Empty); 1056 EXPECT_EQ(Empty.umin(Some), Empty); 1057 EXPECT_EQ(Empty.umin(Wrap), Empty); 1058 EXPECT_EQ(Empty.umin(One), Empty); 1059 EXPECT_EQ(Some.umin(Some), Some); 1060 EXPECT_EQ(Some.umin(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xaaa))); 1061 EXPECT_EQ(Some.umin(One), One); 1062 // TODO: ConstantRange is currently over-conservative here. 1063 EXPECT_EQ(Wrap.umin(Wrap), Full); 1064 EXPECT_EQ(Wrap.umin(One), ConstantRange(APInt(16, 0), APInt(16, 0xb))); 1065 EXPECT_EQ(One.umin(One), One); 1066 } 1067 1068 TEST_F(ConstantRangeTest, SMin) { 1069 EXPECT_EQ(Full.smin(Full), Full); 1070 EXPECT_EQ(Full.smin(Empty), Empty); 1071 EXPECT_EQ(Full.smin(Some), ConstantRange(APInt(16, (uint64_t)INT16_MIN), 1072 APInt(16, 0xaaa))); 1073 EXPECT_EQ(Full.smin(Wrap), Full); 1074 EXPECT_EQ(Empty.smin(Empty), Empty); 1075 EXPECT_EQ(Empty.smin(Some), Empty); 1076 EXPECT_EQ(Empty.smin(Wrap), Empty); 1077 EXPECT_EQ(Empty.smin(One), Empty); 1078 EXPECT_EQ(Some.smin(Some), Some); 1079 EXPECT_EQ(Some.smin(Wrap), ConstantRange(APInt(16, (uint64_t)INT16_MIN), 1080 APInt(16, 0xaaa))); 1081 EXPECT_EQ(Some.smin(One), One); 1082 // TODO: ConstantRange is currently over-conservative here. 1083 EXPECT_EQ(Wrap.smin(Wrap), Full); 1084 EXPECT_EQ(Wrap.smin(One), ConstantRange(APInt(16, (uint64_t)INT16_MIN), 1085 APInt(16, 0xb))); 1086 EXPECT_EQ(One.smin(One), One); 1087 } 1088 1089 TEST_F(ConstantRangeTest, UDiv) { 1090 EXPECT_EQ(Full.udiv(Full), Full); 1091 EXPECT_EQ(Full.udiv(Empty), Empty); 1092 EXPECT_EQ(Full.udiv(One), ConstantRange(APInt(16, 0), 1093 APInt(16, 0xffff / 0xa + 1))); 1094 EXPECT_EQ(Full.udiv(Some), ConstantRange(APInt(16, 0), 1095 APInt(16, 0xffff / 0xa + 1))); 1096 EXPECT_EQ(Full.udiv(Wrap), Full); 1097 EXPECT_EQ(Empty.udiv(Empty), Empty); 1098 EXPECT_EQ(Empty.udiv(One), Empty); 1099 EXPECT_EQ(Empty.udiv(Some), Empty); 1100 EXPECT_EQ(Empty.udiv(Wrap), Empty); 1101 EXPECT_EQ(One.udiv(One), ConstantRange(APInt(16, 1))); 1102 EXPECT_EQ(One.udiv(Some), ConstantRange(APInt(16, 0), APInt(16, 2))); 1103 EXPECT_EQ(One.udiv(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xb))); 1104 EXPECT_EQ(Some.udiv(Some), ConstantRange(APInt(16, 0), APInt(16, 0x111))); 1105 EXPECT_EQ(Some.udiv(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xaaa))); 1106 EXPECT_EQ(Wrap.udiv(Wrap), Full); 1107 1108 1109 ConstantRange Zero(APInt(16, 0)); 1110 EXPECT_EQ(Zero.udiv(One), Zero); 1111 EXPECT_EQ(Zero.udiv(Full), Zero); 1112 1113 EXPECT_EQ(ConstantRange(APInt(16, 0), APInt(16, 99)).udiv(Full), 1114 ConstantRange(APInt(16, 0), APInt(16, 99))); 1115 EXPECT_EQ(ConstantRange(APInt(16, 10), APInt(16, 99)).udiv(Full), 1116 ConstantRange(APInt(16, 0), APInt(16, 99))); 1117 } 1118 1119 TEST_F(ConstantRangeTest, SDiv) { 1120 unsigned Bits = 4; 1121 EnumerateTwoConstantRanges(Bits, [&](const ConstantRange &CR1, 1122 const ConstantRange &CR2) { 1123 // Collect possible results in a bit vector. We store the signed value plus 1124 // a bias to make it unsigned. 1125 int Bias = 1 << (Bits - 1); 1126 BitVector Results(1 << Bits); 1127 ForeachNumInConstantRange(CR1, [&](const APInt &N1) { 1128 ForeachNumInConstantRange(CR2, [&](const APInt &N2) { 1129 // Division by zero is UB. 1130 if (N2 == 0) 1131 return; 1132 1133 // SignedMin / -1 is UB. 1134 if (N1.isMinSignedValue() && N2.isAllOnesValue()) 1135 return; 1136 1137 APInt N = N1.sdiv(N2); 1138 Results.set(N.getSExtValue() + Bias); 1139 }); 1140 }); 1141 1142 ConstantRange CR = CR1.sdiv(CR2); 1143 if (Results.none()) { 1144 EXPECT_TRUE(CR.isEmptySet()); 1145 return; 1146 } 1147 1148 // If there is a non-full signed envelope, that should be the result. 1149 APInt SMin(Bits, Results.find_first() - Bias); 1150 APInt SMax(Bits, Results.find_last() - Bias); 1151 ConstantRange Envelope = ConstantRange::getNonEmpty(SMin, SMax + 1); 1152 if (!Envelope.isFullSet()) { 1153 EXPECT_EQ(Envelope, CR); 1154 return; 1155 } 1156 1157 // If the signed envelope is a full set, try to find a smaller sign wrapped 1158 // set that is separated in negative and positive components (or one which 1159 // can also additionally contain zero). 1160 int LastNeg = Results.find_last_in(0, Bias) - Bias; 1161 int LastPos = Results.find_next(Bias) - Bias; 1162 if (Results[Bias]) { 1163 if (LastNeg == -1) 1164 ++LastNeg; 1165 else if (LastPos == 1) 1166 --LastPos; 1167 } 1168 1169 APInt WMax(Bits, LastNeg); 1170 APInt WMin(Bits, LastPos); 1171 ConstantRange Wrapped = ConstantRange::getNonEmpty(WMin, WMax + 1); 1172 EXPECT_EQ(Wrapped, CR); 1173 }); 1174 } 1175 1176 TEST_F(ConstantRangeTest, URem) { 1177 EXPECT_EQ(Full.urem(Empty), Empty); 1178 EXPECT_EQ(Empty.urem(Full), Empty); 1179 // urem by zero is poison. 1180 EXPECT_EQ(Full.urem(ConstantRange(APInt(16, 0))), Empty); 1181 // urem by full range doesn't contain MaxValue. 1182 EXPECT_EQ(Full.urem(Full), ConstantRange(APInt(16, 0), APInt(16, 0xffff))); 1183 // urem is upper bounded by maximum RHS minus one. 1184 EXPECT_EQ(Full.urem(ConstantRange(APInt(16, 0), APInt(16, 123))), 1185 ConstantRange(APInt(16, 0), APInt(16, 122))); 1186 // urem is upper bounded by maximum LHS. 1187 EXPECT_EQ(ConstantRange(APInt(16, 0), APInt(16, 123)).urem(Full), 1188 ConstantRange(APInt(16, 0), APInt(16, 123))); 1189 // If the LHS is always lower than the RHS, the result is the LHS. 1190 EXPECT_EQ(ConstantRange(APInt(16, 10), APInt(16, 20)) 1191 .urem(ConstantRange(APInt(16, 20), APInt(16, 30))), 1192 ConstantRange(APInt(16, 10), APInt(16, 20))); 1193 // It has to be strictly lower, otherwise the top value may wrap to zero. 1194 EXPECT_EQ(ConstantRange(APInt(16, 10), APInt(16, 20)) 1195 .urem(ConstantRange(APInt(16, 19), APInt(16, 30))), 1196 ConstantRange(APInt(16, 0), APInt(16, 20))); 1197 // [12, 14] % 10 is [2, 4], but we conservatively compute [0, 9]. 1198 EXPECT_EQ(ConstantRange(APInt(16, 12), APInt(16, 15)) 1199 .urem(ConstantRange(APInt(16, 10))), 1200 ConstantRange(APInt(16, 0), APInt(16, 10))); 1201 1202 TestUnsignedBinOpExhaustive( 1203 [](const ConstantRange &CR1, const ConstantRange &CR2) { 1204 return CR1.urem(CR2); 1205 }, 1206 [](const APInt &N1, const APInt &N2) { 1207 return N1.urem(N2); 1208 }, 1209 /* SkipZeroRHS */ true, /* CorrectnessOnly */ true); 1210 } 1211 1212 TEST_F(ConstantRangeTest, SRem) { 1213 EXPECT_EQ(Full.srem(Empty), Empty); 1214 EXPECT_EQ(Empty.srem(Full), Empty); 1215 // srem by zero is UB. 1216 EXPECT_EQ(Full.srem(ConstantRange(APInt(16, 0))), Empty); 1217 // srem by full range doesn't contain SignedMinValue. 1218 EXPECT_EQ(Full.srem(Full), ConstantRange(APInt::getSignedMinValue(16) + 1, 1219 APInt::getSignedMinValue(16))); 1220 1221 ConstantRange PosMod(APInt(16, 10), APInt(16, 21)); // [10, 20] 1222 ConstantRange NegMod(APInt(16, -20), APInt(16, -9)); // [-20, -10] 1223 ConstantRange IntMinMod(APInt::getSignedMinValue(16)); 1224 1225 ConstantRange Expected(16, true); 1226 1227 // srem is bounded by abs(RHS) minus one. 1228 ConstantRange PosLargeLHS(APInt(16, 0), APInt(16, 41)); 1229 Expected = ConstantRange(APInt(16, 0), APInt(16, 20)); 1230 EXPECT_EQ(PosLargeLHS.srem(PosMod), Expected); 1231 EXPECT_EQ(PosLargeLHS.srem(NegMod), Expected); 1232 ConstantRange NegLargeLHS(APInt(16, -40), APInt(16, 1)); 1233 Expected = ConstantRange(APInt(16, -19), APInt(16, 1)); 1234 EXPECT_EQ(NegLargeLHS.srem(PosMod), Expected); 1235 EXPECT_EQ(NegLargeLHS.srem(NegMod), Expected); 1236 ConstantRange PosNegLargeLHS(APInt(16, -32), APInt(16, 38)); 1237 Expected = ConstantRange(APInt(16, -19), APInt(16, 20)); 1238 EXPECT_EQ(PosNegLargeLHS.srem(PosMod), Expected); 1239 EXPECT_EQ(PosNegLargeLHS.srem(NegMod), Expected); 1240 1241 // srem is bounded by LHS. 1242 ConstantRange PosLHS(APInt(16, 0), APInt(16, 16)); 1243 EXPECT_EQ(PosLHS.srem(PosMod), PosLHS); 1244 EXPECT_EQ(PosLHS.srem(NegMod), PosLHS); 1245 EXPECT_EQ(PosLHS.srem(IntMinMod), PosLHS); 1246 ConstantRange NegLHS(APInt(16, -15), APInt(16, 1)); 1247 EXPECT_EQ(NegLHS.srem(PosMod), NegLHS); 1248 EXPECT_EQ(NegLHS.srem(NegMod), NegLHS); 1249 EXPECT_EQ(NegLHS.srem(IntMinMod), NegLHS); 1250 ConstantRange PosNegLHS(APInt(16, -12), APInt(16, 18)); 1251 EXPECT_EQ(PosNegLHS.srem(PosMod), PosNegLHS); 1252 EXPECT_EQ(PosNegLHS.srem(NegMod), PosNegLHS); 1253 EXPECT_EQ(PosNegLHS.srem(IntMinMod), PosNegLHS); 1254 1255 // srem is LHS if it is smaller than RHS. 1256 ConstantRange PosSmallLHS(APInt(16, 3), APInt(16, 8)); 1257 EXPECT_EQ(PosSmallLHS.srem(PosMod), PosSmallLHS); 1258 EXPECT_EQ(PosSmallLHS.srem(NegMod), PosSmallLHS); 1259 EXPECT_EQ(PosSmallLHS.srem(IntMinMod), PosSmallLHS); 1260 ConstantRange NegSmallLHS(APInt(16, -7), APInt(16, -2)); 1261 EXPECT_EQ(NegSmallLHS.srem(PosMod), NegSmallLHS); 1262 EXPECT_EQ(NegSmallLHS.srem(NegMod), NegSmallLHS); 1263 EXPECT_EQ(NegSmallLHS.srem(IntMinMod), NegSmallLHS); 1264 ConstantRange PosNegSmallLHS(APInt(16, -3), APInt(16, 8)); 1265 EXPECT_EQ(PosNegSmallLHS.srem(PosMod), PosNegSmallLHS); 1266 EXPECT_EQ(PosNegSmallLHS.srem(NegMod), PosNegSmallLHS); 1267 EXPECT_EQ(PosNegSmallLHS.srem(IntMinMod), PosNegSmallLHS); 1268 1269 // Example of a suboptimal result: 1270 // [12, 14] srem 10 is [2, 4], but we conservatively compute [0, 9]. 1271 EXPECT_EQ(ConstantRange(APInt(16, 12), APInt(16, 15)) 1272 .srem(ConstantRange(APInt(16, 10))), 1273 ConstantRange(APInt(16, 0), APInt(16, 10))); 1274 1275 TestSignedBinOpExhaustive( 1276 [](const ConstantRange &CR1, const ConstantRange &CR2) { 1277 return CR1.srem(CR2); 1278 }, 1279 [](const APInt &N1, const APInt &N2) { 1280 return N1.srem(N2); 1281 }, 1282 /* SkipZeroRHS */ true, /* CorrectnessOnly */ true); 1283 } 1284 1285 TEST_F(ConstantRangeTest, Shl) { 1286 ConstantRange Some2(APInt(16, 0xfff), APInt(16, 0x8000)); 1287 ConstantRange WrapNullMax(APInt(16, 0x1), APInt(16, 0x0)); 1288 EXPECT_EQ(Full.shl(Full), Full); 1289 EXPECT_EQ(Full.shl(Empty), Empty); 1290 EXPECT_EQ(Full.shl(One), Full); // TODO: [0, (-1 << 0xa) + 1) 1291 EXPECT_EQ(Full.shl(Some), Full); // TODO: [0, (-1 << 0xa) + 1) 1292 EXPECT_EQ(Full.shl(Wrap), Full); 1293 EXPECT_EQ(Empty.shl(Empty), Empty); 1294 EXPECT_EQ(Empty.shl(One), Empty); 1295 EXPECT_EQ(Empty.shl(Some), Empty); 1296 EXPECT_EQ(Empty.shl(Wrap), Empty); 1297 EXPECT_EQ(One.shl(One), ConstantRange(APInt(16, 0xa << 0xa), 1298 APInt(16, (0xa << 0xa) + 1))); 1299 EXPECT_EQ(One.shl(Some), Full); // TODO: [0xa << 0xa, 0) 1300 EXPECT_EQ(One.shl(Wrap), Full); // TODO: [0xa, 0xa << 14 + 1) 1301 EXPECT_EQ(Some.shl(Some), Full); // TODO: [0xa << 0xa, 0xfc01) 1302 EXPECT_EQ(Some.shl(Wrap), Full); // TODO: [0xa, 0x7ff << 0x5 + 1) 1303 EXPECT_EQ(Wrap.shl(Wrap), Full); 1304 EXPECT_EQ( 1305 Some2.shl(ConstantRange(APInt(16, 0x1))), 1306 ConstantRange(APInt(16, 0xfff << 0x1), APInt(16, 0x7fff << 0x1) + 1)); 1307 EXPECT_EQ(One.shl(WrapNullMax), Full); 1308 } 1309 1310 TEST_F(ConstantRangeTest, Lshr) { 1311 EXPECT_EQ(Full.lshr(Full), Full); 1312 EXPECT_EQ(Full.lshr(Empty), Empty); 1313 EXPECT_EQ(Full.lshr(One), ConstantRange(APInt(16, 0), 1314 APInt(16, (0xffff >> 0xa) + 1))); 1315 EXPECT_EQ(Full.lshr(Some), ConstantRange(APInt(16, 0), 1316 APInt(16, (0xffff >> 0xa) + 1))); 1317 EXPECT_EQ(Full.lshr(Wrap), Full); 1318 EXPECT_EQ(Empty.lshr(Empty), Empty); 1319 EXPECT_EQ(Empty.lshr(One), Empty); 1320 EXPECT_EQ(Empty.lshr(Some), Empty); 1321 EXPECT_EQ(Empty.lshr(Wrap), Empty); 1322 EXPECT_EQ(One.lshr(One), ConstantRange(APInt(16, 0))); 1323 EXPECT_EQ(One.lshr(Some), ConstantRange(APInt(16, 0))); 1324 EXPECT_EQ(One.lshr(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xb))); 1325 EXPECT_EQ(Some.lshr(Some), ConstantRange(APInt(16, 0), 1326 APInt(16, (0xaaa >> 0xa) + 1))); 1327 EXPECT_EQ(Some.lshr(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xaaa))); 1328 EXPECT_EQ(Wrap.lshr(Wrap), Full); 1329 } 1330 1331 TEST_F(ConstantRangeTest, Ashr) { 1332 EXPECT_EQ(Full.ashr(Full), Full); 1333 EXPECT_EQ(Full.ashr(Empty), Empty); 1334 EXPECT_EQ(Full.ashr(One), ConstantRange(APInt(16, 0xffe0), 1335 APInt(16, (0x7fff >> 0xa) + 1 ))); 1336 ConstantRange Small(APInt(16, 0xa), APInt(16, 0xb)); 1337 EXPECT_EQ(Full.ashr(Small), ConstantRange(APInt(16, 0xffe0), 1338 APInt(16, (0x7fff >> 0xa) + 1 ))); 1339 EXPECT_EQ(Full.ashr(Some), ConstantRange(APInt(16, 0xffe0), 1340 APInt(16, (0x7fff >> 0xa) + 1 ))); 1341 EXPECT_EQ(Full.ashr(Wrap), Full); 1342 EXPECT_EQ(Empty.ashr(Empty), Empty); 1343 EXPECT_EQ(Empty.ashr(One), Empty); 1344 EXPECT_EQ(Empty.ashr(Some), Empty); 1345 EXPECT_EQ(Empty.ashr(Wrap), Empty); 1346 EXPECT_EQ(One.ashr(One), ConstantRange(APInt(16, 0))); 1347 EXPECT_EQ(One.ashr(Some), ConstantRange(APInt(16, 0))); 1348 EXPECT_EQ(One.ashr(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xb))); 1349 EXPECT_EQ(Some.ashr(Some), ConstantRange(APInt(16, 0), 1350 APInt(16, (0xaaa >> 0xa) + 1))); 1351 EXPECT_EQ(Some.ashr(Wrap), ConstantRange(APInt(16, 0), APInt(16, 0xaaa))); 1352 EXPECT_EQ(Wrap.ashr(Wrap), Full); 1353 ConstantRange Neg(APInt(16, 0xf3f0, true), APInt(16, 0xf7f8, true)); 1354 EXPECT_EQ(Neg.ashr(Small), ConstantRange(APInt(16, 0xfffc, true), 1355 APInt(16, 0xfffe, true))); 1356 } 1357 1358 TEST(ConstantRange, MakeAllowedICmpRegion) { 1359 // PR8250 1360 ConstantRange SMax = ConstantRange(APInt::getSignedMaxValue(32)); 1361 EXPECT_TRUE(ConstantRange::makeAllowedICmpRegion(ICmpInst::ICMP_SGT, SMax) 1362 .isEmptySet()); 1363 } 1364 1365 TEST(ConstantRange, MakeSatisfyingICmpRegion) { 1366 ConstantRange LowHalf(APInt(8, 0), APInt(8, 128)); 1367 ConstantRange HighHalf(APInt(8, 128), APInt(8, 0)); 1368 ConstantRange EmptySet(8, /* isFullSet = */ false); 1369 1370 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_NE, LowHalf), 1371 HighHalf); 1372 1373 EXPECT_EQ( 1374 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_NE, HighHalf), 1375 LowHalf); 1376 1377 EXPECT_TRUE(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_EQ, 1378 HighHalf).isEmptySet()); 1379 1380 ConstantRange UnsignedSample(APInt(8, 5), APInt(8, 200)); 1381 1382 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_ULT, 1383 UnsignedSample), 1384 ConstantRange(APInt(8, 0), APInt(8, 5))); 1385 1386 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_ULE, 1387 UnsignedSample), 1388 ConstantRange(APInt(8, 0), APInt(8, 6))); 1389 1390 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_UGT, 1391 UnsignedSample), 1392 ConstantRange(APInt(8, 200), APInt(8, 0))); 1393 1394 EXPECT_EQ(ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_UGE, 1395 UnsignedSample), 1396 ConstantRange(APInt(8, 199), APInt(8, 0))); 1397 1398 ConstantRange SignedSample(APInt(8, -5), APInt(8, 5)); 1399 1400 EXPECT_EQ( 1401 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SLT, SignedSample), 1402 ConstantRange(APInt(8, -128), APInt(8, -5))); 1403 1404 EXPECT_EQ( 1405 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SLE, SignedSample), 1406 ConstantRange(APInt(8, -128), APInt(8, -4))); 1407 1408 EXPECT_EQ( 1409 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SGT, SignedSample), 1410 ConstantRange(APInt(8, 5), APInt(8, -128))); 1411 1412 EXPECT_EQ( 1413 ConstantRange::makeSatisfyingICmpRegion(ICmpInst::ICMP_SGE, SignedSample), 1414 ConstantRange(APInt(8, 4), APInt(8, -128))); 1415 } 1416 1417 TEST(ConstantRange, MakeGuaranteedNoWrapRegion) { 1418 const int IntMin4Bits = 8; 1419 const int IntMax4Bits = 7; 1420 typedef OverflowingBinaryOperator OBO; 1421 1422 for (int Const : {0, -1, -2, 1, 2, IntMin4Bits, IntMax4Bits}) { 1423 APInt C(4, Const, true /* = isSigned */); 1424 1425 auto NUWRegion = ConstantRange::makeGuaranteedNoWrapRegion( 1426 Instruction::Add, C, OBO::NoUnsignedWrap); 1427 1428 EXPECT_FALSE(NUWRegion.isEmptySet()); 1429 1430 auto NSWRegion = ConstantRange::makeGuaranteedNoWrapRegion( 1431 Instruction::Add, C, OBO::NoSignedWrap); 1432 1433 EXPECT_FALSE(NSWRegion.isEmptySet()); 1434 1435 for (APInt I = NUWRegion.getLower(), E = NUWRegion.getUpper(); I != E; 1436 ++I) { 1437 bool Overflow = false; 1438 (void)I.uadd_ov(C, Overflow); 1439 EXPECT_FALSE(Overflow); 1440 } 1441 1442 for (APInt I = NSWRegion.getLower(), E = NSWRegion.getUpper(); I != E; 1443 ++I) { 1444 bool Overflow = false; 1445 (void)I.sadd_ov(C, Overflow); 1446 EXPECT_FALSE(Overflow); 1447 } 1448 } 1449 1450 for (int Const : {0, -1, -2, 1, 2, IntMin4Bits, IntMax4Bits}) { 1451 APInt C(4, Const, true /* = isSigned */); 1452 1453 auto NUWRegion = ConstantRange::makeGuaranteedNoWrapRegion( 1454 Instruction::Sub, C, OBO::NoUnsignedWrap); 1455 1456 EXPECT_FALSE(NUWRegion.isEmptySet()); 1457 1458 auto NSWRegion = ConstantRange::makeGuaranteedNoWrapRegion( 1459 Instruction::Sub, C, OBO::NoSignedWrap); 1460 1461 EXPECT_FALSE(NSWRegion.isEmptySet()); 1462 1463 for (APInt I = NUWRegion.getLower(), E = NUWRegion.getUpper(); I != E; 1464 ++I) { 1465 bool Overflow = false; 1466 (void)I.usub_ov(C, Overflow); 1467 EXPECT_FALSE(Overflow); 1468 } 1469 1470 for (APInt I = NSWRegion.getLower(), E = NSWRegion.getUpper(); I != E; 1471 ++I) { 1472 bool Overflow = false; 1473 (void)I.ssub_ov(C, Overflow); 1474 EXPECT_FALSE(Overflow); 1475 } 1476 } 1477 1478 auto NSWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion( 1479 Instruction::Add, ConstantRange(32, /* isFullSet = */ true), 1480 OBO::NoSignedWrap); 1481 EXPECT_TRUE(NSWForAllValues.isSingleElement() && 1482 NSWForAllValues.getSingleElement()->isMinValue()); 1483 1484 NSWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion( 1485 Instruction::Sub, ConstantRange(32, /* isFullSet = */ true), 1486 OBO::NoSignedWrap); 1487 EXPECT_TRUE(NSWForAllValues.isSingleElement() && 1488 NSWForAllValues.getSingleElement()->isMaxValue()); 1489 1490 auto NUWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion( 1491 Instruction::Add, ConstantRange(32, /* isFullSet = */ true), 1492 OBO::NoUnsignedWrap); 1493 EXPECT_TRUE(NUWForAllValues.isSingleElement() && 1494 NUWForAllValues.getSingleElement()->isMinValue()); 1495 1496 NUWForAllValues = ConstantRange::makeGuaranteedNoWrapRegion( 1497 Instruction::Sub, ConstantRange(32, /* isFullSet = */ true), 1498 OBO::NoUnsignedWrap); 1499 EXPECT_TRUE(NUWForAllValues.isSingleElement() && 1500 NUWForAllValues.getSingleElement()->isMaxValue()); 1501 1502 EXPECT_TRUE(ConstantRange::makeGuaranteedNoWrapRegion( 1503 Instruction::Add, APInt(32, 0), OBO::NoUnsignedWrap).isFullSet()); 1504 EXPECT_TRUE(ConstantRange::makeGuaranteedNoWrapRegion( 1505 Instruction::Add, APInt(32, 0), OBO::NoSignedWrap).isFullSet()); 1506 EXPECT_TRUE(ConstantRange::makeGuaranteedNoWrapRegion( 1507 Instruction::Sub, APInt(32, 0), OBO::NoUnsignedWrap).isFullSet()); 1508 EXPECT_TRUE(ConstantRange::makeGuaranteedNoWrapRegion( 1509 Instruction::Sub, APInt(32, 0), OBO::NoSignedWrap).isFullSet()); 1510 1511 ConstantRange OneToFive(APInt(32, 1), APInt(32, 6)); 1512 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1513 Instruction::Add, OneToFive, OBO::NoSignedWrap), 1514 ConstantRange(APInt::getSignedMinValue(32), 1515 APInt::getSignedMaxValue(32) - 4)); 1516 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1517 Instruction::Add, OneToFive, OBO::NoUnsignedWrap), 1518 ConstantRange(APInt::getMinValue(32), APInt::getMinValue(32) - 5)); 1519 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1520 Instruction::Sub, OneToFive, OBO::NoSignedWrap), 1521 ConstantRange(APInt::getSignedMinValue(32) + 5, 1522 APInt::getSignedMinValue(32))); 1523 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1524 Instruction::Sub, OneToFive, OBO::NoUnsignedWrap), 1525 ConstantRange(APInt::getMinValue(32) + 5, APInt::getMinValue(32))); 1526 1527 ConstantRange MinusFiveToMinusTwo(APInt(32, -5), APInt(32, -1)); 1528 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1529 Instruction::Add, MinusFiveToMinusTwo, OBO::NoSignedWrap), 1530 ConstantRange(APInt::getSignedMinValue(32) + 5, 1531 APInt::getSignedMinValue(32))); 1532 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1533 Instruction::Add, MinusFiveToMinusTwo, OBO::NoUnsignedWrap), 1534 ConstantRange(APInt(32, 0), APInt(32, 2))); 1535 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1536 Instruction::Sub, MinusFiveToMinusTwo, OBO::NoSignedWrap), 1537 ConstantRange(APInt::getSignedMinValue(32), 1538 APInt::getSignedMaxValue(32) - 4)); 1539 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1540 Instruction::Sub, MinusFiveToMinusTwo, OBO::NoUnsignedWrap), 1541 ConstantRange(APInt::getMaxValue(32) - 1, 1542 APInt::getMinValue(32))); 1543 1544 ConstantRange MinusOneToOne(APInt(32, -1), APInt(32, 2)); 1545 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1546 Instruction::Add, MinusOneToOne, OBO::NoSignedWrap), 1547 ConstantRange(APInt::getSignedMinValue(32) + 1, 1548 APInt::getSignedMinValue(32) - 1)); 1549 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1550 Instruction::Add, MinusOneToOne, OBO::NoUnsignedWrap), 1551 ConstantRange(APInt(32, 0), APInt(32, 1))); 1552 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1553 Instruction::Sub, MinusOneToOne, OBO::NoSignedWrap), 1554 ConstantRange(APInt::getSignedMinValue(32) + 1, 1555 APInt::getSignedMinValue(32) - 1)); 1556 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1557 Instruction::Sub, MinusOneToOne, OBO::NoUnsignedWrap), 1558 ConstantRange(APInt::getMaxValue(32), 1559 APInt::getMinValue(32))); 1560 1561 ConstantRange One(APInt(32, 1), APInt(32, 2)); 1562 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1563 Instruction::Add, One, OBO::NoSignedWrap), 1564 ConstantRange(APInt::getSignedMinValue(32), 1565 APInt::getSignedMaxValue(32))); 1566 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1567 Instruction::Add, One, OBO::NoUnsignedWrap), 1568 ConstantRange(APInt::getMinValue(32), APInt::getMaxValue(32))); 1569 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1570 Instruction::Sub, One, OBO::NoSignedWrap), 1571 ConstantRange(APInt::getSignedMinValue(32) + 1, 1572 APInt::getSignedMinValue(32))); 1573 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1574 Instruction::Sub, One, OBO::NoUnsignedWrap), 1575 ConstantRange(APInt::getMinValue(32) + 1, APInt::getMinValue(32))); 1576 1577 ConstantRange OneLessThanBitWidth(APInt(32, 0), APInt(32, 31) + 1); 1578 ConstantRange UpToBitWidth(APInt(32, 0), APInt(32, 32) + 1); 1579 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1580 Instruction::Shl, UpToBitWidth, OBO::NoUnsignedWrap), 1581 ConstantRange::makeGuaranteedNoWrapRegion( 1582 Instruction::Shl, OneLessThanBitWidth, OBO::NoUnsignedWrap)); 1583 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1584 Instruction::Shl, UpToBitWidth, OBO::NoSignedWrap), 1585 ConstantRange::makeGuaranteedNoWrapRegion( 1586 Instruction::Shl, OneLessThanBitWidth, OBO::NoSignedWrap)); 1587 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1588 Instruction::Shl, UpToBitWidth, OBO::NoUnsignedWrap), 1589 ConstantRange(APInt(32, 0), APInt(32, 1) + 1)); 1590 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1591 Instruction::Shl, UpToBitWidth, OBO::NoSignedWrap), 1592 ConstantRange(APInt(32, -1), APInt(32, 0) + 1)); 1593 1594 EXPECT_EQ( 1595 ConstantRange::makeGuaranteedNoWrapRegion( 1596 Instruction::Shl, ConstantRange::getFull(32), OBO::NoUnsignedWrap), 1597 ConstantRange::makeGuaranteedNoWrapRegion( 1598 Instruction::Shl, OneLessThanBitWidth, OBO::NoUnsignedWrap)); 1599 EXPECT_EQ( 1600 ConstantRange::makeGuaranteedNoWrapRegion( 1601 Instruction::Shl, ConstantRange::getFull(32), OBO::NoSignedWrap), 1602 ConstantRange::makeGuaranteedNoWrapRegion( 1603 Instruction::Shl, OneLessThanBitWidth, OBO::NoSignedWrap)); 1604 1605 ConstantRange IllegalShAmt(APInt(32, 32), APInt(32, 0) + 1); 1606 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1607 Instruction::Shl, IllegalShAmt, OBO::NoUnsignedWrap), 1608 ConstantRange::getFull(32)); 1609 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1610 Instruction::Shl, IllegalShAmt, OBO::NoSignedWrap), 1611 ConstantRange::getFull(32)); 1612 1613 EXPECT_EQ( 1614 ConstantRange::makeGuaranteedNoWrapRegion( 1615 Instruction::Shl, ConstantRange(APInt(32, -32), APInt(32, 16) + 1), 1616 OBO::NoUnsignedWrap), 1617 ConstantRange::makeGuaranteedNoWrapRegion( 1618 Instruction::Shl, ConstantRange(APInt(32, 0), APInt(32, 16) + 1), 1619 OBO::NoUnsignedWrap)); 1620 EXPECT_EQ( 1621 ConstantRange::makeGuaranteedNoWrapRegion( 1622 Instruction::Shl, ConstantRange(APInt(32, -32), APInt(32, 16) + 1), 1623 OBO::NoSignedWrap), 1624 ConstantRange::makeGuaranteedNoWrapRegion( 1625 Instruction::Shl, ConstantRange(APInt(32, 0), APInt(32, 16) + 1), 1626 OBO::NoSignedWrap)); 1627 1628 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1629 Instruction::Shl, 1630 ConstantRange(APInt(32, -32), APInt(32, 16) + 1), 1631 OBO::NoUnsignedWrap), 1632 ConstantRange(APInt(32, 0), APInt(32, 65535) + 1)); 1633 EXPECT_EQ(ConstantRange::makeGuaranteedNoWrapRegion( 1634 Instruction::Shl, 1635 ConstantRange(APInt(32, -32), APInt(32, 16) + 1), 1636 OBO::NoSignedWrap), 1637 ConstantRange(APInt(32, -32768), APInt(32, 32767) + 1)); 1638 } 1639 1640 template<typename Fn> 1641 void TestNoWrapRegionExhaustive(Instruction::BinaryOps BinOp, 1642 unsigned NoWrapKind, Fn OverflowFn) { 1643 unsigned Bits = 5; 1644 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) { 1645 if (CR.isEmptySet()) 1646 return; 1647 if (Instruction::isShift(BinOp) && CR.getUnsignedMax().uge(Bits)) 1648 return; 1649 1650 ConstantRange NoWrap = 1651 ConstantRange::makeGuaranteedNoWrapRegion(BinOp, CR, NoWrapKind); 1652 ConstantRange Full = ConstantRange::getFull(Bits); 1653 ForeachNumInConstantRange(Full, [&](const APInt &N1) { 1654 bool NoOverflow = true; 1655 bool Overflow = true; 1656 ForeachNumInConstantRange(CR, [&](const APInt &N2) { 1657 if (OverflowFn(N1, N2)) 1658 NoOverflow = false; 1659 else 1660 Overflow = false; 1661 }); 1662 EXPECT_EQ(NoOverflow, NoWrap.contains(N1)); 1663 1664 // The no-wrap range is exact for single-element ranges. 1665 if (CR.isSingleElement()) { 1666 EXPECT_EQ(Overflow, !NoWrap.contains(N1)); 1667 } 1668 }); 1669 }); 1670 } 1671 1672 // Show that makeGuaranteedNoWrapRegion() is maximal, and for single-element 1673 // ranges also exact. 1674 TEST(ConstantRange, NoWrapRegionExhaustive) { 1675 TestNoWrapRegionExhaustive( 1676 Instruction::Add, OverflowingBinaryOperator::NoUnsignedWrap, 1677 [](const APInt &N1, const APInt &N2) { 1678 bool Overflow; 1679 (void) N1.uadd_ov(N2, Overflow); 1680 return Overflow; 1681 }); 1682 TestNoWrapRegionExhaustive( 1683 Instruction::Add, OverflowingBinaryOperator::NoSignedWrap, 1684 [](const APInt &N1, const APInt &N2) { 1685 bool Overflow; 1686 (void) N1.sadd_ov(N2, Overflow); 1687 return Overflow; 1688 }); 1689 TestNoWrapRegionExhaustive( 1690 Instruction::Sub, OverflowingBinaryOperator::NoUnsignedWrap, 1691 [](const APInt &N1, const APInt &N2) { 1692 bool Overflow; 1693 (void) N1.usub_ov(N2, Overflow); 1694 return Overflow; 1695 }); 1696 TestNoWrapRegionExhaustive( 1697 Instruction::Sub, OverflowingBinaryOperator::NoSignedWrap, 1698 [](const APInt &N1, const APInt &N2) { 1699 bool Overflow; 1700 (void) N1.ssub_ov(N2, Overflow); 1701 return Overflow; 1702 }); 1703 TestNoWrapRegionExhaustive( 1704 Instruction::Mul, OverflowingBinaryOperator::NoUnsignedWrap, 1705 [](const APInt &N1, const APInt &N2) { 1706 bool Overflow; 1707 (void) N1.umul_ov(N2, Overflow); 1708 return Overflow; 1709 }); 1710 TestNoWrapRegionExhaustive( 1711 Instruction::Mul, OverflowingBinaryOperator::NoSignedWrap, 1712 [](const APInt &N1, const APInt &N2) { 1713 bool Overflow; 1714 (void) N1.smul_ov(N2, Overflow); 1715 return Overflow; 1716 }); 1717 TestNoWrapRegionExhaustive(Instruction::Shl, 1718 OverflowingBinaryOperator::NoUnsignedWrap, 1719 [](const APInt &N1, const APInt &N2) { 1720 bool Overflow; 1721 (void)N1.ushl_ov(N2, Overflow); 1722 return Overflow; 1723 }); 1724 TestNoWrapRegionExhaustive(Instruction::Shl, 1725 OverflowingBinaryOperator::NoSignedWrap, 1726 [](const APInt &N1, const APInt &N2) { 1727 bool Overflow; 1728 (void)N1.sshl_ov(N2, Overflow); 1729 return Overflow; 1730 }); 1731 } 1732 1733 TEST(ConstantRange, GetEquivalentICmp) { 1734 APInt RHS; 1735 CmpInst::Predicate Pred; 1736 1737 EXPECT_TRUE(ConstantRange(APInt::getMinValue(32), APInt(32, 100)) 1738 .getEquivalentICmp(Pred, RHS)); 1739 EXPECT_EQ(Pred, CmpInst::ICMP_ULT); 1740 EXPECT_EQ(RHS, APInt(32, 100)); 1741 1742 EXPECT_TRUE(ConstantRange(APInt::getSignedMinValue(32), APInt(32, 100)) 1743 .getEquivalentICmp(Pred, RHS)); 1744 EXPECT_EQ(Pred, CmpInst::ICMP_SLT); 1745 EXPECT_EQ(RHS, APInt(32, 100)); 1746 1747 EXPECT_TRUE(ConstantRange(APInt(32, 100), APInt::getMinValue(32)) 1748 .getEquivalentICmp(Pred, RHS)); 1749 EXPECT_EQ(Pred, CmpInst::ICMP_UGE); 1750 EXPECT_EQ(RHS, APInt(32, 100)); 1751 1752 EXPECT_TRUE(ConstantRange(APInt(32, 100), APInt::getSignedMinValue(32)) 1753 .getEquivalentICmp(Pred, RHS)); 1754 EXPECT_EQ(Pred, CmpInst::ICMP_SGE); 1755 EXPECT_EQ(RHS, APInt(32, 100)); 1756 1757 EXPECT_TRUE( 1758 ConstantRange(32, /*isFullSet=*/true).getEquivalentICmp(Pred, RHS)); 1759 EXPECT_EQ(Pred, CmpInst::ICMP_UGE); 1760 EXPECT_EQ(RHS, APInt(32, 0)); 1761 1762 EXPECT_TRUE( 1763 ConstantRange(32, /*isFullSet=*/false).getEquivalentICmp(Pred, RHS)); 1764 EXPECT_EQ(Pred, CmpInst::ICMP_ULT); 1765 EXPECT_EQ(RHS, APInt(32, 0)); 1766 1767 EXPECT_FALSE(ConstantRange(APInt(32, 100), APInt(32, 200)) 1768 .getEquivalentICmp(Pred, RHS)); 1769 1770 EXPECT_FALSE(ConstantRange(APInt::getSignedMinValue(32) - APInt(32, 100), 1771 APInt::getSignedMinValue(32) + APInt(32, 100)) 1772 .getEquivalentICmp(Pred, RHS)); 1773 1774 EXPECT_FALSE(ConstantRange(APInt::getMinValue(32) - APInt(32, 100), 1775 APInt::getMinValue(32) + APInt(32, 100)) 1776 .getEquivalentICmp(Pred, RHS)); 1777 1778 EXPECT_TRUE(ConstantRange(APInt(32, 100)).getEquivalentICmp(Pred, RHS)); 1779 EXPECT_EQ(Pred, CmpInst::ICMP_EQ); 1780 EXPECT_EQ(RHS, APInt(32, 100)); 1781 1782 EXPECT_TRUE( 1783 ConstantRange(APInt(32, 100)).inverse().getEquivalentICmp(Pred, RHS)); 1784 EXPECT_EQ(Pred, CmpInst::ICMP_NE); 1785 EXPECT_EQ(RHS, APInt(32, 100)); 1786 1787 EXPECT_TRUE( 1788 ConstantRange(APInt(512, 100)).inverse().getEquivalentICmp(Pred, RHS)); 1789 EXPECT_EQ(Pred, CmpInst::ICMP_NE); 1790 EXPECT_EQ(RHS, APInt(512, 100)); 1791 1792 // NB! It would be correct for the following four calls to getEquivalentICmp 1793 // to return ordered predicates like CmpInst::ICMP_ULT or CmpInst::ICMP_UGT. 1794 // However, that's not the case today. 1795 1796 EXPECT_TRUE(ConstantRange(APInt(32, 0)).getEquivalentICmp(Pred, RHS)); 1797 EXPECT_EQ(Pred, CmpInst::ICMP_EQ); 1798 EXPECT_EQ(RHS, APInt(32, 0)); 1799 1800 EXPECT_TRUE( 1801 ConstantRange(APInt(32, 0)).inverse().getEquivalentICmp(Pred, RHS)); 1802 EXPECT_EQ(Pred, CmpInst::ICMP_NE); 1803 EXPECT_EQ(RHS, APInt(32, 0)); 1804 1805 EXPECT_TRUE(ConstantRange(APInt(32, -1)).getEquivalentICmp(Pred, RHS)); 1806 EXPECT_EQ(Pred, CmpInst::ICMP_EQ); 1807 EXPECT_EQ(RHS, APInt(32, -1)); 1808 1809 EXPECT_TRUE( 1810 ConstantRange(APInt(32, -1)).inverse().getEquivalentICmp(Pred, RHS)); 1811 EXPECT_EQ(Pred, CmpInst::ICMP_NE); 1812 EXPECT_EQ(RHS, APInt(32, -1)); 1813 } 1814 1815 #define EXPECT_MAY_OVERFLOW(op) \ 1816 EXPECT_EQ(ConstantRange::OverflowResult::MayOverflow, (op)) 1817 #define EXPECT_ALWAYS_OVERFLOWS_LOW(op) \ 1818 EXPECT_EQ(ConstantRange::OverflowResult::AlwaysOverflowsLow, (op)) 1819 #define EXPECT_ALWAYS_OVERFLOWS_HIGH(op) \ 1820 EXPECT_EQ(ConstantRange::OverflowResult::AlwaysOverflowsHigh, (op)) 1821 #define EXPECT_NEVER_OVERFLOWS(op) \ 1822 EXPECT_EQ(ConstantRange::OverflowResult::NeverOverflows, (op)) 1823 1824 TEST_F(ConstantRangeTest, UnsignedAddOverflow) { 1825 // Ill-defined - may overflow is a conservative result. 1826 EXPECT_MAY_OVERFLOW(Some.unsignedAddMayOverflow(Empty)); 1827 EXPECT_MAY_OVERFLOW(Empty.unsignedAddMayOverflow(Some)); 1828 1829 // Never overflow despite one full/wrap set. 1830 ConstantRange Zero(APInt::getNullValue(16)); 1831 EXPECT_NEVER_OVERFLOWS(Full.unsignedAddMayOverflow(Zero)); 1832 EXPECT_NEVER_OVERFLOWS(Wrap.unsignedAddMayOverflow(Zero)); 1833 EXPECT_NEVER_OVERFLOWS(Zero.unsignedAddMayOverflow(Full)); 1834 EXPECT_NEVER_OVERFLOWS(Zero.unsignedAddMayOverflow(Wrap)); 1835 1836 // But usually full/wrap always may overflow. 1837 EXPECT_MAY_OVERFLOW(Full.unsignedAddMayOverflow(One)); 1838 EXPECT_MAY_OVERFLOW(Wrap.unsignedAddMayOverflow(One)); 1839 EXPECT_MAY_OVERFLOW(One.unsignedAddMayOverflow(Full)); 1840 EXPECT_MAY_OVERFLOW(One.unsignedAddMayOverflow(Wrap)); 1841 1842 ConstantRange A(APInt(16, 0xfd00), APInt(16, 0xfe00)); 1843 ConstantRange B1(APInt(16, 0x0100), APInt(16, 0x0201)); 1844 ConstantRange B2(APInt(16, 0x0100), APInt(16, 0x0202)); 1845 EXPECT_NEVER_OVERFLOWS(A.unsignedAddMayOverflow(B1)); 1846 EXPECT_MAY_OVERFLOW(A.unsignedAddMayOverflow(B2)); 1847 EXPECT_NEVER_OVERFLOWS(B1.unsignedAddMayOverflow(A)); 1848 EXPECT_MAY_OVERFLOW(B2.unsignedAddMayOverflow(A)); 1849 1850 ConstantRange C1(APInt(16, 0x0299), APInt(16, 0x0400)); 1851 ConstantRange C2(APInt(16, 0x0300), APInt(16, 0x0400)); 1852 EXPECT_MAY_OVERFLOW(A.unsignedAddMayOverflow(C1)); 1853 EXPECT_ALWAYS_OVERFLOWS_HIGH(A.unsignedAddMayOverflow(C2)); 1854 EXPECT_MAY_OVERFLOW(C1.unsignedAddMayOverflow(A)); 1855 EXPECT_ALWAYS_OVERFLOWS_HIGH(C2.unsignedAddMayOverflow(A)); 1856 } 1857 1858 TEST_F(ConstantRangeTest, UnsignedSubOverflow) { 1859 // Ill-defined - may overflow is a conservative result. 1860 EXPECT_MAY_OVERFLOW(Some.unsignedSubMayOverflow(Empty)); 1861 EXPECT_MAY_OVERFLOW(Empty.unsignedSubMayOverflow(Some)); 1862 1863 // Never overflow despite one full/wrap set. 1864 ConstantRange Zero(APInt::getNullValue(16)); 1865 ConstantRange Max(APInt::getAllOnesValue(16)); 1866 EXPECT_NEVER_OVERFLOWS(Full.unsignedSubMayOverflow(Zero)); 1867 EXPECT_NEVER_OVERFLOWS(Wrap.unsignedSubMayOverflow(Zero)); 1868 EXPECT_NEVER_OVERFLOWS(Max.unsignedSubMayOverflow(Full)); 1869 EXPECT_NEVER_OVERFLOWS(Max.unsignedSubMayOverflow(Wrap)); 1870 1871 // But usually full/wrap always may overflow. 1872 EXPECT_MAY_OVERFLOW(Full.unsignedSubMayOverflow(One)); 1873 EXPECT_MAY_OVERFLOW(Wrap.unsignedSubMayOverflow(One)); 1874 EXPECT_MAY_OVERFLOW(One.unsignedSubMayOverflow(Full)); 1875 EXPECT_MAY_OVERFLOW(One.unsignedSubMayOverflow(Wrap)); 1876 1877 ConstantRange A(APInt(16, 0x0000), APInt(16, 0x0100)); 1878 ConstantRange B(APInt(16, 0x0100), APInt(16, 0x0200)); 1879 EXPECT_NEVER_OVERFLOWS(B.unsignedSubMayOverflow(A)); 1880 EXPECT_ALWAYS_OVERFLOWS_LOW(A.unsignedSubMayOverflow(B)); 1881 1882 ConstantRange A1(APInt(16, 0x0000), APInt(16, 0x0101)); 1883 ConstantRange B1(APInt(16, 0x0100), APInt(16, 0x0201)); 1884 EXPECT_NEVER_OVERFLOWS(B1.unsignedSubMayOverflow(A1)); 1885 EXPECT_MAY_OVERFLOW(A1.unsignedSubMayOverflow(B1)); 1886 1887 ConstantRange A2(APInt(16, 0x0000), APInt(16, 0x0102)); 1888 ConstantRange B2(APInt(16, 0x0100), APInt(16, 0x0202)); 1889 EXPECT_MAY_OVERFLOW(B2.unsignedSubMayOverflow(A2)); 1890 EXPECT_MAY_OVERFLOW(A2.unsignedSubMayOverflow(B2)); 1891 } 1892 1893 TEST_F(ConstantRangeTest, SignedAddOverflow) { 1894 // Ill-defined - may overflow is a conservative result. 1895 EXPECT_MAY_OVERFLOW(Some.signedAddMayOverflow(Empty)); 1896 EXPECT_MAY_OVERFLOW(Empty.signedAddMayOverflow(Some)); 1897 1898 // Never overflow despite one full/wrap set. 1899 ConstantRange Zero(APInt::getNullValue(16)); 1900 EXPECT_NEVER_OVERFLOWS(Full.signedAddMayOverflow(Zero)); 1901 EXPECT_NEVER_OVERFLOWS(Wrap.signedAddMayOverflow(Zero)); 1902 EXPECT_NEVER_OVERFLOWS(Zero.signedAddMayOverflow(Full)); 1903 EXPECT_NEVER_OVERFLOWS(Zero.signedAddMayOverflow(Wrap)); 1904 1905 // But usually full/wrap always may overflow. 1906 EXPECT_MAY_OVERFLOW(Full.signedAddMayOverflow(One)); 1907 EXPECT_MAY_OVERFLOW(Wrap.signedAddMayOverflow(One)); 1908 EXPECT_MAY_OVERFLOW(One.signedAddMayOverflow(Full)); 1909 EXPECT_MAY_OVERFLOW(One.signedAddMayOverflow(Wrap)); 1910 1911 ConstantRange A(APInt(16, 0x7d00), APInt(16, 0x7e00)); 1912 ConstantRange B1(APInt(16, 0x0100), APInt(16, 0x0201)); 1913 ConstantRange B2(APInt(16, 0x0100), APInt(16, 0x0202)); 1914 EXPECT_NEVER_OVERFLOWS(A.signedAddMayOverflow(B1)); 1915 EXPECT_MAY_OVERFLOW(A.signedAddMayOverflow(B2)); 1916 ConstantRange B3(APInt(16, 0x8000), APInt(16, 0x0201)); 1917 ConstantRange B4(APInt(16, 0x8000), APInt(16, 0x0202)); 1918 EXPECT_NEVER_OVERFLOWS(A.signedAddMayOverflow(B3)); 1919 EXPECT_MAY_OVERFLOW(A.signedAddMayOverflow(B4)); 1920 ConstantRange B5(APInt(16, 0x0299), APInt(16, 0x0400)); 1921 ConstantRange B6(APInt(16, 0x0300), APInt(16, 0x0400)); 1922 EXPECT_MAY_OVERFLOW(A.signedAddMayOverflow(B5)); 1923 EXPECT_ALWAYS_OVERFLOWS_HIGH(A.signedAddMayOverflow(B6)); 1924 1925 ConstantRange C(APInt(16, 0x8200), APInt(16, 0x8300)); 1926 ConstantRange D1(APInt(16, 0xfe00), APInt(16, 0xff00)); 1927 ConstantRange D2(APInt(16, 0xfd99), APInt(16, 0xff00)); 1928 EXPECT_NEVER_OVERFLOWS(C.signedAddMayOverflow(D1)); 1929 EXPECT_MAY_OVERFLOW(C.signedAddMayOverflow(D2)); 1930 ConstantRange D3(APInt(16, 0xfe00), APInt(16, 0x8000)); 1931 ConstantRange D4(APInt(16, 0xfd99), APInt(16, 0x8000)); 1932 EXPECT_NEVER_OVERFLOWS(C.signedAddMayOverflow(D3)); 1933 EXPECT_MAY_OVERFLOW(C.signedAddMayOverflow(D4)); 1934 ConstantRange D5(APInt(16, 0xfc00), APInt(16, 0xfd02)); 1935 ConstantRange D6(APInt(16, 0xfc00), APInt(16, 0xfd01)); 1936 EXPECT_MAY_OVERFLOW(C.signedAddMayOverflow(D5)); 1937 EXPECT_ALWAYS_OVERFLOWS_LOW(C.signedAddMayOverflow(D6)); 1938 1939 ConstantRange E(APInt(16, 0xff00), APInt(16, 0x0100)); 1940 EXPECT_NEVER_OVERFLOWS(E.signedAddMayOverflow(E)); 1941 ConstantRange F(APInt(16, 0xf000), APInt(16, 0x7000)); 1942 EXPECT_MAY_OVERFLOW(F.signedAddMayOverflow(F)); 1943 } 1944 1945 TEST_F(ConstantRangeTest, SignedSubOverflow) { 1946 // Ill-defined - may overflow is a conservative result. 1947 EXPECT_MAY_OVERFLOW(Some.signedSubMayOverflow(Empty)); 1948 EXPECT_MAY_OVERFLOW(Empty.signedSubMayOverflow(Some)); 1949 1950 // Never overflow despite one full/wrap set. 1951 ConstantRange Zero(APInt::getNullValue(16)); 1952 EXPECT_NEVER_OVERFLOWS(Full.signedSubMayOverflow(Zero)); 1953 EXPECT_NEVER_OVERFLOWS(Wrap.signedSubMayOverflow(Zero)); 1954 1955 // But usually full/wrap always may overflow. 1956 EXPECT_MAY_OVERFLOW(Full.signedSubMayOverflow(One)); 1957 EXPECT_MAY_OVERFLOW(Wrap.signedSubMayOverflow(One)); 1958 EXPECT_MAY_OVERFLOW(One.signedSubMayOverflow(Full)); 1959 EXPECT_MAY_OVERFLOW(One.signedSubMayOverflow(Wrap)); 1960 1961 ConstantRange A(APInt(16, 0x7d00), APInt(16, 0x7e00)); 1962 ConstantRange B1(APInt(16, 0xfe00), APInt(16, 0xff00)); 1963 ConstantRange B2(APInt(16, 0xfd99), APInt(16, 0xff00)); 1964 EXPECT_NEVER_OVERFLOWS(A.signedSubMayOverflow(B1)); 1965 EXPECT_MAY_OVERFLOW(A.signedSubMayOverflow(B2)); 1966 ConstantRange B3(APInt(16, 0xfc00), APInt(16, 0xfd02)); 1967 ConstantRange B4(APInt(16, 0xfc00), APInt(16, 0xfd01)); 1968 EXPECT_MAY_OVERFLOW(A.signedSubMayOverflow(B3)); 1969 EXPECT_ALWAYS_OVERFLOWS_HIGH(A.signedSubMayOverflow(B4)); 1970 1971 ConstantRange C(APInt(16, 0x8200), APInt(16, 0x8300)); 1972 ConstantRange D1(APInt(16, 0x0100), APInt(16, 0x0201)); 1973 ConstantRange D2(APInt(16, 0x0100), APInt(16, 0x0202)); 1974 EXPECT_NEVER_OVERFLOWS(C.signedSubMayOverflow(D1)); 1975 EXPECT_MAY_OVERFLOW(C.signedSubMayOverflow(D2)); 1976 ConstantRange D3(APInt(16, 0x0299), APInt(16, 0x0400)); 1977 ConstantRange D4(APInt(16, 0x0300), APInt(16, 0x0400)); 1978 EXPECT_MAY_OVERFLOW(C.signedSubMayOverflow(D3)); 1979 EXPECT_ALWAYS_OVERFLOWS_LOW(C.signedSubMayOverflow(D4)); 1980 1981 ConstantRange E(APInt(16, 0xff00), APInt(16, 0x0100)); 1982 EXPECT_NEVER_OVERFLOWS(E.signedSubMayOverflow(E)); 1983 ConstantRange F(APInt(16, 0xf000), APInt(16, 0x7001)); 1984 EXPECT_MAY_OVERFLOW(F.signedSubMayOverflow(F)); 1985 } 1986 1987 template<typename Fn1, typename Fn2> 1988 static void TestOverflowExhaustive(Fn1 OverflowFn, Fn2 MayOverflowFn) { 1989 // Constant range overflow checks are tested exhaustively on 4-bit numbers. 1990 unsigned Bits = 4; 1991 EnumerateTwoConstantRanges(Bits, [=](const ConstantRange &CR1, 1992 const ConstantRange &CR2) { 1993 // Loop over all N1 in CR1 and N2 in CR2 and check whether any of the 1994 // operations have overflow / have no overflow. 1995 bool RangeHasOverflowLow = false; 1996 bool RangeHasOverflowHigh = false; 1997 bool RangeHasNoOverflow = false; 1998 ForeachNumInConstantRange(CR1, [&](const APInt &N1) { 1999 ForeachNumInConstantRange(CR2, [&](const APInt &N2) { 2000 bool IsOverflowHigh; 2001 if (!OverflowFn(IsOverflowHigh, N1, N2)) { 2002 RangeHasNoOverflow = true; 2003 return; 2004 } 2005 2006 if (IsOverflowHigh) 2007 RangeHasOverflowHigh = true; 2008 else 2009 RangeHasOverflowLow = true; 2010 }); 2011 }); 2012 2013 ConstantRange::OverflowResult OR = MayOverflowFn(CR1, CR2); 2014 switch (OR) { 2015 case ConstantRange::OverflowResult::AlwaysOverflowsLow: 2016 EXPECT_TRUE(RangeHasOverflowLow); 2017 EXPECT_FALSE(RangeHasOverflowHigh); 2018 EXPECT_FALSE(RangeHasNoOverflow); 2019 break; 2020 case ConstantRange::OverflowResult::AlwaysOverflowsHigh: 2021 EXPECT_TRUE(RangeHasOverflowHigh); 2022 EXPECT_FALSE(RangeHasOverflowLow); 2023 EXPECT_FALSE(RangeHasNoOverflow); 2024 break; 2025 case ConstantRange::OverflowResult::NeverOverflows: 2026 EXPECT_FALSE(RangeHasOverflowLow); 2027 EXPECT_FALSE(RangeHasOverflowHigh); 2028 EXPECT_TRUE(RangeHasNoOverflow); 2029 break; 2030 case ConstantRange::OverflowResult::MayOverflow: 2031 // We return MayOverflow for empty sets as a conservative result, 2032 // but of course neither the RangeHasOverflow nor the 2033 // RangeHasNoOverflow flags will be set. 2034 if (CR1.isEmptySet() || CR2.isEmptySet()) 2035 break; 2036 2037 EXPECT_TRUE(RangeHasOverflowLow || RangeHasOverflowHigh); 2038 EXPECT_TRUE(RangeHasNoOverflow); 2039 break; 2040 } 2041 }); 2042 } 2043 2044 TEST_F(ConstantRangeTest, UnsignedAddOverflowExhaustive) { 2045 TestOverflowExhaustive( 2046 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) { 2047 bool Overflow; 2048 (void) N1.uadd_ov(N2, Overflow); 2049 IsOverflowHigh = true; 2050 return Overflow; 2051 }, 2052 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2053 return CR1.unsignedAddMayOverflow(CR2); 2054 }); 2055 } 2056 2057 TEST_F(ConstantRangeTest, UnsignedSubOverflowExhaustive) { 2058 TestOverflowExhaustive( 2059 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) { 2060 bool Overflow; 2061 (void) N1.usub_ov(N2, Overflow); 2062 IsOverflowHigh = false; 2063 return Overflow; 2064 }, 2065 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2066 return CR1.unsignedSubMayOverflow(CR2); 2067 }); 2068 } 2069 2070 TEST_F(ConstantRangeTest, UnsignedMulOverflowExhaustive) { 2071 TestOverflowExhaustive( 2072 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) { 2073 bool Overflow; 2074 (void) N1.umul_ov(N2, Overflow); 2075 IsOverflowHigh = true; 2076 return Overflow; 2077 }, 2078 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2079 return CR1.unsignedMulMayOverflow(CR2); 2080 }); 2081 } 2082 2083 TEST_F(ConstantRangeTest, SignedAddOverflowExhaustive) { 2084 TestOverflowExhaustive( 2085 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) { 2086 bool Overflow; 2087 (void) N1.sadd_ov(N2, Overflow); 2088 IsOverflowHigh = N1.isNonNegative(); 2089 return Overflow; 2090 }, 2091 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2092 return CR1.signedAddMayOverflow(CR2); 2093 }); 2094 } 2095 2096 TEST_F(ConstantRangeTest, SignedSubOverflowExhaustive) { 2097 TestOverflowExhaustive( 2098 [](bool &IsOverflowHigh, const APInt &N1, const APInt &N2) { 2099 bool Overflow; 2100 (void) N1.ssub_ov(N2, Overflow); 2101 IsOverflowHigh = N1.isNonNegative(); 2102 return Overflow; 2103 }, 2104 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2105 return CR1.signedSubMayOverflow(CR2); 2106 }); 2107 } 2108 2109 TEST_F(ConstantRangeTest, FromKnownBits) { 2110 KnownBits Unknown(16); 2111 EXPECT_EQ(Full, ConstantRange::fromKnownBits(Unknown, /*signed*/false)); 2112 EXPECT_EQ(Full, ConstantRange::fromKnownBits(Unknown, /*signed*/true)); 2113 2114 // .10..01. -> unsigned 01000010 (66) to 11011011 (219) 2115 // -> signed 11000010 (194) to 01011011 (91) 2116 KnownBits Known(8); 2117 Known.Zero = 36; 2118 Known.One = 66; 2119 ConstantRange Unsigned(APInt(8, 66), APInt(8, 219 + 1)); 2120 ConstantRange Signed(APInt(8, 194), APInt(8, 91 + 1)); 2121 EXPECT_EQ(Unsigned, ConstantRange::fromKnownBits(Known, /*signed*/false)); 2122 EXPECT_EQ(Signed, ConstantRange::fromKnownBits(Known, /*signed*/true)); 2123 2124 // 1.10.10. -> 10100100 (164) to 11101101 (237) 2125 Known.Zero = 18; 2126 Known.One = 164; 2127 ConstantRange CR1(APInt(8, 164), APInt(8, 237 + 1)); 2128 EXPECT_EQ(CR1, ConstantRange::fromKnownBits(Known, /*signed*/false)); 2129 EXPECT_EQ(CR1, ConstantRange::fromKnownBits(Known, /*signed*/true)); 2130 2131 // 01.0.1.0 -> 01000100 (68) to 01101110 (110) 2132 Known.Zero = 145; 2133 Known.One = 68; 2134 ConstantRange CR2(APInt(8, 68), APInt(8, 110 + 1)); 2135 EXPECT_EQ(CR2, ConstantRange::fromKnownBits(Known, /*signed*/false)); 2136 EXPECT_EQ(CR2, ConstantRange::fromKnownBits(Known, /*signed*/true)); 2137 } 2138 2139 TEST_F(ConstantRangeTest, FromKnownBitsExhaustive) { 2140 unsigned Bits = 4; 2141 unsigned Max = 1 << Bits; 2142 KnownBits Known(Bits); 2143 for (unsigned Zero = 0; Zero < Max; ++Zero) { 2144 for (unsigned One = 0; One < Max; ++One) { 2145 Known.Zero = Zero; 2146 Known.One = One; 2147 if (Known.hasConflict() || Known.isUnknown()) 2148 continue; 2149 2150 APInt MinUnsigned = APInt::getMaxValue(Bits); 2151 APInt MaxUnsigned = APInt::getMinValue(Bits); 2152 APInt MinSigned = APInt::getSignedMaxValue(Bits); 2153 APInt MaxSigned = APInt::getSignedMinValue(Bits); 2154 for (unsigned N = 0; N < Max; ++N) { 2155 APInt Num(Bits, N); 2156 if ((Num & Known.Zero) != 0 || (~Num & Known.One) != 0) 2157 continue; 2158 2159 if (Num.ult(MinUnsigned)) MinUnsigned = Num; 2160 if (Num.ugt(MaxUnsigned)) MaxUnsigned = Num; 2161 if (Num.slt(MinSigned)) MinSigned = Num; 2162 if (Num.sgt(MaxSigned)) MaxSigned = Num; 2163 } 2164 2165 ConstantRange UnsignedCR(MinUnsigned, MaxUnsigned + 1); 2166 ConstantRange SignedCR(MinSigned, MaxSigned + 1); 2167 EXPECT_EQ(UnsignedCR, ConstantRange::fromKnownBits(Known, false)); 2168 EXPECT_EQ(SignedCR, ConstantRange::fromKnownBits(Known, true)); 2169 } 2170 } 2171 } 2172 2173 TEST_F(ConstantRangeTest, Negative) { 2174 // All elements in an empty set (of which there are none) are both negative 2175 // and non-negative. Empty & full sets checked explicitly for clarity, but 2176 // they are also covered by the exhaustive test below. 2177 EXPECT_TRUE(Empty.isAllNegative()); 2178 EXPECT_TRUE(Empty.isAllNonNegative()); 2179 EXPECT_FALSE(Full.isAllNegative()); 2180 EXPECT_FALSE(Full.isAllNonNegative()); 2181 2182 unsigned Bits = 4; 2183 EnumerateConstantRanges(Bits, [](const ConstantRange &CR) { 2184 bool AllNegative = true; 2185 bool AllNonNegative = true; 2186 ForeachNumInConstantRange(CR, [&](const APInt &N) { 2187 if (!N.isNegative()) 2188 AllNegative = false; 2189 if (!N.isNonNegative()) 2190 AllNonNegative = false; 2191 }); 2192 assert((CR.isEmptySet() || !AllNegative || !AllNonNegative) && 2193 "Only empty set can be both all negative and all non-negative"); 2194 2195 EXPECT_EQ(AllNegative, CR.isAllNegative()); 2196 EXPECT_EQ(AllNonNegative, CR.isAllNonNegative()); 2197 }); 2198 } 2199 2200 TEST_F(ConstantRangeTest, UAddSat) { 2201 TestUnsignedBinOpExhaustive( 2202 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2203 return CR1.uadd_sat(CR2); 2204 }, 2205 [](const APInt &N1, const APInt &N2) { 2206 return N1.uadd_sat(N2); 2207 }); 2208 } 2209 2210 TEST_F(ConstantRangeTest, USubSat) { 2211 TestUnsignedBinOpExhaustive( 2212 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2213 return CR1.usub_sat(CR2); 2214 }, 2215 [](const APInt &N1, const APInt &N2) { 2216 return N1.usub_sat(N2); 2217 }); 2218 } 2219 2220 TEST_F(ConstantRangeTest, UMulSat) { 2221 TestUnsignedBinOpExhaustive( 2222 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2223 return CR1.umul_sat(CR2); 2224 }, 2225 [](const APInt &N1, const APInt &N2) { return N1.umul_sat(N2); }); 2226 } 2227 2228 TEST_F(ConstantRangeTest, UShlSat) { 2229 TestUnsignedBinOpExhaustive( 2230 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2231 return CR1.ushl_sat(CR2); 2232 }, 2233 [](const APInt &N1, const APInt &N2) { return N1.ushl_sat(N2); }); 2234 } 2235 2236 TEST_F(ConstantRangeTest, SAddSat) { 2237 TestSignedBinOpExhaustive( 2238 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2239 return CR1.sadd_sat(CR2); 2240 }, 2241 [](const APInt &N1, const APInt &N2) { 2242 return N1.sadd_sat(N2); 2243 }); 2244 } 2245 2246 TEST_F(ConstantRangeTest, SSubSat) { 2247 TestSignedBinOpExhaustive( 2248 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2249 return CR1.ssub_sat(CR2); 2250 }, 2251 [](const APInt &N1, const APInt &N2) { 2252 return N1.ssub_sat(N2); 2253 }); 2254 } 2255 2256 TEST_F(ConstantRangeTest, SMulSat) { 2257 TestSignedBinOpExhaustive( 2258 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2259 return CR1.smul_sat(CR2); 2260 }, 2261 [](const APInt &N1, const APInt &N2) { return N1.smul_sat(N2); }); 2262 } 2263 2264 TEST_F(ConstantRangeTest, SShlSat) { 2265 TestSignedBinOpExhaustive( 2266 [](const ConstantRange &CR1, const ConstantRange &CR2) { 2267 return CR1.sshl_sat(CR2); 2268 }, 2269 [](const APInt &N1, const APInt &N2) { return N1.sshl_sat(N2); }); 2270 } 2271 2272 TEST_F(ConstantRangeTest, Abs) { 2273 unsigned Bits = 4; 2274 EnumerateConstantRanges(Bits, [&](const ConstantRange &CR) { 2275 // We're working with unsigned integers here, because it makes the signed 2276 // min case non-wrapping. 2277 APInt Min = APInt::getMaxValue(Bits); 2278 APInt Max = APInt::getMinValue(Bits); 2279 ForeachNumInConstantRange(CR, [&](const APInt &N) { 2280 APInt AbsN = N.abs(); 2281 if (AbsN.ult(Min)) 2282 Min = AbsN; 2283 if (AbsN.ugt(Max)) 2284 Max = AbsN; 2285 }); 2286 2287 ConstantRange AbsCR = CR.abs(); 2288 if (Min.ugt(Max)) { 2289 EXPECT_TRUE(AbsCR.isEmptySet()); 2290 return; 2291 } 2292 2293 ConstantRange Exact = ConstantRange::getNonEmpty(Min, Max + 1); 2294 EXPECT_EQ(Exact, AbsCR); 2295 }); 2296 } 2297 2298 TEST_F(ConstantRangeTest, castOps) { 2299 ConstantRange A(APInt(16, 66), APInt(16, 128)); 2300 ConstantRange FpToI8 = A.castOp(Instruction::FPToSI, 8); 2301 EXPECT_EQ(8u, FpToI8.getBitWidth()); 2302 EXPECT_TRUE(FpToI8.isFullSet()); 2303 2304 ConstantRange FpToI16 = A.castOp(Instruction::FPToSI, 16); 2305 EXPECT_EQ(16u, FpToI16.getBitWidth()); 2306 EXPECT_EQ(A, FpToI16); 2307 2308 ConstantRange FPExtToDouble = A.castOp(Instruction::FPExt, 64); 2309 EXPECT_EQ(64u, FPExtToDouble.getBitWidth()); 2310 EXPECT_TRUE(FPExtToDouble.isFullSet()); 2311 2312 ConstantRange PtrToInt = A.castOp(Instruction::PtrToInt, 64); 2313 EXPECT_EQ(64u, PtrToInt.getBitWidth()); 2314 EXPECT_TRUE(PtrToInt.isFullSet()); 2315 2316 ConstantRange IntToPtr = A.castOp(Instruction::IntToPtr, 64); 2317 EXPECT_EQ(64u, IntToPtr.getBitWidth()); 2318 EXPECT_TRUE(IntToPtr.isFullSet()); 2319 } 2320 2321 TEST_F(ConstantRangeTest, binaryXor) { 2322 // Single element ranges. 2323 ConstantRange R16(APInt(8, 16)); 2324 ConstantRange R20(APInt(8, 20)); 2325 EXPECT_EQ(*R16.binaryXor(R16).getSingleElement(), APInt(8, 0)); 2326 EXPECT_EQ(*R16.binaryXor(R20).getSingleElement(), APInt(8, 16 ^ 20)); 2327 2328 // Ranges with more than a single element. Handled conservatively for now. 2329 ConstantRange R16_35(APInt(8, 16), APInt(8, 35)); 2330 ConstantRange R0_99(APInt(8, 0), APInt(8, 99)); 2331 EXPECT_TRUE(R16_35.binaryXor(R16_35).isFullSet()); 2332 EXPECT_TRUE(R16_35.binaryXor(R0_99).isFullSet()); 2333 EXPECT_TRUE(R0_99.binaryXor(R16_35).isFullSet()); 2334 } 2335 2336 } // anonymous namespace 2337