Agent Skills › pytorch/pytorch

pytorch/pytorch

GitHub

用于在PyTorch算子中为uint16/32/64类型添加支持,通过更新AT_DISPATCH_V2宏实现。适用于需要扩展类型覆盖或启用无符号类型的场景。

18 个 Skill 103,183

安装全部 Skills

npx skills add pytorch/pytorch --all -g -y
更多选项

预览集合内 Skills

npx skills add pytorch/pytorch --list

集合内 Skills (18)

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

首页 - Wiki
Copyright © 2011-2026 iteam. Current version is 2.155.2. UTC+08:00, 2026-09-23 15:08
浙ICP备14020137号-1