CommunityProgramación y desarrollogithub.com

pytorch/at-dispatch-v2

Convierte macros AT_DISPATCH de PyTorch al formato AT_DISPATCH_V2 en código C++ de ATen. Útil al migrar AT_DISPATCH_ALL_TYPES_AND*, AT_DISPATCH_FLOATING_TYPES* y otras macros de despacho a la nueva API v2. Para archivos kernel de ATen, kernels CUDA e implementaciones de operadores nativos.

¿Qué es at-dispatch-v2?

Esta habilidad automatiza la conversión de macros AT_DISPATCH heredadas de PyTorch al nuevo formato AT_DISPATCH_V2 en código C++ de ATen. Está diseñada para desarrolladores que trabajan en el backend de PyTorch, específicamente en archivos kernel de ATen, kernels CUDA e implementaciones de operadores nativos. La utilidad principal de esta herramienta es modernizar el código base de PyTorch, reemplazando macros como AT_DISPATCH_ALL_TYPES_AND*, AT_DISPATCH_FLOATING_TYPES* y otras variantes por sus equivalentes en la API v2. El proceso de conversión mantiene intacta la lógica de despacho de tipos mientras actualiza la sintaxis de las macros, asegurando compatibilidad con las últimas prácticas de desarrollo de PyTorch. La habilidad soporta plataformas de desarrollo que utilizan PyTorch y C++, como entornos de compilación de PyTorch desde código fuente, y se basa en el análisis del código fuente para identificar y transformar correctamente los patrones de macros.

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

Installed? Explore more Programación y desarrollo skills: steipete/bluebubbles, steipete/eightctl, steipete/blucli · View all 6 →

Preguntar en tu IA favorita

Abre un nuevo chat con esta habilidad de agente ya precargada.

Documentación

¿Qué hace at-dispatch-v2?

Convierte macros AT_DISPATCH de PyTorch al formato AT_DISPATCH_V2 en código C++ de ATen. Útil al migrar AT_DISPATCH_ALL_TYPES_AND*, AT_DISPATCH_FLOATING_TYPES* y otras macros de despacho a la nueva API v2. Para archivos kernel de ATen, kernels CUDA e implementaciones de operadores nativos.

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 relacionados