ascend-transformer-boost

ascend-transformer-boost Skills Set

Quick Reference

Purpose: Index of skills for the full ATB (Ascend Transformer Boost) development process, from environment setup to operator migration.

Core Commands:

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

Trigger Timing

This index is triggered when the user mentions the following keywords:

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

Detailed Content

Skills Index Table

Skill Directory	Name	Type	Core Function
atb-ops-to-aclnn-migration-workflow/	OPS→ACLNN Migration Standardized Workflow	workflow-template	7-phase standardized workflow template, including HIL user confirmation mechanism
atb-nnal-installer/	CANN NNAL Installation	env-setup	NNAL installation (Toolkit+Kernels provided by cann-operator-env-config)
atb-testframework-build/	ATB Test Framework Compilation	build	Full compilation / incremental compilation, GitHub→gitcode mirror source replacement
atb-aclnn-operator-replacement-designer/	ATB→ACLNN Operator Replacement Design Document Generation	design	Input ATB/ACLNN interface links, generate a 7-chapter structured design document,including HIL user confirmation
atb-aclnn-operator-migration/	ATB→ACLNN Operator Migration Tool	migration	Execute operator migration, enable ACLNN acceleration on 910B/950 devices
atb-csv-tester/	ATB CSV Test Execution	test	Run CSV-format ATB test cases
atb-csv-testcase-generator/	ATB CSV Test Case Generation	testcase	Positive/negative/performance test design, coverage analysis,including HIL user confirmation
atb-debug-guide/	ATB Debugging Guide	debug	Environment troubleshooting, ABI version, memory errors, CSV test failures

Skill Dependencies

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                    ← 辅助技能，独立使用

Standardized Workflow Invocation Sequence

[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] 交付报告

Dependency Matrix

Invoking Skill	Prerequisite Skill	Prerequisite Conditions
`atb-ops-to-aclnn-migration-workflow`	None	Used as an overall process reference
`atb-nnal-installer`	cann-operator-env-config	NPU driver installed, Toolkit+Kernels already installed by cann-operator-env-config
`atb-testframework-build`	atb-nnal-installer	CANN environment, ATB source code, Docker container
`atb-aclnn-operator-replacement-designer`	None	ATB/ACLNN interface document links
`atb-csv-testcase-generator`	None	ATB source code, interface specification document,design document confirmed
`atb-aclnn-operator-migration`	atb-nnal-installer + replacement-designer	CANN environment, ATB source code, design document,CSV test cases confirmed
`atb-csv-tester`	atb-nnal-installer + atb-testframework-build	CANN environment, ATB test framework
`atb-debug-guide`	None	CANN environment, ATB test framework

Compilation Verification Constraints

After modifying ATB operator source code (*_aclnn_runner.cpp、*_operation.cpp , etc.), recompilation and verification are required.

Compilation Commands:

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

Note: Ensure environment variables are set:

ASCEND_TOOLKIT_HOME(CANN installation path)

ATB_REPO_PATH(ATB repository path)

Constraint Rules:

Verify at most 3 times: When compilation fails, record the reason and plan a modification approach
Step by Step: Fix only one problem at a time
When Unable to Resolve: Record the problem and ask the user

Compilation Failure Handling:

Failure Count	Action
1st Failure	Analyze the error, locate the problem, and recompile after fixing
2nd Failure	Analyze in depth and check the reference implementation
3rd Failure	Stop, record the error and modification plan, and ask the user

Invocation Scenario Index

Scenario	Recommended Skill to Invoke	Notes
Install NNAL (ATB acceleration library)	`atb-nnal-installer`	Depends on cann-operator-env-config to provide Toolkit+Kernels
Compile the ATB test framework	`atb-testframework-build`	Requires Docker + CANN environment
Write a design document for operator replacement	`atb-aclnn-operator-replacement-designer`	Requires ATB/ACLNN interface links
Execute operator migration code	`atb-aclnn-operator-migration`	Supports 910B/950 devices; it is recommended to complete the design document first
Run CSV tests to verify operators	`atb-csv-tester`	Requires CANN + test framework
Write operator CSV test cases	`atb-csv-testcase-generator`	Requires interface specifications, covering 910B/950
ATB debugging/troubleshooting	`atb-debug-guide`	Error analysis, environment configuration

About this skill