Communityコーディング&開発github.com

pytorch/add-uint-support

PyTorchオペレーターに符号なし整数(uint)型のサポートを追加します。AT_DISPATCHマクロを更新することで、uint16、uint32、uint64型をオペレーターやカーネルに追加する際に使用します。

add-uint-support とは?

このスキルは、PyTorchのカスタムオペレーターやカーネルに符号なし整数型(uint16、uint32、uint64)のサポートを追加するためのものです。主にAT_DISPATCHマクロを更新することで機能し、新しいデータ型を既存の演算に統合できるようにします。ユーザーが「符号なし型の有効化」「ベアボーン符号なし型」「uintサポート」などに言及した場合に使用します。PyTorchの開発者にとって、型システムを拡張し、メモリ効率や特定の数値範囲を必要とするアプリケーション(例えば画像処理や組み込みシステム)で有用です。このスキルは、Claude CodeやCursorなどのAIエージェントで利用でき、コードベース内の適切なマクロ呼び出しを特定し、必要な型引数を追加する方法を指示します。

対応~Claude Code~Codex CLI~Cursor
npx skills add https://github.com/pytorch/pytorch/tree/main/skills/add-uint-support

Installed? Explore more コーディング&開発 skills: steipete/bluebubbles, steipete/eightctl, steipete/blucli · View all 6 →

お気に入りのAIに質問する

このエージェントスキルを事前に読み込んだ状態で新しいチャットを開きます。

ドキュメント

add-uint-support は何をしますか?

PyTorchオペレーターに符号なし整数(uint)型のサポートを追加するには、AT_DISPATCHマクロを更新します。オペレーターやカーネルにuint16、uint32、uint64型のサポートを追加する場合、またはユーザーが符号なし型の有効化、ベアボーン符号なし型、uintサポートについて言及した場合に使用します。

Individual skills in this repo

This repo contains 8 individual skills — each has its own dedicated page.

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/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.

関連スキル