/riva-asrlib-decoder

Standalone implementation of the CUDA-accelerated WFST Decoder available in Riva

Primary LanguageC++

Riva ASR Library CUDA WFST Decoder

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.

Build and Test

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

Bazel build is not being used. Please don't expect it to work.