1/* { dg-do run } */
2/* { dg-require-effective-target sse4 } */
3/* { dg-options "-O2 -msse4.1" } */
4
5#include "sse4_1-check.h"
6
7#include <smmintrin.h>
8#include <string.h>
9
10#define NUM 20
11
12#undef MASK
13#define MASK 0xfe
14
15static void
16init_pblendw (short *src1, short *src2)
17{
18  int i, sign = 1;
19
20  for (i = 0; i < NUM * 8; i++)
21    {
22      src1[i] = i * i * sign;
23      src2[i] = (i + 20) * sign;
24      sign = -sign;
25    }
26}
27
28static int
29check_pblendw (__m128i *dst, short *src1, short *src2)
30{
31  short tmp[8];
32  int j;
33
34  memcpy (&tmp[0], src1, sizeof (tmp));
35  for (j = 0; j < 8; j++)
36    if ((MASK & (1 << j)))
37      tmp[j] = src2[j];
38
39  return memcmp (dst, &tmp[0], sizeof (tmp));
40}
41
42static void
43sse4_1_test (void)
44{
45  __m128i x, y;
46  union
47    {
48      __m128i x[NUM];
49      short s[NUM * 8];
50    } dst, src1, src2;
51  union
52    {
53      __m128i x;
54      short s[8];
55    } src3;
56  int i;
57
58  init_pblendw (src1.s, src2.s);
59
60  /* Check pblendw imm8, m128, xmm */
61  for (i = 0; i < NUM; i++)
62    {
63      dst.x[i] = _mm_blend_epi16 (src1.x[i], src2.x[i], MASK);
64      if (check_pblendw (&dst.x[i], &src1.s[i * 8], &src2.s[i * 8]))
65	abort ();
66    }
67
68   /* Check pblendw imm8, xmm, xmm */
69  src3.x = _mm_setzero_si128 ();
70
71  x = _mm_blend_epi16 (dst.x[2], src3.x, MASK);
72  y = _mm_blend_epi16 (src3.x, dst.x[2], MASK);
73
74  if (check_pblendw (&x, &dst.s[16], &src3.s[0]))
75    abort ();
76
77  if (check_pblendw (&y, &src3.s[0], &dst.s[16]))
78    abort ();
79}
80