ascend-ai-coding Triton 算子接口文档生成
工作流
- 收集算子信息：名称、功能、公式、接口、参数、约束
- MANDATORY：完整阅读
triton_operator_template.md获取输出格式 - 参考示例：完整阅读
layer-norm.md理解完整文档结构 - 需要确认 API 签名时，完整阅读
triton-api-reference.md - 按模板生成文档，确保包含：
- 产品支持情况表（Ascend 全系列产品）
- 功能说明 + LaTeX 公式
- 函数原型
- 参数说明表
- 约束条件（各平台数据类型支持）
- 调用示例
常见陷阱
| 陷阱 | 症状 | 解决 |
|---|---|---|
| 产品支持表不全 | 文档缺平台行 | 覆盖所有 Ascend 产品线（A3/A2/950/Kirin） |
| 约束表缺少 dtype 组合 | 用户按文档用不支持的类型 | 从 code 的 BLOCK_SIZE/dtype 限制反推，列出所有合法组合 |
| 公式与实现不一致 | 文档说的和代码算的不一样 | 从 kernel 代码逐行验证公式 |
| 调用示例不可运行 | 用户复制粘贴报错 | 示例必须完整可执行，包含 import 和数据构造 |
反模式清单（NEVER）
- ❌ 生成空模板占位符（如
{公式}、{参数描述}未填写）——所有占位符必须替换为实际内容 - ❌ 调用示例缺少 import 语句或数据构造
- ❌ 约束表缺少 dtype × 平台组合
- ❌ 函数原型与实际 kernel 签名不一致
- ❌ 公式使用纯文本而非 LaTeX 语法
- ❌ 产品支持表包含不支持的型号标记为 √