Community程式設計與開發github.com

pytorch/metal-kernel

為 PyTorch 運算子編寫 Metal/MPS 核心。適合在為運算子新增 MPS 裝置支援、實作 Metal 著色器,或將 CUDA 核心移植到 Apple Silicon 時使用。涵蓋 native_functions.yaml 分派、主機端運算子與 Metal 核心實作。

metal-kernel 是什麼?

此技能(pytorch/metal-kernel)旨在協助開發者為 PyTorch 運算子編寫 Metal/MPS 核心,主要應用場景包括為現有運算子新增 MPS 裝置支援、實作高效 Metal 著色器,或將現有的 CUDA 核心移植到 Apple Silicon 平台。它完整涵蓋了 PyTorch 中與 Metal 相關的三個核心層面:透過 native_functions.yaml 進行運算子分派、主機端運算子(host-side operators)的實作,以及底層 Metal 核心的開發。支援的平台以 PyTorch 與 Apple MPS 後端為主,特別適用於 M1/M2/M3 等 Apple Silicon 晶片。使用本技能能大幅簡化開發流程,確保開發者能正確遵循 PyTorch 的 MPS 後端開發慣例,並有效利用 Metal 的平行運算能力來加速深度學習模型推理與訓練。

相容平台~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 中提問

開啟一個已預先載入此 Agent Skill 的新對話。

說明文件

metal-kernel 是做什麼的?

為 PyTorch 運算子編寫 Metal/MPS 核心。適合在為運算子新增 MPS 裝置支援、實作 Metal 著色器,或將 CUDA 核心移植到 Apple Silicon 時使用。涵蓋 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.

相關技能