metal-kernel - 为PyTorch算子实现Metal内核
在Apple Silicon上为PyTorch算子编写原生Metal内核并实现MPS设备支持
标签
更新于: 2026-06-30能力
典型输入
典型输出
该技能可以做什么
- 更新native_functions.yaml调度
- 编写Metal内核文件
- 实现主机端存根
- 注册Metal内核
- 配置内核类型注册
- 编译Metal实现
输入
- native_functions.yaml文件
- PyTorch算子名称
- CUDA内核代码
输出
- Metal内核文件
- 主机端操作文件
- MPS设备调度配置
要求
- Apple Silicon硬件
- Metal支持
- PyTorch源代码
- C++和Metal编程