用途:面向昇腾 ATB 仓库的算子迁移场景,当需要把 ATB 算子替换为 ACLNN 算子并撰写设计文档时使用。
主要能力:解析 ATB 算子与 ACLNN 接口文档,提取参数结构与约束;建立参数映射关系(直接映射、计算映射、废弃参数、新增参数);按模板生成结构化设计文档,包含需求详情、算子 IR 与参数配置、规格说明、开发自测、实现方案、风险评估与结论等章节;完整模板为 7 章,已有 ACLNN Runner 时使用 4 节轻量模板。
必要用法:用户需提供 ATB 算子接口文档链接、ACLNN 算子接口文档链接,以及 CANN 路径 <CANN_PATH> 和 ATB 仓库路径 <ATB_REPO_PATH>,否则技能停止执行。执行时先做规模检测,例如在 ATB 仓库中查找: find <ATB_REPO_PATH>/src/ops/ops_infer/<op_type>/ -name '_aclnn_runner.cpp' -type f 2>/dev/null find <ATB_REPO_PATH>/src/ops/ops_infer/<op_type>/ -name '_aclnn_runner.h' -type f 2>/dev/null 若 *_aclnn_runner.cpp 已存在且已注册 REG_RUNNER_TYPE,则使用轻量模板,否则使用完整模板。
限制:依赖 WebFetch 获取接口文档,任一 URL 获取失败(如 404/403/超时)必须停止并打断用户;禁止从 *_aclnn_runner.cpp 或 infer_op_params.h 反推接口参数与约束。设计文档生成后须经用户确认,未确认前禁止进入 CSV 用例设计阶段。涉及具体算子的数据类型、参数约束与设备支持范围需以接口文档为准。