commit 8c417966ac9f0eb79121fe482a0a31d4be7c3af4
author Marat Dukhan <maratek@google.com> Wed Jul 08 12:27:50 2020 -0700
committer XNNPACK Team <xnnpack-github-robot@google.com> Wed Jul 08 12:28:28 2020 -0700
tree 4d9c4018dfd79b9c1867178b1850fefbd09154a6
parent efc101464f401443aa31a85e355188799b5e9b8c

WAsm SIMD versions of F32 RMAX microkernel

PiperOrigin-RevId: 320241458

test/f32-rmax.cc

diff --git a/test/f32-rmax.cc b/test/f32-rmax.cc
index 5d8bf06..0a8c493 100644
--- a/test/f32-rmax.cc
+++ b/test/f32-rmax.cc
@@ -82,7 +82,9 @@
         .Test(xnn_f32_rmax_ukernel__sse);
     }
   }
+#endif  // XNN_ARCH_X86 || XNN_ARCH_X86_64
 
+#if XNN_ARCH_X86 || XNN_ARCH_X86_64
   TEST(F32_RMAX__AVX, n_lt_32) {
     TEST_REQUIRES_X86_AVX;
     for (size_t n = 1; n < 32; n++) {
@@ -116,7 +118,9 @@
         .Test(xnn_f32_rmax_ukernel__avx);
     }
   }
+#endif  // XNN_ARCH_X86 || XNN_ARCH_X86_64
 
+#if XNN_ARCH_X86 || XNN_ARCH_X86_64
   TEST(F32_RMAX__AVX512F, n_lt_64) {
     TEST_REQUIRES_X86_AVX512F;
     for (size_t n = 1; n < 64; n++) {
@@ -152,6 +156,70 @@
   }
 #endif  // XNN_ARCH_X86 || XNN_ARCH_X86_64
 
+#if XNN_ARCH_WASMSIMD
+  TEST(F32_RMAX__WASMSIMD_ARM, n_lt_16) {
+    for (size_t n = 1; n < 16; n++) {
+      RMaxMicrokernelTester()
+        .n(n)
+        .Test(xnn_f32_rmax_ukernel__wasmsimd_arm);
+    }
+  }
+
+  TEST(F32_RMAX__WASMSIMD_ARM, n_eq_16) {
+    RMaxMicrokernelTester()
+      .n(16)
+      .Test(xnn_f32_rmax_ukernel__wasmsimd_arm);
+  }
+
+  TEST(F32_RMAX__WASMSIMD_ARM, n_div_16) {
+    for (size_t n = 16; n < 128; n += 16) {
+      RMaxMicrokernelTester()
+        .n(n)
+        .Test(xnn_f32_rmax_ukernel__wasmsimd_arm);
+    }
+  }
+
+  TEST(F32_RMAX__WASMSIMD_ARM, n_gt_16) {
+    for (size_t n = 16; n < 32; n++) {
+      RMaxMicrokernelTester()
+        .n(n)
+        .Test(xnn_f32_rmax_ukernel__wasmsimd_arm);
+    }
+  }
+#endif  // XNN_ARCH_WASMSIMD
+
+#if XNN_ARCH_WASMSIMD
+  TEST(F32_RMAX__WASMSIMD_X86, n_lt_16) {
+    for (size_t n = 1; n < 16; n++) {
+      RMaxMicrokernelTester()
+        .n(n)
+        .Test(xnn_f32_rmax_ukernel__wasmsimd_x86);
+    }
+  }
+
+  TEST(F32_RMAX__WASMSIMD_X86, n_eq_16) {
+    RMaxMicrokernelTester()
+      .n(16)
+      .Test(xnn_f32_rmax_ukernel__wasmsimd_x86);
+  }
+
+  TEST(F32_RMAX__WASMSIMD_X86, n_div_16) {
+    for (size_t n = 16; n < 128; n += 16) {
+      RMaxMicrokernelTester()
+        .n(n)
+        .Test(xnn_f32_rmax_ukernel__wasmsimd_x86);
+    }
+  }
+
+  TEST(F32_RMAX__WASMSIMD_X86, n_gt_16) {
+    for (size_t n = 16; n < 32; n++) {
+      RMaxMicrokernelTester()
+        .n(n)
+        .Test(xnn_f32_rmax_ukernel__wasmsimd_x86);
+    }
+  }
+#endif  // XNN_ARCH_WASMSIMD
+
 TEST(F32_RMAX__SCALAR, n_lt_4) {
   for (size_t n = 1; n < 4; n++) {
     RMaxMicrokernelTester()