external-gitcode-ascend-atb-aclnn-operator-migration

name	external-gitcode-ascend-atb-aclnn-operator-migration
description	自动执行 ATB 算子到 ACLNN 的迁移操作，在 910B/950 设备上启用 ACLNN 加速。 支持参数映射、ACLNN Runner 实现、设备检测切换和功能/性能验证全流程。
keywords	["atb","aclnn","migration","operator","昇腾","ascend","910b",950]
metadata	{"author":"ascend-transformer-boost-team + Claude Code + opus4.7","version":"1.2.0","created":"2026-04-17","updated":"2026-04-28","skill-type":"migration"}
hooks	{"PreToolUse":[{"matcher":"Write|Edit|Bash","hooks":"[Truncated]"}]}
original-name	atb-aclnn-operator-migration
synced-from	https://gitcode.com/Ascend/agent-skills
synced-date	2026-05-26
synced-commit	1f7666e7768a0ceb21bb1d40ce4b5179fcb6f1d6
license	UNKNOWN

ATB到ACLNN算子迁移工具

功能概述

该工具帮助开发者将ATB中的算子迁移到ACLNN实现，特别是在 910B/950 设备上获得更好的兼容性优化。

调用时机

在以下情况下调用此技能：

• 用户需要将 ATB 算子迁移到 ACLNN 实现以提升 910B/950 设备性能
• 用户已生成算子替换设计文档（建议先运行 atb-aclnn-operator-replacement-designer）
• 用户需要在 ATB 源码上应用 ACLNN Runner 改造

⚠️ 路径约束（必须执行）

执行此技能前，必须从用户处获取以下路径：

• <CANN_PATH>: CANN 安装路径（如 /usr/local/Ascend/ascend-toolkit/latest）
• <ATB_REPO_PATH>: ATB 仓库路径（如 {your working path}ascend-transformer-boost）

若用户未提供或路径无效，立即停止并报错。

---

前置条件

• CANN 环境已安装（ASCEND_TOOLKIT_HOME 已设置）
• ATB 源码仓库已克隆至本地
• 算子对应的 ATB 接口文档和 ACLNN 接口文档已获取
• 建议已完成 atb-aclnn-operator-replacement-designer 生成设计文档

检查点系统（点击展开）

每个操作步骤末尾的 检查点 标记了验证标准。执行完成后请在 执行结果 章节填写实际状态。

检查点 ID	描述	验证命令
Step0-ENV	CANN 环境 sourced	echo $ASCEND_TOOLKIT_HOME
Step0-DIR	ATB 源码目录存在	[ -d <ATB_REPO_PATH> ]
Step1-DIR	原有 Operation 源文件定位	ls <op>_operation.cpp
Step1-DIR	现有 ACLNN runner 参考定位	ls softmax_aclnn_runner.cpp
Step2-FILE	<op>_aclnn_runner.h 创建	[ -f <op>_aclnn_runner.h ]
Step2-FILE	<op>_aclnn_runner.cpp 创建	[ -f <op>_aclnn_runner.cpp ]
Step2-CODE	BuildAclnnVariantPack 实现	grep "BuildAclnnVariantPack" <op>_aclnn_runner.cpp
Step2-CODE	SetAclNNWorkspaceExecutor 实现	grep "SetAclNNWorkspaceExecutor" <op>_aclnn_runner.cpp