APIntTest.cpp 83 KB

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