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