用途:当需要设计 PyPTO 算子实现方案时使用此技能。基于算子规格信息与相关上下文,生成结构化的算子设计文档 DESIGN.md,涵盖 API 映射、数据规格、Tiling 策略、Loop 结构等设计内容,用于指导后续代码实现。
主要能力:从输入中提取算子名称与规格信息,信息不足时逐步提问补充;检查规格字段完整性,必须字段包括算子名称、数学公式、输入规格、输出规格,建议字段包括典型配置、算法描述、动态轴范围;分析算子特征(含 matmul 判为 Cube,仅逐元素或归约判为 Vector;按复杂度与 quick_ref 判据判断是否需要 Loop);查询 references/quick_ref.md 与 docs/ 获取 API 映射、Tiling 规则与 Loop 策略;基于 templates/design-template.md 生成含 9 个章节的 DESIGN.md。
使用方式:按工作流执行,阶段为输入验证与特征分析、信息收集、生成草稿、质量自检、定向回修、输出文件,输出到当前目录或用户指定位置。生成后按 5 项检查表自检(API 映射是否具体、Tiling/Loop 是否说明理由、验证方案是否覆盖典型配置、风险点是否具体、是否存在空话或占位符),只修复不通过章节。文件已存在时询问是否覆盖。
适用场景:需要生成算子设计方案、生成 DESIGN.md、进行 API 映射或制定 Tiling 策略、设计算子 Loop 结构时。
限制:缺少算子规格信息会报错退出;必须字段缺失需补充;知识库未命中时动态搜索 docs/ 与 models/,docs 查询失败时基于 AI 知识生成并标注需人工确认;信息不足时写“待确认”,不得编造结论。自检通过需 5 项中至少 4 项通过,且 API 映射与 Tiling/Loop 理由两项必须通过,不得残留占位符。