PyTorch 把 dispatch 宏迁移做成了 Claude Skill——框架开发者这次真的可以躺平了

你有没有改过 PyTorch 源码里的 AT_DISPATCH_ALL_TYPES_AND3(...)？那种参数挤成一行、类型拆不开、改完还要担心宏展开对不对的体验，写过 C++ extension 的人都懂。

PyTorch 官方自己也在被这个问题折磨。所以他们在 .claude/skills/ 目录下放了一个 at-dispatch-v2 的 Skill，目标只有一个：把旧 dispatch 宏一键转成新的 AT_DISPATCH_V2 格式。

这个 Skill 在干什么

简单说，它把 PyTorch 框架内部沿用多年的旧 dispatch 宏（AT_DISPATCH_ALL_TYPES_AND*、AT_DISPATCH_FLOATING_TYPES_AND*、AT_DISPATCH_ALL_TYPES_AND_COMPLEX_* 那一整坨）迁移到 aten/src/ATen/Dispatch_v2.h 定义的新版 AT_DISPATCH_V2。

不是简单的「replace string」——它要按 7 步走：

• 加头文件：#include <ATen/Dispatch_v2.h>
• 识别旧宏：扫出 ALL_TYPES / FLOATING_TYPES / INTEGRAL_TYPES / COMPLEX_TYPES 哪个 base
• 映射类型组：用 AT_EXPAND(AT_ALL_TYPES) 这种 V2 宏替换
• 提取 AND3 里的单类型：kHalf、kBFloat16、kBool 这种独立类型变成 trailing 参数
• 调换参数顺序：V2 是 (scalar_type, name, lambda, types…)
• 用 AT_WRAP 包 lambda：处理 lambda 里有逗号的坑（thrust::pair、gpu_reduce_kernel 这种常见）
• 逐项 verify：AT_WRAP 包了吗、AT_EXPAND 用对了吗、include 加了吗

为什么这是开发者该关注的 Skill

三个原因：

• 它来自 pytorch/pytorch 本体——不是社区三方的实验品，是 PyTorch 官方维护在 .claude/skills/ 里的官方流程。这说明 PyTorch 已经把 Claude Skills 当成了框架开发的标准工作流。
• 解决了一个真实痛点：dispatch 宏转换不能出错（类型不全就编译不过、类型多了又重复），人工改一遍很烦。Skill 把这件事固化成可复用的步骤。
• 示范了一种新的框架维护范式：过去框架内部的「迁移 playbook」都是写在 Confluence 或 README 里，需要人去读、记得住、按步骤做；现在它直接挂在了 Claude Skills 里，agent 一看 dispatch 宏上下文就会自动按 7 步迁移。

适配场景

直接说硬场景：

• 你在改 aten/src/ATen/native/ 里的 kernel 文件
• 你在写 PyTorch C++ / CUDA extension，遇到 dispatch 类型不全
• 你在做 PyTorch 框架层面的 PR，需要把整批旧宏迁过去
• 你研究 PyTorch 内部源码，想看 V2 和 V1 在组织类型上的差别

102k+ star 的 PyTorch 主仓维护的 Skill，质量不会差。

📦 GitHub 链接：https://github.com/pytorch/pytorch/tree/main/.claude/skills/at-dispatch-v2

把 PyTorch 的 dispatch 宏迁移交给 Claude 做，你只管 review——这才是框架开发者该有的姿势。

GitHub: https://github.com/pytorch/pytorch/tree/main/.claude/skills/at-dispatch-v2