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