Skills
.Work. You
Rest

metal-kernel

Community

Metal kernels for PyTorch on Apple Silicon.

Software Engineering#dispatch#metal#kernel#pytorch#apple-silicon#mps#host-stub
Authorsteleman
Version1.0.0
Installs0

System Documentation

What problem does it solve?

Metal kernel development for PyTorch on Apple Silicon to enable native GPU acceleration of operators.

Core Features & Use Cases

  • Implement Metal kernels for PyTorch operators in aten/src/ATen/native/mps/kernels/.
  • Add Dispatch mappings to native_functions.yaml for MPS support and ensure proper host-side stubs are wired in aten/src/ATen/native/mps/operations/.
  • Provide battle-tested host-side integration via TensorIterator-based workflows and stepwise migration from MPSGraph to native Metal.

Quick Start

Run the task to implement a Metal kernel and a corresponding host stub for a new PyTorch operator.

Dependency Matrix

Required Modules

None required

Components

Standard package

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: metal-kernel
Download link: https://github.com/steleman/pytorch-cuda-2.11.0/archive/main.zip#metal-kernel

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.