已合并
【PR】:完善pytorch场景下环境部署文档,安装脚本和用例实现 #1398
zzq创建于 7月21日
【PR】:完善pytorch场景下环境部署文档,安装脚本和用例实现 #1398
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 zzq 的贡献)atomgit-bot
7月21日 评论:
7月21日 评论:
变更摘要
此 PR 主要对 autofuse/examples 目录下示例用例进行重组,新增 pytorch 子目录存放 PyTorch 场景用例,并调整了示例中的编译方式:移除 inductor_npu_ext 导入,改用显式的 ascendc 后端选项。同时新增了一个用于安装 torch_npu 每日构建包的自动化脚本。
主要改动
- 示例目录重组: 在
autofuse/examples/下新建pytorch子目录,将af_pointwise/af_add_ge.py和af_reduce/af_mul_reducesum.py移入其中,结构化为pytorch/af_pointwise/和pytorch/af_reduce/。 - 编译后端切换: 两个示例中移除了
import inductor_npu_ext,并将torch.compile调用改为显式传入options={"npu_backend": "ascendc"},使用 AscendC 后端替代原有的 Inductor 后端路径。 - 新增每日构建安装脚本: 新增
scripts/package/graph_autofusion/env_install/pytorch/setup_torch_npu_daily.sh,支持通过参数指定 Conda 环境、torch_npu 版本和下载目录,自动拉取并安装当日的 torch_npu 每日构建 Wheel 包。


atomgit-bot
7月21日 评论:
7月21日 评论:
代码审查
审查总结
审查覆盖的 9 个文件
| 文件 | 审查结果 |
|---|---|
autofuse/examples/README.md(已删除) |
P2:删除导致 4 处外部文档链接失效 |
autofuse/examples/README_en.md(已删除) |
同上(同根因,合并为一个 P2 报告) |
autofuse/examples/pytorch/README.md(新增) |
P3:硬编码 miniconda 路径 |
autofuse/examples/pytorch/README_en.md(新增) |
P3:硬编码 miniconda 路径 |
autofuse/examples/pytorch/af_pointwise/README.md(重命名) |
无新增问题 |
autofuse/examples/pytorch/af_pointwise/README_en.md(重命名) |
无新增问题 |
autofuse/examples/pytorch/af_pointwise/af_add_ge.py(修改) |
P3:移除 import inductor_npu_ext 与主文档不一致 |
autofuse/examples/pytorch/af_reduce/README.md(重命名) |
无新增问题 |
autofuse/examples/pytorch/af_reduce/README_en.md(重命名) |
无新增问题 |
autofuse/examples/pytorch/af_reduce/af_mul_reducesum.py(修改) |
同 af_add_ge.py(同根因,合并为一个 P3 报告) |
scripts/package/graph_autofusion/env_install/pytorch/setup_torch_npu_daily.sh(新增) |
P3:3 个独立问题(缺少 pipefail/-u、WHEEL 空值未校验、硬编码 miniconda 路径) |
按优先级统计
- P0:0
- P1:0
- P2:1(文档链接失效 → 影响
autofuse/README.md、docs/zh/quick_install.md等 4 个文件的导航) - P3:4(shell 脚本健壮性 ×3 + 导入变更与文档一致性 ×1)
整体风险评估
中等偏低风险。核心代码变更(Python 示例文件)逻辑正确,torch.compile 的 options 参数用法符合 PyTorch API 约定。主要风险集中在两点:
- 旧顶层 README 删除后未同步更新外部引用,会导致多处文档导航断裂,建议在合入前补齐链接更新;
- 新增的 shell 安装脚本健壮性不足(缺少
set -uo pipefail、未校验中间变量),在边界场景下可能给出难以排查的错误信息,建议加强防御性检查。
| 类型 | 数量 |
|---|---|
| 🔴 阻塞 | 0 |
| 🟡 建议 | 2 |
💬 仅评论


7月21日 添加了label:cann-cla/yes
此处折叠了118条消息 查看更多
autofuse/examples/README.md
已过期
@@ -2,0 +0,0 @@
2-
删除 examples 顶层 README 后仍有 4 个失效链接
当前仍指向已删除文件:
autofuse/README.md:67
autofuse/README_en.md:81
docs/zh/quick_install.md:123
docs/en/quick_install.md:123
应分别更新到:
autofuse/examples/pytorch/README.md
autofuse/examples/pytorch/README_en.md


zhanj
7月27日 评论:
7月27日 评论:
/approve


7月27日 添加了label:approved
7月27日 合入了pull request
Pull Request
描述
完善pytorch场景下环境部署文档,安装脚本和用例实现。
变更类型
请选择本次引入的变更类型:
关联的Issue
如何测试
描述测试此变更的步骤和前提条件:
1.
2.
核对清单
其他信息
在此添加任何其他关于本次 PR 的说明。