metal-kernel - PyTorch Metal Kernel Implementation
Implement Metal/MPS kernels for PyTorch operators on Apple Silicon
Tags
Updated: 2026-06-30Capabilities
Typical Outputs
What this skill does
- update dispatch configuration
- write Metal kernel code
- implement host-side stub
- register dispatch stub
- compile source code
- remove old implementation
- enable operator testing
Inputs
- PyTorch operator name
- native_functions.yaml file
- TensorIterator base interface
- c10/metal infrastructure
Outputs
- Metal kernel file
- host-side stub file
- dispatch configuration
- compiled code
- test results
Requirements
- Apple Silicon hardware
- Metal runtime support
- PyTorch source code
- C++ compiler
