gcc/testsuite/gcc.target/i386/bmi2-pext64-1.c

   1 /* { dg-do run { target { bmi2 && { ! ia32 } } } } */
   2 /* { dg-options "-mbmi2 -O2" } */
   3
   4 #include <x86intrin.h>
   5 #include "bmi2-check.h"
   6
   7 __attribute__((noinline))
   8 unsigned long long
   9 calc_pext_u64 (unsigned long long a, unsigned long long mask)
  10 {
  11   unsigned long long res = 0;
  12   int i, k = 0;
  13
  14   for (i = 0; i < 64; ++i)
  15     if (mask & (1LL << i)) {
  16       res |= ((a & (1LL << i)) >> i) << k;
  17       ++k;
  18     }
  19
  20   return res;
  21 }
  22
  23 static void
  24 bmi2_test ()
  25 {
  26   unsigned long long i;
  27   unsigned long long src = 0xce7acce7acce7ac;
  28   unsigned long long res, res_ref;
  29
  30   for (i = 0; i < 5; ++i) {
  31     src = src * (i + 1);
  32
  33     res_ref = calc_pext_u64 (src, ~(i * 3));
  34     res = _pext_u64 (src, ~(i * 3));
  35
  36     if (res != res_ref)
  37       abort();
  38   }
  39 }