triton-operator-design について
Triton 演算子の要件・設計書を作成するスキルです。要件分析、API プロトタイプ、仕様制約、実装案を扱い、具体的な Tiling 計算やハードウェア制限、データフロー、精度方針を明記します。
主な機能
- 01要件を分析し機能と数式を定義
- 02API インターフェースのプロトタイプ設計
- 03入出力とハードウェア制約の明示
- 04Tiling 方針と Kernel 実装案の策定
- 05データフロー、アラインメント、精度の記述
活用シーン
- 01Triton 演算子の設計書作成
- 02Ascend NPU 演算子の Tiling 計画
- 03UB 容量とメモリアラインメント制約の評価
インストール
コマンドをコピーしてターミナルで実行します。
$
npx skills add https:/インストール後、Claude Code で/triton-operator-designを入力して読み込まれたか確認します。
同梱ファイルreferences/triton-operator-template.md
詳細
プロジェクトの README よりTriton 算子需求文档生成
工作流
- 需求分析 → 产出：功能定义、数学公式、竞品对比
- 原型设计 → 产出：API 接口定义
- 规格约束 → 产出：输入输出约束、硬件限制
- 特性实现 → 产出：Tiling 策略、Kernel 实现方案
参考资源加载
| 阶段 | 必须加载 | 不要加载 |
|---|---|---|
| 需求分析 | ascend-terminology.md | template, tiling-strategies |
| 原型设计 | triton-api-reference.md | tiling-strategies |
| 特性实现 | tiling-strategies.md, triton-operator-template.md | ascend-terminology |
MANDATORY：阶段 4 前，完整阅读 tiling-strategies.md 和 triton-operator-template.md，不设行数限制。按模板格式输出文档。
关键术语
- GM：全局内存（DDR），UB：Vector Core 高速缓存（192KB），L1：Cube Core 缓存（~1MB）
- AI Core：A2/A3 有 24 个，含 1 Cube + 2 Vector
- 归约操作必须升精度到 FP32
绝对不要做的事
- ❌ 使用模糊术语（"适当切分"、"合理分配"）— 必须给出具体计算方法
- ❌ 忽略 UB 大小（192KB）和对齐要求（32B）
- ❌ 不区分 Vector Core（向量计算）和 Cube Core（矩阵计算）
- ❌ 不标注数据流图中的数据类型和 GM↔UB 传输
- ❌ 归约操作不说明升精度策略
常见陷阱
| 陷阱 | 症状 | 解决 |
|---|---|---|
| UB 超限 | 方案不可实现 | 计算缓冲区总大小 < 192KB |
| 内存未对齐 | 硬件报错 | UB 缓冲区 32B 对齐，单值缓冲区分配 32B |
| 精度损失 | FP16 结果不准 | 归约前升 FP32，完成后降精度 |
| Tiling 不合理 | 性能差/大 shape 不支持 | 按维度切分，避免跨 Core 数据依赖 |
コメント
Markdown 対応