Skip to content

[Perf][Kernel] Tune cooperative topk for medium batch-sizes - #53382

Merged
vllm-bot merged 2 commits into
vllm-project:mainfrom
LopezCastroRoberto:perf/tune_cooperative
Aug 31, 2026
Merged

vllm-bot merged 2 commits into
vllm-project:mainfrom
LopezCastroRoberto:perf/tune_cooperative

Conversation

@LopezCastroRoberto

@LopezCastroRoberto LopezCastroRoberto commented Aug 22, 2026

Copy link
Copy Markdown
Contributor

Speedups w.r.t. FilteredTopK, which is the backend selected at this moment in MAIN for bs>32

topK=512

Sequence length BS=33 BS=40 BS=48 BS=64
64K 2.33x 1.82x 1.79x 1.82x
96K 2.24x 1.66x 1.65x 1.64x
128K 2.52x 1.40x 1.37x 1.37x
160K 2.71x 1.45x 1.45x 1.43x
192K 2.99x 1.52x 1.52x 1.53x
224K 2.38x 1.59x 1.60x 1.57x
250K 2.60x 1.70x 1.69x 1.69x

topK=1024

Sequence length BS=33 BS=40 BS=48 BS=64
64K 2.30x 1.74x 1.75x 1.76x
96K 2.65x 1.93x 1.97x 1.97x
128K 3.08x 1.67x 1.64x 1.65x
160K 3.27x 1.73x 1.72x 1.72x
192K 2.89x 1.50x 1.48x 1.48x
224K 2.36x 1.55x 1.54x 1.53x
250K 2.57x 1.67x 1.67x 1.67x

topK=2048

Sequence length BS=33 BS=40 BS=48 BS=64
64K 2.25x 1.73x 1.71x 1.75x
96K 2.39x 1.78x 1.77x 1.76x
128K 2.84x 1.53x 1.53x 1.53x
160K 3.02x 1.60x 1.62x 1.58x
192K 3.28x 1.73x 1.70x 1.71x
224K 2.68x 1.75x 1.77x 1.75x
250K 2.93x 1.92x 1.90x 1.90x

Co-authored-by: Codex <codex@openai.com>
Signed-off-by: LopezCastroRoberto <rocastro@redhat.com>

@claude claude Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

@LopezCastroRoberto LopezCastroRoberto changed the title [Perf] Tune cooperative topk medium batchsizes [Perf][Kernel] Tune cooperative topk for medium batch-sizes Aug 22, 2026
@mgoin
mgoin enabled auto-merge (squash) August 31, 2026 16:53
@github-actions github-actions Bot added the ready ONLY add when PR is ready to merge/full CI is needed label Aug 31, 2026
@mgoin mgoin added performance Performance-related issues deepseek Related to DeepSeek models labels Aug 31, 2026
@mgoin

mgoin commented Aug 31, 2026

Copy link
Copy Markdown
Member

/ci run

@github-actions

Copy link
Copy Markdown

✅ Triggered Buildkite CI #86392 for commit f81c8ced02f5.

@vllm-bot
vllm-bot merged commit f5c3cc2 into vllm-project:main Aug 31, 2026
271 of 276 checks passed
am-cohere pushed a commit to am-cohere/vllm that referenced this pull request Sep 1, 2026
…ject#53382)

Signed-off-by: LopezCastroRoberto <rocastro@redhat.com>
Co-authored-by: Codex <codex@openai.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
mylibrar pushed a commit to tanyuqian/vllm that referenced this pull request Sep 3, 2026
…ject#53382)

Signed-off-by: LopezCastroRoberto <rocastro@redhat.com>
Co-authored-by: Codex <codex@openai.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
D-G-Dimitrov pushed a commit to D-G-Dimitrov/vllm that referenced this pull request Sep 8, 2026
…ject#53382)

Signed-off-by: LopezCastroRoberto <rocastro@redhat.com>
Co-authored-by: Codex <codex@openai.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
(cherry picked from commit f5c3cc2)
sheralskumar pushed a commit to sheralskumar/vllm that referenced this pull request Sep 8, 2026
…ject#53382)

Signed-off-by: LopezCastroRoberto <rocastro@redhat.com>
Co-authored-by: Codex <codex@openai.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

deepseek Related to DeepSeek models performance Performance-related issues ready ONLY add when PR is ready to merge/full CI is needed

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants