gcc/testsuite/gcc.target/i386/avx512bw-vpacksswb-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 DST_SIZE (AVX512F_LEN / 8)
   8 #define SRC_SIZE (AVX512F_LEN / 16)
   9
  10 #include "limits.h"
  11
  12 #include "avx512f-mask-type.h"
  13
  14 static char
  15 EVAL(short_to_char, AVX512F_LEN,) (short iVal)
  16 {
  17   char sVal;
  18
  19   if (iVal < CHAR_MIN)
  20     sVal = CHAR_MIN;
  21   else if (iVal > CHAR_MAX)
  22     sVal = CHAR_MAX;
  23   else
  24     sVal = iVal;
  25
  26   return sVal;
  27 }
  28
  29 void
  30 CALC (short *src1, short *src2, char *dst)
  31 {
  32   int i;
  33   short *ptr;
  34
  35   for (i = 0; i < DST_SIZE; i++)
  36     {
  37       ptr = (i / 8) % 2 ? src2 : src1;
  38       dst[i] = EVAL(short_to_char, AVX512F_LEN,) (ptr[i % 8 + (i / 16) * 8]);
  39     }
  40 }
  41
  42 void
  43 TEST (void)
  44 {
  45   UNION_TYPE (AVX512F_LEN, i_w) s1, s2;
  46   UNION_TYPE (AVX512F_LEN, i_b) res1, res2, res3;
  47   MASK_TYPE mask = MASK_VALUE;
  48   char dst_ref[DST_SIZE];
  49   int i;
  50
  51   for (i = 0; i < DST_SIZE; i++)
  52     res2.a[i] = DEFAULT_VALUE;
  53
  54   for (i = 0; i < SRC_SIZE; i++)
  55     {
  56       s1.a[i] = i + 10;
  57       s2.a[i] = i + 15;
  58     }
  59
  60   res1.x = INTRINSIC (_packs_epi16) (s1.x, s2.x);
  61   res2.x = INTRINSIC (_mask_packs_epi16) (res2.x, mask, s1.x, s2.x);
  62   res3.x = INTRINSIC (_maskz_packs_epi16) (mask, s1.x, s2.x);
  63
  64   CALC (s1.a, s2.a, dst_ref);
  65
  66   if (UNION_CHECK (AVX512F_LEN, i_b) (res1, dst_ref))
  67     abort ();
  68
  69   MASK_MERGE (i_b) (dst_ref, mask, DST_SIZE);
  70   if (UNION_CHECK (AVX512F_LEN, i_b) (res2, dst_ref))
  71     abort ();
  72
  73   MASK_ZERO (i_b) (dst_ref, mask, DST_SIZE);
  74   if (UNION_CHECK (AVX512F_LEN, i_b) (res3, dst_ref))
  75     abort ();
  76
  77 }