使用方法
- 复制安装命令,在自己的智能体环境安装前先检查源码包。
- 从原始文档核对所需工具、凭据和支持的宿主。
- 说明输入材料与期望交付物,并对照来源检查输出。
npx skills add https://github.com/ascend/agent-skills --skill triton-operator-doc-gen开始前需了解
- 安装不包含模型订阅、API 额度或访问文件的权限。
- 仓库星数和安装量反映活跃情况,不等于独立质量或安全测试。
按源仓库内容呈现,保留标题、案例、代码、表格、链接以及原文引用的演示图片。
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 语法
- ❌ 产品支持表包含不支持的型号标记为 √


