metal-kernel - Implement Metal Kernels for PyTorch Operators
Write native Metal kernels and implement MPS device support for PyTorch operators on Apple Silicon
Tags
Updated: 2026-06-30Capabilities
Typical Inputs
Typical Outputs
What this skill does
- update native_functions.yaml dispatch
- write Metal kernel files
- implement host-side stub
- register Metal kernel
- configure kernel type registration
- compile Metal implementation
Inputs
- native_functions.yaml file
- PyTorch operator name
- CUDA kernel code
Outputs
- Metal kernel file
- host-side operation file
- MPS device dispatch configuration
Requirements
- Apple Silicon hardware
- Metal support
- PyTorch source code
- C++ and Metal programming