#25681 Add support for generic num_tokens_per_bs in TARGET_VERIFY
原始 PR · 作者 jasonjk-park · 合并时间 2026-05-21 02:34
让 TARGET_VERIFY 的每请求 token 数可自定义
该 PR 虽小,但为推测解码的扩展性奠定了基础。建议阅读 spec_registry.py 和 spec_info.py 的新增接口设计,以及 trtllm_mha_backend.py 中元数据计算的统一化。关注后续可能的前向模式重构 PR。
参与讨论