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 是一種編譯器,可將 PyTorch 模型預先編譯為共用物件庫,但執行時可能發生段錯誤(segfaults)、裝置不匹配、常數載入失敗或執行時期例外。本技能提供結構化的診斷建議,包括驗證參數(如 device_id)、檢查常數字典格式與記憶體對齊、確認目標裝置可用性,並確保動態形狀的處理正確。它支援 aot_compile、aot_load、aoti_compile_and_package 和 aoti_load_package 等 API 的常見問題場景,幫助開發者快速定位根本原因,減少反覆嘗試。

相容平台~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.

相關技能