unittests/Support/BlockFrequencyTest.cpp

   1 #include "llvm/Support/BlockFrequency.h"
   2 #include "llvm/Support/BranchProbability.h"
   3 #include "llvm/Support/DataTypes.h"
   4 #include "gtest/gtest.h"
   5 #include <climits>
   6
   7 using namespace llvm;
   8
   9 namespace {
  10
  11 TEST(BlockFrequencyTest, OneToZero) {
  12   BlockFrequency Freq(1);
  13   BranchProbability Prob(UINT32_MAX - 1, UINT32_MAX);
  14   Freq *= Prob;
  15   EXPECT_EQ(Freq.getFrequency(), 0u);
  16
  17   Freq = BlockFrequency(1);
  18   uint32_t Remainder = Freq.scale(Prob);
  19   EXPECT_EQ(Freq.getFrequency(), 0u);
  20   EXPECT_EQ(Remainder, UINT32_MAX - 1);
  21 }
  22
  23 TEST(BlockFrequencyTest, OneToOne) {
  24   BlockFrequency Freq(1);
  25   BranchProbability Prob(UINT32_MAX, UINT32_MAX);
  26   Freq *= Prob;
  27   EXPECT_EQ(Freq.getFrequency(), 1u);
  28
  29   Freq = BlockFrequency(1);
  30   uint32_t Remainder = Freq.scale(Prob);
  31   EXPECT_EQ(Freq.getFrequency(), 1u);
  32   EXPECT_EQ(Remainder, 0u);
  33 }
  34
  35 TEST(BlockFrequencyTest, ThreeToOne) {
  36   BlockFrequency Freq(3);
  37   BranchProbability Prob(3000000, 9000000);
  38   Freq *= Prob;
  39   EXPECT_EQ(Freq.getFrequency(), 1u);
  40
  41   Freq = BlockFrequency(3);
  42   uint32_t Remainder = Freq.scale(Prob);
  43   EXPECT_EQ(Freq.getFrequency(), 1u);
  44   EXPECT_EQ(Remainder, 0u);
  45 }
  46
  47 TEST(BlockFrequencyTest, MaxToHalfMax) {
  48   BlockFrequency Freq(UINT64_MAX);
  49   BranchProbability Prob(UINT32_MAX / 2, UINT32_MAX);
  50   Freq *= Prob;
  51   EXPECT_EQ(Freq.getFrequency(), 9223372034707292159ULL);
  52
  53   Freq = BlockFrequency(UINT64_MAX);
  54   uint32_t Remainder = Freq.scale(Prob);
  55   EXPECT_EQ(Freq.getFrequency(), 9223372034707292159ULL);
  56   EXPECT_EQ(Remainder, 0u);
  57 }
  58
  59 TEST(BlockFrequencyTest, BigToBig) {
  60   const uint64_t Big = 387246523487234346LL;
  61   const uint32_t P = 123456789;
  62   BlockFrequency Freq(Big);
  63   BranchProbability Prob(P, P);
  64   Freq *= Prob;
  65   EXPECT_EQ(Freq.getFrequency(), Big);
  66
  67   Freq = BlockFrequency(Big);
  68   uint32_t Remainder = Freq.scale(Prob);
  69   EXPECT_EQ(Freq.getFrequency(), Big);
  70   EXPECT_EQ(Remainder, 0u);
  71 }
  72
  73 TEST(BlockFrequencyTest, MaxToMax) {
  74   BlockFrequency Freq(UINT64_MAX);
  75   BranchProbability Prob(UINT32_MAX, UINT32_MAX);
  76   Freq *= Prob;
  77   EXPECT_EQ(Freq.getFrequency(), UINT64_MAX);
  78
  79   // This additionally makes sure if we have a value equal to our saturating
  80   // value, we do not signal saturation if the result equals said value, but
  81   // saturating does not occur.
  82   Freq = BlockFrequency(UINT64_MAX);
  83   uint32_t Remainder = Freq.scale(Prob);
  84   EXPECT_EQ(Freq.getFrequency(), UINT64_MAX);
  85   EXPECT_EQ(Remainder, 0u);
  86 }
  87
  88 TEST(BlockFrequencyTest, ScaleResultRemainderTest) {
  89   struct {
  90     uint64_t Freq;
  91     uint32_t Prob[2];
  92     uint64_t ExpectedFreq;
  93     uint32_t ExpectedRemainder;
  94   } Tests[80] = {
  95     // Data for scaling that results in <= 64 bit division.
  96     { 0x1423e2a50ULL, { 0x64819521, 0x7765dd13 }, 0x10f418889ULL, 0x92b9d25 },
  97     { 0x35ef14ceULL, { 0x28ade3c7, 0x304532ae }, 0x2d73c33aULL, 0x2c0fd0b6 },
  98     { 0xd03dbfbe24ULL, { 0x790079, 0xe419f3 }, 0x6e776fc1fdULL, 0x4a06dd },
  99     { 0x21d67410bULL, { 0x302a9dc2, 0x3ddb4442 }, 0x1a5948fd6ULL, 0x265d1c2a },
 100     { 0x8664aeadULL, { 0x3d523513, 0x403523b1 }, 0x805a04cfULL, 0x324c27b8 },
 101     { 0x201db0cf4ULL, { 0x35112a7b, 0x79fc0c74 }, 0xdf8b07f6ULL, 0x490c1dc4 },
 102     { 0x13f1e4430aULL, { 0x21c92bf, 0x21e63aae }, 0x13e0cba15ULL, 0x1df47c30 },
 103     { 0x16c83229ULL, { 0x3793f66f, 0x53180dea }, 0xf3ce7b6ULL, 0x1d0c1b6b },
 104     { 0xc62415be8ULL, { 0x9cc4a63, 0x4327ae9b }, 0x1ce8b71caULL, 0x3f2c696a },
 105     { 0x6fac5e434ULL, { 0xe5f9170, 0x1115e10b }, 0x5df23dd4cULL, 0x4dafc7c },
 106     { 0x1929375f2ULL, { 0x3a851375, 0x76c08456 }, 0xc662b082ULL, 0x343589ee },
 107     { 0x243c89db6ULL, { 0x354ebfc0, 0x450ef197 }, 0x1bf8c1661ULL, 0x4948e49 },
 108     { 0x310e9b31aULL, { 0x1b1b8acf, 0x2d3629f0 }, 0x1d69c93f9ULL, 0x73e3b96 },
 109     { 0xa1fae921dULL, { 0xa7a098c, 0x10469f44 }, 0x684413d6cULL, 0x86a882c },
 110     { 0xc1582d957ULL, { 0x498e061, 0x59856bc }, 0x9edc5f4e7ULL, 0x29b0653 },
 111     { 0x57cfee75ULL, { 0x1d061dc3, 0x7c8bfc17 }, 0x1476a220ULL, 0x2383d33f },
 112     { 0x139220080ULL, { 0x294a6c71, 0x2a2b07c9 }, 0x1329e1c76ULL, 0x7aa5da },
 113     { 0x1665d353cULL, { 0x7080db5, 0xde0d75c }, 0xb590d9fbULL, 0x7ba8c38 },
 114     { 0xe8f14541ULL, { 0x5188e8b2, 0x736527ef }, 0xa4971be5ULL, 0x6b612167 },
 115     { 0x2f4775f29ULL, { 0x254ef0fe, 0x435fcf50 }, 0x1a2e449c1ULL, 0x28bbf5e },
 116     { 0x27b85d8d7ULL, { 0x304c8220, 0x5de678f2 }, 0x146e3bef9ULL, 0x4b27097e },
 117     { 0x1d362e36bULL, { 0x36c85b12, 0x37a66f55 }, 0x1cc19b8e6ULL, 0x688e828 },
 118     { 0x155fd48c7ULL, { 0xf5894d, 0x1256108 }, 0x11e383602ULL, 0x111f0cb },
 119     { 0xb5db2d15ULL, { 0x39bb26c5, 0x5bdcda3e }, 0x72499259ULL, 0x59c4939b },
 120     { 0x153990298ULL, { 0x48921c09, 0x706eb817 }, 0xdb3268e8ULL, 0x66bb8a80 },
 121     { 0x28a7c3ed7ULL, { 0x1f776fd7, 0x349f7a70 }, 0x184f73ae1ULL, 0x28910321 },
 122     { 0x724dbeabULL, { 0x1bd149f5, 0x253a085e }, 0x5569c0b3ULL, 0xff8e2ed },
 123     { 0xd8f0c513ULL, { 0x18c8cc4c, 0x1b72bad0 }, 0xc3e30643ULL, 0xd85e134 },
 124     { 0x17ce3dcbULL, { 0x1e4c6260, 0x233b359e }, 0x1478f4afULL, 0x49ea31e },
 125     { 0x1ce036ce0ULL, { 0x29e3c8af, 0x5318dd4a }, 0xe8e76196ULL, 0x11d5b9c4 },
 126     { 0x1473ae2aULL, { 0x29b897ba, 0x2be29378 }, 0x13718185ULL, 0x6f93b2c },
 127     { 0x1dd41aa68ULL, { 0x3d0a4441, 0x5a0e8f12 }, 0x1437b6bbfULL, 0x54b09ffa },
 128     { 0x1b49e4a53ULL, { 0x3430c1fe, 0x5a204aed }, 0xfcd6852fULL, 0x15ad6ed7 },
 129     { 0x217941b19ULL, { 0x12ced2bd, 0x21b68310 }, 0x12aca65b1ULL, 0x1b2a9565 },
 130     { 0xac6a4dc8ULL, { 0x3ed68da8, 0x6fdca34c }, 0x60da926dULL, 0x22ff53e4 },
 131     { 0x1c503a4e7ULL, { 0xfcbbd32, 0x11e48d17 }, 0x18fec7d38ULL, 0xa8aa816 },
 132     { 0x1c885855ULL, { 0x213e919d, 0x25941897 }, 0x193de743ULL, 0x4ea09c },
 133     { 0x29b9c168eULL, { 0x2b644aea, 0x45725ee7 }, 0x1a122e5d5ULL, 0xbee1099 },
 134     { 0x806a33f2ULL, { 0x30a80a23, 0x5063733a }, 0x4db9a264ULL, 0x1eaed76e },
 135     { 0x282afc96bULL, { 0x143ae554, 0x1a9863ff }, 0x1e8de5204ULL, 0x158d9020 },
 136     // Data for scaling that results in > 64 bit division.
 137     { 0x23ca5f2f672ca41cULL, { 0xecbc641, 0x111373f7 }, 0x1f0301e5e8295ab5ULL, 0xf627f79 },
 138     { 0x5e4f2468142265e3ULL, { 0x1ddf5837, 0x32189233 }, 0x383ca7ba9fdd2c8cULL, 0x1c8f33e1 },
 139     { 0x277a1a6f6b266bf6ULL, { 0x415d81a8, 0x61eb5e1e }, 0x1a5a3e1d41b30c0fULL, 0x29cde3ae },
 140     { 0x1bdbb49a237035cbULL, { 0xea5bf17, 0x1d25ffb3 }, 0xdffc51c53d44b93ULL, 0x5170574 },
 141     { 0x2bce6d29b64fb8ULL, { 0x3bfd5631, 0x7525c9bb }, 0x166ebedda7ac57ULL, 0x3026dfab },
 142     { 0x3a02116103df5013ULL, { 0x2ee18a83, 0x3299aea8 }, 0x35be8922ab1e2a84ULL, 0x298d9919 },
 143     { 0x7b5762390799b18cULL, { 0x12f8e5b9, 0x2563bcd4 }, 0x3e960077aca01209ULL, 0x93afeb8 },
 144     { 0x69cfd72537021579ULL, { 0x4c35f468, 0x6a40feee }, 0x4be4cb3848be98a3ULL, 0x4ff96b9e },
 145     { 0x49dfdf835120f1c1ULL, { 0x8cb3759, 0x559eb891 }, 0x79663f7120edadeULL, 0x51b1fb5b },
 146     { 0x74b5be5c27676381ULL, { 0x47e4c5e0, 0x7c7b19ff }, 0x4367d2dff36a1028ULL, 0x7a7b5608 },
 147     { 0x4f50f97075e7f431ULL, { 0x9a50a17, 0x11cd1185 }, 0x2af952b34c032df4ULL, 0xfddc6a3 },
 148     { 0x2f8b0d712e393be4ULL, { 0x1487e386, 0x15aa356e }, 0x2d0df36478a776aaULL, 0x14e2564c },
 149     { 0x224c1c75999d3deULL, { 0x3b2df0ea, 0x4523b100 }, 0x1d5b481d145f08aULL, 0x15145eec },
 150     { 0x2bcbcea22a399a76ULL, { 0x28b58212, 0x48dd013e }, 0x187814d084c47cabULL, 0x3a38ebe2 },
 151     { 0x1dbfca91257cb2d1ULL, { 0x1a8c04d9, 0x5e92502c }, 0x859cf7d00f77545ULL, 0x7431f4d },
 152     { 0x7f20039b57cda935ULL, { 0xeccf651, 0x323f476e }, 0x25720cd976461a77ULL, 0x202817a3 },
 153     { 0x40512c6a586aa087ULL, { 0x113b0423, 0x398c9eab }, 0x1341c03de8696a7eULL, 0x1e27284b },
 154     { 0x63d802693f050a11ULL, { 0xf50cdd6, 0xfce2a44 }, 0x60c0177bb5e46846ULL, 0xf7ad89e },
 155     { 0x2d956b422838de77ULL, { 0xb2d345b, 0x1321e557 }, 0x1aa0ed16b6aa5319ULL, 0xfe1a5ce },
 156     { 0x5a1cdf0c1657bc91ULL, { 0x1d77bb0c, 0x1f991ff1 }, 0x54097ee94ff87560ULL, 0x11c4a26c },
 157     { 0x3801b26d7e00176bULL, { 0xeed25da, 0x1a819d8b }, 0x1f89e96a3a639526ULL, 0xcd51e7c },
 158     { 0x37655e74338e1e45ULL, { 0x300e170a, 0x5a1595fe }, 0x1d8cfb55fddc0441ULL, 0x3df05434 },
 159     { 0x7b38703f2a84e6ULL, { 0x66d9053, 0xc79b6b9 }, 0x3f7d4c91774094ULL, 0x26d939e },
 160     { 0x2245063c0acb3215ULL, { 0x30ce2f5b, 0x610e7271 }, 0x113b916468389235ULL, 0x1b588512 },
 161     { 0x6bc195877b7b8a7eULL, { 0x392004aa, 0x4a24e60c }, 0x530594fb17db6ba5ULL, 0x35c0a5f0 },
 162     { 0x40a3fde23c7b43dbULL, { 0x4e712195, 0x6553e56e }, 0x320a799bc76a466aULL, 0x5e23a5eb },
 163     { 0x1d3dfc2866fbccbaULL, { 0x5075b517, 0x5fc42245 }, 0x18917f0061595bc3ULL, 0x3fcf4527 },
 164     { 0x19aeb14045a61121ULL, { 0x1bf6edec, 0x707e2f4b }, 0x6626672a070bcc7ULL, 0x3607801f },
 165     { 0x44ff90486c531e9fULL, { 0x66598a, 0x8a90dc }, 0x32f6f2b0525199b0ULL, 0x5ab576 },
 166     { 0x3f3e7121092c5bcbULL, { 0x1c754df7, 0x5951a1b9 }, 0x14267f50b7ef375dULL, 0x221220a8 },
 167     { 0x60e2dafb7e50a67eULL, { 0x4d96c66e, 0x65bd878d }, 0x49e31715ac393f8bULL, 0x4e97b195 },
 168     { 0x656286667e0e6e29ULL, { 0x9d971a2, 0xacda23b }, 0x5c6ee315ead6cb4fULL, 0x516f5bd },
 169     { 0x1114e0974255d507ULL, { 0x1c693, 0x2d6ff }, 0xaae42e4b35f6e60ULL, 0x8b65 },
 170     { 0x508c8baf3a70ff5aULL, { 0x3b26b779, 0x6ad78745 }, 0x2c98387636c4b365ULL, 0x11dc6a51 },
 171     { 0x5b47bc666bf1f9cfULL, { 0x10a87ed6, 0x187d358a }, 0x3e1767155848368bULL, 0xfb871c },
 172     { 0x50954e3744460395ULL, { 0x7a42263, 0xcdaa048 }, 0x2fe739f0aee1fee1ULL, 0xb8add57 },
 173     { 0x20020b406550dd8fULL, { 0x3318539, 0x42eead0 }, 0x186f326325fa346bULL, 0x10d3ae7 },
 174     { 0x5bcb0b872439ffd5ULL, { 0x6f61fb2, 0x9af7344 }, 0x41fa1e3bec3c1b30ULL, 0x4fee45a },
 175     { 0x7a670f365db87a53ULL, { 0x417e102, 0x3bb54c67 }, 0x8642a558304fd9eULL, 0x3b65f514 },
 176     { 0x1ef0db1e7bab1cd0ULL, { 0x2b60cf38, 0x4188f78f }, 0x147ae0d6226b2ee6ULL, 0x336b6106 }
 177   };
 178
 179   for (unsigned i = 0; i < 80; i++) {
 180     BlockFrequency Freq(Tests[i].Freq);
 181     uint32_t Remainder = Freq.scale(BranchProbability(Tests[i].Prob[0],
 182                                                       Tests[i].Prob[1]));
 183     EXPECT_EQ(Tests[i].ExpectedFreq, Freq.getFrequency());
 184     EXPECT_EQ(Tests[i].ExpectedRemainder, Remainder);
 185   }
 186 }
 187
 188 TEST(BlockFrequency, Divide) {
 189   BlockFrequency Freq(0x3333333333333333ULL);
 190   Freq /= BranchProbability(1, 2);
 191   EXPECT_EQ(Freq.getFrequency(), 0x6666666666666666ULL);
 192 }
 193
 194 TEST(BlockFrequencyTest, Saturate) {
 195   BlockFrequency Freq(0x3333333333333333ULL);
 196   Freq /= BranchProbability(100, 300);
 197   EXPECT_EQ(Freq.getFrequency(), 0x9999999999999999ULL);
 198   Freq /= BranchProbability(1, 2);
 199   EXPECT_EQ(Freq.getFrequency(), UINT64_MAX);
 200
 201   Freq = 0x1000000000000000ULL;
 202   Freq /= BranchProbability(10000, 160000);
 203   EXPECT_EQ(Freq.getFrequency(), UINT64_MAX);
 204
 205   // Try to cheat the multiplication overflow check.
 206   Freq = 0x00000001f0000001ull;
 207   Freq /= BranchProbability(1000, 0xf000000f);
 208   EXPECT_EQ(33506781356485509ULL, Freq.getFrequency());
 209 }
 210
 211 TEST(BlockFrequencyTest, ProbabilityCompare) {
 212   BranchProbability A(4, 5);
 213   BranchProbability B(4U << 29, 5U << 29);
 214   BranchProbability C(3, 4);
 215
 216   EXPECT_TRUE(A == B);
 217   EXPECT_FALSE(A != B);
 218   EXPECT_FALSE(A < B);
 219   EXPECT_FALSE(A > B);
 220   EXPECT_TRUE(A <= B);
 221   EXPECT_TRUE(A >= B);
 222
 223   EXPECT_FALSE(B == C);
 224   EXPECT_TRUE(B != C);
 225   EXPECT_FALSE(B < C);
 226   EXPECT_TRUE(B > C);
 227   EXPECT_FALSE(B <= C);
 228   EXPECT_TRUE(B >= C);
 229
 230   BranchProbability BigZero(0, UINT32_MAX);
 231   BranchProbability BigOne(UINT32_MAX, UINT32_MAX);
 232   EXPECT_FALSE(BigZero == BigOne);
 233   EXPECT_TRUE(BigZero != BigOne);
 234   EXPECT_TRUE(BigZero < BigOne);
 235   EXPECT_FALSE(BigZero > BigOne);
 236   EXPECT_TRUE(BigZero <= BigOne);
 237   EXPECT_FALSE(BigZero >= BigOne);
 238 }
 239
 240 }