Community编程与开发github.com

pytorch/aoti-debug

调试 AOTInductor (AOTI) 错误和崩溃。适用于遇到 AOTI 段错误、设备不匹配错误、常量加载失败或来自 aot_compile、aot_load、aoti_compile_and_package 或 aoti_load_package 的运行时错误时。

aoti-debug 是什么?

该技能专门用于调试 PyTorch 的 AOTInductor (AOTI) 编译器和运行时相关的错误与崩溃。它能够帮助开发者快速诊断并解决在使用 AOTI 功能时遇到的常见问题,例如段错误(segfault)、设备不匹配错误(如 CUDA 与 CPU 设备冲突)、常量加载失败,以及调用 `aot_compile`、`aot_load`、`aoti_compile_and_package`、`aoti_load_package` 等 API 时发生的运行时错误。该技能支持在多种开发环境中使用,包括本地开发、CI/CD 流水线以及云端推理部署场景。通过分析错误日志、堆栈跟踪和配置文件,它能够提供针对性的调试建议和修复方案,从而提高 AOTI 模型的稳定性和部署效率。对于依赖 AOTI 进行模型预编译和优化的 AI 工程师来说,这是一个不可或缺的故障诊断工具。

兼容平台~Claude Code~Codex CLI~Cursor
npx skills add https://github.com/pytorch/pytorch/tree/main/skills/aoti-debug

Installed? Explore more 编程与开发 skills: steipete/bluebubbles, steipete/eightctl, steipete/blucli · View all 6 →

在你喜欢的 AI 中提问

打开一个已预加载此 Agent Skill 的新对话。

文档

aoti-debug 是做什么的?

调试 AOTInductor (AOTI) 错误和崩溃。适用于遇到 AOTI 段错误、设备不匹配错误、常量加载失败或来自 aot_compile、aot_load、aoti_compile_and_package 或 aoti_load_package 的运行时错误时。

Individual skills in this repo

This repo contains 8 individual skills — each has its own dedicated page.

pytorch/add-uint-support

Add unsigned integer (uint) type support to PyTorch operators by updating AT_DISPATCH macros. Use when adding support for uint16, uint32, uint64 types to operators, kernels, or when user mentions enabling unsigned types, barebones unsigned types, or uint support.

pytorch/at-dispatch-v2

Convert PyTorch AT_DISPATCH macros to AT_DISPATCH_V2 format in ATen C++ code. Use when porting AT_DISPATCH_ALL_TYPES_AND*, AT_DISPATCH_FLOATING_TYPES*, or other dispatch macros to the new v2 API. For ATen kernel files, CUDA kernels, and native operator implementations.

pytorch/docstring

Write docstrings for PyTorch functions and methods following PyTorch conventions. Use when writing or updating docstrings in PyTorch code.

pytorch/metal-kernel

Write Metal/MPS kernels for PyTorch operators. Use when adding MPS device support to operators, implementing Metal shaders, or porting CUDA kernels to Apple Silicon. Covers native_functions.yaml dispatch, host-side operators, and Metal kernel implementation.

pytorch/pr-review

Review PyTorch pull requests for code quality, test coverage, security, and backward compatibility. Use when reviewing PRs, when asked to review code changes, or when the user mentions "review PR", "code review", or "check this PR".

pytorch/pyrefly-type-coverage

Migrate a file to use stricter Pyrefly type checking with annotations required for all functions, classes, and attributes.

pytorch/skill-writer

Guide users through creating Agent Skills for Claude Code. Use when the user wants to create, write, author, or design a new Skill, or needs help with SKILL.md files, frontmatter, or skill structure.

pytorch/triaging-issues

Triages GitHub issues by routing to oncall teams, applying labels, and closing questions. Use when processing new PyTorch issues or when asked to triage an issue.

相关技能