compiler-rt/test/builtins/Unit/popcountti2_test.c

   1 // RUN: %clang_builtins %s %librt -o %t && %run %t
   2 // REQUIRES: librt_has_popcountti2
   3 // REQUIRES: int128
   4 //===-- popcountti2_test.c - Test __popcountti2 ----------------------------===//
   5 //
   6 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   7 // See https://llvm.org/LICENSE.txt for license information.
   8 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   9 //
  10 //===----------------------------------------------------------------------===//
  11 //
  12 // This file tests __popcountti2 for the compiler_rt library.
  13 //
  14 //===----------------------------------------------------------------------===//
  15
  16 #include "int_lib.h"
  17 #include <stdio.h>
  18 #include <stdlib.h>
  19
  20 #ifdef CRT_HAS_128BIT
  21
  22 // Returns: count of 1 bits
  23
  24 COMPILER_RT_ABI si_int __popcountti2(ti_int a);
  25
  26 int naive_popcount(ti_int a)
  27 {
  28     int r = 0;
  29     for (; a; a = (tu_int)a >> 1)
  30         r += a & 1;
  31     return r;
  32 }
  33
  34 int test__popcountti2(ti_int a)
  35 {
  36     si_int x = __popcountti2(a);
  37     si_int expected = naive_popcount(a);
  38     if (x != expected)
  39     {
  40         twords at;
  41         at.all = a;
  42         printf("error in __popcountti2(0x%.16llX%.16llX) = %d, expected %d\n",
  43                at.s.high, at.s.low, x, expected);
  44     }
  45     return x != expected;
  46 }
  47
  48 char assumption_1[sizeof(ti_int) == 2*sizeof(di_int)] = {0};
  49 char assumption_2[sizeof(di_int)*CHAR_BIT == 64] = {0};
  50
  51 #endif
  52
  53 int main()
  54 {
  55 #ifdef CRT_HAS_128BIT
  56     if (test__popcountti2(0))
  57         return 1;
  58     if (test__popcountti2(1))
  59         return 1;
  60     if (test__popcountti2(2))
  61         return 1;
  62     if (test__popcountti2(0xFFFFFFFFFFFFFFFDLL))
  63         return 1;
  64     if (test__popcountti2(0xFFFFFFFFFFFFFFFELL))
  65         return 1;
  66     if (test__popcountti2(0xFFFFFFFFFFFFFFFFLL))
  67         return 1;
  68     if (test__popcountti2(make_ti(0xFFFFFFFFFFFFFFFFLL, 0xFFFFFFFFFFFFFFFDLL)))
  69         return 1;
  70     if (test__popcountti2(make_ti(0xFFFFFFFFFFFFFFFFLL, 0xFFFFFFFFFFFFFFFELL)))
  71         return 1;
  72     if (test__popcountti2(make_ti(0xFFFFFFFFFFFFFFFFLL, 0xFFFFFFFFFFFFFFFFLL)))
  73         return 1;
  74     int i;
  75     for (i = 0; i < 10000; ++i)
  76         if (test__popcountti2(((ti_int)rand() << 96) | ((ti_int)rand() << 64) |
  77                               ((ti_int)rand() << 32) | rand()))
  78             return 1;
  79
  80 #else
  81     printf("skipped\n");
  82 #endif
  83    return 0;
  84 }