SuperAPI AI 工具集

external-cannbot-ops-pypto-golden-generate

编程与开发

查看项目

更新于 10-15 17:19

技能介绍

用途:基于算子规格信息自动生成 PyTorch golden 参考实现及完整验证代码,作为开发阶段精度验证基准;生成的脚本为独立模块,可被 test_{op}.py 等脚本导入调用。

主要能力:从输入提取算子名称、数学公式、输入输出规格等必要信息,信息不足时逐步提问补充;按固定模板 templates/golden-template.py 生成,纯 torch 实现且禁止引入 pypto;函数名规则为 {算子名}_golden,必要张量参数在前、可选参数在后,dtype 不做参数、精度跟随输入张量;依据规格中的边界条件生成对应逻辑。

置信度与验证:按是否有等效 PyTorch API、是否为已知论文算子、公式转换复杂度划分 ⭐ 等级,⭐⭐⭐ 及以下在 docstring 中加醒目警告,提示人工审查、与论文对比、多组数据验证边界;验证以 python3 {op}_golden.py 的 exit code(0 通过)和验证报告为准,脚本内含 if name == "main": _validate() 入口。验证 shape 来自典型配置(性能 P0、功能 P0 必须通过)与动态轴范围采样(最小、中间、最大值)。检查项含语法、形状一致性、函数签名、值域、特殊点、数值稳定性、数学属性、API 对比等;检查项分 🔴 严重、🟡 警告、🟢 信息三级,严重问题阻止输出。

自动修复:验证失败时按优先级尝试改用 PyTorch 稳定 API、添加数值稳定性保护、使用更稳定的等价形式,最多 3 次尝试。

交互与异常:生成、验证、修复全自动;典型配置缺失时引导提供或确认推荐配置;覆盖已有文件经 AskUserQuestion 确认;缺少规格、解析失败、缺必须字段、多算子未指定时分别报错退出或列出缺失项。

限制:需先提供算子规格信息,缺失则报错退出;生成过程依赖 templates/golden-template.py 模板;性能、收费、价格与平台等资料暂未提供。