1 files changed, 133 insertions, 0 deletions
diff --git a/SingleSource/UnitTests/Vector/AVX512F/roundscale_m512.c b/SingleSource/UnitTests/Vector/AVX512F/roundscale_m512.c
new file mode 100644
index 00000000..4da406cd
--- /dev/null
+++ b/SingleSource/UnitTests/Vector/AVX512F/roundscale_m512.c
@@ -0,0 +1,133 @@
+#include "m512_test_util.h"
+#include <stdio.h>
+
+/*
+ * Here we check for _mm512_[mask|mmaskz]roundscale[_round]_[ss|sd] intrinsics.
+ */
+
+void __declspec(noinline) do_roundscale_pd() {
+  __mmask8 k8;
+  volatile __m512d v1 = _mm512_set1_pd((1.6));
+  volatile __m512d undef = _mm512_set1_pd(3.0);
+  __m512d exp = _mm512_set1_pd((2.0));
+  __m512d exp1 = _mm512_set1_pd((1.0));
+  __m512d expm = _mm512_set_pd((2.0), 3.0, (2.0), 3.0, (2.0), 3.0, (2.0), 3.0);
+  __m512d expzm = _mm512_set_pd((1.0), 0, (1.0), 0, (1.0), 0, (1.0), 0);
+
+  {
+    volatile __m512d r = _mm512_roundscale_pd(v1, 0x8);
+    check_equal_ndf(&r, &exp, 8, "_mm512_roundscale_pd{0x8}", __LINE__);
+    r = _mm512_roundscale_pd(v1, 0x9);
+    check_equal_ndf(&r, &exp1, 8, "_mm512_roundscale_pd{0x9}", __LINE__);
+    k8 = 0xAA;
+    r = _mm512_mask_roundscale_pd(undef, k8, v1, 0xA);
+    check_equal_ndf(&r, &expm, 8, "_mm512_mask_roundscale_pd{1}{0xA}",
+                    __LINE__);
+    r = _mm512_maskz_roundscale_pd(k8, v1, 0xB);
+    check_equal_ndf(&r, &expzm, 8, "_mm512_maskz_roundscale_pd{0}{0xB}",
+                    __LINE__);
+  }
+
+  {
+    volatile __m512d r =
+        _mm512_roundscale_round_pd(v1, 0x8, (_MM_FROUND_CUR_DIRECTION));
+    check_equal_ndf(&r, &exp, 8, "_mm512_roundscale_round_pd{0x8}", __LINE__);
+    r = _mm512_roundscale_round_pd(v1, 0x9, (_MM_FROUND_CUR_DIRECTION));
+    check_equal_ndf(&r, &exp1, 8, "_mm512_roundscale_round_pd{0x9}", __LINE__);
+    k8 = 0xAA;
+    r = _mm512_mask_roundscale_round_pd(undef, k8, v1, 0xA,
+                                        (_MM_FROUND_CUR_DIRECTION));
+    check_equal_ndf(&r, &expm, 8, "_mm512_mask_roundscale_round_pd{1}{0xA}",
+                    __LINE__);
+    r = _mm512_maskz_roundscale_round_pd(k8, v1, 0xB,
+                                         (_MM_FROUND_CUR_DIRECTION));
+    check_equal_ndf(&r, &expzm, 8, "_mm512_maskz_roundscale_round_pd{0}{0xB}",
+                    __LINE__);
+  }
+
+  {
+    volatile __m512d r =
+        _mm512_roundscale_round_pd(v1, 0x8, ((_MM_FROUND_NO_EXC)));
+    check_equal_ndf(&r, &exp, 8, "_mm512_roundscale_round_pd{0x8}", __LINE__);
+    r = _mm512_roundscale_round_pd(v1, 0x9, ((_MM_FROUND_NO_EXC)));
+    check_equal_ndf(&r, &exp1, 8, "_mm512_roundscale_round_pd{0x9}", __LINE__);
+    k8 = 0xAA;
+    r = _mm512_mask_roundscale_round_pd(undef, k8, v1, 0xA,
+                                        ((_MM_FROUND_NO_EXC)));
+    check_equal_ndf(&r, &expm, 8, "_mm512_mask_roundscale_round_pd{1}{0xA}",
+                    __LINE__);
+    r = _mm512_maskz_roundscale_round_pd(k8, v1, 0xB, ((_MM_FROUND_NO_EXC)));
+    check_equal_ndf(&r, &expzm, 8, "_mm512_maskz_roundscale_round_pd{0}{0xB}",
+                    __LINE__);
+  }
+}
+void __declspec(noinline) do_roundscale_ps() {
+  __mmask16 k8;
+  volatile __m512 v1 = _mm512_set1_ps((-1.6f));
+  volatile __m512 undef = _mm512_set1_ps(3.0);
+  __m512 exp = _mm512_set1_ps((-2.0f));
+  __m512 exp1 = _mm512_set1_ps((-2.0f));
+  __m512 expm =
+      _mm512_set_ps((-1.0f), 3.0, (-1.0f), 3.0, (-1.0f), 3.0, (-1.0f), 3.0,
+                    (-1.0f), 3.0, (-1.0f), 3.0, (-1.0f), 3.0, (-1.0f), 3.0);
+  __m512 expzm = _mm512_set_ps((-1.0f), 0, (-1.0f), 0, (-1.0f), 0, (-1.0f), 0,
+                               (-1.0f), 0, (-1.0f), 0, (-1.0f), 0, (-1.0f), 0);
+
+  {
+    volatile __m512 r = _mm512_roundscale_ps(v1, 0x8);
+    check_equal_nsf(&r, &exp, 16, "_mm512_roundscale_{0x8}", __LINE__);
+    r = _mm512_roundscale_ps(v1, 0x9);
+    check_equal_nsf(&r, &exp, 16, "_mm512_roundscale_{0x9}", __LINE__);
+    k8 = 0xAAAA;
+    r = _mm512_mask_roundscale_ps(undef, k8, v1, 0xA);
+    check_equal_nsf(&r, &expm, 16, "_mm512_mask_roundscale_{1}{A}", __LINE__);
+    r = _mm512_maskz_roundscale_ps(k8, v1, 0xB);
+    check_equal_nsf(&r, &expzm, 16, "_mm512_maskz_roundscale_{0}{B}", __LINE__);
+  }
+
+  {
+    volatile __m512 r =
+        _mm512_roundscale_round_ps(v1, 0x8, (_MM_FROUND_CUR_DIRECTION));
+    check_equal_nsf(&r, &exp, 16, "_mm512_roundscale_round_ps", __LINE__);
+    r = _mm512_roundscale_round_ps(v1, 0x9, (_MM_FROUND_CUR_DIRECTION));
+    check_equal_nsf(&r, &exp1, 16, "_mm512_roundscale_round_ps", __LINE__);
+    k8 = 0xAAAA;
+    r = _mm512_mask_roundscale_round_ps(undef, k8, v1, 0xA,
+                                        (_MM_FROUND_CUR_DIRECTION));
+    check_equal_nsf(&r, &expm, 16, "_mm512_mask_roundscale_round_{1}{A}",
+                    __LINE__);
+    r = _mm512_maskz_roundscale_round_ps(k8, v1, 0xB,
+                                         (_MM_FROUND_CUR_DIRECTION));
+    check_equal_nsf(&r, &expzm, 16, "_mm512_maskz_roundscale_round_{0}{B}",
+                    __LINE__);
+  }
+
+  {
+    volatile __m512 r =
+        _mm512_roundscale_round_ps(v1, 0x8, ((_MM_FROUND_NO_EXC)));
+    check_equal_nsf(&r, &exp, 16, "_mm512_roundscale_round_ps", __LINE__);
+    r = _mm512_roundscale_round_ps(v1, 0x9, ((_MM_FROUND_NO_EXC)));
+    check_equal_nsf(&r, &exp1, 16, "_mm512_roundscale_round_ps", __LINE__);
+    k8 = 0xAAAA;
+    r = _mm512_mask_roundscale_round_ps(undef, k8, v1, 0xA,
+                                        ((_MM_FROUND_NO_EXC)));
+    check_equal_nsf(&r, &expm, 16, "_mm512_mask_roundscale_round_{1}{A}",
+                    __LINE__);
+    r = _mm512_maskz_roundscale_round_ps(k8, v1, 0xB, ((_MM_FROUND_NO_EXC)));
+    check_equal_nsf(&r, &expzm, 16, "_mm512_maskz_roundscale_round_{0}{B}",
+                    __LINE__);
+  }
+}
+
+int main(int argc, char *argv[]) {
+  do_roundscale_pd();
+  do_roundscale_ps();
+
+  if (n_errs != 0) {
+    printf("FAILED\n");
+    return 1;
+  }
+
+  printf("PASSED\n");
+  return 0;
+}