49 lines
4.0 KiB
Bash
Executable File
49 lines
4.0 KiB
Bash
Executable File
#!/bin/bash
|
|
INITIAL_MEMORY=${INITIAL_MEMORY:-315mb}
|
|
MAX_THREADS=${MAX_THREADS:-1}
|
|
EMSDK=${EMSDK:-emsdk}
|
|
JOBS=${JOBS:-$(nproc)}
|
|
|
|
if [ "$EMSDK" != emsdk ] && [ ! -d "$EMSDK" ]; then
|
|
echo "Invalid emsdk path"
|
|
exit 1
|
|
fi
|
|
if [ "$MAX_THREADS" -lt 1 ]; then
|
|
echo "MAX_THREADS must be greater than or equal to 1"
|
|
exit 1
|
|
fi
|
|
if [ "$JOBS" -lt 1 ]; then
|
|
echo "JOBS must be greater than or equal to 1"
|
|
exit 1
|
|
fi
|
|
if ! [[ $INITIAL_MEMORY =~ ^[0-9]+([kmgt]b)?$ ]]; then
|
|
echo "INITIAL_MEMORY valid suffixes are kb, mb, gb, tb, none (bytes)"
|
|
exit 1
|
|
fi
|
|
if [ "$EMSDK" = emsdk ] && [ ! -d "$EMSDK" ]; then
|
|
echo "Installing emsdk + Emscripten..."
|
|
git clone --depth=1 https://github.com/emscripten-core/emsdk.git ../emsdk &&
|
|
cd ../emsdk &&
|
|
./emsdk install 3.1.67 &&
|
|
./emsdk activate 3.1.67
|
|
fi
|
|
. $(realpath "$EMSDK")/emsdk_env.sh &&
|
|
export PATH=:$PATH:$(realpath "$EMSDK")/upstream/bin
|
|
|
|
KALDI=$(realpath kaldi)
|
|
VOSK=$(realpath vosk)
|
|
OPENFST=$(realpath openfst)
|
|
CLAPACK_WASM=$(realpath clapack-wasm)
|
|
|
|
cd src &&
|
|
MODE=0 && # 0: Debug, 1: Optimized release
|
|
echo "Mode = $MODE" &&
|
|
if [ "$MODE" = 0 ]; then
|
|
em++ Util.cc CommonModel.cc Recognizer.cc Bindings.cc -O0 -std=c++23 -g3 -Wall -Wextra -Wpedantic -fno-rtti -DEMSCRIPTEN_HAS_UNBOUND_TYPE_NAMES=0 -DMAX_WORKERS="$MAX_THREADS" -sWASMFS -sWASM_BIGINT -sMODULARIZE -sWASM_EXNREF -sTEXTDECODER=2 -sWASM_WORKERS=2 -sEVAL_CTORS=2 -sSTACK_OVERFLOW_CHECK=2 -sASSERTIONS=2 -sINITIAL_MEMORY="$INITIAL_MEMORY" -sALLOW_MEMORY_GROWTH -sRUNTIME_DEBUG -sPOLYFILL=0 -sEXIT_RUNTIME=0 -sINVOKE_RUN=0 -sDISABLE_EXCEPTION_CATCHING=0 -sINCOMING_MODULE_JS_API=wasmMemory,instantiateWasm,wasm -sEXPORT_NAME=loadVosklet -sMALLOC=emmalloc -sENVIRONMENT=web,worker -I. -I"$VOSK"/src -L"$KALDI"/src -l:online2/kaldi-online2.a -l:decoder/kaldi-decoder.a -l:ivector/kaldi-ivector.a -l:gmm/kaldi-gmm.a -l:tree/kaldi-tree.a -l:feat/kaldi-feat.a -l:cudamatrix/kaldi-cudamatrix.a -l:lat/kaldi-lat.a -l:lm/kaldi-lm.a -l:rnnlm/kaldi-rnnlm.a -l:hmm/kaldi-hmm.a -l:nnet3/kaldi-nnet3.a -l:transform/kaldi-transform.a -l:matrix/kaldi-matrix.a -l:fstext/kaldi-fstext.a -l:util/kaldi-util.a -l:base/kaldi-base.a -L"$OPENFST"/lib -l:libfst.a -l:libfstngram.a -L"$CLAPACK_WASM" -l:CBLAS/lib/cblas.a -l:CLAPACK-3.2.1/lapack.a -l:CLAPACK-3.2.1/libcblaswr.a -l:f2c_BLAS-3.8.0/blas.a -l:libf2c/libf2c.a -L"$VOSK"/src -l:vosk.a -lembind -msimd128 -matomics -mbulk-memory -mreference-types -mnontrapping-fptoint -mextended-const -msign-ext -mmutable-globals --pre-js Wrapper.js -o ../test.js
|
|
elif [ "$MODE" = 1 ]; then
|
|
em++ Util.cc CommonModel.cc Recognizer.cc Bindings.cc -O3 -Wall -Wextra -Wpedantic -std=c++23 -fno-rtti -DEMSCRIPTEN_HAS_UNBOUND_TYPE_NAMES=0 -DMAX_WORKERS="$MAX_THREADS" -sWASMFS -sWASM_BIGINT -sMODULARIZE -sWASM_EXNREF -sTEXTDECODER=2 -sWASM_WORKERS=2 -sEVAL_CTORS=2 -sINITIAL_MEMORY="$INITIAL_MEMORY" -sALLOW_MEMORY_GROWTH -sPOLYFILL=0 -sEXIT_RUNTIME=0 -sINVOKE_RUN=0 -sSUPPORT_LONGJMP=0 -sINCOMING_MODULE_JS_API=wasmMemory,instantiateWasm,wasm -sEXPORT_NAME=loadVosklet -sMALLOC=emmalloc -sENVIRONMENT=web,worker -I. -I"$VOSK"/src -L"$KALDI"/src -l:online2/kaldi-online2.a -l:decoder/kaldi-decoder.a -l:ivector/kaldi-ivector.a -l:gmm/kaldi-gmm.a -l:tree/kaldi-tree.a -l:feat/kaldi-feat.a -l:cudamatrix/kaldi-cudamatrix.a -l:lat/kaldi-lat.a -l:lm/kaldi-lm.a -l:rnnlm/kaldi-rnnlm.a -l:hmm/kaldi-hmm.a -l:nnet3/kaldi-nnet3.a -l:transform/kaldi-transform.a -l:matrix/kaldi-matrix.a -l:fstext/kaldi-fstext.a -l:util/kaldi-util.a -l:base/kaldi-base.a -L"$OPENFST"/lib -l:libfst.a -l:libfstngram.a -L"$CLAPACK_WASM" -l:CBLAS/lib/cblas.a -l:CLAPACK-3.2.1/lapack.a -l:CLAPACK-3.2.1/libcblaswr.a -l:f2c_BLAS-3.8.0/blas.a -l:libf2c/libf2c.a -L"$VOSK"/src -l:vosk.a -lembind -flto -msimd128 -matomics -mbulk-memory -mreference-types -mnontrapping-fptoint -mextended-const -msign-ext -mmutable-globals --pre-js Wrapper.js --closure 1 -o ../test.js &&
|
|
tr -d '\n' < ../test.js | tr -s ' ' > /tmp/hehe && mv /tmp/hehe ../test.js
|
|
fi
|
|
cd .. &&
|
|
sed -i 's/.pipeThrough(new CompressionStream("gzip"))//;s/\[1\])/[1].pipeThrough(new DecompressionStream("gzip")))/' test.js &&
|
|
rm -f test.worker.js |