Skip to content

[TRTLLM-13321][feat] Extend rejection sampling to one-model speculative decoding (MTP/DRAFT_TARGET/PARD/DFLASH)#15775

Merged
zhaoyangwang-nvidia merged 8 commits into
NVIDIA:mainfrom
zhaoyangwang-nvidia:rejection-one-model-spec
Jul 16, 2026
Merged

[TRTLLM-13321][feat] Extend rejection sampling to one-model speculative decoding (MTP/DRAFT_TARGET/PARD/DFLASH)#15775
zhaoyangwang-nvidia merged 8 commits into
NVIDIA:mainfrom
zhaoyangwang-nvidia:rejection-one-model-spec

[TRTLLM-13321][fix] Do not TP-gather draft logits under attention DP

f4334c9
Select commit
Loading
Failed to load commit list.