Source code
Revision control
Copy as Markdown
Other Tools
# -*- Mode: python; indent-tabs-mode: nil; tab-width: 40 -*-
# vim: set filetype=python:
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
with Files('**'):
BUG_COMPONENT = ('Core', 'Machine Learning')
# Built into the mozinference shared library alongside the vendored ggml (from
# llama.cpp). Only the parakeet_capi_* symbols are exported, via
# config/external/mozinference/mozinference.symbols.
SOURCES = [
"src/audio_io.cpp",
"src/backend.cpp",
"src/common.cpp",
"src/conformer.cpp",
"src/ctc_decoder.cpp",
"src/encoder.cpp",
"src/fft.cpp",
"src/ggml_graph.cpp",
"src/joint.cpp",
"src/mel.cpp",
"src/mel_gpu.cpp",
"src/model.cpp",
"src/model_loader.cpp",
"src/parakeet_capi.cpp",
# src/parakeet.cpp is intentionally omitted: it is a thin convenience
# wrapper (parakeet_version / parakeet_transcribe_file) whose C symbols
# collide with the other vendored backend. We use the parakeet_capi_* API.
"src/pos_enc.cpp",
"src/prediction.cpp",
"src/prompt_kernel.cpp",
"src/relpos_attention.cpp",
"src/rnnt.cpp",
"src/search.cpp",
"src/streaming.cpp",
"src/streaming_encoder.cpp",
"src/subsampling.cpp",
"src/tdt.cpp",
"src/tokenizer.cpp",
"src/transcription.cpp",
"src/transducer_batch.cpp",
]
EXPORTS += [
"include/parakeet_capi.h",
]
LOCAL_INCLUDES += [
'/third_party/llama.cpp/',
'/third_party/llama.cpp/ggml/',
'/third_party/llama.cpp/ggml/include/',
'/third_party/llama.cpp/ggml/src/',
'/third_party/llama.cpp/ggml/src/ggml-cpu/',
'/third_party/parakeet.cpp/',
'/third_party/parakeet.cpp/include/',
'/third_party/parakeet.cpp/src/',
]
# Vendored third-party code; quiet the warnings it triggers under our flags.
AllowCompilerWarnings()
# The parakeet_capi_* entry points are plain extern "C" without an export
# attribute, so drop Gecko's -fvisibility=hidden for this directory; otherwise
# they end up GLOBAL HIDDEN and the symbols file can't export them. The set
# actually exported is still restricted by mozinference.symbols.
NoVisibilityFlags()
FINAL_LIBRARY = 'mozinference'