已合并
【PR】:完善pytorch场景下环境部署文档,安装脚本和用例实现 #1398
【PR】:完善pytorch场景下环境部署文档,安装脚本和用例实现 #1398
已合并
zzq创建于 7月21日
zzq成员
7月21日

Pull Request

描述

完善pytorch场景下环境部署文档,安装脚本和用例实现。

变更类型

请选择本次引入的变更类型:

关联的Issue

如何测试

描述测试此变更的步骤和前提条件:
1.
2.

核对清单

其他信息

在此添加任何其他关于本次 PR 的说明。

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 zzq 的贡献)
Zzzq成员
7月21日 添加了label:enhancement
Zzzq成员
7月21日 创建了 pull request,commit 89cafae3
atomgit-bot
atomgit-bot
7月21日 评论:

变更摘要

此 PR 主要对 autofuse/examples 目录下示例用例进行重组,新增 pytorch 子目录存放 PyTorch 场景用例,并调整了示例中的编译方式:移除 inductor_npu_ext 导入,改用显式的 ascendc 后端选项。同时新增了一个用于安装 torch_npu 每日构建包的自动化脚本。

主要改动

  • 示例目录重组: 在 autofuse/examples/ 下新建 pytorch 子目录,将 af_pointwise/af_add_ge.pyaf_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 包。
likedislike
atomgit-bot
atomgit-bot
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.mddocs/zh/quick_install.md 等 4 个文件的导航)
  • P3:4(shell 脚本健壮性 ×3 + 导入变更与文档一致性 ×1)

整体风险评估

中等偏低风险。核心代码变更(Python 示例文件)逻辑正确,torch.compileoptions 参数用法符合 PyTorch API 约定。主要风险集中在两点:

  1. 旧顶层 README 删除后未同步更新外部引用,会导致多处文档导航断裂,建议在合入前补齐链接更新;
  2. 新增的 shell 安装脚本健壮性不足(缺少 set -uo pipefail、未校验中间变量),在边界场景下可能给出难以排查的错误信息,建议加强防御性检查。
类型 数量
🔴 阻塞 0
🟡 建议 2

💬 仅评论

likedislike
CANN-robotCANN-robot成员
7月21日 添加了label:cann-cla/yes
此处折叠了118条消息 查看更多
zhang_shengjie成员7月27日进行代码检视1
autofuse/examples/README.md
已过期
@@ -2,0 +0,0 @@
2- 
zhang_shengjie7月27日评论:

删除 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

likedislike
Zzzq成员
7月27日 解决了最后一个问题
zhanj
zhanj成员
7月27日 评论:

/approve

likedislike
CANN-robotCANN-robot成员
7月27日 添加了label:approved
CANN-robotCANN-robot成员
7月27日 合入了pull request