external-cannbot-ops-pypto-precision-compare
Installation
SKILL.md
PyPTO 算子精度问题调试技能
提供两种精度对比方法,用于快速定位 PyPTO 算子中导致精度问题的具体 op。
方法选择
方法对比
| 特性 | 文件保存方法 | 二分对比方法 |
|---|---|---|
| 实现方式 | 使用 pypto.pass_verify_save() 保存到文件,使用 torch.save() 保存 golden |
使用检查点 tensor 作为输入参数,在内存中直接对比 |
| 适用场景 | 初次定位精度问题,需要快速找到问题范围 | 需要对比多个循环迭代数据,减少文件 IO 开销 |
| 循环支持 | 只保存 idx=0 的数据 |
支持保存所有循环迭代数据 |
| 代码修改 | 不需要修改 kernel 函数签名 | 需要修改 kernel 函数签名,添加检查点参数 |
| 数据类型 | 直接保存原始类型,对比时统一转换 | 在内存中直接对比,类型需一致 |
| 使用难度 | 简单,只需添加检查点调用 | 较复杂,需要管理检查点 tensor |