sglang_v0.5.2/pytorch_2.8.0/third_party/XNNPACK/scripts/generate-s32-f32-vcvt.sh

23 lines
1.2 KiB
Bash
Executable File

#!/bin/sh
# Copyright 2024 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 ###################################
tools/xngen src/s32-f32-vcvt/simd.c.in -D BATCH_TILES=4,8,12,16 -D ARCH=neon -o src/s32-f32-vcvt/gen/s32-f32-vcvt-neon.c
################################# x86 AVX2 #################################
tools/xngen src/s32-f32-vcvt/simd.c.in -D BATCH_TILES=8,16,24,32 -D ARCH=avx2 -o src/s32-f32-vcvt/gen/s32-f32-vcvt-avx2.c
################################# x86 AVX512 #################################
tools/xngen src/s32-f32-vcvt/simd.c.in -D BATCH_TILES=16,32,48,64 -D ARCH=avx512f -o src/s32-f32-vcvt/gen/s32-f32-vcvt-avx512f.c
################################## WAsm SIMD ##################################
tools/xngen src/s32-f32-vcvt/simd.c.in -D BATCH_TILES=4,8,12,16 -D ARCH=wasmsimd -o src/s32-f32-vcvt/gen/s32-f32-vcvt-wasmsimd.c
#################################### Scalar ###################################
tools/xngen src/s32-f32-vcvt/simd.c.in -D BATCH_TILES=1,2,3,4 -D ARCH=scalar -o src/s32-f32-vcvt/gen/s32-f32-vcvt-scalar.c
wait