openLiBing可接管PR, PR详情链接:
https://www.openlibing.com/apps/transitionPrDetail?owner=Ascend&repo=pytorch&number=41678&platform=gitcode&codeHostingPlatformFlag=gitcode

变更摘要

此 PR 修复了 NPU 后端 THNPEvent_dealloc 和 THNPStream_dealloc 中因未同步上游 PyTorch 基础类 weakref 支持而导致的内存安全问题。由于 CPython 对静态 PyType 不会自动链式调用父类 tp_dealloc，NPU 的覆盖实现缺少 PyObject_ClearWeakRefs() 调用，导致 weakref 的 wr_object 悬空指向已释放内存（use-after-free）；同时 THNPStream_dealloc 还缺少对 self->context 的清理，造成 lazy stream context 泄漏。修复后添加了相应的回归测试。

主要改动

• THNPEvent_dealloc 添加 weakref 清理：在 torch_npu/csrc/npu/Event.cpp 的 THNPEvent_dealloc 中，析构 npu_event 之后、调用 tp_free 之前插入 PyObject_ClearWeakRefs((PyObject*)self);，确保 weakref 引用在对象释放前被正确清理。
• THNPStream_dealloc 添加 weakref 清理与 context 释放：在 torch_npu/csrc/npu/Stream.cpp 的 THNPStream_dealloc 中，新增 PyObject_ClearWeakRefs((PyObject*)self); 和 Py_CLEAR(self->context);，分别解决悬空指针和 context 泄漏问题。
• 新增 weakref 回调回归测试：在 test/dynamo/test_stream.py 中添加 _assert_weakref_callback_fires 辅助方法及 test_npu_stream_event_weakref_callback 测试用例，验证 torch.npu.Stream 和 torch.npu.Event 对象销毁时 weakref 回调能正常触发。
• 新增 dynamo 注册表悬空指针回归测试：添加 test_dynamo_registry_no_dangling_weakref 测试用例，模拟 torch.compile 捕获 current_stream() 后 wrapper 被释放的场景，验证 index_to_external_object_weakref 注册表中的 weakref 不会保留悬空指针。

代码审查

经过全面审查，我确认此 diff 中的三处更改均正确且必要，未发现错误、安全风险、可靠性问题或回归风险。

变更分析：

1. torch_npu/csrc/npu/Event.cpp 第 121 行：在 THNPEvent_dealloc 的 tp_free 之前添加 PyObject_ClearWeakRefs。当父类 THPEvent 支持 weakref 后，子类自定义的 tp_dealloc 覆盖必须自行调用此函数，否则 weakref 的 wr_object 会悬空指向已释放的内存。此修改正确。

2. torch_npu/csrc/npu/Stream.cpp 第 79-80 行：在 THNPStream_dealloc 中添加 PyObject_ClearWeakRefs 和 Py_CLEAR(self->context)。前者防止 weakref 悬空，后者修复 lazy stream context 泄漏。调用顺序（先清 NPU 资源 → 清 weakref → 清 context → 释放内存）合理：weakref 回调不会访问已析构的 NPU 资源，且 Py_CLEAR 能用 Py_XDECREF 安全处理 NULL。

3. test/dynamo/test_stream.py 第 3-117 行：添加了两个回归测试：

   • test_npu_stream_event_weakref_callback：验证 Stream/Event 析构时 weakref 回调正确触发。
   • test_dynamo_registry_no_dangling_weakref：复现原始 UAF 场景，验证 dynamo registry 中的 weakref 在 wrapper 释放后被正确清空。

   测试逻辑严谨，辅助方法 _assert_weakref_callback_fires 正确使用了 weakref.ref 回调机制和 gc.collect()。

审查结论： 共审查 3 个文件，P0–P3 问题数均为 0。此更改安全，可以合入。

⚠️ 已识别出整体风险，但无法提取行内评论，请参考整体评估。

/lgtm