Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[V1] APC + prompt logprobs unsupported (PR 2/N for v1 sample and prompt logprobs support) #11910

Open
wants to merge 368 commits into
base: main
Choose a base branch
from

Conversation

afeldman-nm
Copy link
Contributor

@afeldman-nm afeldman-nm commented Jan 10, 2025

This PR forces the v1 LLMEngine and AsyncLLM to fail in generate() if a request with prompt logprobs enabled arrives and the engine has prefix caching enabled. Note that while it is possible for prompt logprobs and APC to be compatible, this is left to future work.

This is part of a broader vllm v1 + logprobs workstream:

PR no. 1 (#9880 ) is a dependency of this PR; it adds the infrastructure for sample and prompt logprobs support with limited unit tests

PR no. 2 is this PR

PR no. 3 will port test_completion.py logprobs tests to v1, make any adjustments required to make these tests pass, and fix a compatibility issue with Mistral tokenizer

DarkLight1337 and others added 30 commits December 2, 2024 11:36
…oject#10804)

Signed-off-by: Roger Wang <[email protected]>
Co-authored-by: Chen Zhang <[email protected]>
Co-authored-by: Isotr0py <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
…t#10502)

This PR provides initial support for single-node disaggregated prefill in 1P1D scenario.
Signed-off-by: KuntaiDu <[email protected]>
Co-authored-by: ApostaC <[email protected]>
Co-authored-by: YaoJiayi <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: youkaichao <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: wangxiyuan <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: xffxff <[email protected]>
Signed-off-by: Isotr0py <[email protected]>
Co-authored-by: Isotr0py <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Co-authored-by: Woosuk Kwon <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
afeldman-nm and others added 23 commits January 12, 2025 21:24
Signed-off-by: Andrew Feldman <[email protected]>
…gprobs, and cumulative_logprob to RequestOutput

Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Signed-off-by: Andrew Feldman <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.