Skip to content

[SpecDecode] Add local argmax helpers for Qwen3 drafts - #6

Draft
voipmonitor wants to merge 1300 commits into
mainfrom
sim/pr-qwen3-local-argmax
Draft

[SpecDecode] Add local argmax helpers for Qwen3 drafts#6
voipmonitor wants to merge 1300 commits into
mainfrom
sim/pr-qwen3-local-argmax

Conversation

@voipmonitor

Copy link
Copy Markdown
Owner

Simulated upstream-split PR tracked by #2.

This branch isolates local-argmax helpers for Qwen3 and Qwen3-DFlash draft models so speculative decode can avoid gathering full logits when that is safe.

Scope:

  • Qwen3 get_top_tokens helper
  • Qwen3-DFlash get_top_tokens helper with remapping fallback

This is intended as a follow-up style split relative to the upstream local-argmax work around vllm-project#39419.

jefp and others added 30 commits April 10, 2026 17:21
…llm-project#39435)

Signed-off-by: Jesus Federico <jefp@amazon.com>
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
…er Edition (vllm-project#39183)

Signed-off-by: manu <fortin.emmanuel@gmail.com>
Signed-off-by: Rahul-Tuli <rtuli@redhat.com>
Signed-off-by: Fynn Schmitt-Ulms <fschmitt@redhat.com>
Co-authored-by: Rahul-Tuli <rtuli@redhat.com>
Co-authored-by: Claude <noreply@anthropic.com>
Signed-off-by: Fynn Schmitt-Ulms <fschmitt@redhat.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>
)

Signed-off-by: Yan Ma <yan.ma@intel.com>
Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>
Signed-off-by: Tianyu Guo <guoty9@mail2.sysu.edu.cn>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
Signed-off-by: leeyongjun <jqueen.astro@gmail.com>
Signed-off-by: Ziying Tao <tzzying@outlook.com>
…adding (vllm-project#39064)

Signed-off-by: Vibhav Agarwal <vibhavagarwal5@gmail.com>
Co-authored-by: vibhav-agarwal <vibhav.agarwal@glance.com>
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
…ectly (vllm-project#38844)

Signed-off-by: ShubyM <shubymishra20@gmail.com>
…atch (vllm-project#38919)

Signed-off-by: Itay Etelis <itay.etelis@ibm.com>
Co-authored-by: Itay Etelis <itay.etelis@ibm.com>
Co-authored-by: Michael Goin <mgoin64@gmail.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Xinyu Chen <xinyu1.chen@intel.com>
Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>
…m-project#39344)

Signed-off-by: r266-tech <r266.tech@gmail.com>
Signed-off-by: Roger Wang <hey@rogerw.io>
Co-authored-by: Roger Wang <hey@rogerw.io>
Signed-off-by: Martin Hickey <martin.hickey@ie.ibm.com>
Co-authored-by: Or Ozeri <or@ozery.com>
…roject#38709)

Signed-off-by: Mark McLoughlin <markmc@redhat.com>
…llm-project#39354)

The number of features supported by the connector has grown substantially
and the `nixl_connector.py` file has accumulated a lot of code. Creates a separate
directory and isolates connector/scheduler code in the hope of improving clarity
and maintainability.

Further refactor of components aimed at improving clarity and simplifying code
will follow soon.

Signed-off-by: NickLucche <nlucches@redhat.com>
Signed-off-by: Animesh Jain <anijain@umich.edu>
sighingnow and others added 29 commits April 20, 2026 04:28
…startup errors. (vllm-project#39977)

Signed-off-by: chaojun-zhang <chaojun.zhang@intel.com>
Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>
…project#39916)

Signed-off-by: Julien Denize <julien.denize@mistral.ai>
Signed-off-by: juliendenize <julien.denize@mistral.ai>
…atch (vllm-project#39478)

Signed-off-by: velonica0 <like@mail.nankai.edu.cn>
…roject#40251)

Signed-off-by: wuyingjun <wuyingjun_yewu@cmss.chinamobile.com>
Signed-off-by: milesial <milesial@users.noreply.github.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Fadi Arafeh <fadi.arafeh@arm.com>
)

Signed-off-by: Or Ozeri <oro@il.ibm.com>
Co-authored-by: Nicolò Lucchesi <nlucches@redhat.com>
Signed-off-by: ilmarkov <markovilya197@gmail.com>
Signed-off-by: Markov Ilya <markovilya19@gmail.com>
… TransferTopology (vllm-project#39529)

Signed-off-by: Zhanqiu Hu <zhu@redhat.com>
…m-project#40125)

Signed-off-by: Aleksandar Yanakiev <alexander.yanakiev@discretestack.com>
Co-authored-by: Aleksandar Yanakiev <alexander.yanakiev@discretestack.com>
Signed-off-by: kse <kevin.sejourne@cloud-temple.com>
Co-authored-by: kse <kevin.sejourne@cloud-temple.com>
Signed-off-by: Vasiliy Kuznetsov <vasiliy@meta.com>
…3 speculative decoding on ROCm (vllm-project#39616)

Signed-off-by: larryli2-amd <larryli2@amd.com>
Co-authored-by: TJian <tunjian.tan@embeddedllm.com>
Signed-off-by: yewentao256 <zhyanwentao@126.com>
…vllm-project#39242)

Signed-off-by: Rita Brugarolas Brufau <rita.brugarolasbrufau@amd.com>
Signed-off-by: Hashem Hashemi <hashem.hashemi@amd.com>
…Base (vllm-project#35949)

Signed-off-by: Bill Nell <bnell@redhat.com>
Co-authored-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>
Signed-off-by: Sage Moore <sage@neuralmagic.com>
Co-authored-by: Tyler Michael Smith <tyler@neuralmagic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.