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