Skip to content

[CI][KVConnector][Metrics] Update multi KV connector edge case according to prefill stats changes#39808

Merged
markmc merged 2 commits intovllm-project:mainfrom
ZhanqiuHu:fix/multi-connector-metrics-assertions
Apr 14, 2026
Merged

[CI][KVConnector][Metrics] Update multi KV connector edge case according to prefill stats changes#39808
markmc merged 2 commits intovllm-project:mainfrom
ZhanqiuHu:fix/multi-connector-metrics-assertions

Conversation

@ZhanqiuHu
Copy link
Copy Markdown
Contributor

@ZhanqiuHu ZhanqiuHu commented Apr 14, 2026

Summary

  • PR [Core][Metrics][BugFix] Replace num_cached_tokens/num_external_computed_tokens with PrefillStats #37460 introduced PrefillStats with strict non-overlapping token source accounting (local_compute + local_cache_hit + external_kv_transfer = num_prompt_tokens).
  • Under this new invariant, the last-token recompute in PD disagg cases is no longer separately attributed to local_compute, since PrefillStats.set() is called before the -1 adjustment in _update_waiting_for_remote_kv().
  • Updated three metrics tests in test_multi_connector_edge_cases.py:
    • local_compute: 10
    • local_cache_hit: cached - 1cached

Test

  • run_multi_connector_edge_case_test.sh

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.
  • (Optional) Release notes update. If your change is user facing, please update the release notes draft in the Google Doc.

Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request updates the integration tests for the KV connector to reflect changes in metric reporting. Specifically, it modifies the expected values for local_compute and local_cache_hit across several test cases, ensuring that local_compute is now expected to be 0 and local_cache_hit matches the full cached count. I have no feedback to provide as there were no review comments to evaluate.

@markmc markmc changed the title [CI][KVConnetor] Update multi kv connector edge case according to prefill stats changes [CI][KVConnector][Metrics] Update multi KV connector edge case according to prefill stats changes Apr 14, 2026
@markmc markmc moved this from Backlog to Ready in Metrics & Tracing Apr 14, 2026
@markmc markmc enabled auto-merge (squash) April 14, 2026 16:12
@github-actions github-actions Bot added the ready ONLY add when PR is ready to merge/full CI is needed label Apr 14, 2026
auto-merge was automatically disabled April 14, 2026 16:13

Head branch was pushed to by a user without write access

@ZhanqiuHu ZhanqiuHu force-pushed the fix/multi-connector-metrics-assertions branch from 2da40eb to 83db6c9 Compare April 14, 2026 16:13
@markmc markmc enabled auto-merge (squash) April 14, 2026 16:14
@markmc markmc merged commit 3067931 into vllm-project:main Apr 14, 2026
20 checks passed
@github-project-automation github-project-automation Bot moved this from Ready to Done in Metrics & Tracing Apr 14, 2026
zxd1997066 pushed a commit to zxd1997066/vllm that referenced this pull request Apr 15, 2026
…ing to prefill stats changes (vllm-project#39808)

Signed-off-by: Zhanqiu Hu <zhu@redhat.com>
Signed-off-by: zengxian <xiangdong.zeng@intel.com>
whk-lab pushed a commit to whk-lab/vllm that referenced this pull request Apr 23, 2026
…ing to prefill stats changes (vllm-project#39808)

Signed-off-by: Zhanqiu Hu <zhu@redhat.com>
avinashsingh77 pushed a commit to avinashsingh77/vllm that referenced this pull request Apr 27, 2026
…ing to prefill stats changes (vllm-project#39808)

Signed-off-by: Zhanqiu Hu <zhu@redhat.com>
Signed-off-by: Avinash Singh <avinashsingh.rcoem@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

kv-connector ready ONLY add when PR is ready to merge/full CI is needed v1

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

3 participants