This code repository exposes the CUDA WFST decoder originally
described here https://arxiv.org/abs/1910.10032 as a C++ library and
as a python library (via wrapping the C++ library with pybind11). It
is the same decoder used when you specify --decoder_type=kaldi when
running riva-build
in NVIDIA Riva.
To build just the offline binary:
mkdir build
cd build
cmake -DRIVA_ASRLIB_BUILD_PYTHON_BINDINGS=NO ..
cmake --build --target all --parallel
# Run build/offline-cuda-decode-binary
To build the python bindings and optionally run the python test cases:
# Optionally set up a conda environment first.
pip install -e .[testing]
python -m unittest src/riva/asrlib/decoder/test_decoder.py 2>&1 | tee test.log
Bazel build is not being used. Please don't expect it to work.