【合入来源】

[Usage]: torch.utils._pytree.SUPPORTED_NODES.pop 缺少跨版本的直接 NPU 用例

【修改方案】

上游 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 的调用路径为：

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 云算力日志如下：

torch=2.12.0+cu130
torch_npu=2.12.0.rc1
accelerator=npu
[root@73c35cf5cac2 workspace]# python3 ./test_pytree_registry_2.12.0.py TestPytreeSupportedNodes.test_pop
W0715 20:22:27.166000 31297 site-packages/torch/_native/cutedsl_utils.py:55] CuTeDSL operators require optional Python packages `nvidia-cutlass-dsl` and `apache-tvm-ffi`; missing optional dependency `nvidia_cutlass_dsl` (importlib.util.find_spec(nvidia_cutlass_dsl) failed)
[W715 20:22:28.388390810 NPUCachingAllocator.cpp:199] Warning: The current CANN and Soc versions require processing for 32 padding size, with memory allocation. (function operator())
.
----------------------------------------------------------------------
Ran 1 test in 1.023s

OK

CuTeDSL 信息由可选 Python 包未安装触发，allocator 信息由当前 CANN/Soc 的内存对齐处理触发，均不影响用例结果。

【CheckList】

• 代码注释简洁且为英文
• Tensor 按当前 accelerator 迁移
• PR 标题使用 test 标签
• CI 执行通过