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

29 lines
1.1 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+FP16ARITH
- name: xnn_f16_rsum_ukernel__neonfp16arith_u8
init: xnn_init_f16_scale_scalar_params
- name: xnn_f16_rsum_ukernel__neonfp16arith_u16_acc2
init: xnn_init_f16_scale_scalar_params
- name: xnn_f16_rsum_ukernel__neonfp16arith_u24_acc3
init: xnn_init_f16_scale_scalar_params
- name: xnn_f16_rsum_ukernel__neonfp16arith_u32_acc2
init: xnn_init_f16_scale_scalar_params
- name: xnn_f16_rsum_ukernel__neonfp16arith_u32_acc4
init: xnn_init_f16_scale_scalar_params
# x86 AVX512FP16
- name: xnn_f16_rsum_ukernel__avx512fp16_u32
init: xnn_init_f16_scale_scalar_params
- name: xnn_f16_rsum_ukernel__avx512fp16_u64_acc2
init: xnn_init_f16_scale_scalar_params
- name: xnn_f16_rsum_ukernel__avx512fp16_u96_acc3
init: xnn_init_f16_scale_scalar_params
- name: xnn_f16_rsum_ukernel__avx512fp16_u128_acc2
init: xnn_init_f16_scale_scalar_params
- name: xnn_f16_rsum_ukernel__avx512fp16_u128_acc4
init: xnn_init_f16_scale_scalar_params