xref: /aosp_15_r20/external/XNNPACK/test/f16-gemm-minmax.yaml (revision 4bdc94577ba0e567308109d787f7fec7b531ce36)

# 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.

# AArch64 assembly
- name: xnn_f16_gemm_minmax_ukernel_1x8__aarch64_neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_1x16__aarch64_neonfp16arith_ld32
  init: xnn_init_f16_minmax_neon_params
  k-block: 2
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_1x16__aarch64_neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_4x8__aarch64_neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_4x16__aarch64_neonfp16arith_ld32
  init: xnn_init_f16_minmax_neon_params
  k-block: 2
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_4x16__aarch64_neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_6x8__aarch64_neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_6x16__aarch64_neonfp16arith_cortex_a55
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_6x16__aarch64_neonfp16arith_cortex_a55r0
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_6x16__aarch64_neonfp16arith_cortex_a75
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_6x16__aarch64_neonfp16arith_ld32
  init: xnn_init_f16_minmax_neon_params
  k-block: 2
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_6x16__aarch64_neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
  assembly: true
- name: xnn_f16_gemm_minmax_ukernel_8x8__aarch64_neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
  assembly: true

# ARM NEON+FP16ARITH
- name: xnn_f16_gemm_minmax_ukernel_1x8__neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
- name: xnn_f16_gemm_minmax_ukernel_1x16__neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
- name: xnn_f16_gemm_minmax_ukernel_4x8__neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
- name: xnn_f16_gemm_minmax_ukernel_4x16__neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
- name: xnn_f16_gemm_minmax_ukernel_6x8__neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
- name: xnn_f16_gemm_minmax_ukernel_6x16__neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
- name: xnn_f16_gemm_minmax_ukernel_8x8__neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4
- name: xnn_f16_gemm_minmax_ukernel_8x16__neonfp16arith_ld64
  init: xnn_init_f16_minmax_neon_params
  k-block: 4

# x86 AVX2
- name: xnn_f16_gemm_minmax_ukernel_1x8__avx2_broadcast
  init: xnn_init_f16_minmax_avx_params
  k-block: 1
- name: xnn_f16_gemm_minmax_ukernel_1x16__avx2_broadcast
  init: xnn_init_f16_minmax_avx_params
  k-block: 1
- name: xnn_f16_gemm_minmax_ukernel_3x16__avx2_broadcast
  init: xnn_init_f16_minmax_avx_params
  k-block: 1
- name: xnn_f16_gemm_minmax_ukernel_4x8__avx2_broadcast
  init: xnn_init_f16_minmax_avx_params
  k-block: 1
- name: xnn_f16_gemm_minmax_ukernel_4x16__avx2_broadcast
  init: xnn_init_f16_minmax_avx_params
  k-block: 1
- name: xnn_f16_gemm_minmax_ukernel_5x8__avx2_broadcast
  init: xnn_init_f16_minmax_avx_params
  k-block: 1
- name: xnn_f16_gemm_minmax_ukernel_5x16__avx2_broadcast
  init: xnn_init_f16_minmax_avx_params
  k-block: 1
- name: xnn_f16_gemm_minmax_ukernel_6x8__avx2_broadcast
  init: xnn_init_f16_minmax_avx_params
  k-block: 1
- name: xnn_f16_gemm_minmax_ukernel_7x8__avx2_broadcast
  init: xnn_init_f16_minmax_avx_params
  k-block: 1