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

43 lines
1.5 KiB
YAML

# Copyright 2019 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_spmm_minmax_ukernel_8x1__neonfp16arith
init: xnn_init_f16_minmax_scalar_params
k-block: 1
- name: xnn_f16_spmm_minmax_ukernel_8x1__neonfp16arith_pipelined
init: xnn_init_f16_minmax_scalar_params
k-block: 1
- name: xnn_f16_spmm_minmax_ukernel_8x1__neonfp16arith_x2
init: xnn_init_f16_minmax_scalar_params
k-block: 2
- name: xnn_f16_spmm_minmax_ukernel_16x1__neonfp16arith
init: xnn_init_f16_minmax_scalar_params
k-block: 1
- name: xnn_f16_spmm_minmax_ukernel_16x1__neonfp16arith_pipelined
init: xnn_init_f16_minmax_scalar_params
k-block: 1
- name: xnn_f16_spmm_minmax_ukernel_16x1__neonfp16arith_x2
init: xnn_init_f16_minmax_scalar_params
k-block: 2
- name: xnn_f16_spmm_minmax_ukernel_24x1__neonfp16arith
init: xnn_init_f16_minmax_scalar_params
k-block: 1
- name: xnn_f16_spmm_minmax_ukernel_24x1__neonfp16arith_pipelined
init: xnn_init_f16_minmax_scalar_params
k-block: 1
- name: xnn_f16_spmm_minmax_ukernel_24x1__neonfp16arith_x2
init: xnn_init_f16_minmax_scalar_params
k-block: 2
- name: xnn_f16_spmm_minmax_ukernel_32x1__neonfp16arith
init: xnn_init_f16_minmax_scalar_params
k-block: 1
- name: xnn_f16_spmm_minmax_ukernel_32x1__neonfp16arith_pipelined
init: xnn_init_f16_minmax_scalar_params
k-block: 1
- name: xnn_f16_spmm_minmax_ukernel_32x1__neonfp16arith_x2
init: xnn_init_f16_minmax_scalar_params
k-block: 2