执行摘要

• 一句话：将 getattr 替换为直接属性访问
• 推荐动作：值得精读。该 PR 虽小，但展示了重要的软件工程原则：防御性编码应基于实际必要性，而非习惯；不应为“可能”的不存在而默默吞掉错误，尤其是当该错误会以更隐蔽的方式（如 OOM）表现出来时。PR body 的分析清晰且具有说服力，适合作为代码审查和类型设计参考。

功能与动机

PR body 明确指出：在 preflight 中 mr 总是 ModelRunner，其构造函数无条件设置 self.dflash_draft_num_layers = None，因此 getattr 的默认值永远不会被触发。但一旦下游链中 mr 被替换为 frozen dataclass（如 KVCacheConfigurator），若忘记声明该字段，getattr 会静默返回 None，导致 DFLASH scaling 跳过、KV pool 过分配、GPU OOM，且根因难以定位。PR 旨在将静默错误转为立即抛出的 AttributeError，强制下游类型匹配。

实现拆解

1. 在 python/sglang/srt/model_executor/pool_configurator.py 的 DefaultPoolConfigurator.__init__ 方法中，将第 113 行的 draft_num_layers = getattr(mr, "dflash_draft_num_layers", None) 替换为 draft_num_layers = mr.dflash_draft_num_layers。
2. 后续的 if 语句保持不变，仍检查 draft_num_layers is not None 等条件。

该变更仅修改 1 行代码，不涉及其他文件或配置。

关键文件：

• python/sglang/srt/model_executor/pool_configurator.py（模块 配置器；类别 source；类型 data-contract；符号 DefaultPoolConfigurator.init）: 唯一修改的文件，将 getattr(mr, "dflash_draft_num_layers", None) 替换为 mr.dflash_draft_num_layers，是 PR 的核心变更。

关键符号：DefaultPoolConfigurator.init

关键源码片段

python/sglang/srt/model_executor/pool_configurator.py

唯一修改的文件，将 getattr(mr, "dflash_draft_num_layers", None) 替换为 mr.dflash_draft_num_layers，是 PR 的核心变更。

# python/sglang/srt/model_executor/pool_configurator.py
class DefaultPoolConfigurator(MemoryPoolConfigurator):
    def __init__(self, mr: ModelRunner):
        # ... 前略 (num_layers 计算 ) ...
​
        self._cell_size = self._compute_cell_size(mr, num_layers)
​
        # DFLASH: scale cell_size to account for draft model KV cache
        if mr.spec_algorithm.is_dflash() and not mr.is_draft_worker:
            from sglang.srt.speculative.dflash_utils import (
                scale_kv_cell_size_per_token_for_dflash,
            )
​
            # 变更前 : draft_num_layers = getattr(mr, "dflash_draft_num_layers", None)
            # 变更后 : 直接访问属性，若不存在则立即抛出 AttributeError
            draft_num_layers = mr.dflash_draft_num_layers
            if (
                draft_num_layers is not None
                and int(draft_num_layers) > 0
                and int(num_layers) > 0
            ):
                self._cell_size = scale_kv_cell_size_per_token_for_dflash(
                    target_cell_size_per_token=self._cell_size,
                    target_num_layers=int(num_layers),
                    draft_num_layers=int(draft_num_layers),
                )
​
    # ... 其余方法 ...

评论区精华

该 PR 没有 review 评论。PR body 本身提供了详细的动机阐述，包含具体代码路径和实际发生过的问题实例。

• 暂无高价值评论线程

风险与影响

• 风险：风险极低。原因：1）在 preflight 中，mr.dflash_draft_num_layers 属性始终存在（由 ModelRunner 构造函数保证）；2）行为与原来完全一致（均会读取到 None 或 int）；3）变更已通过编译，且 CI 通过。潜在风险为：如果未来 ModelRunner 的构造函数意外移除了该属性的初始化，会导致 AttributeError，但这正是 PR 期望的效果——暴露问题而非隐藏。
• 影响：影响范围极小：仅修改 pool_configurator.py 中一行代码。对用户和系统无直接功能影响；对开发过程的影响：后续若重构引入新类型替换 mr，将不会因 getattr 静默吞错导致难以调试的 OOM，而是立即抛出 AttributeError，加速根因定位。
• 风险标记：暂无

关联脉络

• PR #24944 Add multi-detokenizer support: 同属 sglang/srt 基础设施重构，且 PR body 提到未来的 KVCacheConfigurator 类型，可能与 multi-detokenizer 相关。
• PR #25449 Convert discarded-value ternary to a plain if statement: 同属移除防御性/冗余代码的清理工作（同期 refactor 系列）。
• PR #25448 Inline the trivial _build_model_config wrapper: 同属简化代码的 refactor 系列。
• PR #25430 Convert local-only self.X attributes to locals: 同属消除冗余代码的 refactor 系列，且由相同作者提交。