sglang_v0.5.2/pytorch_2.8.0/third_party/XNNPACK/test/f32-qc8w-gemm-relu.yaml

81 lines
2.8 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.
# WAsm Relaxed SIMD
- name: xnn_f32_qc8w_gemm_relu_ukernel_1x8__wasmrelaxedsimd_fma_loadsplat
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
- name: xnn_f32_qc8w_gemm_relu_ukernel_1x8__wasmrelaxedsimd_fma_splat
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 4
- name: xnn_f32_qc8w_gemm_relu_ukernel_1x8s4__wasmrelaxedsimd_fma
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 4
- name: xnn_f32_qc8w_gemm_relu_ukernel_3x8__wasmrelaxedsimd_fma_loadsplat
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
- name: xnn_f32_qc8w_gemm_relu_ukernel_3x8__wasmrelaxedsimd_fma_splat
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 4
- name: xnn_f32_qc8w_gemm_relu_ukernel_3x8s4__wasmrelaxedsimd_fma
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 4
- name: xnn_f32_qc8w_gemm_relu_ukernel_4x2c4__wasmrelaxedsimd_fma
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 4
- name: xnn_f32_qc8w_gemm_relu_ukernel_4x8__wasmrelaxedsimd_fma_loadsplat
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
- name: xnn_f32_qc8w_gemm_relu_ukernel_4x8__wasmrelaxedsimd_fma_splat
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 4
- name: xnn_f32_qc8w_gemm_relu_ukernel_4x8s4__wasmrelaxedsimd_fma
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 4
- name: xnn_f32_qc8w_gemm_relu_ukernel_5x8__wasmrelaxedsimd_fma_loadsplat
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
- name: xnn_f32_qc8w_gemm_relu_ukernel_5x8__wasmrelaxedsimd_fma_splat
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 4
- name: xnn_f32_qc8w_gemm_relu_ukernel_5x8s4__wasmrelaxedsimd_fma
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 4
- name: xnn_f32_qc8w_gemm_relu_ukernel_6x8__wasmrelaxedsimd_fma_loadsplat
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
- name: xnn_f32_qc8w_gemm_relu_ukernel_6x8__wasmrelaxedsimd_fma_splat
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 4
- name: xnn_f32_qc8w_gemm_relu_ukernel_6x8s4__wasmrelaxedsimd_fma
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 4
# WAsm
- name: xnn_f32_qc8w_gemm_relu_ukernel_1x4__wasm
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
- name: xnn_f32_qc8w_gemm_relu_ukernel_2x4__wasm
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
- name: xnn_f32_qc8w_gemm_relu_ukernel_4x2__wasm
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
- name: xnn_f32_qc8w_gemm_relu_ukernel_4x4__wasm
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
# Scalar
- name: xnn_f32_qc8w_gemm_relu_ukernel_1x4__scalar
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
- name: xnn_f32_qc8w_gemm_relu_ukernel_2x4__scalar
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
- name: xnn_f32_qc8w_gemm_relu_ukernel_4x2__scalar
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1
- name: xnn_f32_qc8w_gemm_relu_ukernel_4x4__scalar
pack: xnn_pack_f32_qs8w_gemm_goi_w
k-block: 1