at-dispatch-v2 是做什么的?
在 ATen C++ 代码中将 PyTorch AT_DISPATCH 宏转换为 AT_DISPATCH_V2 格式。用于将 AT_DISPATCH_ALL_TYPES_AND*、AT_DISPATCH_FLOATING_TYPES* 或其他分发宏迁移到新的 v2 API。适用于 ATen 内核文件、CUDA 内核和原生算子实现。
将 PyTorch AT_DISPATCH 宏转换为 ATen C++ 代码中的 AT_DISPATCH_V2 格式。用于将 AT_DISPATCH_ALL_TYPES_AND*、AT_DISPATCH_FLOATING_TYPES* 或其他分发宏迁移到新的 v2 API。适用于 ATen 内核文件、CUDA 内核和原生算子实现。
此技能自动将 PyTorch ATen 中旧式 AT_DISPATCH 宏(如 AT_DISPATCH_ALL_TYPES_AND*、AT_DISPATCH_FLOATING_TYPES* 等)转换为新的 AT_DISPATCH_V2 格式,简化了从旧分发 API 到 v2 API 的迁移过程。它对 C++ 文件中匹配的宏进行文本替换和语法调整,同时保留宏参数的结构。该技能支持 ATen 内核文件、CUDA 内核和原生算子实现,是开发者在 PyTorch 代码库中更新分发宏以提高性能或兼容性时的理想自动化工具。
npx skills add https://github.com/pytorch/pytorch/tree/main/skills/at-dispatch-v2Installed? Explore more 编程与开发 skills: steipete/bluebubbles, steipete/eightctl, steipete/blucli · View all 6 →
在 ATen C++ 代码中将 PyTorch AT_DISPATCH 宏转换为 AT_DISPATCH_V2 格式。用于将 AT_DISPATCH_ALL_TYPES_AND*、AT_DISPATCH_FLOATING_TYPES* 或其他分发宏迁移到新的 v2 API。适用于 ATen 内核文件、CUDA 内核和原生算子实现。
This repo contains 8 individual skills — each has its own dedicated page.
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.
Debug AOTInductor (AOTI) errors and crashes. Use when encountering AOTI segfaults, device mismatch errors, constant loading failures, or runtime errors from aot_compile, aot_load, aoti_compile_and_package, or aoti_load_package.
Write docstrings for PyTorch functions and methods following PyTorch conventions. Use when writing or updating docstrings in PyTorch code.
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.
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".
Migrate a file to use stricter Pyrefly type checking with annotations required for all functions, classes, and attributes.
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.
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.
Send and manage iMessages via BlueBubbles, including attachments, tapbacks, edits, replies, and groups.
Control Eight Sleep pods (status, temperature, alarms, schedules).
BluOS CLI (blu) for discovery, playback, grouping, and volume.
Create, search, and manage Bear notes via grizzly CLI.
Capture frames or clips from RTSP/ONVIF cameras.
Search GIF providers with CLI/TUI, download results, and extract stills/sheets.