| name | external-gitcode-ascend-catlass-operator-design |
| description | 将用户基于CATLASS开发算子的需求转变为具体的设计文档 |
| original-name | catlass-operator-design |
| synced-from | https://gitcode.com/Ascend/agent-skills |
| synced-date | 2026-05-26 |
| synced-commit | 1f7666e7768a0ceb21bb1d40ce4b5179fcb6f1d6 |
| license | UNKNOWN |
CATLASS 算子设计
核心工作流
需求分析 → 阅读 catlass 仓库 → 组件选型 → 输出设计文档
前置条件
| 检查项 | 说明 |
|---|
| catlass 仓库 | 须在 OPS_PROJECT_ROOT 下可访问 catlass/(含 include/、examples/) |
| 必读参考 | 设计前必须阅读本 skill 的 references/ 文件 |
必读参考
算子命名(强制)
op_name(snake_case)必须包含子串 catlass,例如:catlass_matmul_add、catlass_fused_gemm_bias
- 禁止单独使用易与框架或其它厂商算子冲突的短名(如
matmul_add、gemm_add)
- OpDef / 类名(CamelCase) 与目录名一致映射,如
CatlassMatmulAdd、ops/catlass_matmul_add/
选型与适配原则
- 从 catlass example 和 include 中确定要用的模板,不是把 example 整份粘贴到算子工程
- 大部分算子工程基于 msopgen 小工程,example 中的 Host 调用代码需要改写为工程中的 op_host/op_kernel 分离结构
- 设计阶段只产出选型决策,代码由 code-gen 生成
NEVER / ALWAYS
NEVER:使用不含 catlass 的算子名定稿设计;把 example 整份照抄;设计文档中写大量代码块;在信息不足时臆测需求
ALWAYS:算子名含 catlass,目录名与类名一致映射;条件不明则追问;设计前存在 catlass 仓库并阅读其文档与代码;设计文档用选型表格描述组件
需求输入
用户可能直接说出需求,也可能不说。执行本 skill 时:若认为信息不足,则继续追问。
需确认的信息:op_name(须含 catlass)、功能要点、I/O 与 dtype、布局、目标 SoC、参考 example(如有)、约束条件。
需求分析
- 确定
op_name:须含 catlass
- 提取:算子功能、数学公式、I/O、dtype、布局、目标 SoC、约束
阅读 catlass 并选型
组件选型
选型结果用概念表格记录在设计文档中,不写代码。
设计文档须显式写出选型表格:ArchTag、BlockMmad(L1/L0 TileShape、数据类型)、BlockEpilogue(各环节 Tile/Block 组件)、BlockScheduler、Kernel 类型。
输出设计文档
文档章节要求
- 概述:功能、场景、数学公式
- 输入输出信息表:变量名、数据类型、Shape、布局、描述
- 核心组件选型:用选型表格描述各层组件
- 参考 Example 与模板选型:选中的 example、选型理由
- TilingKey 分支设计:各 key 对应的条件和 Kernel 分支内容
- Workspace:大小计算逻辑来源
- 接口与使用:aclnn 接口、Host Tiling 流程、Kernel 入口概念
- 扩展性:可替换的组件
- 实现方案纲要:Host 侧与 Kernel 侧要点
输出格式
设计文档须为独立 Markdown 文件,命名建议 design_<op_name>.md,存放于 USER_OP_PROJECT 或用户指定路径。
质量验证
设计文档检查清单
参考资料