已关闭
[Usage]: [v2.12.0] Add NPU validation for global full backward module hook APIs #2323
coconut创建于  6月9日关闭于  6月12日
coconut
coconut
6月9日 创建

在提交新问题之前,请确保您已经在社区中搜索过相关问题,并使用了社区中提供的资源/工具后,仍未找到满意的解决方式。

环境信息

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

使用场景及问题

API名称

  • torch.nn.modules.module.register_module_full_backward_hook
  • torch.nn.modules.module.register_module_full_backward_pre_hook

问题说明

PyTorch 官方社区未提供这 2 个全局 full backward module hook API 的直接测试用例,仅有实例级 hook 用例,因此当前社区验证对这 2 个 API 不够直接,需要新增测试文件补齐。

一、2 个 API 功能如下

  1. torch.nn.modules.module.register_module_full_backward_hook
  2. torch.nn.modules.module.register_module_full_backward_pre_hook

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

  1. 未检索到这 2 个全局 API 的直接官方用例;
  2. 上游仅有实例级 register_full_backward_hookregister_full_backward_pre_hook 相关验证;
  3. 结论:需要新增测试文件做直接验证。

三、NPU 适配

  1. 新增 test/nn/test_global_module_full_backward_hooks.py
  2. 在 NPU 上直接验证 hook 触发和梯度改写结果。

已做验证

Ran 2 tests in 1.234s

OK
EXIT_CODE:0

欢迎加入社区,感谢您对社区的贡献 🎉!

likedislike
ascend-robotascend-robot成员
6月9日 添加了label:usage
coconutcoconut
6月9日 关联了pull request:[v2.12.0] Add NPU tests for global full backward module hook APIs
ascend-robotascend-robot成员
6月12日 关闭了 issue
ascend-robotascend-robot成员
6月12日 添加了label:resolved