hf_text-generation-inference/backends/v3
Daniël de Kok 8deeaca4ff
Add support for prefix caching to the v3 router (#2392)
This change adds support for prefix caching to the v3 router. This
is broken up from the backend support to ease reviewing.

For now prefix caching is only enabled with `USE_PREFIX_CACHING=1`
in this case, the router will switch to `RadixAllocator`. This
allocator uses a radix trie to keep track of prefills that were
seen prior. If a new prefill is a prefix of a previously-seen
prefil, the router will send a request with `prefix_len>0`, which
can be used by the backend to decide to reuse KV blocks from the
cache, rather than recomputing them.

Even though backend support is not added in this PR, the backend
will still work with prefix caching enabled. The prefix lengths
are just ignored and not used.
2024-08-12 14:59:17 +02:00
..
src Add support for prefix caching to the v3 router (#2392) 2024-08-12 14:59:17 +02:00
Cargo.toml Add support for prefix caching to the v3 router (#2392) 2024-08-12 14:59:17 +02:00
build.rs Rebase TRT-llm (#2331) 2024-07-31 10:33:10 +02:00