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[16], res_ref[4] = { 0 };
      30    long long *s1_ptr = s1 + 8;
      31  
      32    for (i = 0; i < ARRAY_SIZE (s1); i++)
      33      {
      34        /* Set some stuff */
      35        s1[i] = 1983 * (i + 1) * (i + 2);
      36      }
      37  
      38    for (i = 0; i < 4; ++i)
      39      {
      40        /* Set src as something different from s1 */
      41        src.a[i] = -s1_ptr[i];
      42  
      43        /* Mask out evens */
      44        ((long long *) mask.a)[i] = i % 2 ? 0 : -1;
      45  
      46        /* About to gather in reverse order,
      47           divide by 2 to demonstrate scale */
      48        idx.a[i] = (16 - (i + 1) * 8) >> 1;
      49      }
      50  
      51    res.x = _mm256_mask_i64gather_epi64 (src.x,
      52  				       (long long int *) s1_ptr,
      53  				       idx.x, mask.x, 2);
      54  
      55    compute_i64gatherq256 (src.a, s1_ptr, idx.a, mask.a, 2, res_ref);
      56  
      57    if (check_union256i_q (res, res_ref) != 0)
      58      abort ();
      59  }