CANN 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 skills/ops-lab/tilelang/teams/tilelang-op-orchestrator # 3. 执行初始化脚本（项目级） bash init.sh project opencode # OpenCode 用户（默认） bash init.sh project claude # Claude Code 用户 # 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 skills/ops-lab/tilelang/teams/tilelang-op-orchestrator # 3. 执行初始化脚本（全局） bash init.sh global opencode # OpenCode 用户（默认） bash init.sh global claude # Claude Code 用户 # 4. 进入 TileLang-Ascend 源码仓库，安装环境 cd tilelang-ascend bash install_ascend.sh cd ..

安装内容

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

环境校验

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

项目级安装：

skills/ops-lab/tilelang/teams/tilelang-op-orchestrator/ ├── .opencode/ │ ├── skills/ # 技能模块 │ │ ├── tilelang-api-best-practices/ │ │ ├── tilelang-programming-model-guide/ │ │ ├── tilelang-op-design/ │ │ ├── tilelang-op-generate/ │ │ └── tilelang-review/ │ ├── agents/ # 子代理（如有） │ ├── AGENTS.md # Agent 配置 │ └── cannbot-manifest.json # 安装清单 ├── tilelang-ascend # tilelang代码仓 ├── init.sh # 初始化脚本 └── quickstart.md # 本文档

二、快速上手

启动

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

opencode # OpenCode 用户

开发算子示例

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

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

核心工作流

按照 3 阶段渐进式开发，确保算子开发质量：

阶段一：需求分析与方案设计 → 阶段二：算子实现 → 阶段三：精度验证与调试

每个阶段的详细说明参见 AGENTS.md。

产出物示例

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

tilelang-ascend/examples/softmax/ ├── softmax.py # TileLang kernel 实现 ├── design.md # 设计文档 └── README.md # 实现说明（可选）

三、可用技能

四、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