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