SuperAPI AI 工具集

triton-operator-doc-gen

编程与开发

查看项目

更新于 10-16 05:00

技能介绍

用途:为昇腾 NPU 上的 Triton 算子生成标准化接口文档,适用于创建算子 README、API 文档、产品支持表与参数说明整理。

主要能力:按固定模板组织文档结构,覆盖产品支持情况表(Ascend 全系列产品)、功能说明与 LaTeX 公式、函数原型、参数说明表、约束条件(各平台数据类型支持)、调用示例与常见陷阱。

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

注意事项与限制:不可生成未填写的占位符;调用示例须包含 import 与数据构造且可运行;约束表需列出 dtype 与平台组合;函数原型须与实际 kernel 签名一致;公式须用 LaTeX;产品支持表不得把不支持的型号标为 √。具体安装命令、收费与平台细节资料暂未提供。