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

pytorch/metal-kernel

PyTorchオペレーター向けのMetal/MPSカーネルを作成します。オペレーターへのMPSデバイスサポート追加、Metalシェーダーの実装、Apple SiliconへのCUDAカーネルの移植に使用します。native_functions.yamlディスパッチ、ホストサイドオペレーター、Metalカーネルの実装をカバーします。

metal-kernel とは?

このスキルは、PyTorchオペレーター向けにMetal Performance Shaders (MPS)カーネルを作成および実装するためのものです。主にApple Silicon(M1/M2/M3)上でPyTorchの演算を高速化するために使用され、既存のCUDAカーネルをMetalシェーダーに移植する場合や、新しいオペレーターのMPSデバイスサポートを追加する場合に役立ちます。このスキルは、native_functions.yamlにおけるディスパッチエントリの設定、C++で記述されたホストサイドオペレーターの実装、そしてMetal Shading Language (MSL)によるGPUカーネルの記述の3つの主要な部分をカバーします。開発者はこのスキルを使用することで、PyTorchのMPSバックエンドに準拠した効率的なカーネルを迅速に開発でき、Apple Silicon上での機械学習ワークロードのパフォーマンスを最大化できます。

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

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

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

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

ドキュメント

metal-kernel は何をしますか?

PyTorchオペレーター向けのMetal/MPSカーネルを作成します。オペレーターへのMPSデバイスサポート追加、Metalシェーダーの実装、Apple SiliconへのCUDAカーネルの移植に使用します。native_functions.yamlディスパッチ、ホストサイドオペレーター、Metalカーネルの実装をカバーします。

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

関連スキル