Skip to content

[Minor] Small pooler output processing optimization#31667

Merged
vllm-bot merged 1 commit intovllm-project:mainfrom
njhill:pool-small-opt
Jan 5, 2026
Merged

[Minor] Small pooler output processing optimization#31667
vllm-bot merged 1 commit intovllm-project:mainfrom
njhill:pool-small-opt

Conversation

@njhill
Copy link
Member

@njhill njhill commented Jan 4, 2026

Small follow-on from #31584, I realized that the final output grouping/slicing can be done prior to the cpu sync since it only contains views.

Signed-off-by: njhill <nickhill123@gmail.com>
@njhill njhill requested a review from DarkLight1337 January 4, 2026 06:41
@mergify mergify bot added the v1 label Jan 4, 2026
Copy link
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 introduces a small but effective optimization for pooler output processing. By moving the output filtering logic to occur before the CPU-GPU synchronization, it allows for better overlap of CPU and GPU work. The changes correctly implement this optimization for both synchronous and asynchronous scheduling paths. The code is clean and the logic is sound. I have no further comments.

@njhill njhill added the ready ONLY add when PR is ready to merge/full CI is needed label Jan 4, 2026
@DarkLight1337 DarkLight1337 enabled auto-merge (squash) January 4, 2026 07:47
@vllm-bot vllm-bot merged commit da436f8 into vllm-project:main Jan 5, 2026
48 of 51 checks passed
@njhill njhill deleted the pool-small-opt branch January 5, 2026 04:35
LucasWilkinson pushed a commit to neuralmagic/vllm that referenced this pull request Jan 6, 2026
yugong333 pushed a commit to yugong333/vllm that referenced this pull request Jan 9, 2026
akh64bit pushed a commit to akh64bit/vllm that referenced this pull request Jan 16, 2026
dsuhinin pushed a commit to dsuhinin/vllm that referenced this pull request Jan 21, 2026
Signed-off-by: njhill <nickhill123@gmail.com>
Signed-off-by: dsuhinin <suhinin.dmitriy@gmail.com>
ItzDEXX pushed a commit to ItzDEXX/vllm that referenced this pull request Feb 19, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

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

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants