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昇腾+开发环境/NPUATB+ACLNN/算子迁移CANN+安装/910B/950ATB+CSV/测试/调试
详细内容
技能索引表
| 技能目录 | 名称 | 类型 | 核心功能 |
|---|---|---|---|
| atb-ops-to-aclnn-migration-workflow/ | OPS→ACLNN 迁移标准化工作流 | workflow-template | 7 阶段标准化工作流模板,含 HIL 用户确认机制 |
| atb-nnal-installer/ | CANN NNAL 安装 | env-setup | NNAL 安装(Toolkit+Kernels 由 cann-operator-env-config 提供) |
| atb-testframework-build/ | ATB 测试框架编译 | build | 全量编译 / 增量编译,GitHub→gitcode 镜像源替换 |
| atb-aclnn-operator-replacement-designer/ | ATB→ACLNN 算子替换设计文档生成 | design | 输入 ATB/ACLNN 接口链接,生成 7 章结构化设计文档,含 HIL 用户确认 |
| atb-aclnn-operator-migration/ | ATB→ACLNN 算子迁移工具 | migration | 执行算子迁移,910B/950 设备启用 ACLNN 加速 |
| atb-csv-tester/ | ATB CSV 测试执行 | test | 运行 CSV 格式 ATB 测试用例 |
| atb-csv-testcase-generator/ | ATB CSV 测试用例生成 | testcase | 正例/反例/性能测试设计,覆盖率分析,含 HIL 用户确认 |
| atb-debug-guide/ | ATB 调试指南 | debug | 环境问题排查、ABI版本、内存错误、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-ops-to-aclnn-migration-workflow |
无 | 作为整体流程参考 |
atb-nnal-installer |
cann-operator-env-config | NPU 驱动已安装,Toolkit+Kernels 已由 cann-operator-env-config 安装 |
atb-testframework-build |
atb-nnal-installer | CANN 环境、ATB 源码、Docker 容器 |
atb-aclnn-operator-replacement-designer |
无 | ATB/ACLNN 接口文档链接 |
atb-csv-testcase-generator |
无 | ATB 源码、接口规格文档、设计文档已确认 |
atb-aclnn-operator-migration |
atb-nnal-installer + replacement-designer | CANN 环境、ATB 源码、设计文档、CSV 用例已确认 |
atb-csv-tester |
atb-nnal-installer + atb-testframework-build | CANN 环境、ATB 测试框架 |
atb-debug-guide |
无 | CANN 环境、ATB 测试框架 |
编译验证约束
当修改 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:每次只修复一个问题
- 无法解决时:记录问题并询问用户
编译失败处理:
| 失败次数 | 操作 |
|---|---|
| 第1次失败 | 分析错误,定位问题,修复后重新编译 |
| 第2次失败 | 深入分析,查看参考实现 |
| 第3次失败 | 停止,记录错误和修改计划,询问用户 |
调用场景索引
| 场景 | 推荐调用技能 | 备注 |
|---|---|---|
| 安装 NNAL(ATB 加速库) | atb-nnal-installer |
依赖 cann-operator-env-config 提供 Toolkit+Kernels |
| 编译 ATB 测试框架 | atb-testframework-build |
需要 Docker + CANN 环境 |
| 为算子替换撰写设计文档 | atb-aclnn-operator-replacement-designer |
需要 ATB/ACLNN 接口链接 |
| 执行算子迁移代码 | atb-aclnn-operator-migration |
支持 910B/950 设备,建议先完成设计文档 |
| 运行 CSV 测试验证算子 | atb-csv-tester |
需要 CANN + 测试框架 |
| 编写算子 CSV 测试用例 | atb-csv-testcase-generator |
需要接口规格,覆盖 910B/950 |
| ATB 调试/问题排查 | atb-debug-guide |
报错分析、环境配置 |
参考文档
More from ascend/agent-skills
triton-operator-code-gen
根据 Ascend NPU 算子设计文档(或直接需求)生成 Triton kernel 代码。当用户需要实现 Triton 算子、将设计文档转为可执行代码时使用。核心产出:kernel 代码 + 基本正确性测试。关键词:Triton kernel、算子实现、代码生成、code generation。
49ascend-inference-repos-copilot
昇腾(Ascend)推理生态开源代码仓库智能问答专家旨在为 vLLM、vLLM-Ascend、MindIE-LLM、MindIE-SD、MindIE-Motor、MindIE-Turbo 以及 msModelSlim (MindStudio-ModelSlim) 等仓库提供专家级且易于理解的解释。在处理昇腾(Ascend)推理生态相关项目的用户询问时,务必触发此技能(Skill),可解答使用方法、部署流程、支持模型、支持特性、系统架构、配置管理、调试、测试、故障排查、性能优化、定制开发、源码解析以及其他技术问题。支持中英文双语回复,并可借助 deepwiki MCP 工具检索仓库知识库,生成具备上下文感知且基于证据的回答。Ascend inference ecosystem open-source code repository intelligent question-and-answer (Q&A) expert. Provide expert-level yet comprehensible explanations for repositories such as vLLM, vLLM-Ascend, MindIE-LLM, MindIE-SD, MindIE-Motor, MindIE-Turbo, and msModelSlim (MindStudio-ModelSlim). Use this skill when addressing user inquiries related to these Ascend inference ecosystem projects, including topics such as usage, deployment process, supported models, supported features, system architecture, configuration management, debugging, testing, troubleshooting, performance optimization, custom development, source code analysis, and any other technical issues about these projects. Support responses in both Chinese and English. Use deepwiki MCP tools to query repository knowledge bases and generate context-aware, evidence-based responses.
47cann-operator-env-config
提供昇腾NPU的CANN安装指导。当用户需要安装CANN、配置昇腾环境或解决安装问题时调用。
46auto-develop-test-gen
开发者测试自动补全技能 - 为函数和类生成高质量单元测试,分析覆盖率盲区并生成高价值补充测试,提升有效覆盖率。
43atc-model-converter
Complete toolkit for Huawei Ascend NPU model conversion and end-to-end inference adaptation. Workflow 1 auto-discovers input shapes and parameters from user source code. Workflow 2 exports PyTorch models to ONNX. Workflow 3 converts ONNX to .om via ATC with multi-CANN version support. Workflow 4 adapts the user's full inference pipeline (preprocessing + model + postprocessing) to run end-to-end on NPU. Workflow 5 verifies precision between ONNX and OM outputs. Workflow 6 generates a reproducible README. Supports any standard PyTorch/ONNX model. Use when converting, testing, or deploying models on Ascend AI processors.
42skill-auditor
Comprehensive security auditor for AI agent skills, prompts, and instructions. Checks for typosquatting, dangerous permissions,
41