sglang_v0.5.2/pytorch_2.8.0/third_party/XNNPACK/test/f32-rsum.yaml

93 lines
3.2 KiB
YAML

# 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.
# ARM NEON
- name: xnn_f32_rsum_ukernel__neon_u4
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__neon_u8_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__neon_u12_acc3
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__neon_u16_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__neon_u16_acc4
init: xnn_init_f32_scale_scalar_params
# x86 SSE
- name: xnn_f32_rsum_ukernel__sse_u4
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__sse_u8_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__sse_u12_acc3
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__sse_u16_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__sse_u16_acc4
init: xnn_init_f32_scale_scalar_params
# x86 AVX
- name: xnn_f32_rsum_ukernel__avx_u8
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__avx_u16_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__avx_u24_acc3
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__avx_u32_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__avx_u32_acc4
init: xnn_init_f32_scale_scalar_params
# x86 AVX512F
- name: xnn_f32_rsum_ukernel__avx512f_u16
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__avx512f_u32_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__avx512f_u48_acc3
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__avx512f_u64_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__avx512f_u64_acc4
init: xnn_init_f32_scale_scalar_params
# Hexagon HVX
- name: xnn_f32_rsum_ukernel__hvx_u32
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__hvx_u64_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__hvx_u96_acc3
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__hvx_u128_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__hvx_u128_acc4
init: xnn_init_f32_scale_scalar_params
# Wasm SIMD
- name: xnn_f32_rsum_ukernel__wasmsimd_u4
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__wasmsimd_u8_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__wasmsimd_u12_acc3
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__wasmsimd_u16_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__wasmsimd_u16_acc4
init: xnn_init_f32_scale_scalar_params
# RISC-V Vector
- name: xnn_f32_rsum_ukernel__rvv_u1v
init: xnn_init_f32_scale_scalar_params
# Scalar
- name: xnn_f32_rsum_ukernel__scalar_u1
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__scalar_u2_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__scalar_u3_acc3
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__scalar_u4_acc2
init: xnn_init_f32_scale_scalar_params
- name: xnn_f32_rsum_ukernel__scalar_u4_acc4
init: xnn_init_f32_scale_scalar_params