已合并
refactor: remove LSTM.forward monkey-patch for v2.10.0 #45456
ymdsall创建于 9 天前
refactor: remove LSTM.forward monkey-patch for v2.10.0 #45456
已合并
Pull Request已成功合入, 合并人@ascend-robot
（感谢 ymdsall 的贡献）9 天前 关联了issue：[RFC]: 移除 LSTM.forward Monkey Patch，将 PackedSequence 适配下沉至 op-plugin
atomgit-bot
9 天前 评论：
9 天前 评论：
变更摘要
本 PR 为重构优化，将主线已合入的修改同步至 v2.10.0 分支：删除 torch_npu/utils/_module.py 中复制自 PyTorch 官方实现的 _lstm_forward 函数及其在 _apply_module_patch() 中对 torch.nn.modules.rnn.LSTM.forward 的 monkey-patch 注册（共删除 139 行、无新增行）。改动后 torch.nn.LSTM.forward 恢复使用 PyTorch 官方实现，PackedSequence LSTM 的设备与输出形状适配交由 op-plugin 负责。
主要改动
- 删除
_lstm_forward函数定义: 移除torch_npu/utils/_module.py中约 138 行的_lstm_forward实现，包括其对 PackedSequence/普通输入的 hx 初始化、permute_hidden排序、torch._VF.lstm调用及 pack 输出的 shape 转换逻辑。 - 移除 monkey-patch 注册: 删除
_apply_module_patch()中的torch.nn.modules.rnn.LSTM.forward = _lstm_forward语句，使 LSTM 前向不再被 torch_npu 侧覆盖。 - 恢复官方实现并移交适配职责:
torch.nn.LSTM.forward恢复为 PyTorch 官方实现，PackedSequence LSTM 的设备与输出形状兼容处理改由固定基线的 op-plugin 后端承担，本 PR 不修改测试文件、PyTorch 官方代码及third_party/op-plugin子模块指针。
AtlasAccount
9 天前 评论：
9 天前 评论：
atomgit-bot
9 天前 评论：
9 天前 评论：
9 天前 添加了label：ascend-cla/yes
ascend-robot
9 天前 评论：
9 天前 评论：
Thanks for your pull-request.
The full list of commands accepted by me can be found at here.
You can get sig-info at here.
You can self-configure the PR merge rules for this repository. For more details, please refer to Here.
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| repo-Ascend/pytorch | ✅ liangsongwei, wanglijun55 (2/2) | ✅ liangsongwei (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
qq_53278473, thanks for your pull request. All authors of the commits have signed the CLA. 👍
ascend-robot
9 天前 评论：
9 天前 评论：
当前仓库存在以下 保护分支 ：
| Protected Branch | Version | Release |
|---|---|---|
| master | ||
| v2.7.1 | ||
| v2.9.0 | ||
| v2.10.0 | ||
| v2.11.0 | ||
| v2.12.0 | ||
| v2.10.0-26.1.0 | ||
| v2.12.0-26.1.0 | ||
| v2.7.1-26.1.0 | ||
| v2.11.0-26.1.0 | ||
| v2.9.0-26.1.0 | ||
| ci-test |
评论 /sync <branch1> <branch2> ... 可将当前 PR 修改同步到其它分支（创建同步 PR）：
a) 如果当前 PR 是 Open 状态，同步操作将延迟到 PR 被合并时执行
b) 如果当前 PR 已经 Merged，将立即执行同步操作
注意：
- /sync 命令可以指定同步到多个分支，仅最后一个 /sync 命令生效
- 如果创建的同步 PR 不正确，可通过向同步 PR 的源分支提交轻量级 PR 完善，或使用 /close 命令关闭
9 天前 添加了label：ci-pipeline-running
ascend-robot
9 天前 评论：
9 天前 评论：
ascend docs pipeline is running...
9 天前 添加了label：docs-ci-pipeline-running
ascend-robot
9 天前 评论：
9 天前 评论：
✅ 跳过 docs ci 检查，没有需要检查的文档文件
9 天前 删除了label：docs-ci-pipeline-running
9 天前 添加了label：docs-ci-pipeline-success
9 天前 删除了label：ci-pipeline-running
9 天前 添加了label：ci-pipeline-passed
AtlasAccount
9 天前 评论：
9 天前 评论：
流水线 PR-pipeline_pytorch#64475 [ commitID：73308fad ] 已完成
>>>代码风格自动修复执行成功（无修复内容）
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | Build_X86 | ✅ | >>> |
| Build_ARM | ✅ | >>> | |
| Build_X86_torchair | 🛑 | >>> | |
| Build_ARM_torchair | 🛑 | >>> | |
| patch_test | 🛑 | >>> | |
| Build_X86_213 | 🛑 | >>> | |
| Build_ARM_213 | 🛑 | >>> | |
| 恶意代码检查 | Antipoison | ✅ | >>> |
| 编码安全与规范检查 | codecheck_pre-commit | ✅ | >>> |
| check_error | ✅ | >>> | |
| lintrunner | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | UT_ARM_A3_Part_01 | 🛑 | >>> |
| UT_ARM_A3_Part_02 | 🛑 | >>> | |
| UT_ARM_A2_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Part_02 | ✅ | >>> | |
| UT_ARM_A2_Part_03 | ✅ | >>> | |
| UT_inductor_Part_01 | 🛑 | >>> | |
| UT_inductor_Part_02 | 🛑 | >>> | |
| UT_inductor_Part_03 | 🛑 | >>> | |
| UT_inductor_Part_04 | 🛑 | >>> | |
| UT_DIST_ARM_Part_01 | 🛑 | >>> | |
| UT_DIST_ARM_Part_02 | 🛑 | >>> | |
| UT_DIST_ARM_Part_03 | 🛑 | >>> | |
| UT_DIST_ARM_Part_04 | 🛑 | >>> | |
| UT_ARM_A2_Select_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Select_Part_02 | ✅ | >>> | |
| UT_ARM_A2_Part_213 | 🛑 | >>> | |
| UT_inductor_Part_213 | 🛑 | >>> | |
| UT_DIST_ARM_Part_213 | 🛑 | >>> | |
| UT_ARM_A2_Select_Part_213 | 🛑 | >>> | |
| 流水线 | PR-pipeline_pytorch | ✅ | >>> |
- compile、compile_inductor、compile_torchair : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线
wanglijun55
9 天前 评论：
9 天前 评论：
/lgtm
6 天前 关联了里程碑：TorchNPU-v2.14.0
梁松伟
6 天前 评论：
6 天前 评论：
/approve
6 天前 添加了label：approvedlgtm
6 天前 删除了label：ci-pipeline-passed
ascend-robot
6 天前 评论：
6 天前 评论：
The following label is not ready.
ci-pipeline-passed: The ci-pipeline-passed label is expired. Please compile again.
ymdsall
6 天前 评论：
6 天前 评论：
compile
6 天前 添加了label：ci-pipeline-running
ascend-robot
6 天前 评论：
6 天前 评论：
ascend docs pipeline is running...
6 天前 删除了label：docs-ci-pipeline-success
6 天前 添加了label：docs-ci-pipeline-running
ascend-robot
6 天前 评论：
6 天前 评论：
✅ 跳过 docs ci 检查，没有需要检查的文档文件
6 天前 删除了label：docs-ci-pipeline-running
6 天前 添加了label：docs-ci-pipeline-success
6 天前 删除了label：ci-pipeline-running
6 天前 添加了label：ci-pipeline-passed
AtlasAccount
6 天前 评论：
6 天前 评论：
流水线 PR-pipeline_pytorch#64914 [ commitID：73308fad ] 已完成
>>>代码风格自动修复执行成功（无修复内容）
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | Build_X86 | ✅ | >>> |
| Build_ARM | ✅ | >>> | |
| Build_X86_torchair | 🛑 | >>> | |
| Build_ARM_torchair | 🛑 | >>> | |
| patch_test | 🛑 | >>> | |
| Build_X86_213 | 🛑 | >>> | |
| Build_ARM_213 | 🛑 | >>> | |
| 恶意代码检查 | Antipoison | ✅ | >>> |
| 编码安全与规范检查 | codecheck_pre-commit | ✅ | >>> |
| check_error | ✅ | >>> | |
| lintrunner | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | UT_ARM_A3_Part_01 | 🛑 | >>> |
| UT_ARM_A3_Part_02 | 🛑 | >>> | |
| UT_ARM_A2_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Part_02 | ✅ | >>> | |
| UT_ARM_A2_Part_03 | ✅ | >>> | |
| UT_inductor_Part_01 | 🛑 | >>> | |
| UT_inductor_Part_02 | 🛑 | >>> | |
| UT_inductor_Part_03 | 🛑 | >>> | |
| UT_inductor_Part_04 | 🛑 | >>> | |
| UT_DIST_ARM_Part_01 | 🛑 | >>> | |
| UT_DIST_ARM_Part_02 | 🛑 | >>> | |
| UT_DIST_ARM_Part_03 | 🛑 | >>> | |
| UT_DIST_ARM_Part_04 | 🛑 | >>> | |
| UT_ARM_A2_Select_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Select_Part_02 | ✅ | >>> | |
| UT_ARM_A2_Part_213 | 🛑 | >>> | |
| UT_inductor_Part_213 | 🛑 | >>> | |
| UT_DIST_ARM_Part_213 | 🛑 | >>> | |
| UT_ARM_A2_Select_Part_213 | 🛑 | >>> | |
| 流水线 | PR-pipeline_pytorch | ✅ | >>> |
- compile、compile_inductor、compile_torchair : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线
6 天前 合入了pull request
AtlasAccount
6 天前 评论：
6 天前 评论：
流水线 pytorch_gitcode_PR_multiVersion#14743 [ commitID：73308fad ] 已完成
【合入来源】
RFC：https://gitcode.com/Ascend/pytorch/issues/3810
主线修改：https://gitcode.com/Ascend/pytorch/merge_requests/43893
本 PR 将主线已合入修改同步至 $version 分支。该分支固定的 op-plugin 基线已包含 PackedSequence LSTM 后端兼容处理。
【修改方案】
torch_npu/utils/_module.py中复制自 PyTorch 官方实现的_lstm_forward。_apply_module_patch()中对torch.nn.modules.rnn.LSTM.forward的 monkey patch 注册。torch.nn.LSTM.forward恢复使用 PyTorch 官方实现；PackedSequence LSTM 的设备与输出形状适配由 op-plugin 负责。third_party/op-plugin子模块指针。【资料变更】
不涉及。
【接口变更】
不涉及。
【功能验证】
v2.10.0基线相比仅修改torch_npu/utils/_module.py，共删除 139 行；git diff --check：通过；_lstm_forward定义、注册和冲突标记残留检查：无残留；本 PR 不新增或修改测试文件，运行时回归由 CI 验证。
【CheckList】