metal-kernel - PyTorch Metal 内核实现
在 Apple Silicon 上为 PyTorch 操作符实现 Metal/MPS 内核
标签
更新于: 2026-06-30能力
典型输入
典型输出
该技能可以做什么
- 更新调度配置
- 编写 Metal 内核代码
- 实现主机端存根
- 注册调度存根
- 编译源代码
- 移除旧实现
- 启用操作符测试
输入
- PyTorch 操作符名称
- native_functions.yaml 文件
- TensorIterator 基础接口
- c10/metal 基础设施
输出
- Metal 内核文件
- 主机端存根文件
- 调度配置
- 编译代码
- 测试结果
要求
- Apple Silicon 硬件
- Metal 运行时支持
- PyTorch 源代码
- C++ 编译器