test / scripts /generate-f16-velu.sh
Androidonnxfork's picture
Upload folder using huggingface_hub
8b7c501
raw
history blame
No virus
969 Bytes
#!/bin/sh
# Copyright 2022 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/f16-velu/neonfp16arith-rr1-p3.c.in -D BATCH_TILE=8 -o src/f16-velu/gen/f16-velu-neonfp16arith-rr1-p3-x8.c &
tools/xngen src/f16-velu/neonfp16arith-rr1-p3.c.in -D BATCH_TILE=16 -o src/f16-velu/gen/f16-velu-neonfp16arith-rr1-p3-x16.c &
################################### x86 AVX2 ##################################
tools/xngen src/f16-velu/avx2-rr1-p3.c.in -D BATCH_TILE=8 -o src/f16-velu/gen/f16-velu-avx2-rr1-p3-x8.c &
tools/xngen src/f16-velu/avx2-rr1-p3.c.in -D BATCH_TILE=16 -o src/f16-velu/gen/f16-velu-avx2-rr1-p3-x16.c &
################################## Unit tests #################################
tools/generate-vunary-test.py --spec test/f16-velu.yaml --output test/f16-velu.cc &
wait