【合入来源】

如有社区issue，请关联issue链接
请勿携带内部流程信息（需求链接、问题单、内部issue等）

• 需求
• 问题单
• issue/工单
• 重构优化
• 资料更新

关联issue：https://gitcode.com/Ascend/pytorch/issues/2323

【修改方案】

一、2 个 API 功能如下

1. torch.nn.modules.module.register_module_full_backward_hook
   • 为所有 nn.Module 注册全局 full backward hook，在完整反向传播阶段触发回调。
2. torch.nn.modules.module.register_module_full_backward_pre_hook
   • 为所有 nn.Module 注册全局 full backward pre hook，在完整反向传播前触发回调。

二、社区用例对这 2 个 API 的验证完整性分析

1. 未检索到这 2 个“全局 API”的直接官方用例；
2. 上游仅覆盖了实例级 API：
   • register_full_backward_hook
   • register_full_backward_pre_hook
3. 因此社区现有用例对这 2 个全局 API 的直接验证不完整，需要新增测试文件补齐。

三、NPU 适配

1. 新增 test/nn/test_global_module_full_backward_hooks.py；
2. 直接验证 torch.nn.modules.module.register_module_full_backward_hook；
3. 直接验证 torch.nn.modules.module.register_module_full_backward_pre_hook；
4. 核心检查点为 hook 是否触发、梯度是否按预期改写；
5. 相关张量均在 NPU 上执行。

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

1. 测试环境

- 操作系统：Ubuntu
- 昇腾硬件信息：NPU
- CANN软件版本：8.5.0
- 安装的对应软件版本：torch 2.12.0+cu130、torch_npu 2.12.0rc1

2. 测试命令

python -u /workspace/torch_npu_v2.12.0/test/nn/test_global_module_full_backward_hooks.py   > /workspace/version_logs/module_hooks_new2_v2.12.0.log 2>&1

3. patch 格式检查真实输出

EXIT_CODE:0

4. 测试真实输出

W0609 14:44:41.813000 33504 venvs/torch2120/lib/python3.11/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)
[W609 14:44:42.301485790 NPUCachingAllocator.cpp:199] Warning: The current CANN and Soc versions require processing for 32 padding size, with memory allocation. (function operator())
..
----------------------------------------------------------------------
Ran 2 tests in 1.234s

OK
EXIT_CODE:0

【CheckList】

PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x]

• 代码注释完备，正确记录错误日志
• 代码实现进行了返回值、空指针等校验
• PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等
• PR持续集成流水线（CI）执行通过，代码检查无异常