Pull Request已成功合入, 合并人@ascend-robot
(感谢 hbhu_bin 的贡献)变更摘要
此 PR 主要完成两件事:一是移除构建脚本中用于开启 DVM 延迟融合编译的环境变量 _TORCH_NPU_ENABLE_DVM,表明该特性已不再需要外部环境变量触发;二是为 DVM 模块新增多个 eager 模式的测试用例,覆盖 viewstore 的原地写入和 native_batch_norm_backward 的反向传播场景。同时更新了 third_party/op-plugin 子模块指针以配合相关底层改动。
主要改动
- 移除
_TORCH_NPU_ENABLE_DVM环境变量:在build_libtorch_npu.py和setup.py中删除了os.environ.setdefault("_TORCH_NPU_ENABLE_DVM", "1")的设置,相关注释一并移除,表明 DVM 延迟融合源码的编译不再依赖该开关。 - 新增 viewstore 原地写入测试:在
test/test_dvm.py中增加test_viewstore_inplace_strided_3d_view和test_viewstore_inplace_sibling_views两个用例,分别验证对非连续 3D 视图的原地add_/tanh_操作和多个兄弟视图的原地mul_/add_/sub_/relu_操作,并通过compare_dvm_on_off对比 DVM 开关下的结果一致性。 - 新增
native_batch_norm_backward测试:在test/test_dvm.py中增加test_native_batch_norm_backward_train_weight_only和test_native_batch_norm_backward_train_masks两个用例,覆盖仅有权重梯度、多种[save_mean, save_var, weight]掩码组合下的反向传播,以及不同输入形状的场景。 - 更新 op-plugin 子模块:将
third_party/op-plugin的 commit 指针从1473204更新至0556ee9,以同步底层插件变更。


代码审查
经过全面审查,我对每个变更文件的分析如下:
文件 1: build_libtorch_npu.py(第 19-20 行删除)
移除了两行设置 _TORCH_NPU_ENABLE_DVM 环境变量的代码。全仓库搜索确认该环境变量已无其他引用。这是与 PR 目标"移除 _TORCH_NPU_ENABLE_DVM"一致的清理操作,无问题。
文件 2: setup.py(第 31-32 行删除)
与 build_libtorch_npu.py 相同的清理操作,移除相同的两行环境变量设置。一致性良好,无问题。
文件 3: test/test_dvm.py(新增 4 个测试方法)
新增的测试方法:
test_viewstore_inplace_strided_3d_view(第 695-710 行):测试对非连续视图的原地写入,张量形状、broadcast 语义均正确。test_viewstore_inplace_sibling_views(第 712-732 行):测试多个非连续兄弟视图的原地更新,所有操作语义正确。test_native_batch_norm_backward_train_weight_only(第 963-977 行):测试仅 weight 梯度的 batch norm 反向传播,参数形状匹配正确。test_native_batch_norm_backward_train_masks(第 979-1004 行):参数化测试多种 output_mask 组合(含 3D 输入),subTest 用法正确,形状计算和save_var正值保证均正确。
所有测试方法都遵循已有模式:
- 使用
compare_dvm_on_off在子进程中对比 DVM 开关结果 - 调用
_assert_match进行浮点容差比较 atol值(1e-6 / 1e-4)与已有测试用例一致
无问题。
文件 4: third_party/op-plugin(子模块指针更新)
子模块 commit 从 147320421eb3b38c8d81bcf7dd4884c25347ca7c 更新到 0556ee97bb1e5dddda60a232f86a96a6f386eee9。这与移除 _TORCH_NPU_ENABLE_DVM 的变更一致——op-plugin 在新 commit 中应已移除对该环境变量的依赖。无问题。
关闭总结
- P0–P2 问题数:0
- P3 问题数:0
- 整体风险判断:低风险。本次变更是一次干净的清理操作:移除已废弃的
_TORCH_NPU_ENABLE_DVM环境变量(含对应的子模块指针更新),并新增 4 个 DVM eager fusion 的测试用例。所有新增测试用例结构正确,与已有测试模式一致。
已审查的文件:
build_libtorch_npu.py— 无问题setup.py— 无问题test/test_dvm.py— 无问题third_party/op-plugin— 无问题
⚠️ 已识别出整体风险,但无法提取行内评论,请参考整体评估。


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-Ascend/pytorch | ✅ liujunzhu, ffmh (2/2) | ✅ liujunzhu (1/1) |
| test | ✅ liujunzhu, ffmh (2/2) | ✅ liujunzhu (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
hbhu_bin, thanks for your pull request. All authors of the commits have signed the CLA. 👍


Linking Issue Notice
@hbhu_bin , the pull request must be linked to at least one issue.
If an issue has already been linked, but the needs-issue label remains, you can remove the label by commenting /check-issue .


当前仓库存在以下 保护分支 :
| Protected Branch | Version | Release |
|---|---|---|
| master | ||
| v2.7.1-26.1.0 | ||
| v2.9.0-26.1.0 | ||
| v2.10.0-26.1.0 | ||
| v2.12.0-26.1.0 | ||
| v2.11.0-26.1.0 | ||
| v2.9.0 | ||
| v2.10.0 | ||
| v2.11.0 | ||
| v2.12.0 | ||
| v2.7.1 | ||
| ci-test |
评论 /sync <branch1> <branch2> ... 可将当前 PR 修改同步到其它分支(创建同步 PR):
a) 如果当前 PR 是 Open 状态,同步操作将延迟到 PR 被合并时执行
b) 如果当前 PR 已经 Merged,将立即执行同步操作
注意:
- /sync 命令可以指定同步到多个分支,仅最后一个 /sync 命令生效
- 如果创建的同步 PR 不正确,可通过向同步 PR 的源分支提交轻量级 PR 完善,或使用 /close 命令关闭


ascend docs pipeline is running...


| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | Build_X86 | 🕚 | >>> |
| Build_ARM | 🕚 | >>> | |
| Build_LibTorch_x86 | 🕚 | >>> | |
| Build_LibTorch_ARM | 🕚 | >>> | |
| Build_X86_torchair | 🕚 | >>> | |
| Build_ARM_torchair | 🕚 | >>> | |
| patch_test | 🕚 | >>> | |
| 恶意代码检查 | Antipoison | 🟨 | >>> |
| 编码安全与规范检查 | CodeCheck | 🟨 | >>> |
| check_error | ✅ | >>> | |
| CodeCheck_lintrunner | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | UT_X86_Part_01 | 🕚 | >>> |
| UT_X86_Part_02 | 🕚 | >>> | |
| UT_ARM_A3_Part_01 | 🕚 | >>> | |
| UT_ARM_A3_Part_02 | 🕚 | >>> | |
| UT_ARM_A2_Part_01 | 🕚 | >>> | |
| UT_ARM_A2_Part_02 | 🕚 | >>> | |
| UT_ARM_A2_Part_03 | 🕚 | >>> | |
| UT_inductor_Part_01 | 🕚 | >>> | |
| UT_inductor_Part_02 | 🕚 | >>> | |
| UT_inductor_Part_03 | 🕚 | >>> | |
| UT_inductor_Part_04 | 🕚 | >>> | |
| UT_DIST_ARM_Part_01 | 🕚 | >>> | |
| UT_DIST_ARM_Part_02 | 🕚 | >>> | |
| UT_DIST_ARM_Part_03 | 🕚 | >>> | |
| UT_DIST_ARM_Part_04 | 🕚 | >>> | |
| UT_ARM_A2_Select_Part_01 | 🕚 | >>> | |
| UT_ARM_A2_Select_Part_02 | 🕚 | >>> | |
| 流水线 | PR-pipeline_pytorch | 🟨 | >>> |
- compile、compile_inductor、compile_torchair : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线


✅ 跳过 docs ci 检查,没有需要检查的文档文件


ascend docs pipeline is running...


✅ 跳过 docs ci 检查,没有需要检查的文档文件


| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | Build_X86 | ✅ | >>> |
| Build_ARM | ✅ | >>> | |
| Build_LibTorch_x86 | ✅ | >>> | |
| Build_LibTorch_ARM | ✅ | >>> | |
| Build_X86_torchair | 🛑 | >>> | |
| Build_ARM_torchair | 🛑 | >>> | |
| patch_test | 🛑 | >>> | |
| 恶意代码检查 | Antipoison | ✅ | >>> |
| 编码安全与规范检查 | CodeCheck | ✅ | >>> |
| check_error | ✅ | >>> | |
| CodeCheck_lintrunner | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | UT_X86_Part_01 | 🛑 | >>> |
| UT_X86_Part_02 | 🛑 | >>> | |
| UT_ARM_A3_Part_01 | 🛑 | >>> | |
| UT_ARM_A3_Part_02 | 🛑 | >>> | |
| UT_ARM_A2_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Part_02 | ✅ | >>> | |
| UT_ARM_A2_Part_03 | ✅ | >>> | |
| UT_inductor_Part_01 | 🛑 | >>> | |
| UT_inductor_Part_02 | 🛑 | >>> | |
| UT_inductor_Part_03 | 🛑 | >>> | |
| UT_inductor_Part_04 | 🛑 | >>> | |
| UT_DIST_ARM_Part_01 | 🛑 | >>> | |
| UT_DIST_ARM_Part_02 | 🛑 | >>> | |
| UT_DIST_ARM_Part_03 | 🛑 | >>> | |
| UT_DIST_ARM_Part_04 | 🛑 | >>> | |
| UT_ARM_A2_Select_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Select_Part_02 | ✅ | >>> | |
| 流水线 | PR-pipeline_pytorch | ✅ | >>> |
- compile、compile_inductor、compile_torchair : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线


/approve


/lgtm




【合入来源】
https://gitcode.com/Ascend/pytorch/issues/2264
【修改方案】
增加DVM测试用例+删除选择编译代码
【资料变更】
不涉及
【接口变更】
不涉及
【功能验证】
不涉及
【CheckList】