已合并
test(npu): add pytree supported nodes pop coverage #41812
jerry lee创建于 7月16日
test(npu): add pytree supported nodes pop coverage #41812
已合并
jerry lee
7月16日
7月16日
【合入来源】
[[Usage]: `torch.utils._pytree.SUPPORTED_NODES.pop` 缺少跨版本的直接 NPU 用例](https://gitcode.com/Ascend/pytorch/issues/2889)
【修改方案】
上游 v2.7.1 的 `torch.onnx._internal.fx.dynamo_graph_extractor._PyTreeExtensionContext.__exit__` 调用 `pop(class_type)`，但没有直接测试；v2.9.0、v2.10.0 没有直接调用；v2.11.0、v2.12.0 仅在注册表清理中调用 `pop(cls, None)`，未验证返回值和删除后的展开语义。因此新增 `test/test_pytree_registry.py`。
`test.test_pytree_registry.TestPytreeSupportedNodes.test_pop` 的调用路径为：
```text
torch.utils._pytree.register_pytree_node(CustomNode, ...)
-> torch.utils._pytree.tree_flatten(CustomNode(npu_tensor))
-> torch.utils._pytree.SUPPORTED_NODES[CustomNode].flatten_fn
-> torch.utils._pytree.SUPPORTED_NODES.pop(CustomNode)
-> torch.utils._pytree.SUPPORTED_NODES.pop(CustomNode, None)
-> torch.utils._pytree.tree_flatten(CustomNode(npu_tensor))
```
用例验证无默认值的 `pop(CustomNode)` 返回原 `NodeDef` 的 `flatten_fn` 并删除注册项，带默认值的 `pop(CustomNode, None)` 返回 `None`，删除后同一节点回退为单个叶子。Tensor 使用 `torch.tensor([1.0]).to(device_type)`；清理使用 `TestCase.addCleanup()` 恢复注册项后再注销，不捕获 API 异常。
【资料变更】
`torch.utils._pytree.SUPPORTED_NODES.pop` 位于 PyTorch 私有模块，不需要更新 native API 支持文档。
【接口变更】
不涉及 API 源码或用户接口，仅新增 torch-npu 测试。
【功能验证】
`python3 -m py_compile test/test_pytree_registry.py` 和 `git diff --check` 已通过。为避免 torch-npu 源码 checkout 遮蔽 pip 安装包，从中立的 `/workspace` 目录执行同一测试文件，NPU 云算力日志如下：
```text
torch=2.10.0+cpu
torch_npu=2.10.0
accelerator=npu
[root@73c35cf5cac2 workspace]# python3 ./test_pytree_registry_2.10.0.py TestPytreeSupportedNodes.test_pop
.
----------------------------------------------------------------------
Ran 1 test in 1.045s
OK
```
【CheckList】
- [x] 代码注释简洁且为英文
- [x] Tensor 按当前 accelerator 迁移
- [x] PR 标题使用 `test` 标签
- [ ] CI 执行通过
Pull Request已成功合入, 合并人@ascend-robot
（感谢 jerry lee 的贡献）7月16日 关联了issue：[Usage]: `torch.utils._pytree.SUPPORTED_NODES.pop` 缺少跨版本的直接 NPU 用例
ascend-robot
7月16日 评论：
7月16日 评论：
atomgit-bot
7月16日 评论：
7月16日 评论：
变更摘要
本次变更新增测试文件 test/test_pytree_registry.py，为 torch.utils._pytree.SUPPORTED_NODES.pop 接口补充直接的用例覆盖。该接口在上游社区缺乏充分验证，且跨越多个 PyTorch 版本（v2.7.1–v2.12.0）调用形式存在差异（有默认值/无默认值、是否验证返回值等）。新增用例完整覆盖 pop 在不同参数形式下的注册项删除行为及删除后 tree_flatten 的回退语义，并通过 device_type 自适应确保在 NPU 和 CPU 上均可执行。
主要改动
- 新增
test/test_pytree_registry.py测试文件：包含一个测试类TestPytreeSupportedNodes，用于验证_pytree.SUPPORTED_NODES.pop的注册表变更行为。 - 新增
test_pop用例：验证无默认值的pop(CustomNode)返回原始NodeDef且其flatten_fn与注册时一致，同时确认节点已从SUPPORTED_NODES中移除。 - 验证带默认值的
pop行为：在节点已被删除后，pop(CustomNode, None)返回None，确保默认值语义正确。 - 验证注销后
tree_flatten回退为叶子语义：删除注册后对同一节点调用tree_flatten，确认spec.is_leaf()为True且返回值为节点本身，覆盖展开逻辑的回退路径。 - NPU/CPU 自适应设备类型：通过
torch.accelerator.current_accelerator()动态获取device_type，测试张量使用.to(device_type)创建，使用例在 NPU 和 CPU 环境均可运行。
atomgit-bot
7月16日 评论：
7月16日 评论：
7月16日 添加了label：ascend-cla/yes
ascend-robot
7月16日 评论：
7月16日 评论：
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 |
|---|---|---|
| test | ✅ liangsongwei, huangjingwei (2/2) | ✅ liangsongwei (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
ascend-robot
7月16日 评论：
7月16日 评论：
ascend-robot
7月16日 评论：
7月16日 评论：
当前仓库存在以下 保护分支 ：
| Protected Branch | Version | Release |
|---|---|---|
| master | ||
| v2.10.0 | ||
| v2.12.0 | ||
| v2.7.1 | ||
| v2.9.0 | ||
| v2.11.0 | ||
| v2.12.0-26.1.0 | ||
| v2.10.0-26.1.0 | ||
| v2.9.0-26.1.0 | ||
| v2.7.1-26.1.0 | ||
| v2.11.0-26.1.0 | ||
| ci-test |
评论 /sync <branch1> <branch2> ... 可将当前 PR 修改同步到其它分支（创建同步 PR）：
a) 如果当前 PR 是 Open 状态，同步操作将延迟到 PR 被合并时执行
b) 如果当前 PR 已经 Merged，将立即执行同步操作
注意：
- /sync 命令可以指定同步到多个分支，仅最后一个 /sync 命令生效
- 如果创建的同步 PR 不正确，可通过向同步 PR 的源分支提交轻量级 PR 完善，或使用 /close 命令关闭
7月16日 添加了label：ci-pipeline-running
ascend-robot
7月16日 评论：
7月16日 评论：
ascend docs pipeline is running...
7月16日 添加了label：docs-ci-pipeline-running
ascend-robot
7月16日 评论：
7月16日 评论：
✅ 跳过 docs ci 检查，没有需要检查的文档文件
7月16日 删除了label：docs-ci-pipeline-running
7月16日 添加了label：docs-ci-pipeline-success
7月16日 删除了label：ci-pipeline-running
7月16日 添加了label：ci-pipeline-passed
ascend-robot
7月16日 评论：
7月16日 评论：
流水线 PR-pipeline_pytorch#45867 [ commitID：b6a3cc47 ] 已完成
>>>代码风格自动修复执行成功（无修复内容）
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | 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 : 停止流水线
Jingwei Huang
28 天前 评论：
28 天前 评论：
/lgtm
梁松伟
27 天前 评论：
27 天前 评论：
/approve
27 天前 添加了label：approvedlgtm
27 天前 删除了label：ci-pipeline-passed
27 天前 合入了pull request
ascend-robot
27 天前 评论：
27 天前 评论：
流水线 pytorch_gitcode_PR_multiVersion#13009 [ commitID：b6a3cc47 ] 已完成