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

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