sglang_v0.5.2/pytorch_2.8.0/third_party/XNNPACK/test/f16-f32acc-rsum.cc

935 lines
33 KiB
C++

// Copyright 2023 Google LLC
//
// This source code is licensed under the BSD-style license found in the
// LICENSE file in the root directory of this source tree.
//
// Auto-generated file. Do not edit!
// Specification: test/f16-f32acc-rsum.yaml
// Generator: tools/generate-reduce-test.py
#include <gtest/gtest.h>
#include "xnnpack/common.h"
#include "xnnpack/isa-checks.h"
#include "xnnpack/microparams-init.h"
#include "xnnpack/reduce.h"
#include "rsum-microkernel-tester.h"
#if XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U4, batch_eq_4) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(4)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u4, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U4, batch_div_4) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 8; batch_size < 40; batch_size += 4) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U4, batch_lt_4) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 1; batch_size < 4; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U4, batch_gt_4) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 5; batch_size < 8; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U4, scale) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(5)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U4, overflow_accumulator) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(512)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u4, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
#if XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U8, batch_eq_8) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(8)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U8, batch_div_8) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 16; batch_size < 80; batch_size += 8) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U8, batch_lt_8) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 1; batch_size < 8; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U8, batch_gt_8) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 9; batch_size < 16; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U8, scale) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(9)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U8, overflow_accumulator) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(1024)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
#if XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U16_ACC2, batch_eq_16) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(16)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U16_ACC2, batch_div_16) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 32; batch_size < 160; batch_size += 16) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U16_ACC2, batch_lt_16) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 1; batch_size < 16; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U16_ACC2, batch_gt_16) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 17; batch_size < 32; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U16_ACC2, scale) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(17)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U16_ACC2, overflow_accumulator) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(2048)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
#if XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U24_ACC3, batch_eq_24) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(24)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U24_ACC3, batch_div_24) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 48; batch_size < 240; batch_size += 24) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U24_ACC3, batch_lt_24) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 1; batch_size < 24; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U24_ACC3, batch_gt_24) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 25; batch_size < 48; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U24_ACC3, scale) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(25)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U24_ACC3, overflow_accumulator) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(3072)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
#if XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC2, batch_eq_32) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(32)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC2, batch_div_32) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 64; batch_size < 320; batch_size += 32) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC2, batch_lt_32) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 1; batch_size < 32; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC2, batch_gt_32) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 33; batch_size < 64; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC2, scale) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(33)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC2, overflow_accumulator) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(4096)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
#if XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC4, batch_eq_32) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(32)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC4, batch_div_32) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 64; batch_size < 320; batch_size += 32) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC4, batch_lt_32) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 1; batch_size < 32; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC4, batch_gt_32) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (size_t batch_size = 33; batch_size < 64; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC4, scale) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(33)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__NEONFP16ARITH_U32_ACC4, overflow_accumulator) {
TEST_REQUIRES_ARM_NEON_FP16_ARITH;
RSumMicrokernelTester()
.batch_size(4096)
.Test(xnn_f16_f32acc_rsum_ukernel__neonfp16arith_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
#if XNN_ARCH_X86 || XNN_ARCH_X86_64
TEST(F16_F32ACC_RSUM__F16C_U8, batch_eq_8) {
TEST_REQUIRES_X86_F16C;
RSumMicrokernelTester()
.batch_size(8)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__F16C_U8, batch_div_8) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 16; batch_size < 80; batch_size += 8) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U8, batch_lt_8) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 1; batch_size < 8; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U8, batch_gt_8) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 9; batch_size < 16; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U8, scale) {
TEST_REQUIRES_X86_F16C;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(9)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U8, overflow_accumulator) {
TEST_REQUIRES_X86_F16C;
RSumMicrokernelTester()
.batch_size(1024)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u8, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ARCH_X86 || XNN_ARCH_X86_64
#if XNN_ARCH_X86 || XNN_ARCH_X86_64
TEST(F16_F32ACC_RSUM__F16C_U16_ACC2, batch_eq_16) {
TEST_REQUIRES_X86_F16C;
RSumMicrokernelTester()
.batch_size(16)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__F16C_U16_ACC2, batch_div_16) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 32; batch_size < 160; batch_size += 16) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U16_ACC2, batch_lt_16) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 1; batch_size < 16; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U16_ACC2, batch_gt_16) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 17; batch_size < 32; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U16_ACC2, scale) {
TEST_REQUIRES_X86_F16C;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(17)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U16_ACC2, overflow_accumulator) {
TEST_REQUIRES_X86_F16C;
RSumMicrokernelTester()
.batch_size(2048)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u16_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ARCH_X86 || XNN_ARCH_X86_64
#if XNN_ARCH_X86 || XNN_ARCH_X86_64
TEST(F16_F32ACC_RSUM__F16C_U24_ACC3, batch_eq_24) {
TEST_REQUIRES_X86_F16C;
RSumMicrokernelTester()
.batch_size(24)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__F16C_U24_ACC3, batch_div_24) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 48; batch_size < 240; batch_size += 24) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U24_ACC3, batch_lt_24) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 1; batch_size < 24; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U24_ACC3, batch_gt_24) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 25; batch_size < 48; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U24_ACC3, scale) {
TEST_REQUIRES_X86_F16C;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(25)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U24_ACC3, overflow_accumulator) {
TEST_REQUIRES_X86_F16C;
RSumMicrokernelTester()
.batch_size(3072)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u24_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ARCH_X86 || XNN_ARCH_X86_64
#if XNN_ARCH_X86 || XNN_ARCH_X86_64
TEST(F16_F32ACC_RSUM__F16C_U32_ACC2, batch_eq_32) {
TEST_REQUIRES_X86_F16C;
RSumMicrokernelTester()
.batch_size(32)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__F16C_U32_ACC2, batch_div_32) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 64; batch_size < 320; batch_size += 32) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U32_ACC2, batch_lt_32) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 1; batch_size < 32; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U32_ACC2, batch_gt_32) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 33; batch_size < 64; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U32_ACC2, scale) {
TEST_REQUIRES_X86_F16C;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(33)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U32_ACC2, overflow_accumulator) {
TEST_REQUIRES_X86_F16C;
RSumMicrokernelTester()
.batch_size(4096)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ARCH_X86 || XNN_ARCH_X86_64
#if XNN_ARCH_X86 || XNN_ARCH_X86_64
TEST(F16_F32ACC_RSUM__F16C_U32_ACC4, batch_eq_32) {
TEST_REQUIRES_X86_F16C;
RSumMicrokernelTester()
.batch_size(32)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__F16C_U32_ACC4, batch_div_32) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 64; batch_size < 320; batch_size += 32) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U32_ACC4, batch_lt_32) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 1; batch_size < 32; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U32_ACC4, batch_gt_32) {
TEST_REQUIRES_X86_F16C;
for (size_t batch_size = 33; batch_size < 64; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U32_ACC4, scale) {
TEST_REQUIRES_X86_F16C;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(33)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__F16C_U32_ACC4, overflow_accumulator) {
TEST_REQUIRES_X86_F16C;
RSumMicrokernelTester()
.batch_size(4096)
.Test(xnn_f16_f32acc_rsum_ukernel__f16c_u32_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ARCH_X86 || XNN_ARCH_X86_64
#if XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
TEST(F16_F32ACC_RSUM__AVX512SKX_U16, batch_eq_16) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(16)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u16, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U16, batch_div_16) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 32; batch_size < 160; batch_size += 16) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u16, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U16, batch_lt_16) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 1; batch_size < 16; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u16, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U16, batch_gt_16) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 17; batch_size < 32; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u16, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U16, scale) {
TEST_REQUIRES_X86_AVX512SKX;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(17)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u16, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U16, overflow_accumulator) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(2048)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u16, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
#if XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
TEST(F16_F32ACC_RSUM__AVX512SKX_U32_ACC2, batch_eq_32) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(32)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U32_ACC2, batch_div_32) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 64; batch_size < 320; batch_size += 32) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U32_ACC2, batch_lt_32) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 1; batch_size < 32; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U32_ACC2, batch_gt_32) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 33; batch_size < 64; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U32_ACC2, scale) {
TEST_REQUIRES_X86_AVX512SKX;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(33)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U32_ACC2, overflow_accumulator) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(4096)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u32_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
#if XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
TEST(F16_F32ACC_RSUM__AVX512SKX_U48_ACC3, batch_eq_48) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(48)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u48_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U48_ACC3, batch_div_48) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 96; batch_size < 480; batch_size += 48) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u48_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U48_ACC3, batch_lt_48) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 1; batch_size < 48; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u48_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U48_ACC3, batch_gt_48) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 49; batch_size < 96; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u48_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U48_ACC3, scale) {
TEST_REQUIRES_X86_AVX512SKX;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(49)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u48_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U48_ACC3, overflow_accumulator) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(6144)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u48_acc3, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
#if XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC2, batch_eq_64) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(64)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC2, batch_div_64) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 128; batch_size < 640; batch_size += 64) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC2, batch_lt_64) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 1; batch_size < 64; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC2, batch_gt_64) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 65; batch_size < 128; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC2, scale) {
TEST_REQUIRES_X86_AVX512SKX;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(65)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC2, overflow_accumulator) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(8192)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc2, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
#if XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC4, batch_eq_64) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(64)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC4, batch_div_64) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 128; batch_size < 640; batch_size += 64) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC4, batch_lt_64) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 1; batch_size < 64; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC4, batch_gt_64) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 65; batch_size < 128; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC4, scale) {
TEST_REQUIRES_X86_AVX512SKX;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(65)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U64_ACC4, overflow_accumulator) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(8192)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u64_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
#if XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
TEST(F16_F32ACC_RSUM__AVX512SKX_U128_ACC4, batch_eq_128) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(128)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u128_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U128_ACC4, batch_div_128) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 256; batch_size < 1280; batch_size += 128) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u128_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U128_ACC4, batch_lt_128) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 1; batch_size < 128; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u128_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U128_ACC4, batch_gt_128) {
TEST_REQUIRES_X86_AVX512SKX;
for (size_t batch_size = 129; batch_size < 256; batch_size++) {
RSumMicrokernelTester()
.batch_size(batch_size)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u128_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U128_ACC4, scale) {
TEST_REQUIRES_X86_AVX512SKX;
for (float scale = 0.3f; scale < 5.0f; scale *= 3.0f) {
RSumMicrokernelTester()
.batch_size(129)
.scale(scale)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u128_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
}
TEST(F16_F32ACC_RSUM__AVX512SKX_U128_ACC4, overflow_accumulator) {
TEST_REQUIRES_X86_AVX512SKX;
RSumMicrokernelTester()
.batch_size(16384)
.Test(xnn_f16_f32acc_rsum_ukernel__avx512skx_u128_acc4, xnn_init_f16_f32acc_scale_scalar_params);
}
#endif // XNN_ENABLE_AVX512SKX && (XNN_ARCH_X86 || XNN_ARCH_X86_64)