执行摘要

• 一句话：profiler 新增 PrivateUse1 活动，支持自定义后端设备
• 推荐动作：值得快速浏览，但不需要精读。可关注两点：一是 module-level 映射对旧 torch 的兼容性，二是是否应对新 activity 补充单元测试或 CI 验证。设计上把一个字符串枚举集中在一处，扩展成本低，是清晰的小改法。

功能与动机

PR body 明确说明：This change enables profiler support of vLLM for custom or third-party backends using PyTorch's PrivateUse1 device / activity namespace. 目标是配合 spyre-inference 之后，能在 vLLM 中对 IBM Spyre 这类自定义设备进行 profiling；此前 profiler 只认 CPU/CUDA/XPU，自定义后端没有可用的 activity 枚举入口。

实现拆解

1. 扩展 activity 白名单：在 vllm/profiler/wrapper.py 中，把 TorchProfilerActivity 的 Literal 从 ["CPU", "CUDA", "XPU"] 改为 ["CPU", "CUDA", "PrivateUse1", "XPU"]，使配置解析路径可接受新值。
2. 补充枚举映射：在 TorchProfilerActivityMap 中新增 "PrivateUse1" 到 torch.profiler.ProfilerActivity.PrivateUse1 的映射，供 profiler 初始化时将字符串转换为 torch 枚举。
3. 无测试与文档配套，也没有其他文件改动；由于映射在模块加载时构造，运行时 torch 版本必须包含 PrivateUse1 枚举，否则会导致导入失败。

关键文件：

• vllm/profiler/wrapper.py（模块 性能分析；类别 source；类型 core-logic；符号 TorchProfilerActivity, TorchProfilerActivityMap）: 这是 vLLM profiler 的活动白名单与 torch 枚举映射的唯一定义处，PR 的全部改动都集中在此，直接决定 PrivateUse1 能否被 profiler 配置解析。

关键符号：未识别

关键源码片段

vllm/profiler/wrapper.py

这是 vLLM profiler 的活动白名单与 torch 枚举映射的唯一定义处，PR 的全部改动都集中在此，直接决定 PrivateUse1 能否被 profiler 配置解析。

# vllm/profiler/wrapper.py
# profiler 活动白名单：把配置字符串映射到 torch profiler 枚举。
# 新增 PrivateUse1，用于支持自定义 / 第三方后端设备。
TorchProfilerActivity = Literal["CPU", "CUDA", "PrivateUse1", "XPU"]
​
TorchProfilerActivityMap = {
    "CPU": torch.profiler.ProfilerActivity.CPU,
    "CUDA": torch.profiler.ProfilerActivity.CUDA,
    # PrivateUse1 是 PyTorch 为自定义后端保留的命名空间，
    # 映射到 torch 官方枚举后即可在 profiler 中直接使用。
    "PrivateUse1": torch.profiler.ProfilerActivity.PrivateUse1,
    "XPU": torch.profiler.ProfilerActivity.XPU,
}

评论区精华

该 PR 来自 fork，claude[bot] 提示自动审查被禁用；tdoublep 多次触发 CI（/ci run、/ci retry、/ci cancel）后给出 LGTM 并批准。没有任何针对实现的 review comments，所以没有可提炼的技术交锋。

• fork PR 的自动 review 与 CI 验证 (other): 未发生技术性讨论，维护者手动触发 CI 后给出 LGTM。

风险与影响

• 风险：主要风险是版本兼容性：TorchProfilerActivityMap 在模块导入时执行，会访问 torch.profiler.ProfilerActivity.PrivateUse1；若运行时的 PyTorch 版本较旧且不含该枚举，会导致 vllm/profiler/wrapper.py 导入失败，进而影响整个 vLLM 启动。其次是缺少测试覆盖，该新 activity 从未在 CI 中验证过。影响范围小，仅限于 profiler 相关路径，正常推理不触发。
• 影响：对用户：使用 PrivateUse1 自定义后端的用户可以直接配置 profiler 活动并采集 trace；其他用户不受影响。对系统：仅 1 个文件 2 行净变更，无配置契约变化；如果 torch 版本过旧则存在导入失败风险。对团队：为后续在 vLLM 中支持更多自定义设备（如 IBM Spyre）铺路，但本 PR 本身没有测试与文档。
• 风险标记：缺少测试覆盖, torch 版本兼容风险

关联脉络

• 暂无明显关联 PR