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 #define SIZE_HALF (AVX512F_LEN_HALF / 16)
12 #include <limits.h>
13 
14 void
CALC(short * r,int * s,int mem)15 CALC (short *r, int *s, int mem)
16 {
17   int i;
18   int len = mem ? SIZE : SIZE_HALF;
19   for (i = 0; i < len; i++)
20     {
21       if (i < SIZE)
22 	{
23 	  if (s[i] < SHRT_MIN)
24 	    r[i] = SHRT_MIN;
25 	  else if (s[i] > SHRT_MAX)
26 	    r[i] = SHRT_MAX;
27 	  else
28 	    r[i] = s[i];
29 	}
30       else
31 	r[i] = 0;
32     }
33 }
34 
35 void
TEST(void)36 TEST (void)
37 {
38   int i, sign;
39   UNION_TYPE (AVX512F_LEN_HALF, i_w) res1, res2, res3;
40   short res4[SIZE_HALF];
41   UNION_TYPE (AVX512F_LEN, i_d) src;
42   MASK_TYPE mask = MASK_VALUE;
43   short res_ref[SIZE_HALF];
44   short res_ref2[SIZE_HALF];
45 
46   sign = -1;
47   for (i = 0; i < SIZE; i++)
48     {
49       src.a[i] = 1 + 34 * i * sign;
50       sign = sign * -1;
51       res2.a[i] = DEFAULT_VALUE;
52       res4[i] = DEFAULT_VALUE;
53     }
54 
55   for (i = SIZE; i < SIZE_HALF; i++)
56     {
57       res_ref2[i] = DEFAULT_VALUE * 2;
58       res4[i] = DEFAULT_VALUE * 2;
59     }
60 
61   res1.x = INTRINSIC (_cvtsepi32_epi16) (src.x);
62   res2.x = INTRINSIC (_mask_cvtsepi32_epi16) (res2.x, mask, src.x);
63   res3.x = INTRINSIC (_maskz_cvtsepi32_epi16) (mask, src.x);
64 
65   CALC (res_ref, src.a, 0);
66 
67   if (UNION_CHECK (AVX512F_LEN_HALF, i_w) (res1, res_ref))
68     abort ();
69 
70   MASK_MERGE (i_w) (res_ref, mask, SIZE);
71   if (UNION_CHECK (AVX512F_LEN_HALF, i_w) (res2, res_ref))
72     abort ();
73 
74   MASK_ZERO (i_w) (res_ref, mask, SIZE);
75   if (UNION_CHECK (AVX512F_LEN_HALF, i_w) (res3, res_ref))
76     abort ();
77 
78   INTRINSIC (_mask_cvtsepi32_storeu_epi16) (res4, mask, src.x);
79   CALC (res_ref2, src.a, 1);
80 
81   MASK_MERGE (i_w) (res_ref2, mask, SIZE);
82   if (checkVs (res4, res_ref2, SIZE_HALF))
83     abort ();
84 }
85