已关闭
[Feature]utils/_module.py的patch消除 #1922
shawnylee233创建于 5月13日关闭于 5月15日
5月13日 添加了label:feature
5月13日 关联了pull request:remove syncbn patch
5月13日 将 shawnylee233 设为负责人
5月13日 关联了pull request:[sync] PR-35045: remove syncbn patch
5月13日 关联了pull request:[sync] PR-35045: remove syncbn patch
5月13日 关联了pull request:[sync] PR-35045: remove syncbn patch
5月13日 关联了pull request:[sync] PR-35045: remove syncbn patch
5月15日 issue状态由 TODO 改变为 DONE
5月15日 关闭了 issue
5月19日 添加了label:resolved
【需求背景】
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后端均已支持,可直接消除与验证