gcc/testsuite/gcc.target/i386/avx2-i64gatherq256-4.c

   1 /* { dg-do run } */
   2 /* { dg-require-effective-target avx2 } */
   3 /* { dg-options "-O2 -mavx2" } */
   4
   5 #include <string.h>
   6 #include "avx2-check.h"
   7
   8 static void
   9 compute_i64gatherq256 (long long *src,
  10                        long long *s1,
  11                        long long *s2,
  12                        long long *mask, int scale, long long *r)
  13 {
  14   long long i;
  15
  16   for (i = 0; i < 4; ++i)
  17     if ((mask[i] >> 63) & 1)
  18       r[i] = *(long long *) (((unsigned char *) s1) + s2[i] * scale);
  19     else
  20       r[i] = src[i];
  21 }
  22
  23 void static
  24 avx2_test (void)
  25 {
  26   long long i;
  27   union256i_q idx;
  28   union256i_q res, src, mask;
  29   long long s1[4], res_ref[4] = { 0 };
  30
  31   for (i = 0; i < 4; ++i)
  32     {
  33       /* Set some stuff */
  34       s1[i] = 1983 * (i + 1) * (i + 2);
  35
  36       /* Set src as something different from s1 */
  37       src.a[i] = -s1[i];
  38
  39       /* Mask out evens */
  40       ((long long *) mask.a)[i] = i % 2 ? 0 : -1;
  41
  42       /* About to gather in reverse order,
  43          divide by 2 to demonstrate scale */
  44       idx.a[i] = (16 - (i + 1) * 8) >> 1;
  45     }
  46
  47   res.x = _mm256_mask_i64gather_epi64 (src.x,
  48                                        (long long int *) s1,
  49                                        idx.x, mask.x, 2);
  50
  51   compute_i64gatherq256 (src.a, s1, idx.a, mask.a, 2, res_ref);
  52
  53   if (check_union256i_q (res, res_ref) != 0)
  54     abort ();
  55 }