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