vllm
[SpecDecode] Reduce TP communication for large-vocab draft models speculative decoding
#39419
Merged

[SpecDecode] Reduce TP communication for large-vocab draft models speculative decoding #39419

EanWang211123
EanWang211123 [feat] add local_argmax_reduction for dflash and qwen3-pard
886b2d5d
EanWang211123 EanWang211123 requested a review from sighingnow sighingnow 99 days ago
EanWang211123 EanWang211123 requested a review from vadiklyutiy vadiklyutiy 99 days ago
EanWang211123 EanWang211123 marked this pull request as draft 99 days ago
mergify mergify added qwen
gemini-code-assist
gemini-code-assist commented on 2026-04-09
EanWang211123 EanWang211123 marked this pull request as ready for review 99 days ago
EanWang211123 EanWang211123 changed the title [SpecDecode] Reduce TP communication for large-vocab draft models in DFlash/PARD [SpecDecode] Reduce TP communication for large-vocab draft models in DFlash/PARD speculative decoding 99 days ago
benchislett
benchislett benchislett added verified
EanWang211123 Merge branch 'main' into local_argmax_for_dflash
c8beb07e
EanWang211123 [feat] add local-argmax for eagle3 llama p-eagle
8b246ce1
mergify mergify added llama
mergify mergify added speculative-decoding
EanWang211123
benchislett
benchislett commented on 2026-04-30
benchislett
benchislett commented on 2026-04-30
benchislett
benchislett commented on 2026-04-30
benchislett
benchislett commented on 2026-04-30
EanWang211123 Merge branch 'main' into local_argmax_for_dflash
8aa572fd
EanWang211123 Merge branch 'main' into local_argmax_for_dflash
fe1453a1
EanWang211123 [feat] refact and add local-argmax logits
db23525c
EanWang211123 [update] support local-argmx for qwen3-5
5587b3de
EanWang211123 EanWang211123 requested a review from luccafong luccafong 46 days ago
EanWang211123 EanWang211123 requested a review from MatthewBonanni MatthewBonanni 46 days ago
EanWang211123 EanWang211123 requested a review from benchislett benchislett 46 days ago
EanWang211123 EanWang211123 changed the title [SpecDecode] Reduce TP communication for large-vocab draft models in DFlash/PARD speculative decoding [SpecDecode] Reduce TP communication for large-vocab draft models speculative decoding 46 days ago
EanWang211123 [update] remove draft_id_to_target_id init logits in llama_eagle3
1137e682
mergify
mergify mergify added deepseek
mergify mergify added v1
EanWang211123 [feat] implement the logits for llama qwen
d97994a6
EanWang211123 [fix] fix MRO conflict
bc039415
mergify
EanWang211123 [fix] fix lint
4aa766c7
EanWang211123
EanWang211123
EanWang211123 Merge branch 'main' into local_argmax_for_dflash
bf3bd55d
EanWang211123 Merge branch 'main' into local_argmax_for_dflash
fafc1304
benchislett
benchislett
benchislett commented on 2026-06-09
benchislett
benchislett approved these changes on 2026-06-10
benchislett benchislett added ready
benchislett benchislett enabled auto-merge (squash) 37 days ago
EanWang211123 Merge branch 'main' into local_argmax_for_dflash
773f68d2
benchislett benchislett merged bd2d83ff into main 37 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone