执行摘要

本PR修复了在macOS/MPS环境中，由于Python 3.12移除find_module()回退机制，导致SGLang的Triton stub无法导入triton.*子模块（如triton.compiler.compiler）的关键bug。通过将_TritonFinder类的导入协议更新为PEP 451的find_spec()，确保了在Python 3.9+上的兼容性，解决了torch 2.9+用户启动失败的问题。变更影响范围有限但重要，风险较低，已通过测试验证。

功能与动机

该PR旨在修复issue #21548中报告的问题：在macOS上，SGLang使用Triton stub模拟triton模块，以便在没有真实Triton的环境下运行。但随Python 3.12发布，CPython移除了sys.meta_path中find_module()的回退机制（当finder未实现find_spec()时），导致旧stub无法处理点分triton.*子模块导入。具体表现为，当PyTorch 2.9+尝试导入triton.compiler.compiler时，会引发ModuleNotFoundError，进而使bench_one_batch等功能失败。PR body中明确指出："On Python 3.12, CPython removed the sys.meta_path fallback to find_module() when a finder does not implement find_spec(), so the old stub no longer materialized those sub-modules."

实现拆解

实现集中在python/sglang/_triton_stub.py文件中的_TritonFinder类。关键改动如下：

• 移除过时方法：删除了find_module和load_module方法，这些是Python遗留导入协议。
• 新增find_spec方法：实现PEP 451的find_spec()，动态处理triton及triton.*子模块导入。代码逻辑包括：
  • 检查sys.modules中是否已存在模块。
  • 若不存在，创建_MockModule实例并注册到sys.modules。
  • 建立父子模块关系，例如当导入triton.compiler时，将其设置为triton的子模块。
• 移除冗余mock：删除了代码中显式注册的triton.compiler和triton.compiler.compilermock，因为find_spec()已能动态处理所有子模块，简化了代码结构。

评论区精华

review讨论中，核心交锋围绕更新导入协议和验证修复：

• yeahdongcn 在Issue评论中提问："I don't see this issue in my environment. It might be related to Python 3.12..." 并建议测试Python 3.11和3.12兼容性。随后在review中强调："Could you please remove find_module/load_module, as described in the PR description?"
• karanb192 回应确认问题是Python 3.12特有，并通过测试矩阵验证修复有效。在后续提交中移除了旧方法，说明："The latest revision removes find_module / load_module entirely and uses only find_spec"。
• 讨论还涉及移除冗余mock，karanb192解释："find_spec() on _TritonFinder already handles dotted triton.* sub-module imports dynamically, so the explicit triton.compiler mocks and their old inline comment are no longer needed."

风险与影响

风险分析：

• 回归风险：find_spec方法需确保正确处理所有子模块导入；代码中通过动态创建mock和检查sys.modules来规避，但缺少单元测试覆盖具体逻辑。
• 兼容性风险：依赖于PEP 451的find_spec，SGLang要求Python 3.9+，find_spec自Python 3.4可用，因此风险可控。移除过时方法未发现副作用。

影响评估：

• 对用户：修复了macOS上Python 3.12+用户在使用torch 2.9+时无法运行SGLang的问题，提升了系统可用性。
• 对系统：改进导入系统健壮性，遵循现代Python标准，降低未来版本升级的兼容性问题。
• 对团队：提供从遗留协议迁移的示例，变更集中，易于维护。

关联脉络

从提供的近期历史PR分析中，未发现直接相关的PR；但issue #21548是本PR的直接源头。变更属于macOS/MPS环境下的基础设施修复，可能与仓库中其他bugfix或infra类PR（如PR 21797修复CI脚本）有间接关联，但无明确跨PR脉络。该修复凸显了对Python版本演进的适应，强调了在维护兼容性时更新导入协议的重要性。