1 /* { dg-do run } */
2 /* { dg-options "-O2 -mavx512vbmi" } */
3 /* { dg-require-effective-target avx512vbmi } */
4 
5 #define AVX512VBMI
6 
7 #include "avx512f-helper.h"
8 
9 #define SIZE (AVX512F_LEN / 8)
10 #include <math.h>
11 #include <limits.h>
12 #include <float.h>
13 #include "avx512f-mask-type.h"
14 
15 #define NUM 32
16 
17 void
CALC(char * dst,char * src1,char * ind,char * src2)18 CALC (char *dst, char *src1, char *ind, char *src2)
19 {
20   int i;
21 
22   for (i = 0; i < SIZE; i++)
23     {
24       unsigned long long offset = ind[i] & (SIZE - 1);
25       unsigned long long cond = ind[i] & SIZE;
26 
27       dst[i] = cond ? src2[offset] : src1[offset];
28     }
29 }
30 
31 void
TEST(void)32 TEST (void)
33 {
34   int i, j;
35   UNION_TYPE (AVX512F_LEN, i_b) s1, s2, res, ind;
36   char res_ref[SIZE];
37 
38   MASK_TYPE mask = MASK_VALUE;
39 
40   for (i = 0; i < NUM; i++)
41     {
42       for (j = 0; j < SIZE; j++)
43 	{
44 	  ind.a[j] = DEFAULT_VALUE;
45 	  s1.a[j] = i * 2 * j + 1;
46 	  s2.a[j] = i * 2 * j;
47 
48 	  res.a[j] = DEFAULT_VALUE;
49 	}
50 
51       CALC (res_ref, s1.a, ind.a, s2.a);
52 
53       res.x =
54 	INTRINSIC (_mask2_permutex2var_epi8) (s1.x, ind.x, mask,
55 					       s2.x);
56 
57       MASK_MERGE (i_b) (res_ref, mask, SIZE);
58       if (UNION_CHECK (AVX512F_LEN, i_b) (res, res_ref))
59 	abort ();
60     }
61 }
62