已合并
feat: Add ACLGraph update plans #36821
luochao60创建于 5月26日
feat: Add ACLGraph update plans #36821
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 luochao60 的贡献)ascend-robot
5月26日 评论:
5月26日 评论:
ascend-robot
5月26日 评论:
5月26日 评论:
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
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| repo-Ascend/pytorch | ✅ adrian07110, chujinjin (2/2) | ✅ adrian07110 (1/1) |
| test | ✅ adrian07110, chujinjin (2/2) | ✅ adrian07110 (1/1) |
| torch_npu/_inductor | ✅ adrian07110, chujinjin (2/2) | ✅ adrian07110 (1/1) |
| torch_npu/npu | ✅ adrian07110, chujinjin (2/2) | ✅ adrian07110 (1/1) |
| torch_npu/utils | ✅ adrian07110, chujinjin (2/2) | ✅ adrian07110 (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
luochao60, thanks for your pull request. All authors of the commits have signed the CLA. 👍


5月26日 添加了label:ascend-cla/yes
ascend-robot
5月26日 评论:
5月26日 评论:
当前仓库存在以下 保护分支 :
| Protected Branch | Version | Release |
|---|---|---|
| master | ||
| v2.11.0 | ||
| v2.7.1 | ||
| v2.9.0 | ||
| v2.10.0 | ||
| v2.12.0 |
评论 /sync <branch1> <branch2> ... 可将当前 PR 修改同步到其它分支(创建同步 PR):
a) 如果当前 PR 是 Open 状态,同步操作将延迟到 PR 被合并时执行
b) 如果当前 PR 已经 Merged,将立即执行同步操作
注意:
- /sync 命令可以指定同步到多个分支,仅最后一个 /sync 命令生效
- 如果创建的同步 PR 不正确,可通过向同步 PR 的源分支提交轻量级 PR 完善,或使用 /close 命令关闭


此处折叠了569条消息 查看更多
6月16日 删除了label:ci-pipeline-passed
ascend-robot
6月16日 评论:
6月16日 评论:
The following label is not ready.
ci-pipeline-passed: The ci-pipeline-passed label is expired. Please compile again.


ascend-robot
6月16日 评论:
6月16日 评论:
The following label is not ready.
ci-pipeline-passed: The ci-pipeline-passed label is expired. Please compile again.


【合入来源】
【修改方案】
torch_npu._inductor.aclgraph_update_plan模块,在 Inductor codegen 阶段生成 ACLGraph update plan,并在运行期根据 callable 属性解析 new_inputs 到 FA 类算子 actual sequence 参数的映射。call或partition_x函数属性。_graph_treerecord/replay 路径消费 update plan,record 后校验 plan 与真实 capture record 的顺序、算子名和可更新 key,replay 前解析为 CPU update input。UPDATE_SPECS声明式描述,并由 base handler 统一更新 args/kwargs。【资料变更】
不涉及对外资料变更。
【接口变更】
不涉及客户面可见 API 变更;新增
torch_npu._inductor.aclgraph_update_plan为内部编译与 ACLGraph 运行期协同模块。【功能验证】
已补充并验证以下单测:
test/npu/test_aclgraph_update_plan.py覆盖 plan 构建、输入映射、常量解析、非法 plan 校验。test/npu/test_npugraph_handler.py覆盖 IFA/IFA v2 handler 的UPDATE_SPECS注册。test/_inductor/test_aclgraph_update_plan_compile.py覆盖 wrapper/graph partition emit、MLIR/DVM wrapper、npugraphify callable 属性保持等非设备单测。【CheckList】