Community코딩 & 개발github.com

pytorch/at-dispatch-v2

ATen C++ 코드에서 PyTorch AT_DISPATCH 매크로를 AT_DISPATCH_V2 형식으로 변환합니다. AT_DISPATCH_ALL_TYPES_AND*, AT_DISPATCH_FLOATING_TYPES* 및 기타 디스패치 매크로를 새로운 v2 API로 포팅할 때 사용합니다. ATen 커널 파일, CUDA 커널 및 네이티브 연산자 구현에 적합합니다.

at-dispatch-v2란 무엇인가요?

이 스킬은 PyTorch의 ATen C++ 코드에서 레거시 AT_DISPATCH 매크로(예: AT_DISPATCH_ALL_TYPES_AND*, AT_DISPATCH_FLOATING_TYPES*)를 새로운 AT_DISPATCH_V2 API 형식으로 자동 변환하는 데 특화되어 있습니다. v2 API는 성능 최적화와 코드 간소화를 위해 도입되었으며, 개발자가 수동으로 변환할 때 발생할 수 있는 오류를 줄여줍니다. 주로 ATen 커널 파일, CUDA 커널 구현, 네이티브 연산자 코드에서 사용되며, PyTorch의 핵심 C++ 코드베이스 현대화 작업에 필수적입니다. 변환 과정에서 매크로 인자, 타입 리스트, 람다 함수 구조를 분석하여 정확한 v2 형식으로 재작성하며, 기존 코드의 동작을 유지하면서 새로운 API의 이점을 활용할 수 있도록 합니다. 이 스킬은 PyTorch 개발자와 AI 엔지니어가 수동 리팩토링 없이 빠르게 v2 API로 마이그레이션할 수 있도록 지원합니다.

지원 대상~Claude Code~Codex CLI~Cursor
npx skills add https://github.com/pytorch/pytorch/tree/main/skills/at-dispatch-v2

Installed? Explore more 코딩 & 개발 skills: steipete/bluebubbles, steipete/eightctl, steipete/blucli · View all 6 →

즐겨 사용하는 AI에게 물어보기

이 에이전트 스킬이 미리 로드된 새 채팅을 엽니다.

문서

at-dispatch-v2은(는) 무엇을 하나요?

ATen C++ 코드에서 PyTorch AT_DISPATCH 매크로를 AT_DISPATCH_V2 형식으로 변환합니다. AT_DISPATCH_ALL_TYPES_AND*, AT_DISPATCH_FLOATING_TYPES* 및 기타 디스패치 매크로를 새로운 v2 API로 포팅할 때 사용합니다. ATen 커널 파일, CUDA 커널 및 네이티브 연산자 구현에 적합합니다.

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.

관련 스킬