执行摘要

• 一句话：增强 Triton JIT 编译诊断日志输出详情
• 推荐动作：建议开发者阅读此 PR，尤其是 _safe_repr、_constant_name 和 _format_extra_compile_info 的设计：安全序列化、参数名推断、跳过内部键等模式值得在类似诊断组件中复用。此外，本 PR 展示了如何通过 Triton jit_post_compile_hook 提取丰富的编译元数据。

功能与动机

Triton JIT 编译日志原有 key 字段仅为原始参数元组，可读性差，难以辅助诊断 warmup 缺失或形状未覆盖导致的延迟尖峰。本 PR 通过结构化信息替换，帮助开发者快速定位需要扩展预热覆盖的内核配置。

实现拆解

1. 导入与类型支持：添加 from collections.abc import Mapping 以支持更精确的类型提示。
2. 辅助函数设计：_safe_repr 安全获取对象的 repr 并自动截断超过 120 字符的长字符串；_get_compile_info 从 kwargs 中安全提取 compile 字典；_constant_name 将 Triton 内部常量路径（如 (0,)）转换为有意义的参数名。
3. 格式化函数：_format_constants、_format_signature、_format_extra_compile_info 分别将编译信息字典中的 constants、signature 及剩余字段格式化为人类可读的键值对字符串，并跳过内部字段（如 key、fn）以避免冗余。
4. 入口重构：_log_triton_jit_compile 在 verbose 模式下调用 _format_verbose_triton_compile_details 生成详细描述并传递给 _handle_jit_event；非 verbose 模式行为不变，仅记录基本事件类型与函数名。
5. 保持兼容：原有事件分类（autotune/warmup 与 kernel JIT）及警告/报错逻辑均未修改。

关键文件：

• vllm/utils/jit_monitor.py（模块 工具层；类别 source；类型 core-logic；符号 _safe_repr, _get_compile_info, _constant_name, _format_constants）: 唯一变更文件，新增 7 个辅助函数并重构核心日志函数，实现诊断信息结构化输出。

关键符号：_safe_repr, _get_compile_info, _constant_name, _format_constants, _format_signature, _format_extra_compile_info, _format_verbose_triton_compile_details, _log_triton_jit_compile

评论区精华

无 reviewer 公开评论。PR 来自 fork，仅由 claude bot 发表自动评论，最终由项目维护者 LucasWilkinson 一键批准合并。

• 暂无高价值评论线程

风险与影响

• 风险：风险极低。变更仅限于日志记录路径，不影响模型推理、warmup 或任何计算路径。新增的 _safe_repr 捕获 repr 异常并优雅回退（返回类型名），_get_compile_info 确保非 dict 输入安全返回空字典。verbose 模式下字符串拼接可能带来微小开销，但该模式默认关闭，仅用于主动调试。未引入新的测试文件（但现有 CI 将覆盖基本日志流程）。
• 影响：对普通用户无影响。开启 --jit-monitor-verbose 后，每次 JIT 编译日志将输出详细的 constexprs、signature、extra_compile_info 等信息，显著提升调试效率。有助于开发者快速识别未在 warmup 阶段覆盖的 Triton kernel 形状，缩短调优周期。
• 风险标记：仅日志改动影响面小, 无新增测试覆盖

关联脉络

• PR #46182 [Feat][1/N] CuTeDSL warmup infrastructure, FA4 MLA: 本 PR 改进的 JIT 编译诊断可用于检测 #46182 引入的 warmup 覆盖情况，两者共同提升 kernel JIT 编译的可观测性。