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