gcc/testsuite/gcc.target/i386/avx2-i64gatherq-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_i64gatherpd (long long *src,
  10                      long long *s1,
  11                      long long *s2, long long *mask, int scale, long long *r)
  12 {
  13   long long i;
  14
  15   for (i = 0; i < 2; ++i)
  16     if ((mask[i] >> 63) & 1)
  17       r[i] = *(long long *) (((unsigned char *) s1) + s2[i] * scale);
  18     else
  19       r[i] = src[i];
  20 }
  21
  22 void static
  23 avx2_test (void)
  24 {
  25   long long i;
  26   union128i_q idx;
  27   union128i_q res, src, mask;
  28   long long s1[2], res_ref[2] = { 0, 0 };
  29
  30   for (i = 0; i < 2; ++i)
  31     {
  32       /* Set some stuff */
  33       s1[i] = 1983 * (i + 1) * (i + 2);
  34
  35       /* Set src as something different from s1 */
  36       src.a[i] = -s1[i];
  37
  38       /* Mask out evens */
  39       mask.a[i] = i % 2 ? 0 : -1;
  40
  41       /* About to gather in reverse order,
  42          divide by 2 to demonstrate scale */
  43       idx.a[i] = (16 - (i + 1) * 8) >> 1;
  44     }
  45
  46   res.x =
  47     _mm_mask_i64gather_epi64 (src.x, (long long int *) s1, idx.x, mask.x, 2);
  48
  49   compute_i64gatherpd (src.a, s1, idx.a, mask.a, 2, res_ref);
  50
  51   if (check_union128i_q (res, res_ref) != 0)
  52     abort ();
  53 }