1 files changed, 73 insertions, 0 deletions
diff --git a/SingleSource/UnitTests/Vector/AVX512F/reduce_add_mul_m512.c b/SingleSource/UnitTests/Vector/AVX512F/reduce_add_mul_m512.c
new file mode 100644
index 00000000..0a9669e5
--- /dev/null
+++ b/SingleSource/UnitTests/Vector/AVX512F/reduce_add_mul_m512.c
@@ -0,0 +1,73 @@
+#include "m512_test_util.h"
+#include <stdio.h>
+#include <string.h>
+
+/*
+ * Here we check for _mm512_[mask_]reduce_[add|mul] intrinsics.
+ */
+
+#define CHECK_PD(op)                                                           \
+  {                                                                            \
+    volatile double r = _mm512_##op##_pd(v1);                                  \
+    check_equal_ndf(&r, &exp1, 1, "_mm512_" #op "_pd", __LINE__);              \
+    k8 = 0xAA;                                                                 \
+    r = _mm512_mask_##op##_pd(k8, v1);                                         \
+    check_equal_ndf(&r, &exp2, 1, "_mm512_mask_" #op "_pd{1}", __LINE__);      \
+  }
+
+#define DECL_PD(op, srcv, out1, out2)                                          \
+  void NOINLINE do_##op##_pd() {                                               \
+    __mmask8 k8;                                                               \
+    volatile __m512d v1 = _mm512_set1_pd((srcv));                              \
+                                                                               \
+    double exp1 = (out1);                                                      \
+    double exp2 = (out2);                                                      \
+                                                                               \
+    CHECK_PD(op);                                                              \
+  }
+
+#define TEST_PD(op) do_##op##_pd()
+
+// PS version starts here.
+
+#define CHECK_PS(op)                                                           \
+  {                                                                            \
+    volatile float r = _mm512_##op##_ps(v1);                                   \
+    check_equal_nsf(&r, &exp1, 1, "_mm512_" #op "_ps", __LINE__);              \
+    k8 = 0xAAAA;                                                               \
+    r = _mm512_mask_##op##_ps(k8, v1);                                         \
+    check_equal_nsf(&r, &exp2, 1, "_mm512_mask_" #op "_ps{1}", __LINE__);      \
+  }
+
+#define DECL_PS(op, srcv, out1, out2)                                          \
+  void NOINLINE do_##op##_ps() {                                               \
+    __mmask16 k8;                                                              \
+    volatile __m512 v1 = _mm512_set1_ps((srcv));                               \
+                                                                               \
+    float exp1 = (out1);                                                       \
+    float exp2 = (out2);                                                       \
+                                                                               \
+    CHECK_PS(op);                                                              \
+  }
+
+#define TEST_PS(op) do_##op##_ps()
+
+DECL_PD(reduce_add, 0.5, 4.0, 2.0)
+DECL_PS(reduce_add, 0.4f, 6.4f, 3.2f)
+DECL_PD(reduce_mul, 1.1, 2.1435f, 1.4641f)
+DECL_PS(reduce_mul, -1.1f, 4.5949f, 2.1435f)
+
+int main(int argc, char *argv[]) {
+  TEST_PD(reduce_add);
+  TEST_PS(reduce_add);
+  TEST_PD(reduce_mul);
+  TEST_PS(reduce_mul);
+
+  if (n_errs != 0) {
+    printf("FAILED\n");
+    return 1;
+  }
+
+  printf("PASSED\n");
+  return 0;
+}