1 files changed, 45 insertions, 0 deletions
diff --git a/gcc/testsuite/gcc.target/i386/avx2-vpsrawi-2.c b/gcc/testsuite/gcc.target/i386/avx2-vpsrawi-2.c
new file mode 100644
index 00000000000..c135833a468
--- /dev/null
+++ b/gcc/testsuite/gcc.target/i386/avx2-vpsrawi-2.c
@@ -0,0 +1,45 @@
+/* { dg-do run } */
+/* { dg-require-effective-target avx2 } */
+/* { dg-options "-O3 -mavx2" } */
+
+#include <string.h>
+#include "avx2-check.h"
+
+#define N 0x5
+
+static void
+compute_psrawi256 (short *s1, short *r)
+{
+  int i;
+
+  memset (r, 0, 32);
+
+  if (N < 16)
+    for (i = 0; i < 16; ++i)
+      r[i] = s1[i] >> N;
+}
+
+
+void static
+avx2_test (void)
+{
+  union256i_w s1, res;
+  short res_ref[16];
+  int i, j;
+  int fail = 0;
+
+  for (i = 0; i < 10; i++)
+    {
+      for (j = 0; j < 16; j++)
+	s1.a[j] = j * i;
+
+      res.x = _mm256_srai_epi16 (s1.x, N);
+
+      compute_psrawi256 (s1.a, res_ref);
+
+      fail += check_union256i_w (res, res_ref);
+    }
+
+  if (fail != 0)
+    abort ();
+}