#38730 Fix custom logit processor params when num_tokens_in_batch is used
原始 PR · 作者 jasonjk-park · 合并时间 2026-09-11 08:52
修复多 token 批量下 custom logit 处理器参数未重复展开的 bug
值得快速精读,尤其是使用 speculative decoding 与自定义 logit 处理器组合的团队。核心启发:当把采样 mask 从请求粒度扩展到 token 粒度时,所有配套的参数列表必须同步扩展,这类长度同步缺陷在批量采样路径中很容易被漏掉。测试用手工构造 SamplingBatchInfo 的方式隔离了模型加载与 server 依赖,写法简洁、可复用,可作为同类采样逻辑单测的模板。