gcc/testsuite/gcc.target/i386/avx512bw-vpalignr-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 #include <string.h>
   8
   9 #define SIZE (AVX512F_LEN / 8)
  10 #include "avx512f-mask-type.h"
  11
  12 #define N 0x3
  13
  14 void
  15 CALC (char *src1, char *src2, char * dst)
  16 {
  17   /* result for EVEX.U1.512 version consists from 4 result block, each of them
  18    * has length of 128 bits. */
  19   unsigned block_len = 16;
  20   unsigned double_block_len = 32;
  21   unsigned shift = 0;
  22   char buf[double_block_len];
  23   char *bout = dst;
  24   int bits, i;
  25
  26   for (bits = 0; bits < AVX512F_LEN; bits += 128)
  27     {
  28       memcpy (&buf[0], src2 + shift, block_len);
  29       memcpy (&buf[block_len], src1 + shift, block_len);
  30
  31       for (i = 0; i < block_len; i++)
  32         /* shift counts larger than 32 produces zero result. */
  33         if (N >= 32 || N + i >= 32)
  34           bout[i] = 0;
  35         else
  36           bout[i] = buf[N + i];
  37
  38       shift += block_len;
  39       bout += block_len;
  40     }
  41 }
  42
  43 void
  44 TEST (void)
  45 {
  46   UNION_TYPE (AVX512F_LEN, i_b) s1, s2, res1, res2, res3;
  47   MASK_TYPE mask = MASK_VALUE;
  48   char res_ref[SIZE];
  49   int i;
  50
  51   for (i = 0; i < SIZE; i++)
  52     {
  53       s1.a[i] = i;
  54       s2.a[i] = i * 2;
  55       res2.a[i] = DEFAULT_VALUE;
  56     }
  57
  58   res1.x = INTRINSIC (_alignr_epi8) (s1.x, s2.x, N);
  59   res2.x = INTRINSIC (_mask_alignr_epi8) (res2.x, mask, s1.x, s2.x, N);
  60   res3.x = INTRINSIC (_maskz_alignr_epi8) (mask, s1.x, s2.x, N);
  61
  62   CALC (s1.a, s2.a, res_ref);
  63
  64   if (UNION_CHECK (AVX512F_LEN, i_b) (res1, res_ref))
  65     abort ();
  66
  67   MASK_MERGE (i_b) (res_ref, mask, SIZE);
  68   if (UNION_CHECK (AVX512F_LEN, i_b) (res2, res_ref))
  69     abort ();
  70
  71   MASK_ZERO (i_b) (res_ref, mask, SIZE);
  72   if (UNION_CHECK (AVX512F_LEN, i_b) (res3, res_ref))
  73     abort ();
  74 }