xref: /aosp_15_r20/external/XNNPACK/scripts/generate-f32-vrelu.sh (revision 4bdc94577ba0e567308109d787f7fec7b531ce36)
1*4bdc9457SAndroid Build Coastguard Worker#!/bin/sh
2*4bdc9457SAndroid Build Coastguard Worker# Copyright 2020 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 Worker### Generic C micro-kernels
9*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/scalar.c.in -D BATCH_TILE=1 -o src/f32-vrelu/gen/vrelu-scalar-x1.c &
10*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/scalar.c.in -D BATCH_TILE=2 -o src/f32-vrelu/gen/vrelu-scalar-x2.c &
11*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/scalar.c.in -D BATCH_TILE=4 -o src/f32-vrelu/gen/vrelu-scalar-x4.c &
12*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/scalar.c.in -D BATCH_TILE=8 -o src/f32-vrelu/gen/vrelu-scalar-x8.c &
13*4bdc9457SAndroid Build Coastguard Worker
14*4bdc9457SAndroid Build Coastguard Worker### WAsm-specific micro-kernels
15*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/wasm.c.in -D BATCH_TILE=1 -o src/f32-vrelu/gen/vrelu-wasm-x1.c &
16*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/wasm.c.in -D BATCH_TILE=2 -o src/f32-vrelu/gen/vrelu-wasm-x2.c &
17*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/wasm.c.in -D BATCH_TILE=4 -o src/f32-vrelu/gen/vrelu-wasm-x4.c &
18*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/wasm.c.in -D BATCH_TILE=8 -o src/f32-vrelu/gen/vrelu-wasm-x8.c &
19*4bdc9457SAndroid Build Coastguard Worker
20*4bdc9457SAndroid Build Coastguard Worker################################## WAsm SIMD ##################################
21*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/wasmsimd.c.in -D BATCH_TILE=4  -o src/f32-vrelu/gen/vrelu-wasmsimd-x4.c &
22*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/wasmsimd.c.in -D BATCH_TILE=8  -o src/f32-vrelu/gen/vrelu-wasmsimd-x8.c &
23*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/wasmsimd.c.in -D BATCH_TILE=16 -o src/f32-vrelu/gen/vrelu-wasmsimd-x16.c &
24*4bdc9457SAndroid Build Coastguard Worker
25*4bdc9457SAndroid Build Coastguard Worker################################### ARM NEON ##################################
26*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/neon.c.in -D BATCH_TILE=4 -o src/f32-vrelu/gen/vrelu-neon-x4.c &
27*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/neon.c.in -D BATCH_TILE=8 -o src/f32-vrelu/gen/vrelu-neon-x8.c &
28*4bdc9457SAndroid Build Coastguard Worker
29*4bdc9457SAndroid Build Coastguard Worker################################# x86 128-bit #################################
30*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/sse.c.in -D BATCH_TILE=4 -o src/f32-vrelu/gen/vrelu-sse-x4.c &
31*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/sse.c.in -D BATCH_TILE=8 -o src/f32-vrelu/gen/vrelu-sse-x8.c &
32*4bdc9457SAndroid Build Coastguard Worker
33*4bdc9457SAndroid Build Coastguard Worker################################# x86 256-bit #################################
34*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/avx.c.in -D BATCH_TILE=8  -o src/f32-vrelu/gen/vrelu-avx-x8.c &
35*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/avx.c.in -D BATCH_TILE=16 -o src/f32-vrelu/gen/vrelu-avx-x16.c &
36*4bdc9457SAndroid Build Coastguard Worker
37*4bdc9457SAndroid Build Coastguard Worker################################# x86 512-bit #################################
38*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/avx512f.c.in -D BATCH_TILE=16 -o src/f32-vrelu/gen/vrelu-avx512f-x16.c &
39*4bdc9457SAndroid Build Coastguard Workertools/xngen src/f32-vrelu/avx512f.c.in -D BATCH_TILE=32 -o src/f32-vrelu/gen/vrelu-avx512f-x32.c &
40*4bdc9457SAndroid Build Coastguard Worker
41*4bdc9457SAndroid Build Coastguard Worker################################## Unit tests #################################
42*4bdc9457SAndroid Build Coastguard Workertools/generate-vunary-test.py --spec test/f32-vrelu.yaml --output test/f32-vrelu.cc &
43*4bdc9457SAndroid Build Coastguard Worker
44*4bdc9457SAndroid Build Coastguard Workerwait
45