gcc/testsuite/gcc.target/i386/avx512vbmi-vpmultishiftqb-2.c

   1 /* { dg-do run } */
   2 /* { dg-options "-O2 -mavx512vbmi" } */
   3 /* { dg-require-effective-target avx512vbmi } */
   4
   5 #define AVX512VBMI
   6
   7 #include "avx512f-helper.h"
   8
   9 #define SIZE (AVX512F_LEN / 8)
  10 #include "avx512f-mask-type.h"
  11
  12 void
  13 CALC (char *r, char *s1, char *s2)
  14 {
  15   int i, j, k;
  16   long long a, b, ctrl;
  17
  18   for (i = 0; i < SIZE / sizeof (long long); i++)
  19     {
  20       union
  21       {
  22         long long x;
  23         char a[sizeof(long long)];
  24       } src;
  25
  26       for (j = 0; j  < sizeof (long long); j++)
  27         src.a[j] = s2[i * sizeof (long long) + j];
  28       for (j = 0; j  < sizeof (long long); j++)
  29         {
  30           ctrl = s1[i * sizeof (long long) + j] & ((1 << sizeof (long long)) - 1);
  31           r[i * sizeof (long long) + j] = 0;
  32           for (k = 0; k < 8; k++)
  33             {
  34                r[i * sizeof (long long) + j] |= ((src.x >> ((ctrl + k) % (sizeof (long long) * 8))) & 1) << k;
  35             }
  36         }
  37     }
  38 }
  39
  40 void
  41 TEST (void)
  42 {
  43   UNION_TYPE (AVX512F_LEN, i_b) src1, src2, dst1, dst2, dst3;
  44   char dst_ref[SIZE];
  45   int i;
  46   MASK_TYPE mask = MASK_VALUE;
  47
  48   for (i = 0; i < SIZE; i++)
  49     {
  50       src1.a[i] = 15 + 3467 * i;
  51       src2.a[i] = 9217 + i;
  52       dst2.a[i] = DEFAULT_VALUE;
  53     }
  54
  55   CALC (dst_ref, src1.a, src2.a);
  56   dst1.x = INTRINSIC (_multishift_epi64_epi8) (src1.x, src2.x);
  57   dst2.x = INTRINSIC (_mask_multishift_epi64_epi8) (dst2.x, mask, src1.x, src2.x);
  58   dst3.x = INTRINSIC (_maskz_multishift_epi64_epi8) (mask, src1.x, src2.x);
  59
  60   if (UNION_CHECK (AVX512F_LEN, i_b) (dst1, dst_ref))
  61     abort ();
  62
  63   MASK_MERGE (i_b) (dst_ref, mask, SIZE);
  64   if (UNION_CHECK (AVX512F_LEN, i_b) (dst2, dst_ref))
  65     abort ();
  66
  67   MASK_ZERO (i_b) (dst_ref, mask, SIZE);
  68   if (UNION_CHECK (AVX512F_LEN, i_b) (dst3, dst_ref))
  69     abort ();
  70 }