Pull Request已成功合入, 合并人@CANN-robot
(感谢 zzq 的贡献)变更摘要
本 PR 主要对 setup_torch_npu_daily.sh 环境安装脚本进行健壮性增强,包括新增 Python 版本参数格式校验以防止越界路径构造、为 pip install 补充 PyTorch CPU 额外索引源、以及增加安装成功提示。同时修复了 4 处文档中的路径错误。
主要改动
- Python 版本参数格式校验:在
setup_torch_npu_daily.sh中新增正则校验^[0-9]+\.[0-9]+\.[0-9]+$,当用户传入的PYTHON_VERSION不符合x.y.z格式时直接报错退出,避免通过非法参数构造越界路径。 - pip install 增加 extra-index-url:为 Daily torch_npu Wheel 的安装命令添加
--extra-index-url https://download.pytorch.org/whl/cpu参数,确保依赖解析时能正确获取 PyTorch CPU 版本的包索引。 - 脚本执行成功提示:在脚本末尾新增
echo "SUCCESS: PyTorch 和 torch_npu 环境安装成功。",便于用户明确判断脚本是否完整执行成功。 - 文档路径错误修复:修正了 4 处文档中的路径错误(涉及 4 个文档文件,具体路径细节未在补丁中展开)。


代码审查
经过对全部 5 个变更文件的逐一排查,我未发现本次 diff 引入的任何正确性、安全、可靠性或破坏性变更问题。以下是逐文件审查结论:
审查详情:
-
scripts/env_install/pytorch/setup_torch_npu_daily.sh— 新增三处改动:- Python 版本格式校验(第 25-28 行):使用正则
^[0-9]+\.[0-9]+\.[0-9]+$校验$PYTHON_VERSION,有效阻止了通过非法参数构造越界路径(如../)的风险。正则写法正确,set -e下校验失败会终止脚本。无问题。 - pip install 增加
--extra-index-url(第 190-192 行):指向 PyTorch 官方 CPU wheel 仓库https://download.pytorch.org/whl/cpu,用于正确解析torch_npudaily 包的torch依赖。URL 使用 HTTPS,是 PyTorch 生态的标准做法。无问题。 - 成功提示(第 194-195 行):在
set -e保护下,仅在所有步骤成功后输出。无问题。
- Python 版本格式校验(第 25-28 行):使用正则
-
autofuse/README.md(第 67 行)— 链接从./examples/README.md修正为./examples/pytorch/README.md。旧文件不存在,新文件存在。修复有效。无问题。 -
autofuse/README_en.md(第 81 行)— 链接从./examples/README.md修正为./examples/pytorch/README_en.md。旧文件不存在,新文件存在。修复有效。无问题。 -
docs/en/quick_install.md(第 123 行)— 链接从../../autofuse/examples/README_en.md修正为../../autofuse/examples/pytorch/README_en.md。旧文件不存在,新文件存在。修复有效。无问题。 -
docs/zh/quick_install.md(第 123 行)— 链接从../../autofuse/examples/README.md修正为../../autofuse/examples/pytorch/README.md。旧文件不存在,新文件存在。修复有效。无问题。
总结:
- P0:0
- P1:0
- P2:0
- P3:0
整体风险判断:低风险。 本次变更均为正向改进——Shell 脚本增加了输入校验和依赖解析能力,增强了健壮性;4 处文档链接修复使原本 404 的路径指向了实际存在的文件。未发现回归风险或安全问题。
⚠️ 已识别出整体风险,但无法提取行内评论,请参考整体评估。


Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| repo-cann/graph-autofusion | ✅ zhanj, xchu42, xuyafei (3/2) | ✅ zhanj (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
gcw_Xt6hqG9y, thanks for your pull request. All authors of the commits have signed the CLA. 👍


流水线任务触发成功
任务链接 [a53d3d0a1cb64017bebd785b55317096][流水线指导]
| 任务名称 | 状态 | 日志 | 下载链接 |
|---|---|---|---|
| UT_Test_Python_superkernel | ✅ SUCCESS | >>>>> | |
| ST_Test_Python_superkernel | ✅ SUCCESS | >>>>> | |
| UT_Test_superkernel | ✅ SUCCESS | >>>>> | |
| UT_Test_autofuse_framework | ✅ SUCCESS | >>>>> | |
| ST_Test_autofuse_framework | ✅ SUCCESS | >>>>> | |
| UT_Test_autofuse_ascendc_api | ✅ SUCCESS | >>>>> | |
| ST_Test_autofuse_ascendc_api | ✅ SUCCESS | >>>>> | |
| ST_Test_autofuse_e2e | ✅ SUCCESS | >>>>> | |
| pre_comment | ✅ SUCCESS | >>>>> |
[2026-07-28 10:33:13] CI执行结束


/lgtm


流水线任务触发成功
任务链接 [fb9498e793aa4e318bd4a3393c6954c7][流水线指导]
| 任务名称 | 状态 | 日志 | 下载链接 |
|---|---|---|---|
| UT_Test_Python_superkernel | ✅ SUCCESS | >>>>> | |
| ST_Test_Python_superkernel | ✅ SUCCESS | >>>>> | |
| UT_Test_superkernel | ✅ SUCCESS | >>>>> | |
| UT_Test_autofuse_framework | ✅ SUCCESS | >>>>> | |
| ST_Test_autofuse_framework | ✅ SUCCESS | >>>>> | |
| UT_Test_autofuse_ascendc_api | ✅ SUCCESS | >>>>> | |
| ST_Test_autofuse_ascendc_api | ✅ SUCCESS | >>>>> | |
| ST_Test_autofuse_e2e | ✅ SUCCESS | >>>>> | |
| pre_comment | ✅ SUCCESS | >>>>> |
[2026-07-28 11:49:39] CI执行结束


/lgtm


/approve


Pull Request
描述
完善脚本,增强健壮性。修复了4处文档中的路径错误。
变更类型
请选择本次引入的变更类型:
关联的Issue
如何测试
描述测试此变更的步骤和前提条件:
1.在CANNLab平台上执行没问题
2.
核对清单
其他信息
在此添加任何其他关于本次 PR 的说明。