1 /* { dg-do run } */
2 /* { dg-options "-O2 -mavx512f" } */
3 /* { dg-require-effective-target avx512f } */
4
5 #define AVX512F
6
7 #include "avx512f-helper.h"
8
9 #define SIZE (AVX512F_LEN / 32)
10 #include "avx512f-mask-type.h"
11
12 static void
CALC(float * s,float * r,MASK_TYPE mask)13 CALC (float *s, float *r, MASK_TYPE mask)
14 {
15 int i, k;
16
17 for (i = 0, k = 0; i < SIZE; i++)
18 {
19 if (mask & (1 << i))
20 r[i] = s[k++];
21 }
22 }
23
24 void
TEST(void)25 TEST (void)
26 {
27 UNION_TYPE (AVX512F_LEN, ) s1, res2, res3, res4, res5;
28 MASK_TYPE mask = MASK_VALUE;
29 float s2[SIZE];
30 float res_ref1[SIZE];
31 float res_ref2[SIZE];
32 float res_ref3[SIZE];
33 int i, sign = 1;
34
35 for (i = 0; i < SIZE; i++)
36 {
37 s1.a[i] = 123.456 * (i + 200) * sign;
38 s2[i] = 789.012 * (i + 300) * sign;
39 res2.a[i] = DEFAULT_VALUE;
40 res4.a[i] = DEFAULT_VALUE;
41 sign = -sign;
42 }
43
44 res2.x = INTRINSIC (_mask_expand_ps) (res2.x, mask, s1.x);
45 res3.x = INTRINSIC (_maskz_expand_ps) (mask, s1.x);
46 res4.x = INTRINSIC (_mask_expandloadu_ps) (res4.x, mask, s2);
47 res5.x = INTRINSIC (_maskz_expandloadu_ps) (mask, s2);
48
49 CALC (s1.a, res_ref1, MASK_ALL_ONES);
50 CALC (s1.a, res_ref2, mask);
51 CALC (s2, res_ref3, mask);
52
53 MASK_MERGE () (res_ref2, mask, SIZE);
54 if (UNION_CHECK (AVX512F_LEN, ) (res2, res_ref2))
55 abort ();
56
57 MASK_ZERO () (res_ref2, mask, SIZE);
58 if (UNION_CHECK (AVX512F_LEN, ) (res3, res_ref2))
59 abort ();
60
61 MASK_MERGE () (res_ref3, mask, SIZE);
62 if (UNION_CHECK (AVX512F_LEN, ) (res4, res_ref3))
63 abort ();
64
65 MASK_ZERO () (res_ref3, mask, SIZE);
66 if (UNION_CHECK (AVX512F_LEN, ) (res5, res_ref3))
67 abort ();
68 }
69