mcpField
A
ascend-transformer-boost

ascend-transformer-boost

github.com/ascend開発5293 回インストール25 週間前に更新
リポジトリ

ascend-transformer-boost について

ascend-transformer-boost は、ATB 開発向けのスキル索引で、CANN/NNAL 環境、テストフレームワークのビルド、演算子の設計・移行、デバッグを扱います。各段階の作業を専門スキルに整理しています。

主な機能

  1. 01ATB 開発スキルの索引
  2. 02NNAL の導入とテストフレームワークのビルド
  3. 03ATB から ACLNN への演算子移行
  4. 04設計文書と CSV テストの生成
  5. 05CSV テストと問題調査

活用シーン

  1. 01演算子移行に必要なスキルの検索
  2. 02CANN と NNAL の開発環境の準備
  3. 03移行設計やテストケースの生成
  4. 04テストフレームワークのビルドと CSV テストの実行

インストール

コマンドをコピーしてターミナルで実行します。

$npx skills add https://github.com/ascend/agent-skills --skill ascend-transformer-boost

インストール後、Claude Code で/ascend-transformer-boostを入力して読み込まれたか確認します。

同梱ファイル

SKILL.md と一緒にインストールされる参考資料やスクリプト。

docs/
agent-skills-background.md
atb-skills-guide.md
rules/
project-description.md
skills/atb-aclnn-operator-replacement-designer/references/
softmax_aclnn_runner_reference.md
skills/atb-aclnn-operator-replacement-designer/templates/
design-template.md
skills/atb-csv-tester/scripts/
run-csv-tests.sh
skills/atb-csv-tester/tests/
run-tests.sh
skills/atb-ops-to-aclnn-migration-workflow/
TASK_TEMPLATE.md

詳細

プロジェクトの README より

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 / 测试 / 调试

详细内容

🔍 查看详细内容
技能索引表
技能目录名称类型核心功能
atb-ops-to-aclnn-migration-workflow/OPS→ACLNN 迁移标准化工作流workflow-template7 阶段标准化工作流模板,含 HIL 用户确认机制
atb-nnal-installer/CANN NNAL 安装env-setupNNAL 安装(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-installercann-operator-env-configNPU 驱动已安装,Toolkit+Kernels 已由 cann-operator-env-config 安装
atb-testframework-buildatb-nnal-installerCANN 环境、ATB 源码、Docker 容器
atb-aclnn-operator-replacement-designer无ATB/ACLNN 接口文档链接
atb-csv-testcase-generator无ATB 源码、接口规格文档、设计文档已确认
atb-aclnn-operator-migrationatb-nnal-installer + replacement-designerCANN 环境、ATB 源码、设计文档、CSV 用例已确认
atb-csv-testeratb-nnal-installer + atb-testframework-buildCANN 环境、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报错分析、环境配置

参考文档

コメント