pytorch/at-dispatch-v2

Convertit les macros AT_DISPATCH de PyTorch vers le format AT_DISPATCH_V2 dans le code C++ d'ATen. Utilisez-le pour porter les macros AT_DISPATCH_ALL_TYPES_AND*, AT_DISPATCH_FLOATING_TYPES* ou autres macros de dispatch vers la nouvelle API v2, pour les fichiers de noyaux ATen, les noyaux CUDA et les implémentations d'opérateurs natifs.

Qu'est-ce que at-dispatch-v2 ?

La compétence pytorch/at-dispatch-v2 automatise la migration des macros de dispatch PyTorch vers la nouvelle API v2 dans ATen. Elle cible spécifiquement les macros comme AT_DISPATCH_ALL_TYPES_AND*, AT_DISPATCH_FLOATING_TYPES* et autres, présentes dans les fichiers C++ d'ATen, les noyaux CUDA et les implémentations d'opérateurs natifs. Cela permet aux développeurs de moderniser leur codebase sans intervention manuelle fastidieuse, en évitant les erreurs et en gagnant du temps. La compétence respecte la structure du code existant tout en appliquant les transformations nécessaires pour adopter le nouveau format, garantissant une compatibilité ascendante.

Compatible avec~Claude Code~Codex CLI~Cursor
npx skills add https://github.com/pytorch/pytorch/tree/main/skills/at-dispatch-v2

Installed? Explore more Développement et programmation skills: steipete/bluebubbles, steipete/eightctl, steipete/blucli · View all 6 →

Demander à votre IA préférée

Ouvre une nouvelle conversation avec cette compétence d'agent déjà préchargée.

Documentation

Que fait at-dispatch-v2 ?

Convertit les macros AT_DISPATCH de PyTorch vers le format AT_DISPATCH_V2 dans le code C++ d'ATen. Utilisez-le pour porter les macros AT_DISPATCH_ALL_TYPES_AND*, AT_DISPATCH_FLOATING_TYPES* ou autres macros de dispatch vers la nouvelle API v2. Pour les fichiers de noyaux ATen, les noyaux CUDA et les implémentations d'opérateurs natifs.

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/aoti-debug

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.

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.

Skills associés