Agent Skills
› pytorch/pytorch
pytorch/pytorch
GitHub用于在PyTorch算子中为uint16/32/64类型添加支持,通过更新AT_DISPATCH_V2宏实现。适用于需要扩展类型覆盖或启用无符号类型的场景。
安装全部 Skills
npx skills add pytorch/pytorch --all -g -y
集合内 Skills (18)
用于在PyTorch算子中为uint16/32/64类型添加支持,通过更新AT_DISPATCH_V2宏实现。适用于需要扩展类型覆盖或启用无符号类型的场景。
添加uint16, uint32, uint64到算子
用户提及unsigned types或uint support
在kernel中启用kUInt16/32/64
npx skills add pytorch/pytorch --skill add-uint-support -g -y
用于诊断和修复AOTInductor (AOTI) 相关的错误与崩溃,如段错误、设备不匹配及常量加载失败。提供错误模式路由及首要检查项(设备与形状匹配),指导用户排查编译、加载及推理阶段的配置问题。
遇到 AOTI 段错误或崩溃
出现设备不匹配错误
常量加载失败
运行时错误来自 aot_compile 或 aot_load
npx skills add pytorch/pytorch --skill aoti-debug -g -y
将 PyTorch ATen C++ 代码中的旧版 AT_DISPATCH 宏转换为新版 AT_DISPATCH_V2 API。适用于 aten/src/ATen/native/ 目录下的内核文件和原生算子实现,支持类型组映射和参数重构。
转换 AT_DISPATCH_* 宏到 AT_DISPATCH_V2
移植 ATen 内核使用新 dispatch API
处理 aten/src/ATen/native/ 中的 dispatch 宏
用户提及 'AT_DISPATCH', 'dispatch v2', 'Dispatch_v2.h' 或宏转换
npx skills add pytorch/pytorch --skill at-dispatch-v2 -g -y
查询PyTorch CI、GitHub Actions及Grafana基础设施指标,用于分析CI耗时、作业失败、队列时间、工作流趋势及Runner健康状态。
询问CI持续时间或作业失败情况
查询队列时间或工作流趋势
检查Runner健康状态或仪表盘数据
获取PyTorch基础设施指标
npx skills add pytorch/pytorch --skill ci-metrics -g -y
指导在PyTorch CUDA内核中处理索引溢出,通过模板化和分发机制智能选择32位或64位索引类型,平衡性能与正确性。
修复CUDA内核大张量索引溢出
决定使用int64_t还是32位索引
npx skills add pytorch/pytorch --skill cuda-index-width -g -y
用于对分布式模块的 Issue 进行二级分类,包括路由至子值班组、标记模块类型及完成状态。通过 GitHub MCP 工具读取和更新 Issue,执行去重检查并应用标签,辅助 OnCall 流程自动化。
需要处理已标记 oncall:distributed 的 Issue
需要对分布式问题进行二次分类和路由
npx skills add pytorch/pytorch --skill distributed-triage -g -y
指导为PyTorch函数和方法编写符合规范的docstring,涵盖Sphinx格式、数学公式、参数说明及交叉引用等结构要求。
编写PyTorch代码文档
更新现有PyTorch函数注释
npx skills add pytorch/pytorch --skill docstring -g -y
用于为 PyTorch 中未文档化的公共 API 生成 Sphinx 文档。通过移除 conf.py 中的忽略列表条目,并在对应的 .md/.rst 文件中添加 autodoc 指令来完成文档化,同时验证函数是否具备有效 docstring。
用户要求移除 conf.py 中的忽略列表条目
需要为 PyTorch 公共 API 补充文档
npx skills add pytorch/pytorch --skill document-public-apis -g -y
自动复现并修复 PyTorch GitHub Issue 中的 Bug。通过克隆仓库、分析 Issue 上下文、定位根因并实施代码修复,确保解决根本问题而非临时绕过。
用户请求修复 PyTorch GitHub Issue
提供 PyTorch Issue URL 或编号
npx skills add pytorch/pytorch --skill fix-issue -g -y
管理PyTorch ghstack堆栈的CI流程,通过智能选择运行时机和标记[no-ci]来优化资源。在创建、提交或更新PR堆栈时,优先对关键PR运行CI,暂缓非关键PR以节省计算成本并加速迭代。
创建或更新ghstack PR堆栈
准备合并PR堆栈
需要诊断CI失败或验证变更
npx skills add pytorch/pytorch --skill ghstack-ci -g -y
指导在 PyTorch 中为 Apple Silicon 编写原生 Metal/MPS 内核,涵盖从 MPSGraph 迁移或新增算子支持。包括更新 native_functions.yaml 分发配置、编写 Metal 着色器及实现主机端存根。
添加 MPS 设备对算子的支持
将 CUDA 内核移植到 Apple Silicon
实施 Metal 着色器代码
npx skills add pytorch/pytorch --skill metal-kernel -g -y
评估PR是否准备好供人工维护者审查的非交互式工具。通过读取代码和差异,判断PR状态并输出JSON格式的结论、摘要及发现项,区分主要与次要问题。
需要快速判断PR是否具备人工审查条件
自动化流程中需获取PR就绪状态
npx skills add pytorch/pytorch --skill pr-review-readiness -g -y
用于审查 PyTorch 项目的 Pull Request,重点关注代码质量、测试覆盖率、安全漏洞及向后兼容性。支持通过 PR 编号、URL 或本地分支进行审查,并能生成详细的逐行评论。
用户请求审查 PR
提及 'review PR' 或 'code review'
提及 'check this PR'
npx skills add pytorch/pytorch --skill pr-review -g -y
用于调试 PyTorch 2 编译器栈(Dynamo、Inductor 等)故障的 Skill。涵盖环境检查、复现最小化、添加单元测试、日志收集、根因分析及修复验证全流程,旨在解决编译错误和精度不匹配问题。
遇到 torch.compile 报错
BackendCompilerFailed 异常
Triton kernel 失败
FX graph 问题
用户提及调试 PT2 或 Dynamo
npx skills add pytorch/pytorch --skill pt2-bug-basher -g -y
指导将Python文件迁移至Pyrefly严格类型检查,包括移除抑制注释、配置子规则、运行检查及修复类型注解错误。
需要将代码文件接入Pyrefly严格类型检查
需要为函数、类或属性添加缺失的类型注解
npx skills add pytorch/pytorch --skill pyrefly-type-coverage -g -y
用于获取、分析、复现并最小化 GitHub Issue 的复现代码。自动评估问题合理性,将复现脚本精简为最小独立版本,并处理标签更新与通知状态保持。
检查 GitHub Issue 是否可复现
最小化 Issue 的复现代码
分析 Bug 报告的可复现性
npx skills add pytorch/pytorch --skill scrub-issue -g -y
辅助用户创建和编写 Claude Code Agent Skills,指导定义范围、选择存储位置、构建目录结构及撰写 SKILL.md 元数据与描述,确保符合最佳实践。
创建新的 Agent Skill
编写或更新 SKILL.md 文件
设计 Skill 结构和 frontmatter
将现有提示词转换为 Skill
npx skills add pytorch/pytorch --skill skill-writer -g -y
用于对 PyTorch GitHub Issue 进行自动化分类、路由和打标。通过 MCP 工具获取问题详情,根据规则分配标签、关闭重复项或重定向至相关团队,确保 Issue 得到正确处理和响应。
处理新的 PyTorch GitHub Issue
被要求对 Issue 进行分类
npx skills add pytorch/pytorch --skill triaging-issues -g -y


