1 //===- llvm/unittest/ADT/APInt.cpp - APInt unit 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/APInt.h" 10 #include "llvm/ADT/ArrayRef.h" 11 #include "llvm/ADT/SmallString.h" 12 #include "llvm/ADT/Twine.h" 13 #include "gtest/gtest.h" 14 #include <array> 15 16 using namespace llvm; 17 18 namespace { 19 20 TEST(APIntTest, ValueInit) { 21 APInt Zero = APInt(); 22 EXPECT_TRUE(!Zero); 23 EXPECT_TRUE(!Zero.zext(64)); 24 EXPECT_TRUE(!Zero.sext(64)); 25 } 26 27 // Test that APInt shift left works when bitwidth > 64 and shiftamt == 0 28 TEST(APIntTest, ShiftLeftByZero) { 29 APInt One = APInt::getZero(65) + 1; 30 APInt Shl = One.shl(0); 31 EXPECT_TRUE(Shl[0]); 32 EXPECT_FALSE(Shl[1]); 33 } 34 35 TEST(APIntTest, i64_ArithmeticRightShiftNegative) { 36 const APInt neg_one(64, static_cast<uint64_t>(-1), true); 37 EXPECT_EQ(neg_one, neg_one.ashr(7)); 38 } 39 40 TEST(APIntTest, i128_NegativeCount) { 41 APInt Minus3(128, static_cast<uint64_t>(-3), true); 42 EXPECT_EQ(126u, Minus3.countLeadingOnes()); 43 EXPECT_EQ(-3, Minus3.getSExtValue()); 44 45 APInt Minus1(128, static_cast<uint64_t>(-1), true); 46 EXPECT_EQ(0u, Minus1.countLeadingZeros()); 47 EXPECT_EQ(128u, Minus1.countLeadingOnes()); 48 EXPECT_EQ(128u, Minus1.getActiveBits()); 49 EXPECT_EQ(0u, Minus1.countTrailingZeros()); 50 EXPECT_EQ(128u, Minus1.countTrailingOnes()); 51 EXPECT_EQ(128u, Minus1.countPopulation()); 52 EXPECT_EQ(-1, Minus1.getSExtValue()); 53 } 54 55 TEST(APIntTest, i33_Count) { 56 APInt i33minus2(33, static_cast<uint64_t>(-2), true); 57 EXPECT_EQ(0u, i33minus2.countLeadingZeros()); 58 EXPECT_EQ(32u, i33minus2.countLeadingOnes()); 59 EXPECT_EQ(33u, i33minus2.getActiveBits()); 60 EXPECT_EQ(1u, i33minus2.countTrailingZeros()); 61 EXPECT_EQ(32u, i33minus2.countPopulation()); 62 EXPECT_EQ(-2, i33minus2.getSExtValue()); 63 EXPECT_EQ(((uint64_t)-2)&((1ull<<33) -1), i33minus2.getZExtValue()); 64 } 65 66 TEST(APIntTest, i61_Count) { 67 APInt i61(61, 1 << 15); 68 EXPECT_EQ(45u, i61.countLeadingZeros()); 69 EXPECT_EQ(0u, i61.countLeadingOnes()); 70 EXPECT_EQ(16u, i61.getActiveBits()); 71 EXPECT_EQ(15u, i61.countTrailingZeros()); 72 EXPECT_EQ(1u, i61.countPopulation()); 73 EXPECT_EQ(static_cast<int64_t>(1 << 15), i61.getSExtValue()); 74 EXPECT_EQ(static_cast<uint64_t>(1 << 15), i61.getZExtValue()); 75 76 i61.setBits(8, 19); 77 EXPECT_EQ(42u, i61.countLeadingZeros()); 78 EXPECT_EQ(0u, i61.countLeadingOnes()); 79 EXPECT_EQ(19u, i61.getActiveBits()); 80 EXPECT_EQ(8u, i61.countTrailingZeros()); 81 EXPECT_EQ(11u, i61.countPopulation()); 82 EXPECT_EQ(static_cast<int64_t>((1 << 19) - (1 << 8)), i61.getSExtValue()); 83 EXPECT_EQ(static_cast<uint64_t>((1 << 19) - (1 << 8)), i61.getZExtValue()); 84 } 85 86 TEST(APIntTest, i65_Count) { 87 APInt i65(65, 0, true); 88 EXPECT_EQ(65u, i65.countLeadingZeros()); 89 EXPECT_EQ(0u, i65.countLeadingOnes()); 90 EXPECT_EQ(0u, i65.getActiveBits()); 91 EXPECT_EQ(1u, i65.getActiveWords()); 92 EXPECT_EQ(65u, i65.countTrailingZeros()); 93 EXPECT_EQ(0u, i65.countPopulation()); 94 95 APInt i65minus(65, 0, true); 96 i65minus.setBit(64); 97 EXPECT_EQ(0u, i65minus.countLeadingZeros()); 98 EXPECT_EQ(1u, i65minus.countLeadingOnes()); 99 EXPECT_EQ(65u, i65minus.getActiveBits()); 100 EXPECT_EQ(64u, i65minus.countTrailingZeros()); 101 EXPECT_EQ(1u, i65minus.countPopulation()); 102 } 103 104 TEST(APIntTest, i128_PositiveCount) { 105 APInt u128max = APInt::getAllOnes(128); 106 EXPECT_EQ(128u, u128max.countLeadingOnes()); 107 EXPECT_EQ(0u, u128max.countLeadingZeros()); 108 EXPECT_EQ(128u, u128max.getActiveBits()); 109 EXPECT_EQ(0u, u128max.countTrailingZeros()); 110 EXPECT_EQ(128u, u128max.countTrailingOnes()); 111 EXPECT_EQ(128u, u128max.countPopulation()); 112 113 APInt u64max(128, static_cast<uint64_t>(-1), false); 114 EXPECT_EQ(64u, u64max.countLeadingZeros()); 115 EXPECT_EQ(0u, u64max.countLeadingOnes()); 116 EXPECT_EQ(64u, u64max.getActiveBits()); 117 EXPECT_EQ(0u, u64max.countTrailingZeros()); 118 EXPECT_EQ(64u, u64max.countTrailingOnes()); 119 EXPECT_EQ(64u, u64max.countPopulation()); 120 EXPECT_EQ((uint64_t)~0ull, u64max.getZExtValue()); 121 122 APInt zero(128, 0, true); 123 EXPECT_EQ(128u, zero.countLeadingZeros()); 124 EXPECT_EQ(0u, zero.countLeadingOnes()); 125 EXPECT_EQ(0u, zero.getActiveBits()); 126 EXPECT_EQ(128u, zero.countTrailingZeros()); 127 EXPECT_EQ(0u, zero.countTrailingOnes()); 128 EXPECT_EQ(0u, zero.countPopulation()); 129 EXPECT_EQ(0u, zero.getSExtValue()); 130 EXPECT_EQ(0u, zero.getZExtValue()); 131 132 APInt one(128, 1, true); 133 EXPECT_EQ(127u, one.countLeadingZeros()); 134 EXPECT_EQ(0u, one.countLeadingOnes()); 135 EXPECT_EQ(1u, one.getActiveBits()); 136 EXPECT_EQ(0u, one.countTrailingZeros()); 137 EXPECT_EQ(1u, one.countTrailingOnes()); 138 EXPECT_EQ(1u, one.countPopulation()); 139 EXPECT_EQ(1, one.getSExtValue()); 140 EXPECT_EQ(1u, one.getZExtValue()); 141 142 APInt s128(128, 2, true); 143 EXPECT_EQ(126u, s128.countLeadingZeros()); 144 EXPECT_EQ(0u, s128.countLeadingOnes()); 145 EXPECT_EQ(2u, s128.getActiveBits()); 146 EXPECT_EQ(1u, s128.countTrailingZeros()); 147 EXPECT_EQ(0u, s128.countTrailingOnes()); 148 EXPECT_EQ(1u, s128.countPopulation()); 149 EXPECT_EQ(2, s128.getSExtValue()); 150 EXPECT_EQ(2u, s128.getZExtValue()); 151 152 // NOP Test 153 s128.setBits(42, 42); 154 EXPECT_EQ(126u, s128.countLeadingZeros()); 155 EXPECT_EQ(0u, s128.countLeadingOnes()); 156 EXPECT_EQ(2u, s128.getActiveBits()); 157 EXPECT_EQ(1u, s128.countTrailingZeros()); 158 EXPECT_EQ(0u, s128.countTrailingOnes()); 159 EXPECT_EQ(1u, s128.countPopulation()); 160 EXPECT_EQ(2, s128.getSExtValue()); 161 EXPECT_EQ(2u, s128.getZExtValue()); 162 163 s128.setBits(3, 32); 164 EXPECT_EQ(96u, s128.countLeadingZeros()); 165 EXPECT_EQ(0u, s128.countLeadingOnes()); 166 EXPECT_EQ(32u, s128.getActiveBits()); 167 EXPECT_EQ(33u, s128.getMinSignedBits()); 168 EXPECT_EQ(1u, s128.countTrailingZeros()); 169 EXPECT_EQ(0u, s128.countTrailingOnes()); 170 EXPECT_EQ(30u, s128.countPopulation()); 171 EXPECT_EQ(static_cast<uint32_t>((~0u << 3) | 2), s128.getZExtValue()); 172 173 s128.setBits(62, 128); 174 EXPECT_EQ(0u, s128.countLeadingZeros()); 175 EXPECT_EQ(66u, s128.countLeadingOnes()); 176 EXPECT_EQ(128u, s128.getActiveBits()); 177 EXPECT_EQ(63u, s128.getMinSignedBits()); 178 EXPECT_EQ(1u, s128.countTrailingZeros()); 179 EXPECT_EQ(0u, s128.countTrailingOnes()); 180 EXPECT_EQ(96u, s128.countPopulation()); 181 EXPECT_EQ(static_cast<int64_t>((3ull << 62) | 182 static_cast<uint32_t>((~0u << 3) | 2)), 183 s128.getSExtValue()); 184 } 185 186 TEST(APIntTest, i256) { 187 APInt s256(256, 15, true); 188 EXPECT_EQ(252u, s256.countLeadingZeros()); 189 EXPECT_EQ(0u, s256.countLeadingOnes()); 190 EXPECT_EQ(4u, s256.getActiveBits()); 191 EXPECT_EQ(0u, s256.countTrailingZeros()); 192 EXPECT_EQ(4u, s256.countTrailingOnes()); 193 EXPECT_EQ(4u, s256.countPopulation()); 194 EXPECT_EQ(15, s256.getSExtValue()); 195 EXPECT_EQ(15u, s256.getZExtValue()); 196 197 s256.setBits(62, 66); 198 EXPECT_EQ(190u, s256.countLeadingZeros()); 199 EXPECT_EQ(0u, s256.countLeadingOnes()); 200 EXPECT_EQ(66u, s256.getActiveBits()); 201 EXPECT_EQ(67u, s256.getMinSignedBits()); 202 EXPECT_EQ(0u, s256.countTrailingZeros()); 203 EXPECT_EQ(4u, s256.countTrailingOnes()); 204 EXPECT_EQ(8u, s256.countPopulation()); 205 206 s256.setBits(60, 256); 207 EXPECT_EQ(0u, s256.countLeadingZeros()); 208 EXPECT_EQ(196u, s256.countLeadingOnes()); 209 EXPECT_EQ(256u, s256.getActiveBits()); 210 EXPECT_EQ(61u, s256.getMinSignedBits()); 211 EXPECT_EQ(0u, s256.countTrailingZeros()); 212 EXPECT_EQ(4u, s256.countTrailingOnes()); 213 EXPECT_EQ(200u, s256.countPopulation()); 214 EXPECT_EQ(static_cast<int64_t>((~0ull << 60) | 15), s256.getSExtValue()); 215 } 216 217 TEST(APIntTest, i1) { 218 const APInt neg_two(1, static_cast<uint64_t>(-2), true); 219 const APInt neg_one(1, static_cast<uint64_t>(-1), true); 220 const APInt zero(1, 0); 221 const APInt one(1, 1); 222 const APInt two(1, 2); 223 224 EXPECT_EQ(0, neg_two.getSExtValue()); 225 EXPECT_EQ(-1, neg_one.getSExtValue()); 226 EXPECT_EQ(1u, neg_one.getZExtValue()); 227 EXPECT_EQ(0u, zero.getZExtValue()); 228 EXPECT_EQ(-1, one.getSExtValue()); 229 EXPECT_EQ(1u, one.getZExtValue()); 230 EXPECT_EQ(0u, two.getZExtValue()); 231 EXPECT_EQ(0, two.getSExtValue()); 232 233 // Basic equalities for 1-bit values. 234 EXPECT_EQ(zero, two); 235 EXPECT_EQ(zero, neg_two); 236 EXPECT_EQ(one, neg_one); 237 EXPECT_EQ(two, neg_two); 238 239 // Min/max signed values. 240 EXPECT_TRUE(zero.isMaxSignedValue()); 241 EXPECT_FALSE(one.isMaxSignedValue()); 242 EXPECT_FALSE(zero.isMinSignedValue()); 243 EXPECT_TRUE(one.isMinSignedValue()); 244 245 // Additions. 246 EXPECT_EQ(two, one + one); 247 EXPECT_EQ(zero, neg_one + one); 248 EXPECT_EQ(neg_two, neg_one + neg_one); 249 250 // Subtractions. 251 EXPECT_EQ(neg_two, neg_one - one); 252 EXPECT_EQ(two, one - neg_one); 253 EXPECT_EQ(zero, one - one); 254 255 // And 256 EXPECT_EQ(zero, zero & zero); 257 EXPECT_EQ(zero, one & zero); 258 EXPECT_EQ(zero, zero & one); 259 EXPECT_EQ(one, one & one); 260 EXPECT_EQ(zero, zero & zero); 261 EXPECT_EQ(zero, neg_one & zero); 262 EXPECT_EQ(zero, zero & neg_one); 263 EXPECT_EQ(neg_one, neg_one & neg_one); 264 265 // Or 266 EXPECT_EQ(zero, zero | zero); 267 EXPECT_EQ(one, one | zero); 268 EXPECT_EQ(one, zero | one); 269 EXPECT_EQ(one, one | one); 270 EXPECT_EQ(zero, zero | zero); 271 EXPECT_EQ(neg_one, neg_one | zero); 272 EXPECT_EQ(neg_one, zero | neg_one); 273 EXPECT_EQ(neg_one, neg_one | neg_one); 274 275 // Xor 276 EXPECT_EQ(zero, zero ^ zero); 277 EXPECT_EQ(one, one ^ zero); 278 EXPECT_EQ(one, zero ^ one); 279 EXPECT_EQ(zero, one ^ one); 280 EXPECT_EQ(zero, zero ^ zero); 281 EXPECT_EQ(neg_one, neg_one ^ zero); 282 EXPECT_EQ(neg_one, zero ^ neg_one); 283 EXPECT_EQ(zero, neg_one ^ neg_one); 284 285 // Shifts. 286 EXPECT_EQ(zero, one << one); 287 EXPECT_EQ(one, one << zero); 288 EXPECT_EQ(zero, one.shl(1)); 289 EXPECT_EQ(one, one.shl(0)); 290 EXPECT_EQ(zero, one.lshr(1)); 291 EXPECT_EQ(one, one.ashr(1)); 292 293 // Rotates. 294 EXPECT_EQ(one, one.rotl(0)); 295 EXPECT_EQ(one, one.rotl(1)); 296 EXPECT_EQ(one, one.rotr(0)); 297 EXPECT_EQ(one, one.rotr(1)); 298 299 // Multiplies. 300 EXPECT_EQ(neg_one, neg_one * one); 301 EXPECT_EQ(neg_one, one * neg_one); 302 EXPECT_EQ(one, neg_one * neg_one); 303 EXPECT_EQ(one, one * one); 304 305 // Divides. 306 EXPECT_EQ(neg_one, one.sdiv(neg_one)); 307 EXPECT_EQ(neg_one, neg_one.sdiv(one)); 308 EXPECT_EQ(one, neg_one.sdiv(neg_one)); 309 EXPECT_EQ(one, one.sdiv(one)); 310 311 EXPECT_EQ(neg_one, one.udiv(neg_one)); 312 EXPECT_EQ(neg_one, neg_one.udiv(one)); 313 EXPECT_EQ(one, neg_one.udiv(neg_one)); 314 EXPECT_EQ(one, one.udiv(one)); 315 316 // Remainders. 317 EXPECT_EQ(zero, neg_one.srem(one)); 318 EXPECT_EQ(zero, neg_one.urem(one)); 319 EXPECT_EQ(zero, one.srem(neg_one)); 320 321 // sdivrem 322 { 323 APInt q(8, 0); 324 APInt r(8, 0); 325 APInt one(8, 1); 326 APInt two(8, 2); 327 APInt nine(8, 9); 328 APInt four(8, 4); 329 330 EXPECT_EQ(nine.srem(two), one); 331 EXPECT_EQ(nine.srem(-two), one); 332 EXPECT_EQ((-nine).srem(two), -one); 333 EXPECT_EQ((-nine).srem(-two), -one); 334 335 APInt::sdivrem(nine, two, q, r); 336 EXPECT_EQ(four, q); 337 EXPECT_EQ(one, r); 338 APInt::sdivrem(-nine, two, q, r); 339 EXPECT_EQ(-four, q); 340 EXPECT_EQ(-one, r); 341 APInt::sdivrem(nine, -two, q, r); 342 EXPECT_EQ(-four, q); 343 EXPECT_EQ(one, r); 344 APInt::sdivrem(-nine, -two, q, r); 345 EXPECT_EQ(four, q); 346 EXPECT_EQ(-one, r); 347 } 348 } 349 350 TEST(APIntTest, compare) { 351 std::array<APInt, 5> testVals{{ 352 APInt{16, 2}, 353 APInt{16, 1}, 354 APInt{16, 0}, 355 APInt{16, (uint64_t)-1, true}, 356 APInt{16, (uint64_t)-2, true}, 357 }}; 358 359 for (auto &arg1 : testVals) 360 for (auto &arg2 : testVals) { 361 auto uv1 = arg1.getZExtValue(); 362 auto uv2 = arg2.getZExtValue(); 363 auto sv1 = arg1.getSExtValue(); 364 auto sv2 = arg2.getSExtValue(); 365 366 EXPECT_EQ(uv1 < uv2, arg1.ult(arg2)); 367 EXPECT_EQ(uv1 <= uv2, arg1.ule(arg2)); 368 EXPECT_EQ(uv1 > uv2, arg1.ugt(arg2)); 369 EXPECT_EQ(uv1 >= uv2, arg1.uge(arg2)); 370 371 EXPECT_EQ(sv1 < sv2, arg1.slt(arg2)); 372 EXPECT_EQ(sv1 <= sv2, arg1.sle(arg2)); 373 EXPECT_EQ(sv1 > sv2, arg1.sgt(arg2)); 374 EXPECT_EQ(sv1 >= sv2, arg1.sge(arg2)); 375 376 EXPECT_EQ(uv1 < uv2, arg1.ult(uv2)); 377 EXPECT_EQ(uv1 <= uv2, arg1.ule(uv2)); 378 EXPECT_EQ(uv1 > uv2, arg1.ugt(uv2)); 379 EXPECT_EQ(uv1 >= uv2, arg1.uge(uv2)); 380 381 EXPECT_EQ(sv1 < sv2, arg1.slt(sv2)); 382 EXPECT_EQ(sv1 <= sv2, arg1.sle(sv2)); 383 EXPECT_EQ(sv1 > sv2, arg1.sgt(sv2)); 384 EXPECT_EQ(sv1 >= sv2, arg1.sge(sv2)); 385 } 386 } 387 388 TEST(APIntTest, compareWithRawIntegers) { 389 EXPECT_TRUE(!APInt(8, 1).uge(256)); 390 EXPECT_TRUE(!APInt(8, 1).ugt(256)); 391 EXPECT_TRUE( APInt(8, 1).ule(256)); 392 EXPECT_TRUE( APInt(8, 1).ult(256)); 393 EXPECT_TRUE(!APInt(8, 1).sge(256)); 394 EXPECT_TRUE(!APInt(8, 1).sgt(256)); 395 EXPECT_TRUE( APInt(8, 1).sle(256)); 396 EXPECT_TRUE( APInt(8, 1).slt(256)); 397 EXPECT_TRUE(!(APInt(8, 0) == 256)); 398 EXPECT_TRUE( APInt(8, 0) != 256); 399 EXPECT_TRUE(!(APInt(8, 1) == 256)); 400 EXPECT_TRUE( APInt(8, 1) != 256); 401 402 auto uint64max = UINT64_MAX; 403 auto int64max = INT64_MAX; 404 auto int64min = INT64_MIN; 405 406 auto u64 = APInt{128, uint64max}; 407 auto s64 = APInt{128, static_cast<uint64_t>(int64max), true}; 408 auto big = u64 + 1; 409 410 EXPECT_TRUE( u64.uge(uint64max)); 411 EXPECT_TRUE(!u64.ugt(uint64max)); 412 EXPECT_TRUE( u64.ule(uint64max)); 413 EXPECT_TRUE(!u64.ult(uint64max)); 414 EXPECT_TRUE( u64.sge(int64max)); 415 EXPECT_TRUE( u64.sgt(int64max)); 416 EXPECT_TRUE(!u64.sle(int64max)); 417 EXPECT_TRUE(!u64.slt(int64max)); 418 EXPECT_TRUE( u64.sge(int64min)); 419 EXPECT_TRUE( u64.sgt(int64min)); 420 EXPECT_TRUE(!u64.sle(int64min)); 421 EXPECT_TRUE(!u64.slt(int64min)); 422 423 EXPECT_TRUE(u64 == uint64max); 424 EXPECT_TRUE(u64 != int64max); 425 EXPECT_TRUE(u64 != int64min); 426 427 EXPECT_TRUE(!s64.uge(uint64max)); 428 EXPECT_TRUE(!s64.ugt(uint64max)); 429 EXPECT_TRUE( s64.ule(uint64max)); 430 EXPECT_TRUE( s64.ult(uint64max)); 431 EXPECT_TRUE( s64.sge(int64max)); 432 EXPECT_TRUE(!s64.sgt(int64max)); 433 EXPECT_TRUE( s64.sle(int64max)); 434 EXPECT_TRUE(!s64.slt(int64max)); 435 EXPECT_TRUE( s64.sge(int64min)); 436 EXPECT_TRUE( s64.sgt(int64min)); 437 EXPECT_TRUE(!s64.sle(int64min)); 438 EXPECT_TRUE(!s64.slt(int64min)); 439 440 EXPECT_TRUE(s64 != uint64max); 441 EXPECT_TRUE(s64 == int64max); 442 EXPECT_TRUE(s64 != int64min); 443 444 EXPECT_TRUE( big.uge(uint64max)); 445 EXPECT_TRUE( big.ugt(uint64max)); 446 EXPECT_TRUE(!big.ule(uint64max)); 447 EXPECT_TRUE(!big.ult(uint64max)); 448 EXPECT_TRUE( big.sge(int64max)); 449 EXPECT_TRUE( big.sgt(int64max)); 450 EXPECT_TRUE(!big.sle(int64max)); 451 EXPECT_TRUE(!big.slt(int64max)); 452 EXPECT_TRUE( big.sge(int64min)); 453 EXPECT_TRUE( big.sgt(int64min)); 454 EXPECT_TRUE(!big.sle(int64min)); 455 EXPECT_TRUE(!big.slt(int64min)); 456 457 EXPECT_TRUE(big != uint64max); 458 EXPECT_TRUE(big != int64max); 459 EXPECT_TRUE(big != int64min); 460 } 461 462 TEST(APIntTest, compareWithInt64Min) { 463 int64_t edge = INT64_MIN; 464 int64_t edgeP1 = edge + 1; 465 int64_t edgeM1 = INT64_MAX; 466 auto a = APInt{64, static_cast<uint64_t>(edge), true}; 467 468 EXPECT_TRUE(!a.slt(edge)); 469 EXPECT_TRUE( a.sle(edge)); 470 EXPECT_TRUE(!a.sgt(edge)); 471 EXPECT_TRUE( a.sge(edge)); 472 EXPECT_TRUE( a.slt(edgeP1)); 473 EXPECT_TRUE( a.sle(edgeP1)); 474 EXPECT_TRUE(!a.sgt(edgeP1)); 475 EXPECT_TRUE(!a.sge(edgeP1)); 476 EXPECT_TRUE( a.slt(edgeM1)); 477 EXPECT_TRUE( a.sle(edgeM1)); 478 EXPECT_TRUE(!a.sgt(edgeM1)); 479 EXPECT_TRUE(!a.sge(edgeM1)); 480 } 481 482 TEST(APIntTest, compareWithHalfInt64Max) { 483 uint64_t edge = 0x4000000000000000; 484 uint64_t edgeP1 = edge + 1; 485 uint64_t edgeM1 = edge - 1; 486 auto a = APInt{64, edge}; 487 488 EXPECT_TRUE(!a.ult(edge)); 489 EXPECT_TRUE( a.ule(edge)); 490 EXPECT_TRUE(!a.ugt(edge)); 491 EXPECT_TRUE( a.uge(edge)); 492 EXPECT_TRUE( a.ult(edgeP1)); 493 EXPECT_TRUE( a.ule(edgeP1)); 494 EXPECT_TRUE(!a.ugt(edgeP1)); 495 EXPECT_TRUE(!a.uge(edgeP1)); 496 EXPECT_TRUE(!a.ult(edgeM1)); 497 EXPECT_TRUE(!a.ule(edgeM1)); 498 EXPECT_TRUE( a.ugt(edgeM1)); 499 EXPECT_TRUE( a.uge(edgeM1)); 500 501 EXPECT_TRUE(!a.slt(edge)); 502 EXPECT_TRUE( a.sle(edge)); 503 EXPECT_TRUE(!a.sgt(edge)); 504 EXPECT_TRUE( a.sge(edge)); 505 EXPECT_TRUE( a.slt(edgeP1)); 506 EXPECT_TRUE( a.sle(edgeP1)); 507 EXPECT_TRUE(!a.sgt(edgeP1)); 508 EXPECT_TRUE(!a.sge(edgeP1)); 509 EXPECT_TRUE(!a.slt(edgeM1)); 510 EXPECT_TRUE(!a.sle(edgeM1)); 511 EXPECT_TRUE( a.sgt(edgeM1)); 512 EXPECT_TRUE( a.sge(edgeM1)); 513 } 514 515 TEST(APIntTest, compareLargeIntegers) { 516 // Make sure all the combinations of signed comparisons work with big ints. 517 auto One = APInt{128, static_cast<uint64_t>(1), true}; 518 auto Two = APInt{128, static_cast<uint64_t>(2), true}; 519 auto MinusOne = APInt{128, static_cast<uint64_t>(-1), true}; 520 auto MinusTwo = APInt{128, static_cast<uint64_t>(-2), true}; 521 522 EXPECT_TRUE(!One.slt(One)); 523 EXPECT_TRUE(!Two.slt(One)); 524 EXPECT_TRUE(MinusOne.slt(One)); 525 EXPECT_TRUE(MinusTwo.slt(One)); 526 527 EXPECT_TRUE(One.slt(Two)); 528 EXPECT_TRUE(!Two.slt(Two)); 529 EXPECT_TRUE(MinusOne.slt(Two)); 530 EXPECT_TRUE(MinusTwo.slt(Two)); 531 532 EXPECT_TRUE(!One.slt(MinusOne)); 533 EXPECT_TRUE(!Two.slt(MinusOne)); 534 EXPECT_TRUE(!MinusOne.slt(MinusOne)); 535 EXPECT_TRUE(MinusTwo.slt(MinusOne)); 536 537 EXPECT_TRUE(!One.slt(MinusTwo)); 538 EXPECT_TRUE(!Two.slt(MinusTwo)); 539 EXPECT_TRUE(!MinusOne.slt(MinusTwo)); 540 EXPECT_TRUE(!MinusTwo.slt(MinusTwo)); 541 } 542 543 TEST(APIntTest, binaryOpsWithRawIntegers) { 544 // Single word check. 545 uint64_t E1 = 0x2CA7F46BF6569915ULL; 546 APInt A1(64, E1); 547 548 EXPECT_EQ(A1 & E1, E1); 549 EXPECT_EQ(A1 & 0, 0); 550 EXPECT_EQ(A1 & 1, 1); 551 EXPECT_EQ(A1 & 5, 5); 552 EXPECT_EQ(A1 & UINT64_MAX, E1); 553 554 EXPECT_EQ(A1 | E1, E1); 555 EXPECT_EQ(A1 | 0, E1); 556 EXPECT_EQ(A1 | 1, E1); 557 EXPECT_EQ(A1 | 2, E1 | 2); 558 EXPECT_EQ(A1 | UINT64_MAX, UINT64_MAX); 559 560 EXPECT_EQ(A1 ^ E1, 0); 561 EXPECT_EQ(A1 ^ 0, E1); 562 EXPECT_EQ(A1 ^ 1, E1 ^ 1); 563 EXPECT_EQ(A1 ^ 7, E1 ^ 7); 564 EXPECT_EQ(A1 ^ UINT64_MAX, ~E1); 565 566 // Multiword check. 567 uint64_t N = 0xEB6EB136591CBA21ULL; 568 APInt::WordType E2[4] = { 569 N, 570 0x7B9358BD6A33F10AULL, 571 0x7E7FFA5EADD8846ULL, 572 0x305F341CA00B613DULL 573 }; 574 APInt A2(APInt::APINT_BITS_PER_WORD*4, E2); 575 576 EXPECT_EQ(A2 & N, N); 577 EXPECT_EQ(A2 & 0, 0); 578 EXPECT_EQ(A2 & 1, 1); 579 EXPECT_EQ(A2 & 5, 1); 580 EXPECT_EQ(A2 & UINT64_MAX, N); 581 582 EXPECT_EQ(A2 | N, A2); 583 EXPECT_EQ(A2 | 0, A2); 584 EXPECT_EQ(A2 | 1, A2); 585 EXPECT_EQ(A2 | 2, A2 + 2); 586 EXPECT_EQ(A2 | UINT64_MAX, A2 - N + UINT64_MAX); 587 588 EXPECT_EQ(A2 ^ N, A2 - N); 589 EXPECT_EQ(A2 ^ 0, A2); 590 EXPECT_EQ(A2 ^ 1, A2 - 1); 591 EXPECT_EQ(A2 ^ 7, A2 + 5); 592 EXPECT_EQ(A2 ^ UINT64_MAX, A2 - N + ~N); 593 } 594 595 TEST(APIntTest, rvalue_arithmetic) { 596 // Test all combinations of lvalue/rvalue lhs/rhs of add/sub 597 598 // Lamdba to return an APInt by value, but also provide the raw value of the 599 // allocated data. 600 auto getRValue = [](const char *HexString, uint64_t const *&RawData) { 601 APInt V(129, HexString, 16); 602 RawData = V.getRawData(); 603 return V; 604 }; 605 606 APInt One(129, "1", 16); 607 APInt Two(129, "2", 16); 608 APInt Three(129, "3", 16); 609 APInt MinusOne = -One; 610 611 const uint64_t *RawDataL = nullptr; 612 const uint64_t *RawDataR = nullptr; 613 614 { 615 // 1 + 1 = 2 616 APInt AddLL = One + One; 617 EXPECT_EQ(AddLL, Two); 618 619 APInt AddLR = One + getRValue("1", RawDataR); 620 EXPECT_EQ(AddLR, Two); 621 EXPECT_EQ(AddLR.getRawData(), RawDataR); 622 623 APInt AddRL = getRValue("1", RawDataL) + One; 624 EXPECT_EQ(AddRL, Two); 625 EXPECT_EQ(AddRL.getRawData(), RawDataL); 626 627 APInt AddRR = getRValue("1", RawDataL) + getRValue("1", RawDataR); 628 EXPECT_EQ(AddRR, Two); 629 EXPECT_EQ(AddRR.getRawData(), RawDataR); 630 631 // LValue's and constants 632 APInt AddLK = One + 1; 633 EXPECT_EQ(AddLK, Two); 634 635 APInt AddKL = 1 + One; 636 EXPECT_EQ(AddKL, Two); 637 638 // RValue's and constants 639 APInt AddRK = getRValue("1", RawDataL) + 1; 640 EXPECT_EQ(AddRK, Two); 641 EXPECT_EQ(AddRK.getRawData(), RawDataL); 642 643 APInt AddKR = 1 + getRValue("1", RawDataR); 644 EXPECT_EQ(AddKR, Two); 645 EXPECT_EQ(AddKR.getRawData(), RawDataR); 646 } 647 648 { 649 // 0x0,FFFF...FFFF + 0x2 = 0x100...0001 650 APInt AllOnes(129, "0FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", 16); 651 APInt HighOneLowOne(129, "100000000000000000000000000000001", 16); 652 653 APInt AddLL = AllOnes + Two; 654 EXPECT_EQ(AddLL, HighOneLowOne); 655 656 APInt AddLR = AllOnes + getRValue("2", RawDataR); 657 EXPECT_EQ(AddLR, HighOneLowOne); 658 EXPECT_EQ(AddLR.getRawData(), RawDataR); 659 660 APInt AddRL = getRValue("FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataL) + Two; 661 EXPECT_EQ(AddRL, HighOneLowOne); 662 EXPECT_EQ(AddRL.getRawData(), RawDataL); 663 664 APInt AddRR = getRValue("FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataL) + 665 getRValue("2", RawDataR); 666 EXPECT_EQ(AddRR, HighOneLowOne); 667 EXPECT_EQ(AddRR.getRawData(), RawDataR); 668 669 // LValue's and constants 670 APInt AddLK = AllOnes + 2; 671 EXPECT_EQ(AddLK, HighOneLowOne); 672 673 APInt AddKL = 2 + AllOnes; 674 EXPECT_EQ(AddKL, HighOneLowOne); 675 676 // RValue's and constants 677 APInt AddRK = getRValue("FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataL) + 2; 678 EXPECT_EQ(AddRK, HighOneLowOne); 679 EXPECT_EQ(AddRK.getRawData(), RawDataL); 680 681 APInt AddKR = 2 + getRValue("FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataR); 682 EXPECT_EQ(AddKR, HighOneLowOne); 683 EXPECT_EQ(AddKR.getRawData(), RawDataR); 684 } 685 686 { 687 // 2 - 1 = 1 688 APInt SubLL = Two - One; 689 EXPECT_EQ(SubLL, One); 690 691 APInt SubLR = Two - getRValue("1", RawDataR); 692 EXPECT_EQ(SubLR, One); 693 EXPECT_EQ(SubLR.getRawData(), RawDataR); 694 695 APInt SubRL = getRValue("2", RawDataL) - One; 696 EXPECT_EQ(SubRL, One); 697 EXPECT_EQ(SubRL.getRawData(), RawDataL); 698 699 APInt SubRR = getRValue("2", RawDataL) - getRValue("1", RawDataR); 700 EXPECT_EQ(SubRR, One); 701 EXPECT_EQ(SubRR.getRawData(), RawDataR); 702 703 // LValue's and constants 704 APInt SubLK = Two - 1; 705 EXPECT_EQ(SubLK, One); 706 707 APInt SubKL = 2 - One; 708 EXPECT_EQ(SubKL, One); 709 710 // RValue's and constants 711 APInt SubRK = getRValue("2", RawDataL) - 1; 712 EXPECT_EQ(SubRK, One); 713 EXPECT_EQ(SubRK.getRawData(), RawDataL); 714 715 APInt SubKR = 2 - getRValue("1", RawDataR); 716 EXPECT_EQ(SubKR, One); 717 EXPECT_EQ(SubKR.getRawData(), RawDataR); 718 } 719 720 { 721 // 0x100...0001 - 0x0,FFFF...FFFF = 0x2 722 APInt AllOnes(129, "0FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", 16); 723 APInt HighOneLowOne(129, "100000000000000000000000000000001", 16); 724 725 APInt SubLL = HighOneLowOne - AllOnes; 726 EXPECT_EQ(SubLL, Two); 727 728 APInt SubLR = HighOneLowOne - 729 getRValue("0FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataR); 730 EXPECT_EQ(SubLR, Two); 731 EXPECT_EQ(SubLR.getRawData(), RawDataR); 732 733 APInt SubRL = getRValue("100000000000000000000000000000001", RawDataL) - 734 AllOnes; 735 EXPECT_EQ(SubRL, Two); 736 EXPECT_EQ(SubRL.getRawData(), RawDataL); 737 738 APInt SubRR = getRValue("100000000000000000000000000000001", RawDataL) - 739 getRValue("0FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataR); 740 EXPECT_EQ(SubRR, Two); 741 EXPECT_EQ(SubRR.getRawData(), RawDataR); 742 743 // LValue's and constants 744 // 0x100...0001 - 0x2 = 0x0,FFFF...FFFF 745 APInt SubLK = HighOneLowOne - 2; 746 EXPECT_EQ(SubLK, AllOnes); 747 748 // 2 - (-1) = 3 749 APInt SubKL = 2 - MinusOne; 750 EXPECT_EQ(SubKL, Three); 751 752 // RValue's and constants 753 // 0x100...0001 - 0x2 = 0x0,FFFF...FFFF 754 APInt SubRK = getRValue("100000000000000000000000000000001", RawDataL) - 2; 755 EXPECT_EQ(SubRK, AllOnes); 756 EXPECT_EQ(SubRK.getRawData(), RawDataL); 757 758 APInt SubKR = 2 - getRValue("1FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataR); 759 EXPECT_EQ(SubKR, Three); 760 EXPECT_EQ(SubKR.getRawData(), RawDataR); 761 } 762 } 763 764 TEST(APIntTest, rvalue_bitwise) { 765 // Test all combinations of lvalue/rvalue lhs/rhs of and/or/xor 766 767 // Lamdba to return an APInt by value, but also provide the raw value of the 768 // allocated data. 769 auto getRValue = [](const char *HexString, uint64_t const *&RawData) { 770 APInt V(129, HexString, 16); 771 RawData = V.getRawData(); 772 return V; 773 }; 774 775 APInt Ten(129, "A", 16); 776 APInt Twelve(129, "C", 16); 777 778 const uint64_t *RawDataL = nullptr; 779 const uint64_t *RawDataR = nullptr; 780 781 { 782 // 12 & 10 = 8 783 APInt AndLL = Ten & Twelve; 784 EXPECT_EQ(AndLL, 0x8); 785 786 APInt AndLR = Ten & getRValue("C", RawDataR); 787 EXPECT_EQ(AndLR, 0x8); 788 EXPECT_EQ(AndLR.getRawData(), RawDataR); 789 790 APInt AndRL = getRValue("A", RawDataL) & Twelve; 791 EXPECT_EQ(AndRL, 0x8); 792 EXPECT_EQ(AndRL.getRawData(), RawDataL); 793 794 APInt AndRR = getRValue("A", RawDataL) & getRValue("C", RawDataR); 795 EXPECT_EQ(AndRR, 0x8); 796 EXPECT_EQ(AndRR.getRawData(), RawDataR); 797 798 // LValue's and constants 799 APInt AndLK = Ten & 0xc; 800 EXPECT_EQ(AndLK, 0x8); 801 802 APInt AndKL = 0xa & Twelve; 803 EXPECT_EQ(AndKL, 0x8); 804 805 // RValue's and constants 806 APInt AndRK = getRValue("A", RawDataL) & 0xc; 807 EXPECT_EQ(AndRK, 0x8); 808 EXPECT_EQ(AndRK.getRawData(), RawDataL); 809 810 APInt AndKR = 0xa & getRValue("C", RawDataR); 811 EXPECT_EQ(AndKR, 0x8); 812 EXPECT_EQ(AndKR.getRawData(), RawDataR); 813 } 814 815 { 816 // 12 | 10 = 14 817 APInt OrLL = Ten | Twelve; 818 EXPECT_EQ(OrLL, 0xe); 819 820 APInt OrLR = Ten | getRValue("C", RawDataR); 821 EXPECT_EQ(OrLR, 0xe); 822 EXPECT_EQ(OrLR.getRawData(), RawDataR); 823 824 APInt OrRL = getRValue("A", RawDataL) | Twelve; 825 EXPECT_EQ(OrRL, 0xe); 826 EXPECT_EQ(OrRL.getRawData(), RawDataL); 827 828 APInt OrRR = getRValue("A", RawDataL) | getRValue("C", RawDataR); 829 EXPECT_EQ(OrRR, 0xe); 830 EXPECT_EQ(OrRR.getRawData(), RawDataR); 831 832 // LValue's and constants 833 APInt OrLK = Ten | 0xc; 834 EXPECT_EQ(OrLK, 0xe); 835 836 APInt OrKL = 0xa | Twelve; 837 EXPECT_EQ(OrKL, 0xe); 838 839 // RValue's and constants 840 APInt OrRK = getRValue("A", RawDataL) | 0xc; 841 EXPECT_EQ(OrRK, 0xe); 842 EXPECT_EQ(OrRK.getRawData(), RawDataL); 843 844 APInt OrKR = 0xa | getRValue("C", RawDataR); 845 EXPECT_EQ(OrKR, 0xe); 846 EXPECT_EQ(OrKR.getRawData(), RawDataR); 847 } 848 849 { 850 // 12 ^ 10 = 6 851 APInt XorLL = Ten ^ Twelve; 852 EXPECT_EQ(XorLL, 0x6); 853 854 APInt XorLR = Ten ^ getRValue("C", RawDataR); 855 EXPECT_EQ(XorLR, 0x6); 856 EXPECT_EQ(XorLR.getRawData(), RawDataR); 857 858 APInt XorRL = getRValue("A", RawDataL) ^ Twelve; 859 EXPECT_EQ(XorRL, 0x6); 860 EXPECT_EQ(XorRL.getRawData(), RawDataL); 861 862 APInt XorRR = getRValue("A", RawDataL) ^ getRValue("C", RawDataR); 863 EXPECT_EQ(XorRR, 0x6); 864 EXPECT_EQ(XorRR.getRawData(), RawDataR); 865 866 // LValue's and constants 867 APInt XorLK = Ten ^ 0xc; 868 EXPECT_EQ(XorLK, 0x6); 869 870 APInt XorKL = 0xa ^ Twelve; 871 EXPECT_EQ(XorKL, 0x6); 872 873 // RValue's and constants 874 APInt XorRK = getRValue("A", RawDataL) ^ 0xc; 875 EXPECT_EQ(XorRK, 0x6); 876 EXPECT_EQ(XorRK.getRawData(), RawDataL); 877 878 APInt XorKR = 0xa ^ getRValue("C", RawDataR); 879 EXPECT_EQ(XorKR, 0x6); 880 EXPECT_EQ(XorKR.getRawData(), RawDataR); 881 } 882 } 883 884 TEST(APIntTest, rvalue_invert) { 885 // Lamdba to return an APInt by value, but also provide the raw value of the 886 // allocated data. 887 auto getRValue = [](const char *HexString, uint64_t const *&RawData) { 888 APInt V(129, HexString, 16); 889 RawData = V.getRawData(); 890 return V; 891 }; 892 893 APInt One(129, 1); 894 APInt NegativeTwo(129, -2ULL, true); 895 896 const uint64_t *RawData = nullptr; 897 898 { 899 // ~1 = -2 900 APInt NegL = ~One; 901 EXPECT_EQ(NegL, NegativeTwo); 902 903 APInt NegR = ~getRValue("1", RawData); 904 EXPECT_EQ(NegR, NegativeTwo); 905 EXPECT_EQ(NegR.getRawData(), RawData); 906 } 907 } 908 909 // Tests different div/rem varaints using scheme (a * b + c) / a 910 void testDiv(APInt a, APInt b, APInt c) { 911 ASSERT_TRUE(a.uge(b)); // Must: a >= b 912 ASSERT_TRUE(a.ugt(c)); // Must: a > c 913 914 auto p = a * b + c; 915 916 auto q = p.udiv(a); 917 auto r = p.urem(a); 918 EXPECT_EQ(b, q); 919 EXPECT_EQ(c, r); 920 APInt::udivrem(p, a, q, r); 921 EXPECT_EQ(b, q); 922 EXPECT_EQ(c, r); 923 q = p.sdiv(a); 924 r = p.srem(a); 925 EXPECT_EQ(b, q); 926 EXPECT_EQ(c, r); 927 APInt::sdivrem(p, a, q, r); 928 EXPECT_EQ(b, q); 929 EXPECT_EQ(c, r); 930 931 if (b.ugt(c)) { // Test also symmetric case 932 q = p.udiv(b); 933 r = p.urem(b); 934 EXPECT_EQ(a, q); 935 EXPECT_EQ(c, r); 936 APInt::udivrem(p, b, q, r); 937 EXPECT_EQ(a, q); 938 EXPECT_EQ(c, r); 939 q = p.sdiv(b); 940 r = p.srem(b); 941 EXPECT_EQ(a, q); 942 EXPECT_EQ(c, r); 943 APInt::sdivrem(p, b, q, r); 944 EXPECT_EQ(a, q); 945 EXPECT_EQ(c, r); 946 } 947 } 948 949 TEST(APIntTest, divrem_big1) { 950 // Tests KnuthDiv rare step D6 951 testDiv({256, "1ffffffffffffffff", 16}, 952 {256, "1ffffffffffffffff", 16}, 953 {256, 0}); 954 } 955 956 TEST(APIntTest, divrem_big2) { 957 // Tests KnuthDiv rare step D6 958 testDiv({1024, "112233ceff" 959 "cecece000000ffffffffffffffffffff" 960 "ffffffffffffffffffffffffffffffff" 961 "ffffffffffffffffffffffffffffffff" 962 "ffffffffffffffffffffffffffffff33", 16}, 963 {1024, "111111ffffffffffffffff" 964 "ffffffffffffffffffffffffffffffff" 965 "fffffffffffffffffffffffffffffccf" 966 "ffffffffffffffffffffffffffffff00", 16}, 967 {1024, 7919}); 968 } 969 970 TEST(APIntTest, divrem_big3) { 971 // Tests KnuthDiv case without shift 972 testDiv({256, "80000001ffffffffffffffff", 16}, 973 {256, "ffffffffffffff0000000", 16}, 974 {256, 4219}); 975 } 976 977 TEST(APIntTest, divrem_big4) { 978 // Tests heap allocation in divide() enfoced by huge numbers 979 testDiv(APInt{4096, 5}.shl(2001), 980 APInt{4096, 1}.shl(2000), 981 APInt{4096, 4219*13}); 982 } 983 984 TEST(APIntTest, divrem_big5) { 985 // Tests one word divisor case of divide() 986 testDiv(APInt{1024, 19}.shl(811), 987 APInt{1024, 4356013}, // one word 988 APInt{1024, 1}); 989 } 990 991 TEST(APIntTest, divrem_big6) { 992 // Tests some rare "borrow" cases in D4 step 993 testDiv(APInt{512, "ffffffffffffffff00000000000000000000000001", 16}, 994 APInt{512, "10000000000000001000000000000001", 16}, 995 APInt{512, "10000000000000000000000000000000", 16}); 996 } 997 998 TEST(APIntTest, divrem_big7) { 999 // Yet another test for KnuthDiv rare step D6. 1000 testDiv({224, "800000008000000200000005", 16}, 1001 {224, "fffffffd", 16}, 1002 {224, "80000000800000010000000f", 16}); 1003 } 1004 1005 void testDiv(APInt a, uint64_t b, APInt c) { 1006 auto p = a * b + c; 1007 1008 APInt q; 1009 uint64_t r; 1010 // Unsigned division will only work if our original number wasn't negative. 1011 if (!a.isNegative()) { 1012 q = p.udiv(b); 1013 r = p.urem(b); 1014 EXPECT_EQ(a, q); 1015 EXPECT_EQ(c, r); 1016 APInt::udivrem(p, b, q, r); 1017 EXPECT_EQ(a, q); 1018 EXPECT_EQ(c, r); 1019 } 1020 q = p.sdiv(b); 1021 r = p.srem(b); 1022 EXPECT_EQ(a, q); 1023 if (c.isNegative()) 1024 EXPECT_EQ(-c, -r); // Need to negate so the uint64_t compare will work. 1025 else 1026 EXPECT_EQ(c, r); 1027 int64_t sr; 1028 APInt::sdivrem(p, b, q, sr); 1029 EXPECT_EQ(a, q); 1030 if (c.isNegative()) 1031 EXPECT_EQ(-c, -sr); // Need to negate so the uint64_t compare will work. 1032 else 1033 EXPECT_EQ(c, sr); 1034 } 1035 1036 TEST(APIntTest, divremuint) { 1037 // Single word APInt 1038 testDiv(APInt{64, 9}, 1039 2, 1040 APInt{64, 1}); 1041 1042 // Single word negative APInt 1043 testDiv(-APInt{64, 9}, 1044 2, 1045 -APInt{64, 1}); 1046 1047 // Multiword dividend with only one significant word. 1048 testDiv(APInt{256, 9}, 1049 2, 1050 APInt{256, 1}); 1051 1052 // Negative dividend. 1053 testDiv(-APInt{256, 9}, 1054 2, 1055 -APInt{256, 1}); 1056 1057 // Multiword dividend 1058 testDiv(APInt{1024, 19}.shl(811), 1059 4356013, // one word 1060 APInt{1024, 1}); 1061 } 1062 1063 TEST(APIntTest, divrem_simple) { 1064 // Test simple cases. 1065 APInt A(65, 2), B(65, 2); 1066 APInt Q, R; 1067 1068 // X / X 1069 APInt::sdivrem(A, B, Q, R); 1070 EXPECT_EQ(Q, APInt(65, 1)); 1071 EXPECT_EQ(R, APInt(65, 0)); 1072 APInt::udivrem(A, B, Q, R); 1073 EXPECT_EQ(Q, APInt(65, 1)); 1074 EXPECT_EQ(R, APInt(65, 0)); 1075 1076 // 0 / X 1077 APInt O(65, 0); 1078 APInt::sdivrem(O, B, Q, R); 1079 EXPECT_EQ(Q, APInt(65, 0)); 1080 EXPECT_EQ(R, APInt(65, 0)); 1081 APInt::udivrem(O, B, Q, R); 1082 EXPECT_EQ(Q, APInt(65, 0)); 1083 EXPECT_EQ(R, APInt(65, 0)); 1084 1085 // X / 1 1086 APInt I(65, 1); 1087 APInt::sdivrem(A, I, Q, R); 1088 EXPECT_EQ(Q, A); 1089 EXPECT_EQ(R, APInt(65, 0)); 1090 APInt::udivrem(A, I, Q, R); 1091 EXPECT_EQ(Q, A); 1092 EXPECT_EQ(R, APInt(65, 0)); 1093 } 1094 1095 TEST(APIntTest, fromString) { 1096 EXPECT_EQ(APInt(32, 0), APInt(32, "0", 2)); 1097 EXPECT_EQ(APInt(32, 1), APInt(32, "1", 2)); 1098 EXPECT_EQ(APInt(32, 2), APInt(32, "10", 2)); 1099 EXPECT_EQ(APInt(32, 3), APInt(32, "11", 2)); 1100 EXPECT_EQ(APInt(32, 4), APInt(32, "100", 2)); 1101 1102 EXPECT_EQ(APInt(32, 0), APInt(32, "+0", 2)); 1103 EXPECT_EQ(APInt(32, 1), APInt(32, "+1", 2)); 1104 EXPECT_EQ(APInt(32, 2), APInt(32, "+10", 2)); 1105 EXPECT_EQ(APInt(32, 3), APInt(32, "+11", 2)); 1106 EXPECT_EQ(APInt(32, 4), APInt(32, "+100", 2)); 1107 1108 EXPECT_EQ(APInt(32, uint64_t(-0LL)), APInt(32, "-0", 2)); 1109 EXPECT_EQ(APInt(32, uint64_t(-1LL)), APInt(32, "-1", 2)); 1110 EXPECT_EQ(APInt(32, uint64_t(-2LL)), APInt(32, "-10", 2)); 1111 EXPECT_EQ(APInt(32, uint64_t(-3LL)), APInt(32, "-11", 2)); 1112 EXPECT_EQ(APInt(32, uint64_t(-4LL)), APInt(32, "-100", 2)); 1113 1114 EXPECT_EQ(APInt(32, 0), APInt(32, "0", 8)); 1115 EXPECT_EQ(APInt(32, 1), APInt(32, "1", 8)); 1116 EXPECT_EQ(APInt(32, 7), APInt(32, "7", 8)); 1117 EXPECT_EQ(APInt(32, 8), APInt(32, "10", 8)); 1118 EXPECT_EQ(APInt(32, 15), APInt(32, "17", 8)); 1119 EXPECT_EQ(APInt(32, 16), APInt(32, "20", 8)); 1120 1121 EXPECT_EQ(APInt(32, +0), APInt(32, "+0", 8)); 1122 EXPECT_EQ(APInt(32, +1), APInt(32, "+1", 8)); 1123 EXPECT_EQ(APInt(32, +7), APInt(32, "+7", 8)); 1124 EXPECT_EQ(APInt(32, +8), APInt(32, "+10", 8)); 1125 EXPECT_EQ(APInt(32, +15), APInt(32, "+17", 8)); 1126 EXPECT_EQ(APInt(32, +16), APInt(32, "+20", 8)); 1127 1128 EXPECT_EQ(APInt(32, uint64_t(-0LL)), APInt(32, "-0", 8)); 1129 EXPECT_EQ(APInt(32, uint64_t(-1LL)), APInt(32, "-1", 8)); 1130 EXPECT_EQ(APInt(32, uint64_t(-7LL)), APInt(32, "-7", 8)); 1131 EXPECT_EQ(APInt(32, uint64_t(-8LL)), APInt(32, "-10", 8)); 1132 EXPECT_EQ(APInt(32, uint64_t(-15LL)), APInt(32, "-17", 8)); 1133 EXPECT_EQ(APInt(32, uint64_t(-16LL)), APInt(32, "-20", 8)); 1134 1135 EXPECT_EQ(APInt(32, 0), APInt(32, "0", 10)); 1136 EXPECT_EQ(APInt(32, 1), APInt(32, "1", 10)); 1137 EXPECT_EQ(APInt(32, 9), APInt(32, "9", 10)); 1138 EXPECT_EQ(APInt(32, 10), APInt(32, "10", 10)); 1139 EXPECT_EQ(APInt(32, 19), APInt(32, "19", 10)); 1140 EXPECT_EQ(APInt(32, 20), APInt(32, "20", 10)); 1141 1142 EXPECT_EQ(APInt(32, uint64_t(-0LL)), APInt(32, "-0", 10)); 1143 EXPECT_EQ(APInt(32, uint64_t(-1LL)), APInt(32, "-1", 10)); 1144 EXPECT_EQ(APInt(32, uint64_t(-9LL)), APInt(32, "-9", 10)); 1145 EXPECT_EQ(APInt(32, uint64_t(-10LL)), APInt(32, "-10", 10)); 1146 EXPECT_EQ(APInt(32, uint64_t(-19LL)), APInt(32, "-19", 10)); 1147 EXPECT_EQ(APInt(32, uint64_t(-20LL)), APInt(32, "-20", 10)); 1148 1149 EXPECT_EQ(APInt(32, 0), APInt(32, "0", 16)); 1150 EXPECT_EQ(APInt(32, 1), APInt(32, "1", 16)); 1151 EXPECT_EQ(APInt(32, 15), APInt(32, "F", 16)); 1152 EXPECT_EQ(APInt(32, 16), APInt(32, "10", 16)); 1153 EXPECT_EQ(APInt(32, 31), APInt(32, "1F", 16)); 1154 EXPECT_EQ(APInt(32, 32), APInt(32, "20", 16)); 1155 1156 EXPECT_EQ(APInt(32, uint64_t(-0LL)), APInt(32, "-0", 16)); 1157 EXPECT_EQ(APInt(32, uint64_t(-1LL)), APInt(32, "-1", 16)); 1158 EXPECT_EQ(APInt(32, uint64_t(-15LL)), APInt(32, "-F", 16)); 1159 EXPECT_EQ(APInt(32, uint64_t(-16LL)), APInt(32, "-10", 16)); 1160 EXPECT_EQ(APInt(32, uint64_t(-31LL)), APInt(32, "-1F", 16)); 1161 EXPECT_EQ(APInt(32, uint64_t(-32LL)), APInt(32, "-20", 16)); 1162 1163 EXPECT_EQ(APInt(32, 0), APInt(32, "0", 36)); 1164 EXPECT_EQ(APInt(32, 1), APInt(32, "1", 36)); 1165 EXPECT_EQ(APInt(32, 35), APInt(32, "Z", 36)); 1166 EXPECT_EQ(APInt(32, 36), APInt(32, "10", 36)); 1167 EXPECT_EQ(APInt(32, 71), APInt(32, "1Z", 36)); 1168 EXPECT_EQ(APInt(32, 72), APInt(32, "20", 36)); 1169 1170 EXPECT_EQ(APInt(32, uint64_t(-0LL)), APInt(32, "-0", 36)); 1171 EXPECT_EQ(APInt(32, uint64_t(-1LL)), APInt(32, "-1", 36)); 1172 EXPECT_EQ(APInt(32, uint64_t(-35LL)), APInt(32, "-Z", 36)); 1173 EXPECT_EQ(APInt(32, uint64_t(-36LL)), APInt(32, "-10", 36)); 1174 EXPECT_EQ(APInt(32, uint64_t(-71LL)), APInt(32, "-1Z", 36)); 1175 EXPECT_EQ(APInt(32, uint64_t(-72LL)), APInt(32, "-20", 36)); 1176 } 1177 1178 TEST(APIntTest, SaturatingMath) { 1179 APInt AP_10 = APInt(8, 10); 1180 APInt AP_42 = APInt(8, 42); 1181 APInt AP_100 = APInt(8, 100); 1182 APInt AP_200 = APInt(8, 200); 1183 1184 EXPECT_EQ(APInt(7, 100), AP_100.truncUSat(7)); 1185 EXPECT_EQ(APInt(6, 63), AP_100.truncUSat(6)); 1186 EXPECT_EQ(APInt(5, 31), AP_100.truncUSat(5)); 1187 1188 EXPECT_EQ(APInt(7, 127), AP_200.truncUSat(7)); 1189 EXPECT_EQ(APInt(6, 63), AP_200.truncUSat(6)); 1190 EXPECT_EQ(APInt(5, 31), AP_200.truncUSat(5)); 1191 1192 EXPECT_EQ(APInt(7, 42), AP_42.truncSSat(7)); 1193 EXPECT_EQ(APInt(6, 31), AP_42.truncSSat(6)); 1194 EXPECT_EQ(APInt(5, 15), AP_42.truncSSat(5)); 1195 1196 EXPECT_EQ(APInt(7, -56), AP_200.truncSSat(7)); 1197 EXPECT_EQ(APInt(6, -32), AP_200.truncSSat(6)); 1198 EXPECT_EQ(APInt(5, -16), AP_200.truncSSat(5)); 1199 1200 EXPECT_EQ(APInt(8, 200), AP_100.uadd_sat(AP_100)); 1201 EXPECT_EQ(APInt(8, 255), AP_100.uadd_sat(AP_200)); 1202 EXPECT_EQ(APInt(8, 255), APInt(8, 255).uadd_sat(APInt(8, 255))); 1203 1204 EXPECT_EQ(APInt(8, 110), AP_10.sadd_sat(AP_100)); 1205 EXPECT_EQ(APInt(8, 127), AP_100.sadd_sat(AP_100)); 1206 EXPECT_EQ(APInt(8, -128), (-AP_100).sadd_sat(-AP_100)); 1207 EXPECT_EQ(APInt(8, -128), APInt(8, -128).sadd_sat(APInt(8, -128))); 1208 1209 EXPECT_EQ(APInt(8, 90), AP_100.usub_sat(AP_10)); 1210 EXPECT_EQ(APInt(8, 0), AP_100.usub_sat(AP_200)); 1211 EXPECT_EQ(APInt(8, 0), APInt(8, 0).usub_sat(APInt(8, 255))); 1212 1213 EXPECT_EQ(APInt(8, -90), AP_10.ssub_sat(AP_100)); 1214 EXPECT_EQ(APInt(8, 127), AP_100.ssub_sat(-AP_100)); 1215 EXPECT_EQ(APInt(8, -128), (-AP_100).ssub_sat(AP_100)); 1216 EXPECT_EQ(APInt(8, -128), APInt(8, -128).ssub_sat(APInt(8, 127))); 1217 1218 EXPECT_EQ(APInt(8, 250), APInt(8, 50).umul_sat(APInt(8, 5))); 1219 EXPECT_EQ(APInt(8, 255), APInt(8, 50).umul_sat(APInt(8, 6))); 1220 EXPECT_EQ(APInt(8, 255), APInt(8, -128).umul_sat(APInt(8, 3))); 1221 EXPECT_EQ(APInt(8, 255), APInt(8, 3).umul_sat(APInt(8, -128))); 1222 EXPECT_EQ(APInt(8, 255), APInt(8, -128).umul_sat(APInt(8, -128))); 1223 1224 EXPECT_EQ(APInt(8, 125), APInt(8, 25).smul_sat(APInt(8, 5))); 1225 EXPECT_EQ(APInt(8, 127), APInt(8, 25).smul_sat(APInt(8, 6))); 1226 EXPECT_EQ(APInt(8, 127), APInt(8, 127).smul_sat(APInt(8, 127))); 1227 EXPECT_EQ(APInt(8, -125), APInt(8, -25).smul_sat(APInt(8, 5))); 1228 EXPECT_EQ(APInt(8, -125), APInt(8, 25).smul_sat(APInt(8, -5))); 1229 EXPECT_EQ(APInt(8, 125), APInt(8, -25).smul_sat(APInt(8, -5))); 1230 EXPECT_EQ(APInt(8, 125), APInt(8, 25).smul_sat(APInt(8, 5))); 1231 EXPECT_EQ(APInt(8, -128), APInt(8, -25).smul_sat(APInt(8, 6))); 1232 EXPECT_EQ(APInt(8, -128), APInt(8, 25).smul_sat(APInt(8, -6))); 1233 EXPECT_EQ(APInt(8, 127), APInt(8, -25).smul_sat(APInt(8, -6))); 1234 EXPECT_EQ(APInt(8, 127), APInt(8, 25).smul_sat(APInt(8, 6))); 1235 1236 EXPECT_EQ(APInt(8, 128), APInt(8, 4).ushl_sat(APInt(8, 5))); 1237 EXPECT_EQ(APInt(8, 255), APInt(8, 4).ushl_sat(APInt(8, 6))); 1238 EXPECT_EQ(APInt(8, 128), APInt(8, 1).ushl_sat(APInt(8, 7))); 1239 EXPECT_EQ(APInt(8, 255), APInt(8, 1).ushl_sat(APInt(8, 8))); 1240 EXPECT_EQ(APInt(8, 255), APInt(8, -128).ushl_sat(APInt(8, 2))); 1241 EXPECT_EQ(APInt(8, 255), APInt(8, 64).ushl_sat(APInt(8, 2))); 1242 EXPECT_EQ(APInt(8, 255), APInt(8, 64).ushl_sat(APInt(8, -2))); 1243 1244 EXPECT_EQ(APInt(8, 64), APInt(8, 4).sshl_sat(APInt(8, 4))); 1245 EXPECT_EQ(APInt(8, 127), APInt(8, 4).sshl_sat(APInt(8, 5))); 1246 EXPECT_EQ(APInt(8, 127), APInt(8, 1).sshl_sat(APInt(8, 8))); 1247 EXPECT_EQ(APInt(8, -64), APInt(8, -4).sshl_sat(APInt(8, 4))); 1248 EXPECT_EQ(APInt(8, -128), APInt(8, -4).sshl_sat(APInt(8, 5))); 1249 EXPECT_EQ(APInt(8, -128), APInt(8, -4).sshl_sat(APInt(8, 6))); 1250 EXPECT_EQ(APInt(8, -128), APInt(8, -1).sshl_sat(APInt(8, 7))); 1251 EXPECT_EQ(APInt(8, -128), APInt(8, -1).sshl_sat(APInt(8, 8))); 1252 } 1253 1254 TEST(APIntTest, FromArray) { 1255 EXPECT_EQ(APInt(32, uint64_t(1)), APInt(32, ArrayRef<uint64_t>(1))); 1256 } 1257 1258 TEST(APIntTest, StringBitsNeeded2) { 1259 EXPECT_EQ(1U, APInt::getBitsNeeded( "0", 2)); 1260 EXPECT_EQ(1U, APInt::getBitsNeeded( "1", 2)); 1261 EXPECT_EQ(2U, APInt::getBitsNeeded( "10", 2)); 1262 EXPECT_EQ(2U, APInt::getBitsNeeded( "11", 2)); 1263 EXPECT_EQ(3U, APInt::getBitsNeeded("100", 2)); 1264 1265 EXPECT_EQ(1U, APInt::getBitsNeeded( "+0", 2)); 1266 EXPECT_EQ(1U, APInt::getBitsNeeded( "+1", 2)); 1267 EXPECT_EQ(2U, APInt::getBitsNeeded( "+10", 2)); 1268 EXPECT_EQ(2U, APInt::getBitsNeeded( "+11", 2)); 1269 EXPECT_EQ(3U, APInt::getBitsNeeded("+100", 2)); 1270 1271 EXPECT_EQ(2U, APInt::getBitsNeeded( "-0", 2)); 1272 EXPECT_EQ(2U, APInt::getBitsNeeded( "-1", 2)); 1273 EXPECT_EQ(3U, APInt::getBitsNeeded( "-10", 2)); 1274 EXPECT_EQ(3U, APInt::getBitsNeeded( "-11", 2)); 1275 EXPECT_EQ(4U, APInt::getBitsNeeded("-100", 2)); 1276 } 1277 1278 TEST(APIntTest, StringBitsNeeded8) { 1279 EXPECT_EQ(3U, APInt::getBitsNeeded( "0", 8)); 1280 EXPECT_EQ(3U, APInt::getBitsNeeded( "7", 8)); 1281 EXPECT_EQ(6U, APInt::getBitsNeeded("10", 8)); 1282 EXPECT_EQ(6U, APInt::getBitsNeeded("17", 8)); 1283 EXPECT_EQ(6U, APInt::getBitsNeeded("20", 8)); 1284 1285 EXPECT_EQ(3U, APInt::getBitsNeeded( "+0", 8)); 1286 EXPECT_EQ(3U, APInt::getBitsNeeded( "+7", 8)); 1287 EXPECT_EQ(6U, APInt::getBitsNeeded("+10", 8)); 1288 EXPECT_EQ(6U, APInt::getBitsNeeded("+17", 8)); 1289 EXPECT_EQ(6U, APInt::getBitsNeeded("+20", 8)); 1290 1291 EXPECT_EQ(4U, APInt::getBitsNeeded( "-0", 8)); 1292 EXPECT_EQ(4U, APInt::getBitsNeeded( "-7", 8)); 1293 EXPECT_EQ(7U, APInt::getBitsNeeded("-10", 8)); 1294 EXPECT_EQ(7U, APInt::getBitsNeeded("-17", 8)); 1295 EXPECT_EQ(7U, APInt::getBitsNeeded("-20", 8)); 1296 } 1297 1298 TEST(APIntTest, StringBitsNeeded10) { 1299 EXPECT_EQ(1U, APInt::getBitsNeeded( "0", 10)); 1300 EXPECT_EQ(2U, APInt::getBitsNeeded( "3", 10)); 1301 EXPECT_EQ(4U, APInt::getBitsNeeded( "9", 10)); 1302 EXPECT_EQ(4U, APInt::getBitsNeeded("10", 10)); 1303 EXPECT_EQ(5U, APInt::getBitsNeeded("19", 10)); 1304 EXPECT_EQ(5U, APInt::getBitsNeeded("20", 10)); 1305 1306 EXPECT_EQ(1U, APInt::getBitsNeeded( "+0", 10)); 1307 EXPECT_EQ(4U, APInt::getBitsNeeded( "+9", 10)); 1308 EXPECT_EQ(4U, APInt::getBitsNeeded("+10", 10)); 1309 EXPECT_EQ(5U, APInt::getBitsNeeded("+19", 10)); 1310 EXPECT_EQ(5U, APInt::getBitsNeeded("+20", 10)); 1311 1312 EXPECT_EQ(2U, APInt::getBitsNeeded( "-0", 10)); 1313 EXPECT_EQ(5U, APInt::getBitsNeeded( "-9", 10)); 1314 EXPECT_EQ(5U, APInt::getBitsNeeded("-10", 10)); 1315 EXPECT_EQ(6U, APInt::getBitsNeeded("-19", 10)); 1316 EXPECT_EQ(6U, APInt::getBitsNeeded("-20", 10)); 1317 1318 EXPECT_EQ(1U, APInt::getBitsNeeded("-1", 10)); 1319 EXPECT_EQ(2U, APInt::getBitsNeeded("-2", 10)); 1320 EXPECT_EQ(3U, APInt::getBitsNeeded("-4", 10)); 1321 EXPECT_EQ(4U, APInt::getBitsNeeded("-8", 10)); 1322 EXPECT_EQ(5U, APInt::getBitsNeeded("-16", 10)); 1323 EXPECT_EQ(6U, APInt::getBitsNeeded("-23", 10)); 1324 EXPECT_EQ(6U, APInt::getBitsNeeded("-32", 10)); 1325 EXPECT_EQ(7U, APInt::getBitsNeeded("-64", 10)); 1326 EXPECT_EQ(8U, APInt::getBitsNeeded("-127", 10)); 1327 EXPECT_EQ(8U, APInt::getBitsNeeded("-128", 10)); 1328 EXPECT_EQ(9U, APInt::getBitsNeeded("-255", 10)); 1329 EXPECT_EQ(9U, APInt::getBitsNeeded("-256", 10)); 1330 EXPECT_EQ(10U, APInt::getBitsNeeded("-512", 10)); 1331 EXPECT_EQ(11U, APInt::getBitsNeeded("-1024", 10)); 1332 EXPECT_EQ(12U, APInt::getBitsNeeded("-1025", 10)); 1333 } 1334 1335 TEST(APIntTest, StringBitsNeeded16) { 1336 EXPECT_EQ(4U, APInt::getBitsNeeded( "0", 16)); 1337 EXPECT_EQ(4U, APInt::getBitsNeeded( "F", 16)); 1338 EXPECT_EQ(8U, APInt::getBitsNeeded("10", 16)); 1339 EXPECT_EQ(8U, APInt::getBitsNeeded("1F", 16)); 1340 EXPECT_EQ(8U, APInt::getBitsNeeded("20", 16)); 1341 1342 EXPECT_EQ(4U, APInt::getBitsNeeded( "+0", 16)); 1343 EXPECT_EQ(4U, APInt::getBitsNeeded( "+F", 16)); 1344 EXPECT_EQ(8U, APInt::getBitsNeeded("+10", 16)); 1345 EXPECT_EQ(8U, APInt::getBitsNeeded("+1F", 16)); 1346 EXPECT_EQ(8U, APInt::getBitsNeeded("+20", 16)); 1347 1348 EXPECT_EQ(5U, APInt::getBitsNeeded( "-0", 16)); 1349 EXPECT_EQ(5U, APInt::getBitsNeeded( "-F", 16)); 1350 EXPECT_EQ(9U, APInt::getBitsNeeded("-10", 16)); 1351 EXPECT_EQ(9U, APInt::getBitsNeeded("-1F", 16)); 1352 EXPECT_EQ(9U, APInt::getBitsNeeded("-20", 16)); 1353 } 1354 1355 TEST(APIntTest, toString) { 1356 SmallString<16> S; 1357 bool isSigned; 1358 1359 APInt(8, 0).toString(S, 2, true, true); 1360 EXPECT_EQ(std::string(S), "0b0"); 1361 S.clear(); 1362 APInt(8, 0).toString(S, 8, true, true); 1363 EXPECT_EQ(std::string(S), "00"); 1364 S.clear(); 1365 APInt(8, 0).toString(S, 10, true, true); 1366 EXPECT_EQ(std::string(S), "0"); 1367 S.clear(); 1368 APInt(8, 0).toString(S, 16, true, true); 1369 EXPECT_EQ(std::string(S), "0x0"); 1370 S.clear(); 1371 APInt(8, 0).toString(S, 36, true, false); 1372 EXPECT_EQ(std::string(S), "0"); 1373 S.clear(); 1374 1375 isSigned = false; 1376 APInt(8, 255, isSigned).toString(S, 2, isSigned, true); 1377 EXPECT_EQ(std::string(S), "0b11111111"); 1378 S.clear(); 1379 APInt(8, 255, isSigned).toString(S, 8, isSigned, true); 1380 EXPECT_EQ(std::string(S), "0377"); 1381 S.clear(); 1382 APInt(8, 255, isSigned).toString(S, 10, isSigned, true); 1383 EXPECT_EQ(std::string(S), "255"); 1384 S.clear(); 1385 APInt(8, 255, isSigned).toString(S, 16, isSigned, true); 1386 EXPECT_EQ(std::string(S), "0xFF"); 1387 S.clear(); 1388 APInt(8, 255, isSigned).toString(S, 36, isSigned, false); 1389 EXPECT_EQ(std::string(S), "73"); 1390 S.clear(); 1391 1392 isSigned = true; 1393 APInt(8, 255, isSigned).toString(S, 2, isSigned, true); 1394 EXPECT_EQ(std::string(S), "-0b1"); 1395 S.clear(); 1396 APInt(8, 255, isSigned).toString(S, 8, isSigned, true); 1397 EXPECT_EQ(std::string(S), "-01"); 1398 S.clear(); 1399 APInt(8, 255, isSigned).toString(S, 10, isSigned, true); 1400 EXPECT_EQ(std::string(S), "-1"); 1401 S.clear(); 1402 APInt(8, 255, isSigned).toString(S, 16, isSigned, true); 1403 EXPECT_EQ(std::string(S), "-0x1"); 1404 S.clear(); 1405 APInt(8, 255, isSigned).toString(S, 36, isSigned, false); 1406 EXPECT_EQ(std::string(S), "-1"); 1407 S.clear(); 1408 } 1409 1410 TEST(APIntTest, Log2) { 1411 EXPECT_EQ(APInt(15, 7).logBase2(), 2U); 1412 EXPECT_EQ(APInt(15, 7).ceilLogBase2(), 3U); 1413 EXPECT_EQ(APInt(15, 7).exactLogBase2(), -1); 1414 EXPECT_EQ(APInt(15, 8).logBase2(), 3U); 1415 EXPECT_EQ(APInt(15, 8).ceilLogBase2(), 3U); 1416 EXPECT_EQ(APInt(15, 8).exactLogBase2(), 3); 1417 EXPECT_EQ(APInt(15, 9).logBase2(), 3U); 1418 EXPECT_EQ(APInt(15, 9).ceilLogBase2(), 4U); 1419 EXPECT_EQ(APInt(15, 9).exactLogBase2(), -1); 1420 } 1421 1422 #ifdef GTEST_HAS_DEATH_TEST 1423 #ifndef NDEBUG 1424 TEST(APIntTest, StringDeath) { 1425 EXPECT_DEATH((void)APInt(32, "", 0), "Invalid string length"); 1426 EXPECT_DEATH((void)APInt(32, "0", 0), "Radix should be 2, 8, 10, 16, or 36!"); 1427 EXPECT_DEATH((void)APInt(32, "", 10), "Invalid string length"); 1428 EXPECT_DEATH((void)APInt(32, "-", 10), "String is only a sign, needs a value."); 1429 EXPECT_DEATH((void)APInt(1, "1234", 10), "Insufficient bit width"); 1430 EXPECT_DEATH((void)APInt(32, "\0", 10), "Invalid string length"); 1431 EXPECT_DEATH((void)APInt(32, StringRef("1\02", 3), 10), "Invalid character in digit string"); 1432 EXPECT_DEATH((void)APInt(32, "1L", 10), "Invalid character in digit string"); 1433 } 1434 #endif 1435 #endif 1436 1437 TEST(APIntTest, mul_clear) { 1438 APInt ValA(65, -1ULL); 1439 APInt ValB(65, 4); 1440 APInt ValC(65, 0); 1441 ValC = ValA * ValB; 1442 ValA *= ValB; 1443 SmallString<16> StrA, StrC; 1444 ValA.toString(StrA, 10, false); 1445 ValC.toString(StrC, 10, false); 1446 EXPECT_EQ(std::string(StrA), std::string(StrC)); 1447 } 1448 1449 TEST(APIntTest, Rotate) { 1450 EXPECT_EQ(APInt(8, 1), APInt(8, 1).rotl(0)); 1451 EXPECT_EQ(APInt(8, 2), APInt(8, 1).rotl(1)); 1452 EXPECT_EQ(APInt(8, 4), APInt(8, 1).rotl(2)); 1453 EXPECT_EQ(APInt(8, 16), APInt(8, 1).rotl(4)); 1454 EXPECT_EQ(APInt(8, 1), APInt(8, 1).rotl(8)); 1455 1456 EXPECT_EQ(APInt(8, 16), APInt(8, 16).rotl(0)); 1457 EXPECT_EQ(APInt(8, 32), APInt(8, 16).rotl(1)); 1458 EXPECT_EQ(APInt(8, 64), APInt(8, 16).rotl(2)); 1459 EXPECT_EQ(APInt(8, 1), APInt(8, 16).rotl(4)); 1460 EXPECT_EQ(APInt(8, 16), APInt(8, 16).rotl(8)); 1461 1462 EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(33)); 1463 EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(APInt(32, 33))); 1464 1465 EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(33)); 1466 EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(APInt(32, 33))); 1467 EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(APInt(33, 33))); 1468 EXPECT_EQ(APInt(32, (1 << 8)), APInt(32, 1).rotl(APInt(32, 40))); 1469 EXPECT_EQ(APInt(32, (1 << 30)), APInt(32, 1).rotl(APInt(31, 30))); 1470 EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotl(APInt(31, 31))); 1471 1472 EXPECT_EQ(APInt(32, 1), APInt(32, 1).rotl(APInt(1, 0))); 1473 EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(APInt(1, 1))); 1474 1475 EXPECT_EQ(APInt(32, 16), APInt(32, 1).rotl(APInt(3, 4))); 1476 1477 EXPECT_EQ(APInt(32, 1), APInt(32, 1).rotl(APInt(64, 64))); 1478 EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(APInt(64, 65))); 1479 1480 EXPECT_EQ(APInt(7, 24), APInt(7, 3).rotl(APInt(7, 3))); 1481 EXPECT_EQ(APInt(7, 24), APInt(7, 3).rotl(APInt(7, 10))); 1482 EXPECT_EQ(APInt(7, 24), APInt(7, 3).rotl(APInt(5, 10))); 1483 EXPECT_EQ(APInt(7, 6), APInt(7, 3).rotl(APInt(12, 120))); 1484 1485 EXPECT_EQ(APInt(8, 16), APInt(8, 16).rotr(0)); 1486 EXPECT_EQ(APInt(8, 8), APInt(8, 16).rotr(1)); 1487 EXPECT_EQ(APInt(8, 4), APInt(8, 16).rotr(2)); 1488 EXPECT_EQ(APInt(8, 1), APInt(8, 16).rotr(4)); 1489 EXPECT_EQ(APInt(8, 16), APInt(8, 16).rotr(8)); 1490 1491 EXPECT_EQ(APInt(8, 1), APInt(8, 1).rotr(0)); 1492 EXPECT_EQ(APInt(8, 128), APInt(8, 1).rotr(1)); 1493 EXPECT_EQ(APInt(8, 64), APInt(8, 1).rotr(2)); 1494 EXPECT_EQ(APInt(8, 16), APInt(8, 1).rotr(4)); 1495 EXPECT_EQ(APInt(8, 1), APInt(8, 1).rotr(8)); 1496 1497 EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(33)); 1498 EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(APInt(32, 33))); 1499 1500 EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(33)); 1501 EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(APInt(32, 33))); 1502 EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(APInt(33, 33))); 1503 EXPECT_EQ(APInt(32, (1 << 24)), APInt(32, 1).rotr(APInt(32, 40))); 1504 1505 EXPECT_EQ(APInt(32, (1 << 2)), APInt(32, 1).rotr(APInt(31, 30))); 1506 EXPECT_EQ(APInt(32, (1 << 1)), APInt(32, 1).rotr(APInt(31, 31))); 1507 1508 EXPECT_EQ(APInt(32, 1), APInt(32, 1).rotr(APInt(1, 0))); 1509 EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(APInt(1, 1))); 1510 1511 EXPECT_EQ(APInt(32, (1 << 28)), APInt(32, 1).rotr(APInt(3, 4))); 1512 1513 EXPECT_EQ(APInt(32, 1), APInt(32, 1).rotr(APInt(64, 64))); 1514 EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(APInt(64, 65))); 1515 1516 EXPECT_EQ(APInt(7, 48), APInt(7, 3).rotr(APInt(7, 3))); 1517 EXPECT_EQ(APInt(7, 48), APInt(7, 3).rotr(APInt(7, 10))); 1518 EXPECT_EQ(APInt(7, 48), APInt(7, 3).rotr(APInt(5, 10))); 1519 EXPECT_EQ(APInt(7, 65), APInt(7, 3).rotr(APInt(12, 120))); 1520 1521 APInt Big(256, "00004000800000000000000000003fff8000000000000003", 16); 1522 APInt Rot(256, "3fff80000000000000030000000000000000000040008000", 16); 1523 EXPECT_EQ(Rot, Big.rotr(144)); 1524 1525 EXPECT_EQ(APInt(32, 8), APInt(32, 1).rotl(Big)); 1526 EXPECT_EQ(APInt(32, (1 << 29)), APInt(32, 1).rotr(Big)); 1527 } 1528 1529 TEST(APIntTest, Splat) { 1530 APInt ValA(8, 0x01); 1531 EXPECT_EQ(ValA, APInt::getSplat(8, ValA)); 1532 EXPECT_EQ(APInt(64, 0x0101010101010101ULL), APInt::getSplat(64, ValA)); 1533 1534 APInt ValB(3, 5); 1535 EXPECT_EQ(APInt(4, 0xD), APInt::getSplat(4, ValB)); 1536 EXPECT_EQ(APInt(15, 0xDB6D), APInt::getSplat(15, ValB)); 1537 } 1538 1539 TEST(APIntTest, tcDecrement) { 1540 // Test single word decrement. 1541 1542 // No out borrow. 1543 { 1544 APInt::WordType singleWord = ~APInt::WordType(0) << (APInt::APINT_BITS_PER_WORD - 1); 1545 APInt::WordType carry = APInt::tcDecrement(&singleWord, 1); 1546 EXPECT_EQ(carry, APInt::WordType(0)); 1547 EXPECT_EQ(singleWord, ~APInt::WordType(0) >> 1); 1548 } 1549 1550 // With out borrow. 1551 { 1552 APInt::WordType singleWord = 0; 1553 APInt::WordType carry = APInt::tcDecrement(&singleWord, 1); 1554 EXPECT_EQ(carry, APInt::WordType(1)); 1555 EXPECT_EQ(singleWord, ~APInt::WordType(0)); 1556 } 1557 1558 // Test multiword decrement. 1559 1560 // No across word borrow, no out borrow. 1561 { 1562 APInt::WordType test[4] = {0x1, 0x1, 0x1, 0x1}; 1563 APInt::WordType expected[4] = {0x0, 0x1, 0x1, 0x1}; 1564 APInt::tcDecrement(test, 4); 1565 EXPECT_EQ(APInt::tcCompare(test, expected, 4), 0); 1566 } 1567 1568 // 1 across word borrow, no out borrow. 1569 { 1570 APInt::WordType test[4] = {0x0, 0xF, 0x1, 0x1}; 1571 APInt::WordType expected[4] = {~APInt::WordType(0), 0xE, 0x1, 0x1}; 1572 APInt::WordType carry = APInt::tcDecrement(test, 4); 1573 EXPECT_EQ(carry, APInt::WordType(0)); 1574 EXPECT_EQ(APInt::tcCompare(test, expected, 4), 0); 1575 } 1576 1577 // 2 across word borrow, no out borrow. 1578 { 1579 APInt::WordType test[4] = {0x0, 0x0, 0xC, 0x1}; 1580 APInt::WordType expected[4] = {~APInt::WordType(0), ~APInt::WordType(0), 0xB, 0x1}; 1581 APInt::WordType carry = APInt::tcDecrement(test, 4); 1582 EXPECT_EQ(carry, APInt::WordType(0)); 1583 EXPECT_EQ(APInt::tcCompare(test, expected, 4), 0); 1584 } 1585 1586 // 3 across word borrow, no out borrow. 1587 { 1588 APInt::WordType test[4] = {0x0, 0x0, 0x0, 0x1}; 1589 APInt::WordType expected[4] = {~APInt::WordType(0), ~APInt::WordType(0), ~APInt::WordType(0), 0x0}; 1590 APInt::WordType carry = APInt::tcDecrement(test, 4); 1591 EXPECT_EQ(carry, APInt::WordType(0)); 1592 EXPECT_EQ(APInt::tcCompare(test, expected, 4), 0); 1593 } 1594 1595 // 3 across word borrow, with out borrow. 1596 { 1597 APInt::WordType test[4] = {0x0, 0x0, 0x0, 0x0}; 1598 APInt::WordType expected[4] = {~APInt::WordType(0), ~APInt::WordType(0), ~APInt::WordType(0), ~APInt::WordType(0)}; 1599 APInt::WordType carry = APInt::tcDecrement(test, 4); 1600 EXPECT_EQ(carry, APInt::WordType(1)); 1601 EXPECT_EQ(APInt::tcCompare(test, expected, 4), 0); 1602 } 1603 } 1604 1605 TEST(APIntTest, arrayAccess) { 1606 // Single word check. 1607 uint64_t E1 = 0x2CA7F46BF6569915ULL; 1608 APInt A1(64, E1); 1609 for (unsigned i = 0, e = 64; i < e; ++i) { 1610 EXPECT_EQ(bool(E1 & (1ULL << i)), 1611 A1[i]); 1612 } 1613 1614 // Multiword check. 1615 APInt::WordType E2[4] = { 1616 0xEB6EB136591CBA21ULL, 1617 0x7B9358BD6A33F10AULL, 1618 0x7E7FFA5EADD8846ULL, 1619 0x305F341CA00B613DULL 1620 }; 1621 APInt A2(APInt::APINT_BITS_PER_WORD*4, E2); 1622 for (unsigned i = 0; i < 4; ++i) { 1623 for (unsigned j = 0; j < APInt::APINT_BITS_PER_WORD; ++j) { 1624 EXPECT_EQ(bool(E2[i] & (1ULL << j)), 1625 A2[i*APInt::APINT_BITS_PER_WORD + j]); 1626 } 1627 } 1628 } 1629 1630 TEST(APIntTest, LargeAPIntConstruction) { 1631 // Check that we can properly construct very large APInt. It is very 1632 // unlikely that people will ever do this, but it is a legal input, 1633 // so we should not crash on it. 1634 APInt A9(UINT32_MAX, 0); 1635 EXPECT_FALSE(A9.getBoolValue()); 1636 } 1637 1638 TEST(APIntTest, nearestLogBase2) { 1639 // Single word check. 1640 1641 // Test round up. 1642 uint64_t I1 = 0x1800001; 1643 APInt A1(64, I1); 1644 EXPECT_EQ(A1.nearestLogBase2(), A1.ceilLogBase2()); 1645 1646 // Test round down. 1647 uint64_t I2 = 0x1000011; 1648 APInt A2(64, I2); 1649 EXPECT_EQ(A2.nearestLogBase2(), A2.logBase2()); 1650 1651 // Test ties round up. 1652 uint64_t I3 = 0x1800000; 1653 APInt A3(64, I3); 1654 EXPECT_EQ(A3.nearestLogBase2(), A3.ceilLogBase2()); 1655 1656 // Multiple word check. 1657 1658 // Test round up. 1659 APInt::WordType I4[4] = {0x0, 0xF, 0x18, 0x0}; 1660 APInt A4(APInt::APINT_BITS_PER_WORD*4, I4); 1661 EXPECT_EQ(A4.nearestLogBase2(), A4.ceilLogBase2()); 1662 1663 // Test round down. 1664 APInt::WordType I5[4] = {0x0, 0xF, 0x10, 0x0}; 1665 APInt A5(APInt::APINT_BITS_PER_WORD*4, I5); 1666 EXPECT_EQ(A5.nearestLogBase2(), A5.logBase2()); 1667 1668 // Test ties round up. 1669 uint64_t I6[4] = {0x0, 0x0, 0x0, 0x18}; 1670 APInt A6(APInt::APINT_BITS_PER_WORD*4, I6); 1671 EXPECT_EQ(A6.nearestLogBase2(), A6.ceilLogBase2()); 1672 1673 // Test BitWidth == 1 special cases. 1674 APInt A7(1, 1); 1675 EXPECT_EQ(A7.nearestLogBase2(), 0ULL); 1676 APInt A8(1, 0); 1677 EXPECT_EQ(A8.nearestLogBase2(), UINT32_MAX); 1678 1679 // Test the zero case when we have a bit width large enough such 1680 // that the bit width is larger than UINT32_MAX-1. 1681 APInt A9(UINT32_MAX, 0); 1682 EXPECT_EQ(A9.nearestLogBase2(), UINT32_MAX); 1683 } 1684 1685 TEST(APIntTest, IsSplat) { 1686 APInt A(32, 0x01010101); 1687 EXPECT_FALSE(A.isSplat(1)); 1688 EXPECT_FALSE(A.isSplat(2)); 1689 EXPECT_FALSE(A.isSplat(4)); 1690 EXPECT_TRUE(A.isSplat(8)); 1691 EXPECT_TRUE(A.isSplat(16)); 1692 EXPECT_TRUE(A.isSplat(32)); 1693 1694 APInt B(24, 0xAAAAAA); 1695 EXPECT_FALSE(B.isSplat(1)); 1696 EXPECT_TRUE(B.isSplat(2)); 1697 EXPECT_TRUE(B.isSplat(4)); 1698 EXPECT_TRUE(B.isSplat(8)); 1699 EXPECT_TRUE(B.isSplat(24)); 1700 1701 APInt C(24, 0xABAAAB); 1702 EXPECT_FALSE(C.isSplat(1)); 1703 EXPECT_FALSE(C.isSplat(2)); 1704 EXPECT_FALSE(C.isSplat(4)); 1705 EXPECT_FALSE(C.isSplat(8)); 1706 EXPECT_TRUE(C.isSplat(24)); 1707 1708 APInt D(32, 0xABBAABBA); 1709 EXPECT_FALSE(D.isSplat(1)); 1710 EXPECT_FALSE(D.isSplat(2)); 1711 EXPECT_FALSE(D.isSplat(4)); 1712 EXPECT_FALSE(D.isSplat(8)); 1713 EXPECT_TRUE(D.isSplat(16)); 1714 EXPECT_TRUE(D.isSplat(32)); 1715 1716 APInt E(32, 0); 1717 EXPECT_TRUE(E.isSplat(1)); 1718 EXPECT_TRUE(E.isSplat(2)); 1719 EXPECT_TRUE(E.isSplat(4)); 1720 EXPECT_TRUE(E.isSplat(8)); 1721 EXPECT_TRUE(E.isSplat(16)); 1722 EXPECT_TRUE(E.isSplat(32)); 1723 } 1724 1725 TEST(APIntTest, isMask) { 1726 EXPECT_FALSE(APInt(32, 0x01010101).isMask()); 1727 EXPECT_FALSE(APInt(32, 0xf0000000).isMask()); 1728 EXPECT_FALSE(APInt(32, 0xffff0000).isMask()); 1729 EXPECT_FALSE(APInt(32, 0xff << 1).isMask()); 1730 1731 for (int N : { 1, 2, 3, 4, 7, 8, 16, 32, 64, 127, 128, 129, 256 }) { 1732 EXPECT_FALSE(APInt(N, 0).isMask()); 1733 1734 APInt One(N, 1); 1735 for (int I = 1; I <= N; ++I) { 1736 APInt MaskVal = One.shl(I) - 1; 1737 EXPECT_TRUE(MaskVal.isMask()); 1738 EXPECT_TRUE(MaskVal.isMask(I)); 1739 } 1740 } 1741 } 1742 1743 TEST(APIntTest, isShiftedMask) { 1744 EXPECT_FALSE(APInt(32, 0x01010101).isShiftedMask()); 1745 EXPECT_TRUE(APInt(32, 0xf0000000).isShiftedMask()); 1746 EXPECT_TRUE(APInt(32, 0xffff0000).isShiftedMask()); 1747 EXPECT_TRUE(APInt(32, 0xff << 1).isShiftedMask()); 1748 1749 unsigned MaskIdx, MaskLen; 1750 EXPECT_FALSE(APInt(32, 0x01010101).isShiftedMask(MaskIdx, MaskLen)); 1751 EXPECT_TRUE(APInt(32, 0xf0000000).isShiftedMask(MaskIdx, MaskLen)); 1752 EXPECT_EQ(28, MaskIdx); 1753 EXPECT_EQ(4, MaskLen); 1754 EXPECT_TRUE(APInt(32, 0xffff0000).isShiftedMask(MaskIdx, MaskLen)); 1755 EXPECT_EQ(16, MaskIdx); 1756 EXPECT_EQ(16, MaskLen); 1757 EXPECT_TRUE(APInt(32, 0xff << 1).isShiftedMask(MaskIdx, MaskLen)); 1758 EXPECT_EQ(1, MaskIdx); 1759 EXPECT_EQ(8, MaskLen); 1760 1761 for (int N : { 1, 2, 3, 4, 7, 8, 16, 32, 64, 127, 128, 129, 256 }) { 1762 EXPECT_FALSE(APInt(N, 0).isShiftedMask()); 1763 EXPECT_FALSE(APInt(N, 0).isShiftedMask(MaskIdx, MaskLen)); 1764 1765 APInt One(N, 1); 1766 for (int I = 1; I < N; ++I) { 1767 APInt MaskVal = One.shl(I) - 1; 1768 EXPECT_TRUE(MaskVal.isShiftedMask()); 1769 EXPECT_TRUE(MaskVal.isShiftedMask(MaskIdx, MaskLen)); 1770 EXPECT_EQ(0, MaskIdx); 1771 EXPECT_EQ(I, MaskLen); 1772 } 1773 for (int I = 1; I < N - 1; ++I) { 1774 APInt MaskVal = One.shl(I); 1775 EXPECT_TRUE(MaskVal.isShiftedMask()); 1776 EXPECT_TRUE(MaskVal.isShiftedMask(MaskIdx, MaskLen)); 1777 EXPECT_EQ(I, MaskIdx); 1778 EXPECT_EQ(1, MaskLen); 1779 } 1780 for (int I = 1; I < N; ++I) { 1781 APInt MaskVal = APInt::getHighBitsSet(N, I); 1782 EXPECT_TRUE(MaskVal.isShiftedMask()); 1783 EXPECT_TRUE(MaskVal.isShiftedMask(MaskIdx, MaskLen)); 1784 EXPECT_EQ(N - I, MaskIdx); 1785 EXPECT_EQ(I, MaskLen); 1786 } 1787 } 1788 } 1789 1790 TEST(APIntTest, isPowerOf2) { 1791 EXPECT_FALSE(APInt(5, 0x00).isPowerOf2()); 1792 EXPECT_FALSE(APInt(32, 0x11).isPowerOf2()); 1793 EXPECT_TRUE(APInt(17, 0x01).isPowerOf2()); 1794 EXPECT_TRUE(APInt(32, (unsigned)(0xffu << 31)).isPowerOf2()); 1795 1796 for (int N : {1, 2, 3, 4, 7, 8, 16, 32, 64, 127, 128, 129, 256}) { 1797 EXPECT_FALSE(APInt(N, 0).isPowerOf2()); 1798 EXPECT_TRUE(APInt::getSignedMinValue(N).isPowerOf2()); 1799 1800 APInt One(N, 1); 1801 for (int I = 1; I < N - 1; ++I) { 1802 EXPECT_TRUE(APInt::getOneBitSet(N, I).isPowerOf2()); 1803 1804 APInt MaskVal = One.shl(I); 1805 EXPECT_TRUE(MaskVal.isPowerOf2()); 1806 } 1807 } 1808 } 1809 1810 TEST(APIntTest, isNegatedPowerOf2) { 1811 EXPECT_FALSE(APInt(5, 0x00).isNegatedPowerOf2()); 1812 EXPECT_TRUE(APInt(15, 0x7ffe).isNegatedPowerOf2()); 1813 EXPECT_TRUE(APInt(16, 0xfffc).isNegatedPowerOf2()); 1814 EXPECT_TRUE(APInt(32, 0xffffffff).isNegatedPowerOf2()); 1815 1816 for (int N : {1, 2, 3, 4, 7, 8, 16, 32, 64, 127, 128, 129, 256}) { 1817 EXPECT_FALSE(APInt(N, 0).isNegatedPowerOf2()); 1818 EXPECT_TRUE(APInt::getAllOnes(N).isNegatedPowerOf2()); 1819 EXPECT_TRUE(APInt::getSignedMinValue(N).isNegatedPowerOf2()); 1820 EXPECT_TRUE((-APInt::getSignedMinValue(N)).isNegatedPowerOf2()); 1821 1822 APInt One(N, 1); 1823 for (int I = 1; I < N - 1; ++I) { 1824 EXPECT_FALSE(APInt::getOneBitSet(N, I).isNegatedPowerOf2()); 1825 EXPECT_TRUE((-APInt::getOneBitSet(N, I)).isNegatedPowerOf2()); 1826 1827 APInt MaskVal = One.shl(I); 1828 EXPECT_TRUE((-MaskVal).isNegatedPowerOf2()); 1829 1830 APInt ShiftMaskVal = One.getHighBitsSet(N, I); 1831 EXPECT_TRUE(ShiftMaskVal.isNegatedPowerOf2()); 1832 } 1833 } 1834 } 1835 1836 // Test that self-move works with EXPENSIVE_CHECKS. It calls std::shuffle which 1837 // does self-move on some platforms. 1838 #ifdef EXPENSIVE_CHECKS 1839 #if defined(__clang__) 1840 // Disable the pragma warning from versions of Clang without -Wself-move 1841 #pragma clang diagnostic push 1842 #pragma clang diagnostic ignored "-Wunknown-pragmas" 1843 // Disable the warning that triggers on exactly what is being tested. 1844 #pragma clang diagnostic push 1845 #pragma clang diagnostic ignored "-Wself-move" 1846 #endif 1847 TEST(APIntTest, SelfMoveAssignment) { 1848 APInt X(32, 0xdeadbeef); 1849 X = std::move(X); 1850 EXPECT_EQ(32u, X.getBitWidth()); 1851 EXPECT_EQ(0xdeadbeefULL, X.getLimitedValue()); 1852 1853 uint64_t Bits[] = {0xdeadbeefdeadbeefULL, 0xdeadbeefdeadbeefULL}; 1854 APInt Y(128, Bits); 1855 Y = std::move(Y); 1856 EXPECT_EQ(128u, Y.getBitWidth()); 1857 EXPECT_EQ(~0ULL, Y.getLimitedValue()); 1858 const uint64_t *Raw = Y.getRawData(); 1859 EXPECT_EQ(2u, Y.getNumWords()); 1860 EXPECT_EQ(0xdeadbeefdeadbeefULL, Raw[0]); 1861 EXPECT_EQ(0xdeadbeefdeadbeefULL, Raw[1]); 1862 } 1863 #if defined(__clang__) 1864 #pragma clang diagnostic pop 1865 #pragma clang diagnostic pop 1866 #endif 1867 #endif // EXPENSIVE_CHECKS 1868 1869 TEST(APIntTest, byteSwap) { 1870 EXPECT_EQ(0x00000000, APInt(16, 0x0000).byteSwap()); 1871 EXPECT_EQ(0x0000010f, APInt(16, 0x0f01).byteSwap()); 1872 EXPECT_EQ(0x00ff8000, APInt(24, 0x0080ff).byteSwap()); 1873 EXPECT_EQ(0x117700ff, APInt(32, 0xff007711).byteSwap()); 1874 EXPECT_EQ(0x228811aaffULL, APInt(40, 0xffaa118822ULL).byteSwap()); 1875 EXPECT_EQ(0x050403020100ULL, APInt(48, 0x000102030405ULL).byteSwap()); 1876 EXPECT_EQ(0xff050403020100ULL, APInt(56, 0x000102030405ffULL).byteSwap()); 1877 EXPECT_EQ(0xff050403020100aaULL, APInt(64, 0xaa000102030405ffULL).byteSwap()); 1878 1879 for (unsigned N : {16, 24, 32, 48, 56, 64, 72, 80, 96, 112, 128, 248, 256, 1880 1024, 1032, 1040}) { 1881 for (unsigned I = 0; I < N; I += 8) { 1882 APInt X = APInt::getBitsSet(N, I, I + 8); 1883 APInt Y = APInt::getBitsSet(N, N - I - 8, N - I); 1884 EXPECT_EQ(Y, X.byteSwap()); 1885 EXPECT_EQ(X, Y.byteSwap()); 1886 } 1887 } 1888 } 1889 1890 TEST(APIntTest, reverseBits) { 1891 EXPECT_EQ(1, APInt(1, 1).reverseBits()); 1892 EXPECT_EQ(0, APInt(1, 0).reverseBits()); 1893 1894 EXPECT_EQ(3, APInt(2, 3).reverseBits()); 1895 EXPECT_EQ(3, APInt(2, 3).reverseBits()); 1896 1897 EXPECT_EQ(0xb, APInt(4, 0xd).reverseBits()); 1898 EXPECT_EQ(0xd, APInt(4, 0xb).reverseBits()); 1899 EXPECT_EQ(0xf, APInt(4, 0xf).reverseBits()); 1900 1901 EXPECT_EQ(0x30, APInt(7, 0x6).reverseBits()); 1902 EXPECT_EQ(0x5a, APInt(7, 0x2d).reverseBits()); 1903 1904 EXPECT_EQ(0x0f, APInt(8, 0xf0).reverseBits()); 1905 EXPECT_EQ(0xf0, APInt(8, 0x0f).reverseBits()); 1906 1907 EXPECT_EQ(0x0f0f, APInt(16, 0xf0f0).reverseBits()); 1908 EXPECT_EQ(0xf0f0, APInt(16, 0x0f0f).reverseBits()); 1909 1910 EXPECT_EQ(0x0f0f0f0f, APInt(32, 0xf0f0f0f0).reverseBits()); 1911 EXPECT_EQ(0xf0f0f0f0, APInt(32, 0x0f0f0f0f).reverseBits()); 1912 1913 EXPECT_EQ(0x402880a0 >> 1, APInt(31, 0x05011402).reverseBits()); 1914 1915 EXPECT_EQ(0x0f0f0f0f, APInt(32, 0xf0f0f0f0).reverseBits()); 1916 EXPECT_EQ(0xf0f0f0f0, APInt(32, 0x0f0f0f0f).reverseBits()); 1917 1918 EXPECT_EQ(0x0f0f0f0f0f0f0f0f, APInt(64, 0xf0f0f0f0f0f0f0f0).reverseBits()); 1919 EXPECT_EQ(0xf0f0f0f0f0f0f0f0, APInt(64, 0x0f0f0f0f0f0f0f0f).reverseBits()); 1920 1921 for (unsigned N : { 1, 8, 16, 24, 31, 32, 33, 1922 63, 64, 65, 127, 128, 257, 1024 }) { 1923 for (unsigned I = 0; I < N; ++I) { 1924 APInt X = APInt::getOneBitSet(N, I); 1925 APInt Y = APInt::getOneBitSet(N, N - (I + 1)); 1926 EXPECT_EQ(Y, X.reverseBits()); 1927 EXPECT_EQ(X, Y.reverseBits()); 1928 } 1929 } 1930 } 1931 1932 TEST(APIntTest, insertBits) { 1933 APInt iSrc(31, 0x00123456); 1934 1935 // Direct copy. 1936 APInt i31(31, 0x76543210ull); 1937 i31.insertBits(iSrc, 0); 1938 EXPECT_EQ(static_cast<int64_t>(0x00123456ull), i31.getSExtValue()); 1939 1940 // Single word src/dst insertion. 1941 APInt i63(63, 0x01234567FFFFFFFFull); 1942 i63.insertBits(iSrc, 4); 1943 EXPECT_EQ(static_cast<int64_t>(0x012345600123456Full), i63.getSExtValue()); 1944 1945 // Zero width insert is a noop. 1946 i31.insertBits(APInt::getZeroWidth(), 1); 1947 EXPECT_EQ(static_cast<int64_t>(0x00123456ull), i31.getSExtValue()); 1948 1949 // Insert single word src into one word of dst. 1950 APInt i120(120, UINT64_MAX, true); 1951 i120.insertBits(iSrc, 8); 1952 EXPECT_EQ(static_cast<int64_t>(0xFFFFFF80123456FFull), i120.getSExtValue()); 1953 1954 // Insert single word src into two words of dst. 1955 APInt i127(127, UINT64_MAX, true); 1956 i127.insertBits(iSrc, 48); 1957 EXPECT_EQ(i127.extractBits(64, 0).getZExtValue(), 0x3456FFFFFFFFFFFFull); 1958 EXPECT_EQ(i127.extractBits(63, 64).getZExtValue(), 0x7FFFFFFFFFFF8012ull); 1959 1960 // Insert on word boundaries. 1961 APInt i128(128, 0); 1962 i128.insertBits(APInt(64, UINT64_MAX, true), 0); 1963 i128.insertBits(APInt(64, UINT64_MAX, true), 64); 1964 EXPECT_EQ(-1, i128.getSExtValue()); 1965 1966 APInt i256(256, UINT64_MAX, true); 1967 i256.insertBits(APInt(65, 0), 0); 1968 i256.insertBits(APInt(69, 0), 64); 1969 i256.insertBits(APInt(128, 0), 128); 1970 EXPECT_EQ(0u, i256.getSExtValue()); 1971 1972 APInt i257(257, 0); 1973 i257.insertBits(APInt(96, UINT64_MAX, true), 64); 1974 EXPECT_EQ(i257.extractBits(64, 0).getZExtValue(), 0x0000000000000000ull); 1975 EXPECT_EQ(i257.extractBits(64, 64).getZExtValue(), 0xFFFFFFFFFFFFFFFFull); 1976 EXPECT_EQ(i257.extractBits(64, 128).getZExtValue(), 0x00000000FFFFFFFFull); 1977 EXPECT_EQ(i257.extractBits(65, 192).getZExtValue(), 0x0000000000000000ull); 1978 1979 // General insertion. 1980 APInt i260(260, UINT64_MAX, true); 1981 i260.insertBits(APInt(129, 1ull << 48), 15); 1982 EXPECT_EQ(i260.extractBits(64, 0).getZExtValue(), 0x8000000000007FFFull); 1983 EXPECT_EQ(i260.extractBits(64, 64).getZExtValue(), 0x0000000000000000ull); 1984 EXPECT_EQ(i260.extractBits(64, 128).getZExtValue(), 0xFFFFFFFFFFFF0000ull); 1985 EXPECT_EQ(i260.extractBits(64, 192).getZExtValue(), 0xFFFFFFFFFFFFFFFFull); 1986 EXPECT_EQ(i260.extractBits(4, 256).getZExtValue(), 0x000000000000000Full); 1987 } 1988 1989 TEST(APIntTest, insertBitsUInt64) { 1990 // Tests cloned from insertBits but adapted to the numBits <= 64 constraint 1991 uint64_t iSrc = 0x00123456; 1992 1993 // Direct copy. 1994 APInt i31(31, 0x76543210ull); 1995 i31.insertBits(iSrc, 0, 31); 1996 EXPECT_EQ(static_cast<int64_t>(0x00123456ull), i31.getSExtValue()); 1997 1998 // Single word src/dst insertion. 1999 APInt i63(63, 0x01234567FFFFFFFFull); 2000 i63.insertBits(iSrc, 4, 31); 2001 EXPECT_EQ(static_cast<int64_t>(0x012345600123456Full), i63.getSExtValue()); 2002 2003 // Insert single word src into one word of dst. 2004 APInt i120(120, UINT64_MAX, true); 2005 i120.insertBits(iSrc, 8, 31); 2006 EXPECT_EQ(static_cast<int64_t>(0xFFFFFF80123456FFull), i120.getSExtValue()); 2007 2008 // Insert single word src into two words of dst. 2009 APInt i127(127, UINT64_MAX, true); 2010 i127.insertBits(iSrc, 48, 31); 2011 EXPECT_EQ(i127.extractBits(64, 0).getZExtValue(), 0x3456FFFFFFFFFFFFull); 2012 EXPECT_EQ(i127.extractBits(63, 64).getZExtValue(), 0x7FFFFFFFFFFF8012ull); 2013 2014 // Insert on word boundaries. 2015 APInt i128(128, 0); 2016 i128.insertBits(UINT64_MAX, 0, 64); 2017 i128.insertBits(UINT64_MAX, 64, 64); 2018 EXPECT_EQ(-1, i128.getSExtValue()); 2019 2020 APInt i256(256, UINT64_MAX, true); 2021 i256.insertBits(0, 0, 64); 2022 i256.insertBits(0, 64, 1); 2023 i256.insertBits(0, 64, 64); 2024 i256.insertBits(0, 128, 5); 2025 i256.insertBits(0, 128, 64); 2026 i256.insertBits(0, 192, 64); 2027 EXPECT_EQ(0u, i256.getSExtValue()); 2028 2029 APInt i257(257, 0); 2030 i257.insertBits(APInt(96, UINT64_MAX, true), 64); 2031 EXPECT_EQ(i257.extractBitsAsZExtValue(64, 0), 0x0000000000000000ull); 2032 EXPECT_EQ(i257.extractBitsAsZExtValue(64, 64), 0xFFFFFFFFFFFFFFFFull); 2033 EXPECT_EQ(i257.extractBitsAsZExtValue(64, 128), 0x00000000FFFFFFFFull); 2034 EXPECT_EQ(i257.extractBitsAsZExtValue(64, 192), 0x0000000000000000ull); 2035 EXPECT_EQ(i257.extractBitsAsZExtValue(1, 256), 0x0000000000000000ull); 2036 2037 // General insertion. 2038 APInt i260(260, UINT64_MAX, true); 2039 i260.insertBits(APInt(129, 1ull << 48), 15); 2040 EXPECT_EQ(i260.extractBitsAsZExtValue(64, 0), 0x8000000000007FFFull); 2041 EXPECT_EQ(i260.extractBitsAsZExtValue(64, 64), 0x0000000000000000ull); 2042 EXPECT_EQ(i260.extractBitsAsZExtValue(64, 128), 0xFFFFFFFFFFFF0000ull); 2043 EXPECT_EQ(i260.extractBitsAsZExtValue(64, 192), 0xFFFFFFFFFFFFFFFFull); 2044 EXPECT_EQ(i260.extractBitsAsZExtValue(4, 256), 0x000000000000000Full); 2045 } 2046 2047 TEST(APIntTest, extractBits) { 2048 APInt i32(32, 0x1234567); 2049 EXPECT_EQ(0x3456, i32.extractBits(16, 4)); 2050 2051 APInt i64(64, 0x01234567FFFFFFFFull); 2052 EXPECT_EQ(0xFFFFFFFF, i64.extractBits(32, 0)); 2053 EXPECT_EQ(0xFFFFFFFF, i64.trunc(32)); 2054 EXPECT_EQ(0x01234567, i64.extractBits(32, 32)); 2055 EXPECT_EQ(0x01234567, i64.lshr(32).trunc(32)); 2056 2057 APInt i257(257, 0xFFFFFFFFFF0000FFull, true); 2058 EXPECT_EQ(0xFFu, i257.extractBits(16, 0)); 2059 EXPECT_EQ(0xFFu, i257.lshr(0).trunc(16)); 2060 EXPECT_EQ((0xFFu >> 1), i257.extractBits(16, 1)); 2061 EXPECT_EQ((0xFFu >> 1), i257.lshr(1).trunc(16)); 2062 EXPECT_EQ(-1, i257.extractBits(32, 64).getSExtValue()); 2063 EXPECT_EQ(-1, i257.lshr(64).trunc(32).getSExtValue()); 2064 EXPECT_EQ(-1, i257.extractBits(128, 128).getSExtValue()); 2065 EXPECT_EQ(-1, i257.lshr(128).trunc(128).getSExtValue()); 2066 EXPECT_EQ(-1, i257.extractBits(66, 191).getSExtValue()); 2067 EXPECT_EQ(-1, i257.lshr(191).trunc(66).getSExtValue()); 2068 EXPECT_EQ(static_cast<int64_t>(0xFFFFFFFFFF80007Full), 2069 i257.extractBits(128, 1).getSExtValue()); 2070 EXPECT_EQ(static_cast<int64_t>(0xFFFFFFFFFF80007Full), 2071 i257.lshr(1).trunc(128).getSExtValue()); 2072 EXPECT_EQ(static_cast<int64_t>(0xFFFFFFFFFF80007Full), 2073 i257.extractBits(129, 1).getSExtValue()); 2074 EXPECT_EQ(static_cast<int64_t>(0xFFFFFFFFFF80007Full), 2075 i257.lshr(1).trunc(129).getSExtValue()); 2076 2077 EXPECT_EQ(APInt(48, 0), 2078 APInt(144, "281474976710655", 10).extractBits(48, 48)); 2079 EXPECT_EQ(APInt(48, 0), 2080 APInt(144, "281474976710655", 10).lshr(48).trunc(48)); 2081 EXPECT_EQ(APInt(48, 0x0000ffffffffffffull), 2082 APInt(144, "281474976710655", 10).extractBits(48, 0)); 2083 EXPECT_EQ(APInt(48, 0x0000ffffffffffffull), 2084 APInt(144, "281474976710655", 10).lshr(0).trunc(48)); 2085 EXPECT_EQ(APInt(48, 0x00007fffffffffffull), 2086 APInt(144, "281474976710655", 10).extractBits(48, 1)); 2087 EXPECT_EQ(APInt(48, 0x00007fffffffffffull), 2088 APInt(144, "281474976710655", 10).lshr(1).trunc(48)); 2089 } 2090 2091 TEST(APIntTest, extractBitsAsZExtValue) { 2092 // Tests based on extractBits 2093 APInt i32(32, 0x1234567); 2094 EXPECT_EQ(0x3456u, i32.extractBitsAsZExtValue(16, 4)); 2095 2096 APInt i257(257, 0xFFFFFFFFFF0000FFull, true); 2097 EXPECT_EQ(0xFFu, i257.extractBitsAsZExtValue(16, 0)); 2098 EXPECT_EQ((0xFFu >> 1), i257.extractBitsAsZExtValue(16, 1)); 2099 EXPECT_EQ(0xFFFFFFFFull, i257.extractBitsAsZExtValue(32, 64)); 2100 EXPECT_EQ(0xFFFFFFFFFFFFFFFFull, i257.extractBitsAsZExtValue(64, 128)); 2101 EXPECT_EQ(0xFFFFFFFFFFFFFFFFull, i257.extractBitsAsZExtValue(64, 192)); 2102 EXPECT_EQ(0xFFFFFFFFFFFFFFFFull, i257.extractBitsAsZExtValue(64, 191)); 2103 EXPECT_EQ(0x3u, i257.extractBitsAsZExtValue(2, 255)); 2104 EXPECT_EQ(0xFFFFFFFFFF80007Full, i257.extractBitsAsZExtValue(64, 1)); 2105 EXPECT_EQ(0xFFFFFFFFFFFFFFFFull, i257.extractBitsAsZExtValue(64, 65)); 2106 EXPECT_EQ(0xFFFFFFFFFF80007Full, i257.extractBitsAsZExtValue(64, 1)); 2107 EXPECT_EQ(0xFFFFFFFFFFFFFFFFull, i257.extractBitsAsZExtValue(64, 65)); 2108 EXPECT_EQ(0x1ull, i257.extractBitsAsZExtValue(1, 129)); 2109 2110 EXPECT_EQ(APInt(48, 0), 2111 APInt(144, "281474976710655", 10).extractBitsAsZExtValue(48, 48)); 2112 EXPECT_EQ(APInt(48, 0x0000ffffffffffffull), 2113 APInt(144, "281474976710655", 10).extractBitsAsZExtValue(48, 0)); 2114 EXPECT_EQ(APInt(48, 0x00007fffffffffffull), 2115 APInt(144, "281474976710655", 10).extractBitsAsZExtValue(48, 1)); 2116 } 2117 2118 TEST(APIntTest, getLowBitsSet) { 2119 APInt i128lo64 = APInt::getLowBitsSet(128, 64); 2120 EXPECT_EQ(0u, i128lo64.countLeadingOnes()); 2121 EXPECT_EQ(64u, i128lo64.countLeadingZeros()); 2122 EXPECT_EQ(64u, i128lo64.getActiveBits()); 2123 EXPECT_EQ(0u, i128lo64.countTrailingZeros()); 2124 EXPECT_EQ(64u, i128lo64.countTrailingOnes()); 2125 EXPECT_EQ(64u, i128lo64.countPopulation()); 2126 } 2127 2128 TEST(APIntTest, getBitsSet) { 2129 APInt i64hi1lo1 = APInt::getBitsSet(64, 1, 63); 2130 EXPECT_EQ(0u, i64hi1lo1.countLeadingOnes()); 2131 EXPECT_EQ(1u, i64hi1lo1.countLeadingZeros()); 2132 EXPECT_EQ(63u, i64hi1lo1.getActiveBits()); 2133 EXPECT_EQ(1u, i64hi1lo1.countTrailingZeros()); 2134 EXPECT_EQ(0u, i64hi1lo1.countTrailingOnes()); 2135 EXPECT_EQ(62u, i64hi1lo1.countPopulation()); 2136 2137 APInt i127hi1lo1 = APInt::getBitsSet(127, 1, 126); 2138 EXPECT_EQ(0u, i127hi1lo1.countLeadingOnes()); 2139 EXPECT_EQ(1u, i127hi1lo1.countLeadingZeros()); 2140 EXPECT_EQ(126u, i127hi1lo1.getActiveBits()); 2141 EXPECT_EQ(1u, i127hi1lo1.countTrailingZeros()); 2142 EXPECT_EQ(0u, i127hi1lo1.countTrailingOnes()); 2143 EXPECT_EQ(125u, i127hi1lo1.countPopulation()); 2144 } 2145 2146 TEST(APIntTest, getBitsSetWithWrap) { 2147 APInt i64hi1lo1 = APInt::getBitsSetWithWrap(64, 1, 63); 2148 EXPECT_EQ(0u, i64hi1lo1.countLeadingOnes()); 2149 EXPECT_EQ(1u, i64hi1lo1.countLeadingZeros()); 2150 EXPECT_EQ(63u, i64hi1lo1.getActiveBits()); 2151 EXPECT_EQ(1u, i64hi1lo1.countTrailingZeros()); 2152 EXPECT_EQ(0u, i64hi1lo1.countTrailingOnes()); 2153 EXPECT_EQ(62u, i64hi1lo1.countPopulation()); 2154 2155 APInt i127hi1lo1 = APInt::getBitsSetWithWrap(127, 1, 126); 2156 EXPECT_EQ(0u, i127hi1lo1.countLeadingOnes()); 2157 EXPECT_EQ(1u, i127hi1lo1.countLeadingZeros()); 2158 EXPECT_EQ(126u, i127hi1lo1.getActiveBits()); 2159 EXPECT_EQ(1u, i127hi1lo1.countTrailingZeros()); 2160 EXPECT_EQ(0u, i127hi1lo1.countTrailingOnes()); 2161 EXPECT_EQ(125u, i127hi1lo1.countPopulation()); 2162 2163 APInt i64hi1lo1wrap = APInt::getBitsSetWithWrap(64, 63, 1); 2164 EXPECT_EQ(1u, i64hi1lo1wrap.countLeadingOnes()); 2165 EXPECT_EQ(0u, i64hi1lo1wrap.countLeadingZeros()); 2166 EXPECT_EQ(64u, i64hi1lo1wrap.getActiveBits()); 2167 EXPECT_EQ(0u, i64hi1lo1wrap.countTrailingZeros()); 2168 EXPECT_EQ(1u, i64hi1lo1wrap.countTrailingOnes()); 2169 EXPECT_EQ(2u, i64hi1lo1wrap.countPopulation()); 2170 2171 APInt i127hi1lo1wrap = APInt::getBitsSetWithWrap(127, 126, 1); 2172 EXPECT_EQ(1u, i127hi1lo1wrap.countLeadingOnes()); 2173 EXPECT_EQ(0u, i127hi1lo1wrap.countLeadingZeros()); 2174 EXPECT_EQ(127u, i127hi1lo1wrap.getActiveBits()); 2175 EXPECT_EQ(0u, i127hi1lo1wrap.countTrailingZeros()); 2176 EXPECT_EQ(1u, i127hi1lo1wrap.countTrailingOnes()); 2177 EXPECT_EQ(2u, i127hi1lo1wrap.countPopulation()); 2178 2179 APInt i32hiequallowrap = APInt::getBitsSetWithWrap(32, 10, 10); 2180 EXPECT_EQ(32u, i32hiequallowrap.countLeadingOnes()); 2181 EXPECT_EQ(0u, i32hiequallowrap.countLeadingZeros()); 2182 EXPECT_EQ(32u, i32hiequallowrap.getActiveBits()); 2183 EXPECT_EQ(0u, i32hiequallowrap.countTrailingZeros()); 2184 EXPECT_EQ(32u, i32hiequallowrap.countTrailingOnes()); 2185 EXPECT_EQ(32u, i32hiequallowrap.countPopulation()); 2186 } 2187 2188 TEST(APIntTest, getHighBitsSet) { 2189 APInt i64hi32 = APInt::getHighBitsSet(64, 32); 2190 EXPECT_EQ(32u, i64hi32.countLeadingOnes()); 2191 EXPECT_EQ(0u, i64hi32.countLeadingZeros()); 2192 EXPECT_EQ(64u, i64hi32.getActiveBits()); 2193 EXPECT_EQ(32u, i64hi32.countTrailingZeros()); 2194 EXPECT_EQ(0u, i64hi32.countTrailingOnes()); 2195 EXPECT_EQ(32u, i64hi32.countPopulation()); 2196 } 2197 2198 TEST(APIntTest, getBitsSetFrom) { 2199 APInt i64hi31 = APInt::getBitsSetFrom(64, 33); 2200 EXPECT_EQ(31u, i64hi31.countLeadingOnes()); 2201 EXPECT_EQ(0u, i64hi31.countLeadingZeros()); 2202 EXPECT_EQ(64u, i64hi31.getActiveBits()); 2203 EXPECT_EQ(33u, i64hi31.countTrailingZeros()); 2204 EXPECT_EQ(0u, i64hi31.countTrailingOnes()); 2205 EXPECT_EQ(31u, i64hi31.countPopulation()); 2206 } 2207 2208 TEST(APIntTest, setLowBits) { 2209 APInt i64lo32(64, 0); 2210 i64lo32.setLowBits(32); 2211 EXPECT_EQ(0u, i64lo32.countLeadingOnes()); 2212 EXPECT_EQ(32u, i64lo32.countLeadingZeros()); 2213 EXPECT_EQ(32u, i64lo32.getActiveBits()); 2214 EXPECT_EQ(0u, i64lo32.countTrailingZeros()); 2215 EXPECT_EQ(32u, i64lo32.countTrailingOnes()); 2216 EXPECT_EQ(32u, i64lo32.countPopulation()); 2217 2218 APInt i128lo64(128, 0); 2219 i128lo64.setLowBits(64); 2220 EXPECT_EQ(0u, i128lo64.countLeadingOnes()); 2221 EXPECT_EQ(64u, i128lo64.countLeadingZeros()); 2222 EXPECT_EQ(64u, i128lo64.getActiveBits()); 2223 EXPECT_EQ(0u, i128lo64.countTrailingZeros()); 2224 EXPECT_EQ(64u, i128lo64.countTrailingOnes()); 2225 EXPECT_EQ(64u, i128lo64.countPopulation()); 2226 2227 APInt i128lo24(128, 0); 2228 i128lo24.setLowBits(24); 2229 EXPECT_EQ(0u, i128lo24.countLeadingOnes()); 2230 EXPECT_EQ(104u, i128lo24.countLeadingZeros()); 2231 EXPECT_EQ(24u, i128lo24.getActiveBits()); 2232 EXPECT_EQ(0u, i128lo24.countTrailingZeros()); 2233 EXPECT_EQ(24u, i128lo24.countTrailingOnes()); 2234 EXPECT_EQ(24u, i128lo24.countPopulation()); 2235 2236 APInt i128lo104(128, 0); 2237 i128lo104.setLowBits(104); 2238 EXPECT_EQ(0u, i128lo104.countLeadingOnes()); 2239 EXPECT_EQ(24u, i128lo104.countLeadingZeros()); 2240 EXPECT_EQ(104u, i128lo104.getActiveBits()); 2241 EXPECT_EQ(0u, i128lo104.countTrailingZeros()); 2242 EXPECT_EQ(104u, i128lo104.countTrailingOnes()); 2243 EXPECT_EQ(104u, i128lo104.countPopulation()); 2244 2245 APInt i128lo0(128, 0); 2246 i128lo0.setLowBits(0); 2247 EXPECT_EQ(0u, i128lo0.countLeadingOnes()); 2248 EXPECT_EQ(128u, i128lo0.countLeadingZeros()); 2249 EXPECT_EQ(0u, i128lo0.getActiveBits()); 2250 EXPECT_EQ(128u, i128lo0.countTrailingZeros()); 2251 EXPECT_EQ(0u, i128lo0.countTrailingOnes()); 2252 EXPECT_EQ(0u, i128lo0.countPopulation()); 2253 2254 APInt i80lo79(80, 0); 2255 i80lo79.setLowBits(79); 2256 EXPECT_EQ(0u, i80lo79.countLeadingOnes()); 2257 EXPECT_EQ(1u, i80lo79.countLeadingZeros()); 2258 EXPECT_EQ(79u, i80lo79.getActiveBits()); 2259 EXPECT_EQ(0u, i80lo79.countTrailingZeros()); 2260 EXPECT_EQ(79u, i80lo79.countTrailingOnes()); 2261 EXPECT_EQ(79u, i80lo79.countPopulation()); 2262 } 2263 2264 TEST(APIntTest, setHighBits) { 2265 APInt i64hi32(64, 0); 2266 i64hi32.setHighBits(32); 2267 EXPECT_EQ(32u, i64hi32.countLeadingOnes()); 2268 EXPECT_EQ(0u, i64hi32.countLeadingZeros()); 2269 EXPECT_EQ(64u, i64hi32.getActiveBits()); 2270 EXPECT_EQ(32u, i64hi32.countTrailingZeros()); 2271 EXPECT_EQ(0u, i64hi32.countTrailingOnes()); 2272 EXPECT_EQ(32u, i64hi32.countPopulation()); 2273 2274 APInt i128hi64(128, 0); 2275 i128hi64.setHighBits(64); 2276 EXPECT_EQ(64u, i128hi64.countLeadingOnes()); 2277 EXPECT_EQ(0u, i128hi64.countLeadingZeros()); 2278 EXPECT_EQ(128u, i128hi64.getActiveBits()); 2279 EXPECT_EQ(64u, i128hi64.countTrailingZeros()); 2280 EXPECT_EQ(0u, i128hi64.countTrailingOnes()); 2281 EXPECT_EQ(64u, i128hi64.countPopulation()); 2282 2283 APInt i128hi24(128, 0); 2284 i128hi24.setHighBits(24); 2285 EXPECT_EQ(24u, i128hi24.countLeadingOnes()); 2286 EXPECT_EQ(0u, i128hi24.countLeadingZeros()); 2287 EXPECT_EQ(128u, i128hi24.getActiveBits()); 2288 EXPECT_EQ(104u, i128hi24.countTrailingZeros()); 2289 EXPECT_EQ(0u, i128hi24.countTrailingOnes()); 2290 EXPECT_EQ(24u, i128hi24.countPopulation()); 2291 2292 APInt i128hi104(128, 0); 2293 i128hi104.setHighBits(104); 2294 EXPECT_EQ(104u, i128hi104.countLeadingOnes()); 2295 EXPECT_EQ(0u, i128hi104.countLeadingZeros()); 2296 EXPECT_EQ(128u, i128hi104.getActiveBits()); 2297 EXPECT_EQ(24u, i128hi104.countTrailingZeros()); 2298 EXPECT_EQ(0u, i128hi104.countTrailingOnes()); 2299 EXPECT_EQ(104u, i128hi104.countPopulation()); 2300 2301 APInt i128hi0(128, 0); 2302 i128hi0.setHighBits(0); 2303 EXPECT_EQ(0u, i128hi0.countLeadingOnes()); 2304 EXPECT_EQ(128u, i128hi0.countLeadingZeros()); 2305 EXPECT_EQ(0u, i128hi0.getActiveBits()); 2306 EXPECT_EQ(128u, i128hi0.countTrailingZeros()); 2307 EXPECT_EQ(0u, i128hi0.countTrailingOnes()); 2308 EXPECT_EQ(0u, i128hi0.countPopulation()); 2309 2310 APInt i80hi1(80, 0); 2311 i80hi1.setHighBits(1); 2312 EXPECT_EQ(1u, i80hi1.countLeadingOnes()); 2313 EXPECT_EQ(0u, i80hi1.countLeadingZeros()); 2314 EXPECT_EQ(80u, i80hi1.getActiveBits()); 2315 EXPECT_EQ(79u, i80hi1.countTrailingZeros()); 2316 EXPECT_EQ(0u, i80hi1.countTrailingOnes()); 2317 EXPECT_EQ(1u, i80hi1.countPopulation()); 2318 2319 APInt i32hi16(32, 0); 2320 i32hi16.setHighBits(16); 2321 EXPECT_EQ(16u, i32hi16.countLeadingOnes()); 2322 EXPECT_EQ(0u, i32hi16.countLeadingZeros()); 2323 EXPECT_EQ(32u, i32hi16.getActiveBits()); 2324 EXPECT_EQ(16u, i32hi16.countTrailingZeros()); 2325 EXPECT_EQ(0u, i32hi16.countTrailingOnes()); 2326 EXPECT_EQ(16u, i32hi16.countPopulation()); 2327 } 2328 2329 TEST(APIntTest, setBitsFrom) { 2330 APInt i64from63(64, 0); 2331 i64from63.setBitsFrom(63); 2332 EXPECT_EQ(1u, i64from63.countLeadingOnes()); 2333 EXPECT_EQ(0u, i64from63.countLeadingZeros()); 2334 EXPECT_EQ(64u, i64from63.getActiveBits()); 2335 EXPECT_EQ(63u, i64from63.countTrailingZeros()); 2336 EXPECT_EQ(0u, i64from63.countTrailingOnes()); 2337 EXPECT_EQ(1u, i64from63.countPopulation()); 2338 } 2339 2340 TEST(APIntTest, setAllBits) { 2341 APInt i32(32, 0); 2342 i32.setAllBits(); 2343 EXPECT_EQ(32u, i32.countLeadingOnes()); 2344 EXPECT_EQ(0u, i32.countLeadingZeros()); 2345 EXPECT_EQ(32u, i32.getActiveBits()); 2346 EXPECT_EQ(0u, i32.countTrailingZeros()); 2347 EXPECT_EQ(32u, i32.countTrailingOnes()); 2348 EXPECT_EQ(32u, i32.countPopulation()); 2349 2350 APInt i64(64, 0); 2351 i64.setAllBits(); 2352 EXPECT_EQ(64u, i64.countLeadingOnes()); 2353 EXPECT_EQ(0u, i64.countLeadingZeros()); 2354 EXPECT_EQ(64u, i64.getActiveBits()); 2355 EXPECT_EQ(0u, i64.countTrailingZeros()); 2356 EXPECT_EQ(64u, i64.countTrailingOnes()); 2357 EXPECT_EQ(64u, i64.countPopulation()); 2358 2359 APInt i96(96, 0); 2360 i96.setAllBits(); 2361 EXPECT_EQ(96u, i96.countLeadingOnes()); 2362 EXPECT_EQ(0u, i96.countLeadingZeros()); 2363 EXPECT_EQ(96u, i96.getActiveBits()); 2364 EXPECT_EQ(0u, i96.countTrailingZeros()); 2365 EXPECT_EQ(96u, i96.countTrailingOnes()); 2366 EXPECT_EQ(96u, i96.countPopulation()); 2367 2368 APInt i128(128, 0); 2369 i128.setAllBits(); 2370 EXPECT_EQ(128u, i128.countLeadingOnes()); 2371 EXPECT_EQ(0u, i128.countLeadingZeros()); 2372 EXPECT_EQ(128u, i128.getActiveBits()); 2373 EXPECT_EQ(0u, i128.countTrailingZeros()); 2374 EXPECT_EQ(128u, i128.countTrailingOnes()); 2375 EXPECT_EQ(128u, i128.countPopulation()); 2376 } 2377 2378 TEST(APIntTest, getLoBits) { 2379 APInt i32(32, 0xfa); 2380 i32.setHighBits(1); 2381 EXPECT_EQ(0xa, i32.getLoBits(4)); 2382 APInt i128(128, 0xfa); 2383 i128.setHighBits(1); 2384 EXPECT_EQ(0xa, i128.getLoBits(4)); 2385 } 2386 2387 TEST(APIntTest, getHiBits) { 2388 APInt i32(32, 0xfa); 2389 i32.setHighBits(2); 2390 EXPECT_EQ(0xc, i32.getHiBits(4)); 2391 APInt i128(128, 0xfa); 2392 i128.setHighBits(2); 2393 EXPECT_EQ(0xc, i128.getHiBits(4)); 2394 } 2395 2396 TEST(APIntTest, clearLowBits) { 2397 APInt i64hi32 = APInt::getAllOnes(64); 2398 i64hi32.clearLowBits(32); 2399 EXPECT_EQ(32u, i64hi32.countLeadingOnes()); 2400 EXPECT_EQ(0u, i64hi32.countLeadingZeros()); 2401 EXPECT_EQ(64u, i64hi32.getActiveBits()); 2402 EXPECT_EQ(32u, i64hi32.countTrailingZeros()); 2403 EXPECT_EQ(0u, i64hi32.countTrailingOnes()); 2404 EXPECT_EQ(32u, i64hi32.countPopulation()); 2405 2406 APInt i128hi64 = APInt::getAllOnes(128); 2407 i128hi64.clearLowBits(64); 2408 EXPECT_EQ(64u, i128hi64.countLeadingOnes()); 2409 EXPECT_EQ(0u, i128hi64.countLeadingZeros()); 2410 EXPECT_EQ(128u, i128hi64.getActiveBits()); 2411 EXPECT_EQ(64u, i128hi64.countTrailingZeros()); 2412 EXPECT_EQ(0u, i128hi64.countTrailingOnes()); 2413 EXPECT_EQ(64u, i128hi64.countPopulation()); 2414 2415 APInt i128hi24 = APInt::getAllOnes(128); 2416 i128hi24.clearLowBits(104); 2417 EXPECT_EQ(24u, i128hi24.countLeadingOnes()); 2418 EXPECT_EQ(0u, i128hi24.countLeadingZeros()); 2419 EXPECT_EQ(128u, i128hi24.getActiveBits()); 2420 EXPECT_EQ(104u, i128hi24.countTrailingZeros()); 2421 EXPECT_EQ(0u, i128hi24.countTrailingOnes()); 2422 EXPECT_EQ(24u, i128hi24.countPopulation()); 2423 2424 APInt i128hi104 = APInt::getAllOnes(128); 2425 i128hi104.clearLowBits(24); 2426 EXPECT_EQ(104u, i128hi104.countLeadingOnes()); 2427 EXPECT_EQ(0u, i128hi104.countLeadingZeros()); 2428 EXPECT_EQ(128u, i128hi104.getActiveBits()); 2429 EXPECT_EQ(24u, i128hi104.countTrailingZeros()); 2430 EXPECT_EQ(0u, i128hi104.countTrailingOnes()); 2431 EXPECT_EQ(104u, i128hi104.countPopulation()); 2432 2433 APInt i128hi0 = APInt::getAllOnes(128); 2434 i128hi0.clearLowBits(128); 2435 EXPECT_EQ(0u, i128hi0.countLeadingOnes()); 2436 EXPECT_EQ(128u, i128hi0.countLeadingZeros()); 2437 EXPECT_EQ(0u, i128hi0.getActiveBits()); 2438 EXPECT_EQ(128u, i128hi0.countTrailingZeros()); 2439 EXPECT_EQ(0u, i128hi0.countTrailingOnes()); 2440 EXPECT_EQ(0u, i128hi0.countPopulation()); 2441 2442 APInt i80hi1 = APInt::getAllOnes(80); 2443 i80hi1.clearLowBits(79); 2444 EXPECT_EQ(1u, i80hi1.countLeadingOnes()); 2445 EXPECT_EQ(0u, i80hi1.countLeadingZeros()); 2446 EXPECT_EQ(80u, i80hi1.getActiveBits()); 2447 EXPECT_EQ(79u, i80hi1.countTrailingZeros()); 2448 EXPECT_EQ(0u, i80hi1.countTrailingOnes()); 2449 EXPECT_EQ(1u, i80hi1.countPopulation()); 2450 2451 APInt i32hi16 = APInt::getAllOnes(32); 2452 i32hi16.clearLowBits(16); 2453 EXPECT_EQ(16u, i32hi16.countLeadingOnes()); 2454 EXPECT_EQ(0u, i32hi16.countLeadingZeros()); 2455 EXPECT_EQ(32u, i32hi16.getActiveBits()); 2456 EXPECT_EQ(16u, i32hi16.countTrailingZeros()); 2457 EXPECT_EQ(0u, i32hi16.countTrailingOnes()); 2458 EXPECT_EQ(16u, i32hi16.countPopulation()); 2459 } 2460 2461 TEST(APIntTest, GCD) { 2462 using APIntOps::GreatestCommonDivisor; 2463 2464 for (unsigned Bits : {1, 2, 32, 63, 64, 65}) { 2465 // Test some corner cases near zero. 2466 APInt Zero(Bits, 0), One(Bits, 1); 2467 EXPECT_EQ(GreatestCommonDivisor(Zero, Zero), Zero); 2468 EXPECT_EQ(GreatestCommonDivisor(Zero, One), One); 2469 EXPECT_EQ(GreatestCommonDivisor(One, Zero), One); 2470 EXPECT_EQ(GreatestCommonDivisor(One, One), One); 2471 2472 if (Bits > 1) { 2473 APInt Two(Bits, 2); 2474 EXPECT_EQ(GreatestCommonDivisor(Zero, Two), Two); 2475 EXPECT_EQ(GreatestCommonDivisor(One, Two), One); 2476 EXPECT_EQ(GreatestCommonDivisor(Two, Two), Two); 2477 2478 // Test some corner cases near the highest representable value. 2479 APInt Max(Bits, 0); 2480 Max.setAllBits(); 2481 EXPECT_EQ(GreatestCommonDivisor(Zero, Max), Max); 2482 EXPECT_EQ(GreatestCommonDivisor(One, Max), One); 2483 EXPECT_EQ(GreatestCommonDivisor(Two, Max), One); 2484 EXPECT_EQ(GreatestCommonDivisor(Max, Max), Max); 2485 2486 APInt MaxOver2 = Max.udiv(Two); 2487 EXPECT_EQ(GreatestCommonDivisor(MaxOver2, Max), One); 2488 // Max - 1 == Max / 2 * 2, because Max is odd. 2489 EXPECT_EQ(GreatestCommonDivisor(MaxOver2, Max - 1), MaxOver2); 2490 } 2491 } 2492 2493 // Compute the 20th Mersenne prime. 2494 const unsigned BitWidth = 4450; 2495 APInt HugePrime = APInt::getLowBitsSet(BitWidth, 4423); 2496 2497 // 9931 and 123456 are coprime. 2498 APInt A = HugePrime * APInt(BitWidth, 9931); 2499 APInt B = HugePrime * APInt(BitWidth, 123456); 2500 APInt C = GreatestCommonDivisor(A, B); 2501 EXPECT_EQ(C, HugePrime); 2502 } 2503 2504 TEST(APIntTest, LogicalRightShift) { 2505 APInt i256(APInt::getHighBitsSet(256, 2)); 2506 2507 i256.lshrInPlace(1); 2508 EXPECT_EQ(1U, i256.countLeadingZeros()); 2509 EXPECT_EQ(253U, i256.countTrailingZeros()); 2510 EXPECT_EQ(2U, i256.countPopulation()); 2511 2512 i256.lshrInPlace(62); 2513 EXPECT_EQ(63U, i256.countLeadingZeros()); 2514 EXPECT_EQ(191U, i256.countTrailingZeros()); 2515 EXPECT_EQ(2U, i256.countPopulation()); 2516 2517 i256.lshrInPlace(65); 2518 EXPECT_EQ(128U, i256.countLeadingZeros()); 2519 EXPECT_EQ(126U, i256.countTrailingZeros()); 2520 EXPECT_EQ(2U, i256.countPopulation()); 2521 2522 i256.lshrInPlace(64); 2523 EXPECT_EQ(192U, i256.countLeadingZeros()); 2524 EXPECT_EQ(62U, i256.countTrailingZeros()); 2525 EXPECT_EQ(2U, i256.countPopulation()); 2526 2527 i256.lshrInPlace(63); 2528 EXPECT_EQ(255U, i256.countLeadingZeros()); 2529 EXPECT_EQ(0U, i256.countTrailingZeros()); 2530 EXPECT_EQ(1U, i256.countPopulation()); 2531 2532 // Ensure we handle large shifts of multi-word. 2533 const APInt neg_one(128, static_cast<uint64_t>(-1), true); 2534 EXPECT_EQ(0, neg_one.lshr(128)); 2535 } 2536 2537 TEST(APIntTest, ArithmeticRightShift) { 2538 APInt i72(APInt::getHighBitsSet(72, 1)); 2539 i72.ashrInPlace(46); 2540 EXPECT_EQ(47U, i72.countLeadingOnes()); 2541 EXPECT_EQ(25U, i72.countTrailingZeros()); 2542 EXPECT_EQ(47U, i72.countPopulation()); 2543 2544 i72 = APInt::getHighBitsSet(72, 1); 2545 i72.ashrInPlace(64); 2546 EXPECT_EQ(65U, i72.countLeadingOnes()); 2547 EXPECT_EQ(7U, i72.countTrailingZeros()); 2548 EXPECT_EQ(65U, i72.countPopulation()); 2549 2550 APInt i128(APInt::getHighBitsSet(128, 1)); 2551 i128.ashrInPlace(64); 2552 EXPECT_EQ(65U, i128.countLeadingOnes()); 2553 EXPECT_EQ(63U, i128.countTrailingZeros()); 2554 EXPECT_EQ(65U, i128.countPopulation()); 2555 2556 // Ensure we handle large shifts of multi-word. 2557 const APInt signmin32(APInt::getSignedMinValue(32)); 2558 EXPECT_TRUE(signmin32.ashr(32).isAllOnes()); 2559 2560 // Ensure we handle large shifts of multi-word. 2561 const APInt umax32(APInt::getSignedMaxValue(32)); 2562 EXPECT_EQ(0, umax32.ashr(32)); 2563 2564 // Ensure we handle large shifts of multi-word. 2565 const APInt signmin128(APInt::getSignedMinValue(128)); 2566 EXPECT_TRUE(signmin128.ashr(128).isAllOnes()); 2567 2568 // Ensure we handle large shifts of multi-word. 2569 const APInt umax128(APInt::getSignedMaxValue(128)); 2570 EXPECT_EQ(0, umax128.ashr(128)); 2571 } 2572 2573 TEST(APIntTest, LeftShift) { 2574 APInt i256(APInt::getLowBitsSet(256, 2)); 2575 2576 i256 <<= 1; 2577 EXPECT_EQ(253U, i256.countLeadingZeros()); 2578 EXPECT_EQ(1U, i256.countTrailingZeros()); 2579 EXPECT_EQ(2U, i256.countPopulation()); 2580 2581 i256 <<= 62; 2582 EXPECT_EQ(191U, i256.countLeadingZeros()); 2583 EXPECT_EQ(63U, i256.countTrailingZeros()); 2584 EXPECT_EQ(2U, i256.countPopulation()); 2585 2586 i256 <<= 65; 2587 EXPECT_EQ(126U, i256.countLeadingZeros()); 2588 EXPECT_EQ(128U, i256.countTrailingZeros()); 2589 EXPECT_EQ(2U, i256.countPopulation()); 2590 2591 i256 <<= 64; 2592 EXPECT_EQ(62U, i256.countLeadingZeros()); 2593 EXPECT_EQ(192U, i256.countTrailingZeros()); 2594 EXPECT_EQ(2U, i256.countPopulation()); 2595 2596 i256 <<= 63; 2597 EXPECT_EQ(0U, i256.countLeadingZeros()); 2598 EXPECT_EQ(255U, i256.countTrailingZeros()); 2599 EXPECT_EQ(1U, i256.countPopulation()); 2600 2601 // Ensure we handle large shifts of multi-word. 2602 const APInt neg_one(128, static_cast<uint64_t>(-1), true); 2603 EXPECT_EQ(0, neg_one.shl(128)); 2604 } 2605 2606 TEST(APIntTest, isSubsetOf) { 2607 APInt i32_1(32, 1); 2608 APInt i32_2(32, 2); 2609 APInt i32_3(32, 3); 2610 EXPECT_FALSE(i32_3.isSubsetOf(i32_1)); 2611 EXPECT_TRUE(i32_1.isSubsetOf(i32_3)); 2612 EXPECT_FALSE(i32_2.isSubsetOf(i32_1)); 2613 EXPECT_FALSE(i32_1.isSubsetOf(i32_2)); 2614 EXPECT_TRUE(i32_3.isSubsetOf(i32_3)); 2615 2616 APInt i128_1(128, 1); 2617 APInt i128_2(128, 2); 2618 APInt i128_3(128, 3); 2619 EXPECT_FALSE(i128_3.isSubsetOf(i128_1)); 2620 EXPECT_TRUE(i128_1.isSubsetOf(i128_3)); 2621 EXPECT_FALSE(i128_2.isSubsetOf(i128_1)); 2622 EXPECT_FALSE(i128_1.isSubsetOf(i128_2)); 2623 EXPECT_TRUE(i128_3.isSubsetOf(i128_3)); 2624 2625 i128_1 <<= 64; 2626 i128_2 <<= 64; 2627 i128_3 <<= 64; 2628 EXPECT_FALSE(i128_3.isSubsetOf(i128_1)); 2629 EXPECT_TRUE(i128_1.isSubsetOf(i128_3)); 2630 EXPECT_FALSE(i128_2.isSubsetOf(i128_1)); 2631 EXPECT_FALSE(i128_1.isSubsetOf(i128_2)); 2632 EXPECT_TRUE(i128_3.isSubsetOf(i128_3)); 2633 } 2634 2635 TEST(APIntTest, sext) { 2636 EXPECT_EQ(0, APInt(1, 0).sext(64)); 2637 EXPECT_EQ(~uint64_t(0), APInt(1, 1).sext(64)); 2638 2639 APInt i32_max(APInt::getSignedMaxValue(32).sext(63)); 2640 EXPECT_EQ(32U, i32_max.countLeadingZeros()); 2641 EXPECT_EQ(0U, i32_max.countTrailingZeros()); 2642 EXPECT_EQ(31U, i32_max.countPopulation()); 2643 2644 APInt i32_min(APInt::getSignedMinValue(32).sext(63)); 2645 EXPECT_EQ(32U, i32_min.countLeadingOnes()); 2646 EXPECT_EQ(31U, i32_min.countTrailingZeros()); 2647 EXPECT_EQ(32U, i32_min.countPopulation()); 2648 2649 APInt i32_neg1(APInt(32, ~uint64_t(0)).sext(63)); 2650 EXPECT_EQ(63U, i32_neg1.countLeadingOnes()); 2651 EXPECT_EQ(0U, i32_neg1.countTrailingZeros()); 2652 EXPECT_EQ(63U, i32_neg1.countPopulation()); 2653 } 2654 2655 TEST(APIntTest, truncOrSelf) { 2656 APInt val(32, 0xFFFFFFFF); 2657 EXPECT_EQ(0xFFFF, val.truncOrSelf(16)); 2658 EXPECT_EQ(0xFFFFFFFF, val.truncOrSelf(32)); 2659 EXPECT_EQ(0xFFFFFFFF, val.truncOrSelf(64)); 2660 } 2661 2662 TEST(APIntTest, concat) { 2663 APInt Int1(4, 0x1ULL); 2664 APInt Int3(4, 0x3ULL); 2665 2666 EXPECT_EQ(0x31, Int3.concat(Int1)); 2667 EXPECT_EQ(APInt(12, 0x313), Int3.concat(Int1).concat(Int3)); 2668 EXPECT_EQ(APInt(16, 0x3313), Int3.concat(Int3).concat(Int1).concat(Int3)); 2669 2670 APInt I64(64, 0x3ULL); 2671 EXPECT_EQ(I64, I64.concat(I64).lshr(64).trunc(64)); 2672 2673 APInt I65(65, 0x3ULL); 2674 APInt I0 = APInt::getZeroWidth(); 2675 EXPECT_EQ(I65, I65.concat(I0)); 2676 EXPECT_EQ(I65, I0.concat(I65)); 2677 } 2678 2679 TEST(APIntTest, multiply) { 2680 APInt i64(64, 1234); 2681 2682 EXPECT_EQ(7006652, i64 * 5678); 2683 EXPECT_EQ(7006652, 5678 * i64); 2684 2685 APInt i128 = APInt::getOneBitSet(128, 64); 2686 APInt i128_1234(128, 1234); 2687 i128_1234 <<= 64; 2688 EXPECT_EQ(i128_1234, i128 * 1234); 2689 EXPECT_EQ(i128_1234, 1234 * i128); 2690 2691 APInt i96 = APInt::getOneBitSet(96, 64); 2692 i96 *= ~0ULL; 2693 EXPECT_EQ(32U, i96.countLeadingOnes()); 2694 EXPECT_EQ(32U, i96.countPopulation()); 2695 EXPECT_EQ(64U, i96.countTrailingZeros()); 2696 } 2697 2698 TEST(APIntTest, RoundingUDiv) { 2699 for (uint64_t Ai = 1; Ai <= 255; Ai++) { 2700 APInt A(8, Ai); 2701 APInt Zero(8, 0); 2702 EXPECT_EQ(0, APIntOps::RoundingUDiv(Zero, A, APInt::Rounding::UP)); 2703 EXPECT_EQ(0, APIntOps::RoundingUDiv(Zero, A, APInt::Rounding::DOWN)); 2704 EXPECT_EQ(0, APIntOps::RoundingUDiv(Zero, A, APInt::Rounding::TOWARD_ZERO)); 2705 2706 for (uint64_t Bi = 1; Bi <= 255; Bi++) { 2707 APInt B(8, Bi); 2708 { 2709 APInt Quo = APIntOps::RoundingUDiv(A, B, APInt::Rounding::UP); 2710 auto Prod = Quo.zext(16) * B.zext(16); 2711 EXPECT_TRUE(Prod.uge(Ai)); 2712 if (Prod.ugt(Ai)) { 2713 EXPECT_TRUE(((Quo - 1).zext(16) * B.zext(16)).ult(Ai)); 2714 } 2715 } 2716 { 2717 APInt Quo = A.udiv(B); 2718 EXPECT_EQ(Quo, APIntOps::RoundingUDiv(A, B, APInt::Rounding::TOWARD_ZERO)); 2719 EXPECT_EQ(Quo, APIntOps::RoundingUDiv(A, B, APInt::Rounding::DOWN)); 2720 } 2721 } 2722 } 2723 } 2724 2725 TEST(APIntTest, RoundingSDiv) { 2726 for (int64_t Ai = -128; Ai <= 127; Ai++) { 2727 APInt A(8, Ai); 2728 2729 if (Ai != 0) { 2730 APInt Zero(8, 0); 2731 EXPECT_EQ(0, APIntOps::RoundingSDiv(Zero, A, APInt::Rounding::UP)); 2732 EXPECT_EQ(0, APIntOps::RoundingSDiv(Zero, A, APInt::Rounding::DOWN)); 2733 EXPECT_EQ(0, APIntOps::RoundingSDiv(Zero, A, APInt::Rounding::TOWARD_ZERO)); 2734 } 2735 2736 for (int64_t Bi = -128; Bi <= 127; Bi++) { 2737 if (Bi == 0) 2738 continue; 2739 2740 APInt B(8, Bi); 2741 APInt QuoTowardZero = A.sdiv(B); 2742 { 2743 APInt Quo = APIntOps::RoundingSDiv(A, B, APInt::Rounding::UP); 2744 if (A.srem(B).isNullValue()) { 2745 EXPECT_EQ(QuoTowardZero, Quo); 2746 } else if (A.isNegative() != 2747 B.isNegative()) { // if the math quotient is negative. 2748 EXPECT_EQ(QuoTowardZero, Quo); 2749 } else { 2750 EXPECT_EQ(QuoTowardZero + 1, Quo); 2751 } 2752 } 2753 { 2754 APInt Quo = APIntOps::RoundingSDiv(A, B, APInt::Rounding::DOWN); 2755 if (A.srem(B).isNullValue()) { 2756 EXPECT_EQ(QuoTowardZero, Quo); 2757 } else if (A.isNegative() != 2758 B.isNegative()) { // if the math quotient is negative. 2759 EXPECT_EQ(QuoTowardZero - 1, Quo); 2760 } else { 2761 EXPECT_EQ(QuoTowardZero, Quo); 2762 } 2763 } 2764 EXPECT_EQ(QuoTowardZero, 2765 APIntOps::RoundingSDiv(A, B, APInt::Rounding::TOWARD_ZERO)); 2766 } 2767 } 2768 } 2769 2770 TEST(APIntTest, umul_ov) { 2771 const std::pair<uint64_t, uint64_t> Overflows[] = { 2772 {0x8000000000000000, 2}, 2773 {0x5555555555555556, 3}, 2774 {4294967296, 4294967296}, 2775 {4294967295, 4294967298}, 2776 }; 2777 const std::pair<uint64_t, uint64_t> NonOverflows[] = { 2778 {0x7fffffffffffffff, 2}, 2779 {0x5555555555555555, 3}, 2780 {4294967295, 4294967297}, 2781 }; 2782 2783 bool Overflow; 2784 for (auto &X : Overflows) { 2785 APInt A(64, X.first); 2786 APInt B(64, X.second); 2787 (void)A.umul_ov(B, Overflow); 2788 EXPECT_TRUE(Overflow); 2789 } 2790 for (auto &X : NonOverflows) { 2791 APInt A(64, X.first); 2792 APInt B(64, X.second); 2793 (void)A.umul_ov(B, Overflow); 2794 EXPECT_FALSE(Overflow); 2795 } 2796 2797 for (unsigned Bits = 1; Bits <= 5; ++Bits) 2798 for (unsigned A = 0; A != 1u << Bits; ++A) 2799 for (unsigned B = 0; B != 1u << Bits; ++B) { 2800 APInt N1 = APInt(Bits, A), N2 = APInt(Bits, B); 2801 APInt Narrow = N1.umul_ov(N2, Overflow); 2802 APInt Wide = N1.zext(2 * Bits) * N2.zext(2 * Bits); 2803 EXPECT_EQ(Wide.trunc(Bits), Narrow); 2804 EXPECT_EQ(Narrow.zext(2 * Bits) != Wide, Overflow); 2805 } 2806 } 2807 2808 TEST(APIntTest, smul_ov) { 2809 for (unsigned Bits = 1; Bits <= 5; ++Bits) 2810 for (unsigned A = 0; A != 1u << Bits; ++A) 2811 for (unsigned B = 0; B != 1u << Bits; ++B) { 2812 bool Overflow; 2813 APInt N1 = APInt(Bits, A), N2 = APInt(Bits, B); 2814 APInt Narrow = N1.smul_ov(N2, Overflow); 2815 APInt Wide = N1.sext(2 * Bits) * N2.sext(2 * Bits); 2816 EXPECT_EQ(Wide.trunc(Bits), Narrow); 2817 EXPECT_EQ(Narrow.sext(2 * Bits) != Wide, Overflow); 2818 } 2819 } 2820 2821 TEST(APIntTest, SolveQuadraticEquationWrap) { 2822 // Verify that "Solution" is the first non-negative integer that solves 2823 // Ax^2 + Bx + C = "0 or overflow", i.e. that it is a correct solution 2824 // as calculated by SolveQuadraticEquationWrap. 2825 auto Validate = [] (int A, int B, int C, unsigned Width, int Solution) { 2826 int Mask = (1 << Width) - 1; 2827 2828 // Solution should be non-negative. 2829 EXPECT_GE(Solution, 0); 2830 2831 auto OverflowBits = [] (int64_t V, unsigned W) { 2832 return V & -(1 << W); 2833 }; 2834 2835 int64_t Over0 = OverflowBits(C, Width); 2836 2837 auto IsZeroOrOverflow = [&] (int X) { 2838 int64_t ValueAtX = A*X*X + B*X + C; 2839 int64_t OverX = OverflowBits(ValueAtX, Width); 2840 return (ValueAtX & Mask) == 0 || OverX != Over0; 2841 }; 2842 2843 auto EquationToString = [&] (const char *X_str) { 2844 return (Twine(A) + Twine(X_str) + Twine("^2 + ") + Twine(B) + 2845 Twine(X_str) + Twine(" + ") + Twine(C) + Twine(", bitwidth: ") + 2846 Twine(Width)).str(); 2847 }; 2848 2849 auto IsSolution = [&] (const char *X_str, int X) { 2850 if (IsZeroOrOverflow(X)) 2851 return ::testing::AssertionSuccess() 2852 << X << " is a solution of " << EquationToString(X_str); 2853 return ::testing::AssertionFailure() 2854 << X << " is not an expected solution of " 2855 << EquationToString(X_str); 2856 }; 2857 2858 auto IsNotSolution = [&] (const char *X_str, int X) { 2859 if (!IsZeroOrOverflow(X)) 2860 return ::testing::AssertionSuccess() 2861 << X << " is not a solution of " << EquationToString(X_str); 2862 return ::testing::AssertionFailure() 2863 << X << " is an unexpected solution of " 2864 << EquationToString(X_str); 2865 }; 2866 2867 // This is the important part: make sure that there is no solution that 2868 // is less than the calculated one. 2869 if (Solution > 0) { 2870 for (int X = 1; X < Solution-1; ++X) 2871 EXPECT_PRED_FORMAT1(IsNotSolution, X); 2872 } 2873 2874 // Verify that the calculated solution is indeed a solution. 2875 EXPECT_PRED_FORMAT1(IsSolution, Solution); 2876 }; 2877 2878 // Generate all possible quadratic equations with Width-bit wide integer 2879 // coefficients, get the solution from SolveQuadraticEquationWrap, and 2880 // verify that the solution is correct. 2881 auto Iterate = [&] (unsigned Width) { 2882 assert(1 < Width && Width < 32); 2883 int Low = -(1 << (Width-1)); 2884 int High = (1 << (Width-1)); 2885 2886 for (int A = Low; A != High; ++A) { 2887 if (A == 0) 2888 continue; 2889 for (int B = Low; B != High; ++B) { 2890 for (int C = Low; C != High; ++C) { 2891 Optional<APInt> S = APIntOps::SolveQuadraticEquationWrap( 2892 APInt(Width, A), APInt(Width, B), 2893 APInt(Width, C), Width); 2894 if (S.hasValue()) 2895 Validate(A, B, C, Width, S->getSExtValue()); 2896 } 2897 } 2898 } 2899 }; 2900 2901 // Test all widths in [2..6]. 2902 for (unsigned i = 2; i <= 6; ++i) 2903 Iterate(i); 2904 } 2905 2906 TEST(APIntTest, MultiplicativeInverseExaustive) { 2907 for (unsigned BitWidth = 1; BitWidth <= 16; ++BitWidth) { 2908 for (unsigned Value = 0; Value < (1u << BitWidth); ++Value) { 2909 APInt V = APInt(BitWidth, Value); 2910 APInt MulInv = 2911 V.zext(BitWidth + 1) 2912 .multiplicativeInverse(APInt::getSignedMinValue(BitWidth + 1)) 2913 .trunc(BitWidth); 2914 APInt One = V * MulInv; 2915 if (!V.isNullValue() && V.countTrailingZeros() == 0) { 2916 // Multiplicative inverse exists for all odd numbers. 2917 EXPECT_TRUE(One.isOneValue()); 2918 } else { 2919 // Multiplicative inverse does not exist for even numbers (and 0). 2920 EXPECT_TRUE(MulInv.isNullValue()); 2921 } 2922 } 2923 } 2924 } 2925 2926 TEST(APIntTest, GetMostSignificantDifferentBit) { 2927 EXPECT_EQ(APIntOps::GetMostSignificantDifferentBit(APInt(8, 0), APInt(8, 0)), 2928 llvm::None); 2929 EXPECT_EQ( 2930 APIntOps::GetMostSignificantDifferentBit(APInt(8, 42), APInt(8, 42)), 2931 llvm::None); 2932 EXPECT_EQ(*APIntOps::GetMostSignificantDifferentBit(APInt(8, 0), APInt(8, 1)), 2933 0u); 2934 EXPECT_EQ(*APIntOps::GetMostSignificantDifferentBit(APInt(8, 0), APInt(8, 2)), 2935 1u); 2936 EXPECT_EQ(*APIntOps::GetMostSignificantDifferentBit(APInt(8, 0), APInt(8, 3)), 2937 1u); 2938 EXPECT_EQ(*APIntOps::GetMostSignificantDifferentBit(APInt(8, 1), APInt(8, 0)), 2939 0u); 2940 EXPECT_EQ(APIntOps::GetMostSignificantDifferentBit(APInt(8, 1), APInt(8, 1)), 2941 llvm::None); 2942 EXPECT_EQ(*APIntOps::GetMostSignificantDifferentBit(APInt(8, 1), APInt(8, 2)), 2943 1u); 2944 EXPECT_EQ(*APIntOps::GetMostSignificantDifferentBit(APInt(8, 1), APInt(8, 3)), 2945 1u); 2946 EXPECT_EQ( 2947 *APIntOps::GetMostSignificantDifferentBit(APInt(8, 42), APInt(8, 112)), 2948 6u); 2949 } 2950 2951 TEST(APIntTest, GetMostSignificantDifferentBitExaustive) { 2952 auto GetHighestDifferentBitBruteforce = 2953 [](const APInt &V0, const APInt &V1) -> llvm::Optional<unsigned> { 2954 assert(V0.getBitWidth() == V1.getBitWidth() && "Must have same bitwidth"); 2955 if (V0 == V1) 2956 return llvm::None; // Bitwise identical. 2957 // There is a mismatch. Let's find the most significant different bit. 2958 for (int Bit = V0.getBitWidth() - 1; Bit >= 0; --Bit) { 2959 if (V0[Bit] == V1[Bit]) 2960 continue; 2961 return Bit; 2962 } 2963 llvm_unreachable("Must have found bit mismatch."); 2964 }; 2965 2966 for (unsigned BitWidth = 1; BitWidth <= 8; ++BitWidth) { 2967 for (unsigned V0 = 0; V0 < (1u << BitWidth); ++V0) { 2968 for (unsigned V1 = 0; V1 < (1u << BitWidth); ++V1) { 2969 APInt A = APInt(BitWidth, V0); 2970 APInt B = APInt(BitWidth, V1); 2971 2972 auto Bit = APIntOps::GetMostSignificantDifferentBit(A, B); 2973 EXPECT_EQ(Bit, GetHighestDifferentBitBruteforce(A, B)); 2974 2975 if (!Bit.hasValue()) 2976 EXPECT_EQ(A, B); 2977 else { 2978 EXPECT_NE(A, B); 2979 for (unsigned NumLowBits = 0; NumLowBits <= BitWidth; ++NumLowBits) { 2980 APInt Adash = A; 2981 Adash.clearLowBits(NumLowBits); 2982 APInt Bdash = B; 2983 Bdash.clearLowBits(NumLowBits); 2984 // Clearing only low bits up to and including *Bit is sufficient 2985 // to make values equal. 2986 if (NumLowBits >= 1 + *Bit) 2987 EXPECT_EQ(Adash, Bdash); 2988 else 2989 EXPECT_NE(Adash, Bdash); 2990 } 2991 } 2992 } 2993 } 2994 } 2995 } 2996 2997 TEST(APIntTest, SignbitZeroChecks) { 2998 EXPECT_TRUE(APInt(8, -1).isNegative()); 2999 EXPECT_FALSE(APInt(8, -1).isNonNegative()); 3000 EXPECT_FALSE(APInt(8, -1).isStrictlyPositive()); 3001 EXPECT_TRUE(APInt(8, -1).isNonPositive()); 3002 3003 EXPECT_FALSE(APInt(8, 0).isNegative()); 3004 EXPECT_TRUE(APInt(8, 0).isNonNegative()); 3005 EXPECT_FALSE(APInt(8, 0).isStrictlyPositive()); 3006 EXPECT_TRUE(APInt(8, 0).isNonPositive()); 3007 3008 EXPECT_FALSE(APInt(8, 1).isNegative()); 3009 EXPECT_TRUE(APInt(8, 1).isNonNegative()); 3010 EXPECT_TRUE(APInt(8, 1).isStrictlyPositive()); 3011 EXPECT_FALSE(APInt(8, 1).isNonPositive()); 3012 } 3013 3014 TEST(APIntTest, ZeroWidth) { 3015 // Zero width Constructors. 3016 auto ZW = APInt::getZeroWidth(); 3017 EXPECT_EQ(0U, ZW.getBitWidth()); 3018 EXPECT_EQ(0U, APInt(0, ArrayRef<uint64_t>({0, 1, 2})).getBitWidth()); 3019 EXPECT_EQ(0U, APInt(0, "0", 10).getBitWidth()); 3020 3021 // Default constructor is single bit wide. 3022 EXPECT_EQ(1U, APInt().getBitWidth()); 3023 3024 // Copy ctor (move is down below). 3025 APInt ZW2(ZW); 3026 EXPECT_EQ(0U, ZW2.getBitWidth()); 3027 // Assignment 3028 ZW = ZW2; 3029 EXPECT_EQ(0U, ZW.getBitWidth()); 3030 3031 // Methods like getLowBitsSet work with zero bits. 3032 EXPECT_EQ(0U, APInt::getLowBitsSet(0, 0).getBitWidth()); 3033 EXPECT_EQ(0U, APInt::getSplat(0, ZW).getBitWidth()); 3034 EXPECT_EQ(0U, APInt(4, 10).extractBits(0, 2).getBitWidth()); 3035 3036 // Logical operators. 3037 ZW |= ZW2; 3038 ZW &= ZW2; 3039 ZW ^= ZW2; 3040 ZW |= 42; // These ignore high bits of the literal. 3041 ZW &= 42; 3042 ZW ^= 42; 3043 EXPECT_EQ(1, ZW.isIntN(0)); 3044 3045 // Modulo Arithmetic. Divide/Rem aren't defined on division by zero, so they 3046 // aren't supported. 3047 ZW += ZW2; 3048 ZW -= ZW2; 3049 ZW *= ZW2; 3050 3051 // Logical Shifts and rotates, the amount must be <= bitwidth. 3052 ZW <<= 0; 3053 ZW.lshrInPlace(0); 3054 (void)ZW.rotl(0); 3055 (void)ZW.rotr(0); 3056 3057 // Comparisons. 3058 EXPECT_EQ(1, ZW == ZW); 3059 EXPECT_EQ(0, ZW != ZW); 3060 EXPECT_EQ(0, ZW.ult(ZW)); 3061 3062 // Mutations. 3063 ZW.setBitsWithWrap(0, 0); 3064 ZW.setBits(0, 0); 3065 ZW.clearAllBits(); 3066 ZW.flipAllBits(); 3067 3068 // Leading, trailing, ctpop, etc 3069 EXPECT_EQ(0U, ZW.countLeadingZeros()); 3070 EXPECT_EQ(0U, ZW.countLeadingOnes()); 3071 EXPECT_EQ(0U, ZW.countPopulation()); 3072 EXPECT_EQ(0U, ZW.reverseBits().getBitWidth()); 3073 EXPECT_EQ(0U, ZW.getHiBits(0).getBitWidth()); 3074 EXPECT_EQ(0U, ZW.getLoBits(0).getBitWidth()); 3075 EXPECT_EQ(0, ZW.zext(4)); 3076 EXPECT_EQ(0U, APInt(4, 3).trunc(0).getBitWidth()); 3077 EXPECT_TRUE(ZW.isAllOnes()); 3078 3079 // Zero extension. 3080 EXPECT_EQ(0U, ZW.getZExtValue()); 3081 3082 SmallString<42> STR; 3083 ZW.toStringUnsigned(STR); 3084 EXPECT_EQ("0", STR); 3085 3086 // Move ctor (keep at the end of the method since moves are destructive). 3087 APInt MZW1(std::move(ZW)); 3088 EXPECT_EQ(0U, MZW1.getBitWidth()); 3089 // Move Assignment 3090 MZW1 = std::move(ZW2); 3091 EXPECT_EQ(0U, MZW1.getBitWidth()); 3092 } 3093 3094 TEST(APIntTest, ScaleBitMask) { 3095 EXPECT_EQ(APIntOps::ScaleBitMask(APInt(2, 0x00), 8), APInt(8, 0x00)); 3096 EXPECT_EQ(APIntOps::ScaleBitMask(APInt(2, 0x01), 8), APInt(8, 0x0F)); 3097 EXPECT_EQ(APIntOps::ScaleBitMask(APInt(2, 0x02), 8), APInt(8, 0xF0)); 3098 EXPECT_EQ(APIntOps::ScaleBitMask(APInt(2, 0x03), 8), APInt(8, 0xFF)); 3099 3100 EXPECT_EQ(APIntOps::ScaleBitMask(APInt(8, 0x00), 4), APInt(4, 0x00)); 3101 EXPECT_EQ(APIntOps::ScaleBitMask(APInt(8, 0xFF), 4), APInt(4, 0x0F)); 3102 EXPECT_EQ(APIntOps::ScaleBitMask(APInt(8, 0xE4), 4), APInt(4, 0x0E)); 3103 3104 EXPECT_EQ(APIntOps::ScaleBitMask(APInt(8, 0x00), 8), APInt(8, 0x00)); 3105 3106 EXPECT_EQ(APIntOps::ScaleBitMask(APInt::getNullValue(1024), 4096), 3107 APInt::getNullValue(4096)); 3108 EXPECT_EQ(APIntOps::ScaleBitMask(APInt::getAllOnes(4096), 256), 3109 APInt::getAllOnes(256)); 3110 EXPECT_EQ(APIntOps::ScaleBitMask(APInt::getOneBitSet(4096, 32), 256), 3111 APInt::getOneBitSet(256, 2)); 3112 } 3113 3114 } // end anonymous namespace 3115