xref: /aosp_15_r20/external/XNNPACK/scripts/generate-x8-lut.sh (revision 4bdc94577ba0e567308109d787f7fec7b531ce36)
1*4bdc9457SAndroid Build Coastguard Worker#!/bin/sh
2*4bdc9457SAndroid Build Coastguard Worker# Copyright 2021 Google LLC
3*4bdc9457SAndroid Build Coastguard Worker#
4*4bdc9457SAndroid Build Coastguard Worker# This source code is licensed under the BSD-style license found in the
5*4bdc9457SAndroid Build Coastguard Worker# LICENSE file in the root directory of this source tree.
6*4bdc9457SAndroid Build Coastguard Worker
7*4bdc9457SAndroid Build Coastguard Worker#################################### Scalar ###################################
8*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/scalar.c.in -D BATCH_TILE=1  -o src/x8-lut/gen/lut-scalar-x1.c &
9*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/scalar.c.in -D BATCH_TILE=2  -o src/x8-lut/gen/lut-scalar-x2.c &
10*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/scalar.c.in -D BATCH_TILE=4  -o src/x8-lut/gen/lut-scalar-x4.c &
11*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/scalar.c.in -D BATCH_TILE=8  -o src/x8-lut/gen/lut-scalar-x8.c &
12*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/scalar.c.in -D BATCH_TILE=16 -o src/x8-lut/gen/lut-scalar-x16.c &
13*4bdc9457SAndroid Build Coastguard Worker
14*4bdc9457SAndroid Build Coastguard Worker################################## WAsm SIMD ##################################
15*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/wasmsimd.c.in -D BATCH_TILE=16 -o src/x8-lut/gen/lut-wasmsimd-x16.c &
16*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/wasmsimd.c.in -D BATCH_TILE=32 -o src/x8-lut/gen/lut-wasmsimd-x32.c &
17*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/wasmsimd.c.in -D BATCH_TILE=48 -o src/x8-lut/gen/lut-wasmsimd-x48.c &
18*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/wasmsimd.c.in -D BATCH_TILE=64 -o src/x8-lut/gen/lut-wasmsimd-x64.c &
19*4bdc9457SAndroid Build Coastguard Worker
20*4bdc9457SAndroid Build Coastguard Worker################################## ARM64 NEON #################################
21*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/neon-tbx128x4.c.in -D BATCH_TILE=16 -o src/x8-lut/gen/lut-neon-tbx128x4-x16.c &
22*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/neon-tbx128x4.c.in -D BATCH_TILE=32 -o src/x8-lut/gen/lut-neon-tbx128x4-x32.c &
23*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/neon-tbx128x4.c.in -D BATCH_TILE=48 -o src/x8-lut/gen/lut-neon-tbx128x4-x48.c &
24*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/neon-tbx128x4.c.in -D BATCH_TILE=64 -o src/x8-lut/gen/lut-neon-tbx128x4-x64.c &
25*4bdc9457SAndroid Build Coastguard Worker
26*4bdc9457SAndroid Build Coastguard Worker################################### x86 SSE ###################################
27*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/ssse3.c.in -D AVX=0 -D BATCH_TILE=16 -o src/x8-lut/gen/lut-ssse3-x16.c &
28*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/ssse3.c.in -D AVX=0 -D BATCH_TILE=32 -o src/x8-lut/gen/lut-ssse3-x32.c &
29*4bdc9457SAndroid Build Coastguard Worker
30*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/ssse3.c.in -D AVX=1 -D BATCH_TILE=16 -o src/x8-lut/gen/lut-avx-x16.c &
31*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/ssse3.c.in -D AVX=1 -D BATCH_TILE=32 -o src/x8-lut/gen/lut-avx-x32.c &
32*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/ssse3.c.in -D AVX=1 -D BATCH_TILE=48 -o src/x8-lut/gen/lut-avx-x48.c &
33*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/ssse3.c.in -D AVX=1 -D BATCH_TILE=64 -o src/x8-lut/gen/lut-avx-x64.c &
34*4bdc9457SAndroid Build Coastguard Worker
35*4bdc9457SAndroid Build Coastguard Worker################################### x86 AVX2 ##################################
36*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/avx2.c.in -D BATCH_TILE=32  -o src/x8-lut/gen/lut-avx2-x32.c &
37*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/avx2.c.in -D BATCH_TILE=64  -o src/x8-lut/gen/lut-avx2-x64.c &
38*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/avx2.c.in -D BATCH_TILE=96  -o src/x8-lut/gen/lut-avx2-x96.c &
39*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/avx2.c.in -D BATCH_TILE=128 -o src/x8-lut/gen/lut-avx2-x128.c &
40*4bdc9457SAndroid Build Coastguard Worker
41*4bdc9457SAndroid Build Coastguard Worker################################## x86 AVX512 #################################
42*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/avx512skx-vpshufb.c.in -D BATCH_TILE=64  -o src/x8-lut/gen/lut-avx512skx-vpshufb-x64.c &
43*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/avx512skx-vpshufb.c.in -D BATCH_TILE=128 -o src/x8-lut/gen/lut-avx512skx-vpshufb-x128.c &
44*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/avx512skx-vpshufb.c.in -D BATCH_TILE=192 -o src/x8-lut/gen/lut-avx512skx-vpshufb-x192.c &
45*4bdc9457SAndroid Build Coastguard Workertools/xngen src/x8-lut/avx512skx-vpshufb.c.in -D BATCH_TILE=256 -o src/x8-lut/gen/lut-avx512skx-vpshufb-x256.c &
46*4bdc9457SAndroid Build Coastguard Worker
47*4bdc9457SAndroid Build Coastguard Worker################################## Unit tests #################################
48*4bdc9457SAndroid Build Coastguard Workertools/generate-lut-test.py --spec test/x8-lut.yaml --output test/x8-lut.cc &
49*4bdc9457SAndroid Build Coastguard Worker
50*4bdc9457SAndroid Build Coastguard Workerwait
51