SuperAPI AI 工具集

external-gitcode-ascend-triton-operator-doc-gen

编程与开发

查看项目

更新于 10-15 17:19

技能介绍

用途:面向昇腾 NPU 的 Triton 算子编写标准化接口文档,覆盖 README、API 文档、产品支持表和参数说明等场景。

工作流:先收集算子信息(名称、功能、公式、接口、参数、约束);完整阅读 triton_operator_template.md 获取输出格式;参考 layer-norm.md 理解完整文档结构;需要确认 API 签名时阅读 triton-api-reference.md;最后按模板生成文档。

文档应包含:产品支持情况表(覆盖 Ascend 全系列产品线,如 A3/A2/950/Kirin)、功能说明与 LaTeX 公式、函数原型、参数说明表、约束条件(各平台数据类型支持)、可运行的调用示例、常见陷阱与解决方式。

限制与注意:不得留空模板占位符;调用示例须含 import 与数据构造且可执行;约束表须列全 dtype × 平台组合;函数原型须与实际 kernel 签名一致;公式须用 LaTeX 语法并逐行对照 kernel 代码验证;产品支持表不得把不支持的型号标为支持。