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_i64gatherq256(long long * src,long long * s1,long long * s2,long long * mask,int scale,long long * r)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
avx2_test(void)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 }
56