cann/cannbot-skills TileLang算子开发指南

CANNBot TileLang 算子开发快速入门指南

【免费下载链接】cannbot-skillsCANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体，本仓库为其提供可复用的 Skills 模块。项目地址: https://gitcode.com/cann/cannbot-skills

概述

CANNBot TileLang 算子开发模式适用于通过TileLang-Ascend框架开发自定义算子。基于 TVM 编译器基础设施，使用 Python DSL +@tilelang.jit编写 AI 计算 kernel，支持 Developer 模式（自动化）和 Expert 模式（手动控制）两种编程范式。

一、环境搭建

操作步骤

方式一：项目级安装（推荐）

在项目目录下安装，配置仅对当前项目生效。

# 1. 克隆 CANN Skills 仓库 git clone https://gitcode.com/cann/cannbot-skills.git # 2. 进入 TileLang 算子开发目录 cd cannbot-skills/plugins-official/tilelang-op-orchestrator # 3. 执行初始化脚本（项目级） bash init.sh project opencode # OpenCode 用户（默认） bash init.sh project claude # Claude Code 用户 bash init.sh project trae # Trae 用户 bash init.sh project cursor # Cursor 用户 bash init.sh project copilot # Copilot 用户 # 4. 进入 TileLang-Ascend 源码仓库，安装环境 cd tilelang-ascend bash install_ascend.sh cd ..

方式二：全局安装

在用户目录下安装，配置全局生效。

# 1. 克隆 CANN Skills 仓库 git clone https://gitcode.com/cann/cannbot-skills.git # 2. 进入 TileLang 算子开发目录 cd cannbot-skills/plugins-official/tilelang-op-orchestrator # 3. 执行初始化脚本（全局） bash init.sh global opencode # OpenCode 用户（默认） bash init.sh global claude # Claude Code 用户 bash init.sh global trae # Trae 用户 bash init.sh global cursor # Cursor 用户 bash init.sh global copilot # Copilot 用户 # 4. 进入 TileLang-Ascend 源码仓库，安装环境 cd tilelang-ascend bash install_ascend.sh cd ..

安装内容

init.sh 脚本会完成以下操作：

Cursor 安装路径

Copilot 安装路径

环境校验

执行完上述步骤后，检查目录结构是否符合以下规范：

项目级安装：

cannbot-skills/plugins-official/tilelang-op-orchestrator/ ├── .opencode/ │ ├── skills/ # 技能模块（9 个） │ │ ├── tilelang-env-check/ │ │ ├── tilelang-submodule-pull/ │ │ ├── tilelang-op-design/ │ │ ├── tilelang-op-develop/ │ │ ├── tilelang-op-test-design/ │ │ ├── tilelang-perf-optimization/ │ │ ├── tilelang-api-best-practices/ │ │ ├── tilelang-programming-model-guide/ │ │ └── tilelang-review/ │ ├── agents/ # 3 个子代理（analyst / developer / perf-tuner） │ ├── AGENTS.md # 编排器（Primary）配置 │ └── cannbot-manifest.json # 安装清单 ├── tilelang-ascend # tilelang代码仓 ├── init.sh # 初始化脚本 └── quickstart.md # 本文档

二、快速上手

启动

在初始化完成的目录下执行：

opencode # OpenCode 用户

开发算子示例

在交互界面中输入算子开发需求，CANNBot 会按照“算子方案设计-->算子代码实现-->算子精度验证”分阶段开发流程引导你完成：

帮我开发一个 softmax 算子方案设计

核心工作流

采用 3 阶段状态机编排，由 orchestrator 统一调度，确保算子开发质量：

Stage 1 算子设计（含需求理解） → Stage 2 代码实现 + 测试 + 精度调试（一站式） → Stage 3 性能调优（可选）

每阶段通过工件门禁校验后才进入下一阶段；Stage 2 内部完成"生成代码 → 跑测试 → 精度调试"全部循环，精度通过后才询问是否进入 Stage 3。支持断点续跑、失败恢复与设计回退（Subagent 返回[DESIGN_ERROR]时回退到 Stage 1 重做设计），详见 AGENTS.md。

产出物示例

TileLang 算子开发模式下，CANNBot 会在examples/{operator}/目录下生成文件：

examples/softmax/ ├── DESIGN.md # Stage 1 设计文档 ├── example_softmax.py # Stage 2 kernel + 内嵌 golden + test 用例 + main 块 ├── README.md # 实现说明（可选） ├── perf_tuning/ # Stage 3 性能调优产物（可选） ├── history_version/ # 设计回退 / 精度调试备份 └── .orchestrator_state.json # 流程状态（自动维护，支持断点续跑）

三、可用技能与代理

四、Developer 模式 vs Expert 模式

TileLang-Ascend 支持两种编程范式，开发前需先确认使用哪种模式：

详细对照参见tilelang-programming-model-guideskill。

五、常见问题

Q: 如何查看帮助信息？

bash init.sh --help

Q: 项目级和全局安装如何选择？

• 项目级：适合多项目开发，每个项目可以有不同配置
• 全局：适合单一项目，全局生效

Q: 如何更新技能模块？

重新执行 init.sh 即可，脚本会自动覆盖旧版本。

Q: 如何选择 Developer 模式还是 Expert 模式？

总结

1. TileLang 算子开发模式通过 Python DSL 实现昇腾 NPU 算子的快速开发
2. 环境搭建核心两步：克隆仓库 → 执行 init.sh
3. opencode/claude是核心交互指令
4. 开发前必须确认使用 Developer / Expert / 混合模式

【免费下载链接】cannbot-skillsCANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体，本仓库为其提供可复用的 Skills 模块。项目地址: https://gitcode.com/cann/cannbot-skills