triton-npu-operator-doc-gen
Triton NPU 算子接口文档生成器
功能说明
该Skill用于为昇腾NPU平台的Triton算子生成标准化的接口文档。生成的文档遵循昇腾官方文档格式,包含产品支持情况、功能说明、函数原型、参数说明、约束条件和调用示例等内容。
使用方法
基础用法
- 提供算子名称和基本信息
- 输入算子的功能描述和计算公式
- 提供函数原型和参数说明
- 定义约束条件和支持的数据类型
- 提供调用示例代码
文档结构模板
生成的文档将包含以下章节:
# {算子名称}
[📄 查看源码]({源码链接})
## 产品支持情况
|产品 | 是否支持 |
|:-------------------------|:----------:|
| <term>Ascend 950PR/Ascend 950DT</term> | {是否支持} |
| <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term> | {是否支持} |
| <term>Atlas A2 训练系列产品/Atlas A2 推理系列产品</term> | {是否支持} |
| <term>Atlas 200I/500 A2 推理产品</term> | {是否支持} |
| <term>Atlas 推理系列产品</term> | {是否支持} |
| <term>Atlas 训练系列产品</term> | {是否支持} |
## 功能说明
- 接口功能:{算子功能详细描述}
- 计算公式:
$$
{LaTeX格式的计算公式}
$$
## 函数原型
{函数原型}
## 参数说明
<table style="undefined;table-layout: fixed; width: 953px"><colgroup>
<col style="width: 173px">
<col style="width: 112px">
<col style="width: 668px">
</colgroup>
<thead>
<tr>
<th>参数名</th>
<th>输入/输出</th>
<th>描述</th>
</tr></thead>
<tbody>
<tr>
<td>{参数名}</td>
<td>{输入/输出}</td>
<td>{参数描述}</td>
</tr>
</tbody>
</table>
## 约束说明
- 各平台支持数据类型说明:
- <term>Atlas A2 训练系列产品/Atlas A2 推理系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR/Ascend 950DT</term>:
| `{参数1}`数据类型 | `{参数2}`数据类型 | `{参数3}`数据类型 | `{参数4}`数据类型 |
| -------- | -------- | -------- | -------- |
| {数据类型1} | {数据类型2} | {数据类型3} | {数据类型4} |
## 调用示例
```python
{Python调用示例代码}
More from ascend/agent-skills
ascendc-operator-dev
AscendC算子端到端开发编排器。当用户需要开发新算子、实现自定义算子、或完成从需求到测试的完整流程时使用。关键词:算子开发、operator development、端到端、完整流程、工作流编排、新建算子。
56ascendc-operator-doc-gen
为AscendC算子生成PyTorch风格的接口文档(README.md)。触发场景:编译调试通过后需要生成接口文档,或用户提到"生成算子文档"、"创建README"、"文档化算子"、"帮我写文档"(算子上下文)、"算子文档"时使用。
55ascendc-operator-design
完成AscendC算子设计 - 帮助用户完成算子的架构设计、接口定义和性能规划。当用户提到算子设计、算子开发、tiling策略、内存规划、AscendC kernel设计、两级tiling、核间切分、核内切分时,使用此skill。
55ascendc-operator-precision-eval
AscendC算子精度评估。对已编译安装的算子生成全面的精度测试用例集(≥30例),运行并生成精度验证报告。关键词:精度测试、precision evaluation、精度报告、accuracy、误差分析。执行完成后 MUST 在当前对话中展示总览、失败摘要与关键发现,不得仅附报告路径。
54ascendc-operator-testcase-gen
完成AscendC算子验证用例生成 - 帮助用户完成testcase设计。当用户提到用例设计、泛化用例生成、算子标杆、UT用例、精度用例、性能用例时,使用此skill。
54ascendc-operator-project-init
初始化 AscendC 算子工程并创建可编译的算子骨架。触发场景:(1) 用户要求创建新算子;(2) 关键词:ascendc算子、新建算子、算子目录、算子初始化;(3) 需要基于 ascend-kernel 模板快速落地。本 skill 不只建目录,还输出“可继续开发”的标准文件与检查清单。
54