1 /* { dg-do run } */
2 /* { dg-require-effective-target avx2 } */
3 /* { dg-options "-O3 -mavx2" } */
4 
5 #include <string.h>
6 #include "avx2-check.h"
7 
8 #define N 0x5
9 
10 static void
compute_psrawi256(short * s1,short * r)11 compute_psrawi256 (short *s1, short *r)
12 {
13   int i;
14 
15   memset (r, 0, 32);
16 
17   if (N < 16)
18     for (i = 0; i < 16; ++i)
19       r[i] = s1[i] >> N;
20 }
21 
22 
23 void static
avx2_test(void)24 avx2_test (void)
25 {
26   union256i_w s1, res;
27   short res_ref[16];
28   int i, j;
29   int fail = 0;
30 
31   for (i = 0; i < 10; i++)
32     {
33       for (j = 0; j < 16; j++)
34 	s1.a[j] = j * i;
35 
36       res.x = _mm256_srai_epi16 (s1.x, N);
37 
38       compute_psrawi256 (s1.a, res_ref);
39 
40       fail += check_union256i_w (res, res_ref);
41     }
42 
43   if (fail != 0)
44     abort ();
45 }
46