metal-kernel

作者： pytorch

为PyTorch算子编写Metal/MPS内核。在向算子添加MPS设备支持、实现Metal着色器或将CUDA内核移植到Apple…时使用。

npx skills add https://github.com/pytorch/pytorch --skill metal-kernel

下载 ZIP GitHub

来自 pytorch 的更多技能

add-uint-support

通过更新AT_DISPATCH宏，为PyTorch算子添加无符号整数（uint）类型支持。在需要为uint16、uint32、uint64类型添加支持时使用…

调试AOTInductor（AOTI）的错误和崩溃。在遇到AOTI段错误、设备不匹配错误、常量加载失败或运行时错误时使用…

将PyTorch的AT_DISPATCH宏转换为ATen C++代码中的AT_DISPATCH_V2格式。适用于移植AT_DISPATCH_ALL_TYPES_AND*、AT_DISPATCH_FLOATING_TYPES*或…

distributed-triage

在oncall:distributed队列中通过分配分布式模块标签、路由到子值班组并标记已分类来对问题进行子分类。当问题具有…时使用。

为PyTorch函数和方法编写符合PyTorch惯例的文档字符串。在编写或更新PyTorch代码中的文档字符串时使用。

document-public-apis

通过从docs/source/conf.py中的coverage_ignore_functions和coverage_ignore_classes中移除函数，记录PyTorch中未文档化的公共API，然后运行…

审查PyTorch拉取请求的代码质量、测试覆盖率、安全性和向后兼容性。在审查PR、被要求审查代码变更时使用…

调试PyTorch 2编译器堆栈故障，包括Dynamo图断裂、Inductor代码生成错误、AOTAutograd崩溃以及精度不匹配。当…时使用