执行摘要

• 一句话：内联仅被测试使用的 split-prefill 方法
• 推荐动作：该 PR 属于轻微重构，值得了解但无需深入精读。体现了清除无用抽象、保持代码简洁的良好实践。

功能与动机

PR body 指出该方法仅被测试文件使用，且生产 PDMUX 已内联设置 forward mode，因此该包装方法没有价值。通过内联可以减少代码量并消除不必要的间接层。

实现拆解

1. 在 python/sglang/srt/managers/schedule_batch.py 中删除 prepare_for_split_prefill 方法（5 行代码）。该方法的逻辑只是依次调用 prepare_for_extend() 并设置 forward_mode = ForwardMode.SPLIT_PREFILL。
2. 在 test/manual/test_forward_split_prefill.py 中，将原先条件分支的 batch.prepare_for_split_prefill() 改为直接调用 batch.prepare_for_extend() 后再单独设置 batch.forward_mode = ForwardMode.SPLIT_PREFILL，并调整 import 以导入 ForwardMode。

关键文件：

• python/sglang/srt/managers/schedule_batch.py（模块 调度批处理；类别 source；类型 core-logic；符号 prepare_for_split_prefill）: 删除了 prepare_for_split_prefill 方法，这是本次重构的核心操作。
• test/manual/test_forward_split_prefill.py（模块 测试；类别 test；类型 test-coverage）: 调整了测试代码以直接内联设置 forward mode，并导入 ForwardMode。

关键符号：prepare_for_split_prefill

关键源码片段

python/sglang/srt/managers/schedule_batch.py

删除了 prepare_for_split_prefill 方法，这是本次重构的核心操作。

# python/sglang/srt/managers/schedule_batch.py
# 以下方法被删除，因为仅被测试文件使用，而生产代码已内联其逻辑
# def prepare_for_split_prefill(self):
# self.prepare_for_extend()
# # For split prefill, we need to set the forward mode to SPLIT_PREFILL
# self.forward_mode = ForwardMode.SPLIT_PREFILL

test/manual/test_forward_split_prefill.py

调整了测试代码以直接内联设置 forward mode，并导入 ForwardMode。

# test/manual/test_forward_split_prefill.py
# 变更前：
# if is_split_prefill:
# batch.prepare_for_split_prefill()
# else:
# batch.prepare_for_extend()
​
# 变更后：
batch.prepare_for_extend()
if is_split_prefill:
    # For split prefill, we need to set the forward mode to SPLIT_PREFILL
    batch.forward_mode = ForwardMode.SPLIT_PREFILL

评论区精华

无 review 讨论。

• 暂无高价值评论线程

风险与影响

• 风险：风险极低。该变更仅涉及删除一个仅供测试使用的包装方法和调整测试代码。生产代码不受影响，因为生产路径已经在复用器或调度器中直接设置 forward mode。测试行为完全等价。
• 影响：影响范围极小。只影响一个测试文件和一个源码文件。代码库减少了 5 行代码，没有引入任何功能变化。
• 风险标记：暂无

关联脉络

• 暂无明显关联 PR