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

36 lines
1.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_f16_rmax_ukernel__neonfp16arith_u8
- name: xnn_f16_rmax_ukernel__neonfp16arith_u16_acc2
- name: xnn_f16_rmax_ukernel__neonfp16arith_u24_acc3
- name: xnn_f16_rmax_ukernel__neonfp16arith_u32_acc2
- name: xnn_f16_rmax_ukernel__neonfp16arith_u32_acc4
# x86 AVX512FP16
- name: xnn_f16_rmax_ukernel__avx512fp16_u32
- name: xnn_f16_rmax_ukernel__avx512fp16_u64_acc2
- name: xnn_f16_rmax_ukernel__avx512fp16_u96_acc3
- name: xnn_f16_rmax_ukernel__avx512fp16_u128_acc2
- name: xnn_f16_rmax_ukernel__avx512fp16_u128_acc4
# x86 AVX512SKX
- name: xnn_f16_rmax_ukernel__avx512skx_u16
- name: xnn_f16_rmax_ukernel__avx512skx_u32_acc2
- name: xnn_f16_rmax_ukernel__avx512skx_u48_acc3
- name: xnn_f16_rmax_ukernel__avx512skx_u64_acc2
- name: xnn_f16_rmax_ukernel__avx512skx_u64_acc4
# x86 F16C
- name: xnn_f16_rmax_ukernel__f16c_u32
# Scalar
- name: xnn_f16_rmax_ukernel__scalar_u1
- name: xnn_f16_rmax_ukernel__scalar_u2_acc2
- name: xnn_f16_rmax_ukernel__scalar_u3_acc3
- name: xnn_f16_rmax_ukernel__scalar_u4_acc2
- name: xnn_f16_rmax_ukernel__scalar_u4_acc4