gcc/testsuite/gcc.target/i386/avx512bw-vpshufhw-2.c

   1 /* { dg-do run } */
   2 /* { dg-options "-O2 -mavx512bw -DAVX512BW" } */
   3 /* { dg-require-effective-target avx512bw } */
   4
   5 #include "avx512f-helper.h"
   6
   7 #define SIZE (AVX512F_LEN / 16)
   8 #include "avx512f-mask-type.h"
   9
  10 void
  11 CALC (short *s, unsigned char imm, short *r)
  12 {
  13   int i;
  14
  15   for (i = 0; i < SIZE / 8; i++)
  16     {
  17       r[8 * i] = s[8 * i];
  18       r[8 * i + 1] = s[8 * i + 1];
  19       r[8 * i + 2] = s[8 * i + 2];
  20       r[8 * i + 3] = s[8 * i + 3];
  21       r[8 * i + 4] = s[8 * i + (imm >> 0 & 3) + 4];
  22       r[8 * i + 5] = s[8 * i + (imm >> 2 & 3) + 4];
  23       r[8 * i + 6] = s[8 * i + (imm >> 4 & 3) + 4];
  24       r[8 * i + 7] = s[8 * i + (imm >> 6 & 3) + 4];
  25     }
  26 }
  27
  28 void
  29 TEST (void)
  30 {
  31   UNION_TYPE (AVX512F_LEN, i_w) s1, res1, res2, res3;
  32   short res_ref[SIZE];
  33   int i, sign = 1;
  34   MASK_TYPE mask = MASK_VALUE;
  35
  36   for (i = 0; i < SIZE; i++)
  37     {
  38       s1.a[i] = i * i * sign;
  39       res1.a[i] = DEFAULT_VALUE;
  40       res2.a[i] = DEFAULT_VALUE;
  41       res3.a[i] = DEFAULT_VALUE;
  42       sign = -sign;
  43     }
  44
  45   res1.x = INTRINSIC (_shufflehi_epi16) (s1.x, 0xec);
  46   res2.x =
  47     INTRINSIC (_mask_shufflehi_epi16) (res2.x, mask, s1.x, 0xec);
  48   res3.x = INTRINSIC (_maskz_shufflehi_epi16) (mask, s1.x, 0xec);
  49
  50   CALC (s1.a, 0xec, res_ref);
  51
  52   if (UNION_CHECK (AVX512F_LEN, i_w) (res1, res_ref))
  53     abort ();
  54
  55   MASK_MERGE (i_w) (res_ref, mask, SIZE);
  56   if (UNION_CHECK (AVX512F_LEN, i_w) (res2, res_ref))
  57     abort ();
  58
  59   MASK_ZERO (i_w) (res_ref, mask, SIZE);
  60   if (UNION_CHECK (AVX512F_LEN, i_w) (res3, res_ref))
  61     abort ();
  62 }