Skip to content

Feature/prophet kv#8

Open
almogtavor wants to merge 2 commits into
segmented_spansfrom
feature/prophet_kv
Open

Feature/prophet kv#8
almogtavor wants to merge 2 commits into
segmented_spansfrom
feature/prophet_kv

Conversation

@almogtavor

@almogtavor almogtavor commented Apr 29, 2026

Copy link
Copy Markdown
Collaborator

Purpose

Test Plan

Test Result


Essential Elements of an Effective PR Description Checklist
  • The purpose of the PR, such as "Fix some issue (link existing issues this PR will resolve)".
  • The test plan, such as providing test command.
  • The test results, such as pasting the results comparison before and after, or e2e results
  • (Optional) The necessary documentation update, such as updating supported_models.md and examples for a new model.

…mputation

- Added query capturing functionality in the Attention layer to collect queries for ProphetKV.
- Introduced new parameters in the AsyncScheduler and related classes to manage cached tokens and recompute requests.
- Implemented gap recomputation logic in the Scheduler to handle virtual requests efficiently.
- Enhanced NewRequestData to include cached token counts and support for gap recompute requests.
- Updated ProphetKVGapPolicy to include new tunables for gap thresholds and maximum spans.
- Refactored GPUModelRunner to support ProphetKV query collection and scoring, ensuring proper handling of cached tokens.
- Improved error handling and logging for ProphetKV operations, including warnings for unsupported configurations.
almogtavor pushed a commit that referenced this pull request Jun 2, 2026
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.

1 participant