External Gitcode Ascend Ascend Transformer Boost

ascend-transformer-boost 技能集

Quick Reference

用途: ATB（Ascend Transformer Boost）开发全流程技能索引，从环境搭建到算子迁移。

核心命令:

# 迁移标准化工作流
atb-ops-to-aclnn-migration-workflow

调用时机

当用户提到以下关键词时触发此索引：

• ATB / ascend-transformer-boost / ascend transformer boost
• 昇腾 + 开发环境 / NPU
• ATB + ACLNN / 算子迁移
• CANN + 安装 / 910B / 950
• ATB + CSV / 测试 / 调试

详细内容

技能索引表

技能依赖关系

ascend-transformer-boost (Index Skill)
│
├── atb-ops-to-aclnn-migration-workflow         ← 标准化工作流模板
│       │
├── atb-nnal-installer                 ← 依赖 cann-operator-env-config（提供 Toolkit+Kernels）
│       │
├── atb-testframework-build             ← 依赖 atb-nnal-installer
│       │
├── atb-aclnn-operator-replacement-designer  ← 独立，迁移前必需
│       │ (含 HIL 用户确认 Gate 1)
│       │
├── atb-csv-testcase-generator          ← 依赖设计文档
│       │ (含 HIL 用户确认 Gate 2)
│       │
├── atb-aclnn-operator-migration        ← 依赖 atb-nnal-installer + 设计文档
│       │ (依赖 Gate 1 & Gate 2 通过)
│       │
├── atb-csv-tester                      ← 依赖 atb-nnal-installer + 测试框架
│       │
└── atb-debug-guide                    ← 辅助技能，独立使用

标准化工作流调用时序

[Phase 0] 前置学习
    │
    ▼
[Phase 1] 设计文档生成
    └─ atb-aclnn-operator-replacement-designer
        │
        └── [Gate 1: 用户确认] ⭐
                │
                ▼
[Phase 2] CSV用例设计
    └─ atb-csv-testcase-generator
        │
        └── [Gate 2: 用户确认] ⭐
                │
                ▼
[Phase 3] 实际迁移
    └─ atb-aclnn-operator-migration
        │
        ▼
[Phase 4] 编译验证
    └─ atb-testframework-build
        │ (失败) → atb-debug-guide
        ▼
[Phase 5a] 小范围验证（双文件）
    ├─ Step 1: atb-csv-tester（前台 _func.csv，正例+反例）
    └─ Step 2: subagent（后台 _perf.csv，逐个 -n N:N 避免 OOM）
        │ (失败) → atb-debug-guide
        ▼
[Phase 5b] 全量回归（汇合 subagent 结果）
    └─ atb-csv-tester (Gate 5 全量通过)
        │ (失败) → atb-debug-guide
        ▼
[Phase 6] 交付报告

依赖矩阵

编译验证约束

当修改 ATB 算子源码（*_aclnn_runner.cpp、*_operation.cpp 等）后需要重新编译验证。

编译命令:

# 在 Docker 容器内执行
source $ASCEND_TOOLKIT_HOME/set_env.sh
cd $ATB_REPO_PATH
bash scripts/build.sh testframework 2>&1 | tail -50

注意：确保环境变量已设置：

• ASCEND_TOOLKIT_HOME（CANN 安装路径）
• ATB_REPO_PATH（ATB 仓库路径）

约束规则:

• 最多验证 3 次：编译失败时记录原因，规划修改方案
• Step by Step：每次只修复一个问题
• 无法解决时：记录问题并询问用户

编译失败处理:

调用场景索引

参考文档

• ATB Skills 开发者指南
• Agent Skills 背景知识
• 项目描述规则