执行摘要

• 一句话：优化 draft extend 的 select_index 计算
• 推荐动作：该 PR 极小且安全，建议直接合并。虽无测试覆盖，但由于逻辑等价且改动简单，风险可控。对于追求极致性能的工程师，可关注此类微优化；对于一般开发者，可快速略过。

功能与动机

在 Eagle 投机解码的 _draft_extend_for_decode 方法中，select_index 的计算涉及一个乘法操作。通过将 torch.arange 的调用从 torch.arange(len(batch.seq_lens), device=self.device) * self.speculative_num_draft_tokens 改为使用显式的 start、stop、step 参数，可以消除不必要的乘法，并提升代码可读性。PR body 未详细说明，但标题和改动明确指出这是一次性能微优化。

实现拆解

1. 入口：修改位于 python/sglang/srt/speculative/eagle_worker_v2.py 文件的 EagleWorkerV2._draft_extend_for_decode 方法。
2. 核心变更：将 select_index 的计算方式从先 torch.arange(len(batch.seq_lens)) 再乘以 self.speculative_num_draft_tokens，改为直接使用 torch.arange(0, len(batch.seq_lens) * self.speculative_num_draft_tokens, self.speculative_num_draft_tokens, device=self.device)。新方式一步生成等间隔序列，避免了乘法指令和中间张量，同时语义上更直接地表达了“生成从 0 开始、步长为 speculative_num_draft_tokens、长度为 batch size 的序列”。
3. 测试与配置：无测试文件或配置变更，仅源码级微优化。

关键文件：

• python/sglang/srt/speculative/eagle_worker_v2.py（模块 投机解码；类别 source；类型 core-logic；符号 EagleWorkerV2._draft_extend_for_decode）: 核心变更文件，修改了 _draft_extend_for_decode 方法中 select_index 张量的计算逻辑，消除了乘法运算。

关键符号：EagleWorkerV2._draft_extend_for_decode

评论区精华

该 PR 无 review 评论或讨论。

• 暂无高价值评论线程

风险与影响

• 风险：风险极低。变更仅涉及一行张量计算，且逻辑等价——新旧两种方式在数学上生成完全相同的序列（旧方式：torch.arange(N) * S；新方式：torch.arange(0, N*S, S)）。唯一细微差异在于 torch.arange 的数据类型：旧方式默认使用 int64（对于 len(batch.seq_lens) 的步长乘积可能超过 int32 范围时更安全），但新方式显式指定了 device 参数而未指定 dtype，仍保持默认 int64，因此无类型变化风险。无需担心回归或兼容性问题。
• 影响：影响范围：仅限于 Eagle 投机解码的 draft extend 路径，影响所有使用 EagleWorkerV2 的 speculative decoding 场景。影响程度：微优化，可减少一次张量乘法和一个中间张量的创建，对于大规模 batch 可能有微小性能提升，但整体收益较低。无需用户感知或配置调整。
• 风险标记：缺少测试覆盖

关联脉络

• 暂无明显关联 PR