rework: added test helper functions for argmaxpool and formatting cha…

…nges
umadevimcw · Feb 21, 2025 · 8faed2d · 8faed2d
1 parent 579c490
commit 8faed2d
Show file tree

Hide file tree

Showing 3 changed files with 205 additions and 500 deletions.
diff --git a/src/f32-argmaxpool/f32-argmaxpool-multipass.h b/src/f32-argmaxpool/f32-argmaxpool-multipass.h
@@ -20,7 +20,7 @@ XNN_UKERNEL_WITH_PARAMS(0, xnn_f32_argmaxpool_ukernel_9p8x__scalar_c1, 9, 8, 1,
 
 
 #if (XNN_ARCH_ARM || XNN_ARCH_ARM64)
-XNN_UKERNEL_WITH_PARAMS(xnn_arch_arm_neon, xnn_f32_argmaxpool_ukernel_9p8x__neon_c4, 9, 8, 4,false, float, struct xnn_f32_default_params, NULL)
+XNN_UKERNEL_WITH_PARAMS(xnn_arch_arm_neon, xnn_f32_argmaxpool_ukernel_9p8x__neon_c4, 9, 8, 4, false, float, struct xnn_f32_default_params, NULL)
 
 
 #endif  // XNN_ARCH_ARM || XNN_ARCH_ARM64
@@ -36,7 +36,7 @@ XNN_UKERNEL_WITH_PARAMS(0, xnn_f32_argmaxpool_ukernel_9p8x__wasmsimd_c4, 9, 8, 4
 
 #if XNN_ARCH_RISCV && (XNN_ENABLE_RISCV_VECTOR)
 
-XNN_UKERNEL_WITH_PARAMS(xnn_arch_riscv_vector, xnn_f32_argmaxpool_ukernel_9p8x__rvv_u1v,  9, 8, 1, true, float, struct xnn_f32_default_params, NULL)
+XNN_UKERNEL_WITH_PARAMS(xnn_arch_riscv_vector, xnn_f32_argmaxpool_ukernel_9p8x__rvv_u1v, 9, 8, 1, true, float, struct xnn_f32_default_params, NULL)
 #endif // XNN_ARCH_RISCV && XNN_ENABLE_RISCV_VECTOR
 
 #ifdef XNN_DEFINED_UKERNEL_WITH_PARAMS

diff --git a/src/f32-argmaxpool/f32-argmaxpool-unipass.h b/src/f32-argmaxpool/f32-argmaxpool-unipass.h
@@ -20,14 +20,14 @@ XNN_UKERNEL_WITH_PARAMS(0, xnn_f32_argmaxpool_ukernel_4x__scalar_c1, 4, 0, 1, fa
 XNN_UKERNEL_WITH_PARAMS(0, xnn_f32_argmaxpool_ukernel_9x__scalar_c1, 9, 0, 1, false, float, struct xnn_f32_default_params, NULL)
 
 #if (XNN_ARCH_ARM || XNN_ARCH_ARM64)
-XNN_UKERNEL_WITH_PARAMS(xnn_arch_arm_neon, xnn_f32_argmaxpool_ukernel_4x__neon_c4, 4, 0, 4,false, float, struct xnn_f32_default_params, NULL)
-XNN_UKERNEL_WITH_PARAMS(xnn_arch_arm_neon, xnn_f32_argmaxpool_ukernel_9x__neon_c4, 9, 0, 4, false,float, struct xnn_f32_default_params, NULL)
+XNN_UKERNEL_WITH_PARAMS(xnn_arch_arm_neon, xnn_f32_argmaxpool_ukernel_4x__neon_c4, 4, 0, 4, false, float, struct xnn_f32_default_params, NULL)
+XNN_UKERNEL_WITH_PARAMS(xnn_arch_arm_neon, xnn_f32_argmaxpool_ukernel_9x__neon_c4, 9, 0, 4, false, float, struct xnn_f32_default_params, NULL)
 
 #endif  // XNN_ARCH_ARM || XNN_ARCH_ARM64
 
 #if XNN_ARCH_X86 || XNN_ARCH_X86_64
 XNN_UKERNEL_WITH_PARAMS(0, xnn_f32_argmaxpool_ukernel_4x__sse2_c4, 4, 0, 4, false, float, struct xnn_f32_default_params, NULL)
-XNN_UKERNEL_WITH_PARAMS(0, xnn_f32_argmaxpool_ukernel_9x__sse2_c4,  9, 0, 4, false, float, struct xnn_f32_default_params, NULL)
+XNN_UKERNEL_WITH_PARAMS(0, xnn_f32_argmaxpool_ukernel_9x__sse2_c4, 9, 0, 4, false, float, struct xnn_f32_default_params, NULL)
 #endif  // XNN_ARCH_X86 || XNN_ARCH_X86_64
 
 #if XNN_ARCH_WASMSIMD || XNN_ARCH_WASMRELAXEDSIMD
@@ -37,7 +37,7 @@ XNN_UKERNEL_WITH_PARAMS(0, xnn_f32_argmaxpool_ukernel_9x__wasmsimd_c4, 9, 0, 4,
 
 #if XNN_ARCH_RISCV && (XNN_ENABLE_RISCV_VECTOR)
 
-XNN_UKERNEL_WITH_PARAMS(xnn_arch_riscv_vector, xnn_f32_argmaxpool_ukernel_4x__rvv_u1v,  4, 0, 1, true, float, struct xnn_f32_default_params, NULL)
+XNN_UKERNEL_WITH_PARAMS(xnn_arch_riscv_vector, xnn_f32_argmaxpool_ukernel_4x__rvv_u1v, 4, 0, 1, true, float, struct xnn_f32_default_params, NULL)
 XNN_UKERNEL_WITH_PARAMS(xnn_arch_riscv_vector, xnn_f32_argmaxpool_ukernel_9x__rvv_u1v, 9, 0, 1, true, float, struct xnn_f32_default_params, NULL)
 
 #endif // XNN_ARCH_RISCV && XNN_ENABLE_RISCV_VECTOR