已关闭
[Feature]utils/_module.py的patch消除 #1922
shawnylee233创建于  5月13日关闭于  5月15日
shawnylee233成员
5月13日 创建

【需求背景】
torch_npu为了使PyTorch兼容NPU平台,功能上添加了patch。这些patch随着PyTorch的版本演进,在配套使用时存在风险,需要消除。
【详细需求】
torch_npu/torch_npu/utils/_module.py中torch.nn.modules.batchnorm.SyncBatchNorm.forward完全重写原生 forward 以接入 NPU 集合通信。完全重写原生 forward 以接入 NPU 集合通信。
patch里在syncbatchnorm函数中使用了torch_npu接口,当前需要用的接口npu后端均已支持,可直接消除与验证

likedislike
ascend-robotascend-robot成员
5月13日 添加了label:feature
Sshawnylee233成员
5月13日 关联了pull request:remove syncbn patch
Sshawnylee233成员
5月13日 将 shawnylee233 设为负责人
ascend-robotascend-robot成员
5月13日 关联了pull request:[sync] PR-35045: remove syncbn patch
ascend-robotascend-robot成员
5月13日 关联了pull request:[sync] PR-35045: remove syncbn patch
ascend-robotascend-robot成员
5月13日 关联了pull request:[sync] PR-35045: remove syncbn patch
ascend-robotascend-robot成员
5月13日 关联了pull request:[sync] PR-35045: remove syncbn patch
Hhuangyunlong成员
5月15日 issue状态由 TODO 改变为 DONE
Hhuangyunlong成员
5月15日 关闭了 issue
ascend-robotascend-robot成员
5月19日 添加了label:resolved