metal-kernel - Apple Silicon 的 PyTorch Metal 内核
使用原生 c10/metal 基础设施为 PyTorch 操作实现 Metal/MPS 内核
标签
更新于: 2026-06-30能力
典型输入
典型输出
该技能可以做什么
- 编写 Metal 内核
- 更新 native_functions.yaml
- 实现主机端存根
- 注册调度
- 编译 Metal 代码
- 移除 MPSGraph 实现
- 启用操作测试
输入
- PyTorch 操作符
- CUDA 内核
- native_functions.yaml
- 张量数据
输出
- Metal 内核文件
- 主机端存根
- 更新的 native_functions.yaml
- 编译的二进制
要求
- Apple Silicon 硬件
- macOS
- Xcode
- Clang
- PyTorch 源码仓库