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#include "string.h"
12
13static void
14CALC (int *e, UNION_TYPE (AVX512F_LEN, i_d) s1,
15      UNION_TYPE (AVX512F_LEN, i_d) s2, int imm)
16{
17  int i, offset, selector;
18  int *source;
19  for (i = 0; i < SIZE / 4; i++)
20    {
21
22#if AVX512F_LEN == 512
23      selector = (imm >> i * 2) & 0x3;
24#else
25      selector = (imm >> i) & 0x1;
26#endif
27
28      offset = i * 4;
29      source = i * 4 * 32 < AVX512F_LEN / 2 ? s1.a : s2.a;
30      memcpy (e + offset, source + selector * 4, 16);
31    }
32}
33
34void
35TEST (void)
36{
37  UNION_TYPE (AVX512F_LEN, i_d) u1, u2, u3, s1, s2;
38  MASK_TYPE mask = MASK_VALUE;
39  int e[SIZE];
40  int i;
41#if AVX512F_LEN == 512
42  int imm = 203;
43#else
44  int imm = 1;
45#endif
46
47  for (i = 0; i < SIZE; i++)
48    {
49      s1.a[i] = 1.2 / (i + 0.378);
50      s2.a[i] = 91.02 / (i + 4.3578);
51      u1.a[i] = DEFAULT_VALUE;
52      u2.a[i] = DEFAULT_VALUE;
53      u3.a[i] = DEFAULT_VALUE;
54    }
55
56  u1.x = INTRINSIC (_shuffle_i32x4) (s1.x, s2.x, imm);
57  u2.x = INTRINSIC (_mask_shuffle_i32x4) (u2.x, mask, s1.x, s2.x, imm);
58  u3.x = INTRINSIC (_maskz_shuffle_i32x4) (mask, s1.x, s2.x, imm);
59
60  CALC (e, s1, s2, imm);
61
62  if (UNION_CHECK (AVX512F_LEN, i_d) (u1, e))
63    abort ();
64
65  MASK_MERGE (i_d) (e, mask, SIZE);
66  if (UNION_CHECK (AVX512F_LEN, i_d) (u2, e))
67    abort ();
68
69  MASK_ZERO (i_d) (e, mask, SIZE);
70  if (UNION_CHECK (AVX512F_LEN, i_d) (u3, e))
71    abort ();
72}
73