已合并
feat: Add ACLGraph update plans #38049
luochao60创建于 6月10日
feat: Add ACLGraph update plans #38049
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 luochao60 的贡献)ascend-robot
6月10日 评论:
6月10日 评论:
Ascend docs pipeline is running...


6月10日 添加了label:docs-ci-pipeline-running
此处折叠了343条消息 查看更多
6月18日 添加了label:ci-pipeline-passed
ascend-robot
6月18日 评论:
6月18日 评论:
流水线 PR-pipeline_pytorch#33047 (重试第1次)已完成
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | Build_X86 | ✅ | >>> |
| Build_ARM | ✅ | >>> | |
| Build_LibTorch_x86 | ✅ | >>> | |
| Build_LibTorch_ARM | ✅ | >>> | |
| Build_X86_torchair | 🛑 | >>> | |
| Build_ARM_torchair | 🛑 | >>> | |
| patch_test | 🛑 | >>> | |
| 恶意代码检查 | Antipoison | ✅ | >>> |
| 编码安全与规范检查 | CodeCheck | ✅ | >>> |
| check_error | ✅ | >>> | |
| CodeCheck_lintrunner | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | UT_X86_Part_01 | 🛑 | >>> |
| UT_X86_Part_02 | 🛑 | >>> | |
| 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 | ✅ | >>> | |
| 流水线 | PR-pipeline_pytorch | ✅ | >>> |


6月18日 合入了pull request
ascend-robot
6月18日 评论:
6月18日 评论:
Pull Request 已合并或已关闭。
If you want to solve this problem, you can click here to do it in the FAQs.


ascend-robot
6月18日 评论:
6月18日 评论:
流水线 pytorch_gitcode_PR_multiVersion#10921 [ commitID:3358cad6 ] 运行失败


【合入来源】
【修改方案】
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】