CANNBot torch-compile 快速入门

CANNBot torch-compile 快速入门

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

概述

torch-compile是 PyTorchtorch.compile图模式编排入口，用agents/下的 Subagent 组织不同图模式能力。当前已提供torch-npugraph-exSubagent，面向昇腾 NPUnpugraph_ex/aclgraph模式，覆盖 torch.compile + TorchAir 的配置指导、脚本迁移、自定义算子入图、编译错误和运行时错误诊断。

安装

Claude Code

/plugin marketplace add https://gitcode.com/cann/skills.git /plugin install torch-compile@cannbot /reload-plugins

安装后新开会话，或在当前会话执行/clear触发SessionStart。torch-compile是主对话入口，会把AGENTS.md注入当前 Claude 上下文，并通过agents/torch-npugraph-ex.md处理 npugraph_ex 专项工作。

验证：

claude plugin list # 应看到 torch-compile@cannbot ✔ enabled

OpenCode

opencode plugin cannbot@git+https://gitcode.com/cann/skills.git

如需只启用该插件，在.opencode/opencode.json中配置：

{ "plugin": [["cannbot@git+https://gitcode.com/cann/skills.git", {"team": "torch-compile"}]] }

配置后重启 OpenCode。OpenCode 会注册torch-npugraph-exSubagent，并通过插件消息 transform 把AGENTS.md注入主对话上下文。

如果更新插件后仍只看到旧的 skills、agents 或上下文没有变化，可能是 OpenCode 复用了旧插件缓存。可清理缓存后重新安装并重启：

rm -rf ~/.cache/opencode/ opencode plugin cannbot@git+https://gitcode.com/cann/skills.git

使用示例

我想用 torch.compile 在昇腾 NPU 上加速推理，应该怎么配置 npugraph_ex？

我的模型使用 npugraph_ex 编译失败了，帮我分析这段报错日志。

怎么把自定义算子加入 npugraph_ex 图编译？

可用 Agents

可用 Skills

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