WAsm SIMD versions of HSWISH microkernels
PiperOrigin-RevId: 318408478
diff --git a/test/f32-hswish.cc b/test/f32-hswish.cc
index 3964293..b50f287 100644
--- a/test/f32-hswish.cc
+++ b/test/f32-hswish.cc
@@ -675,6 +675,174 @@
#endif // !XNN_ARCH_ASMJS && !XNN_ARCH_WASM && !XNN_COMPILER_MSVC && !XNN_COMPILER_ICC
+#if XNN_ARCH_WASMSIMD
+ TEST(F32_HSWISH__WASMSIMD_ARM_X4, batch_eq_4) {
+ HSwishMicrokernelTester()
+ .batch_size(4)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_arm_x4, HSwishMicrokernelTester::Variant::Scalar);
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_ARM_X4, batch_div_4) {
+ for (size_t batch_size = 8; batch_size < 40; batch_size += 4) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_arm_x4, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_ARM_X4, batch_lt_4) {
+ for (size_t batch_size = 1; batch_size < 4; batch_size++) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_arm_x4, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_ARM_X4, batch_gt_4) {
+ for (size_t batch_size = 5; batch_size < 8; batch_size++) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_arm_x4, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_ARM_X4, inplace) {
+ for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .inplace(true)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_arm_x4, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+#endif // XNN_ARCH_WASMSIMD
+
+
+#if XNN_ARCH_WASMSIMD
+ TEST(F32_HSWISH__WASMSIMD_ARM_X8, batch_eq_8) {
+ HSwishMicrokernelTester()
+ .batch_size(8)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_arm_x8, HSwishMicrokernelTester::Variant::Scalar);
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_ARM_X8, batch_div_8) {
+ for (size_t batch_size = 16; batch_size < 80; batch_size += 8) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_arm_x8, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_ARM_X8, batch_lt_8) {
+ for (size_t batch_size = 1; batch_size < 8; batch_size++) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_arm_x8, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_ARM_X8, batch_gt_8) {
+ for (size_t batch_size = 9; batch_size < 16; batch_size++) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_arm_x8, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_ARM_X8, inplace) {
+ for (size_t batch_size = 1; batch_size <= 40; batch_size += 7) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .inplace(true)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_arm_x8, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+#endif // XNN_ARCH_WASMSIMD
+
+
+#if XNN_ARCH_WASMSIMD
+ TEST(F32_HSWISH__WASMSIMD_X86_X4, batch_eq_4) {
+ HSwishMicrokernelTester()
+ .batch_size(4)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_x86_x4, HSwishMicrokernelTester::Variant::Scalar);
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_X86_X4, batch_div_4) {
+ for (size_t batch_size = 8; batch_size < 40; batch_size += 4) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_x86_x4, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_X86_X4, batch_lt_4) {
+ for (size_t batch_size = 1; batch_size < 4; batch_size++) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_x86_x4, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_X86_X4, batch_gt_4) {
+ for (size_t batch_size = 5; batch_size < 8; batch_size++) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_x86_x4, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_X86_X4, inplace) {
+ for (size_t batch_size = 1; batch_size <= 20; batch_size += 3) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .inplace(true)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_x86_x4, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+#endif // XNN_ARCH_WASMSIMD
+
+
+#if XNN_ARCH_WASMSIMD
+ TEST(F32_HSWISH__WASMSIMD_X86_X8, batch_eq_8) {
+ HSwishMicrokernelTester()
+ .batch_size(8)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_x86_x8, HSwishMicrokernelTester::Variant::Scalar);
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_X86_X8, batch_div_8) {
+ for (size_t batch_size = 16; batch_size < 80; batch_size += 8) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_x86_x8, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_X86_X8, batch_lt_8) {
+ for (size_t batch_size = 1; batch_size < 8; batch_size++) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_x86_x8, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_X86_X8, batch_gt_8) {
+ for (size_t batch_size = 9; batch_size < 16; batch_size++) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_x86_x8, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+
+ TEST(F32_HSWISH__WASMSIMD_X86_X8, inplace) {
+ for (size_t batch_size = 1; batch_size <= 40; batch_size += 7) {
+ HSwishMicrokernelTester()
+ .batch_size(batch_size)
+ .inplace(true)
+ .Test(xnn_f32_hswish_ukernel__wasmsimd_x86_x8, HSwishMicrokernelTester::Variant::Scalar);
+ }
+ }
+#endif // XNN_ARCH_WASMSIMD
+
+
#if XNN_ARCH_WASM
TEST(F32_HSWISH__WASM_X1, batch_eq_1) {
HSwishMicrokernelTester()