已合并
【PR】: [feat] [autofuse] 增加图级schedInfo. #4530
邢智雄创建于 29 天前
【PR】: [feat] [autofuse] 增加图级schedInfo. #4530
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 邢智雄 的贡献)29 天前 添加了label:enhancement
atomgit-bot
29 天前 评论:
29 天前 评论:
变更摘要
本 PR 为 autofuse 模块增加图级调度信息(graph-level SchedInfo)。核心改动是在图属性上新增独立的图级调度结构 GraphSchedInfo,用于接收外部统一调度轴 axis,而 loop_axis 保留给内部流程使用;该结构在 ascendc_ir_def.h 中定义并与节点级 SchedInfo 分开,避免将节点级 exec_condition 提升到图级;同时在 af_ir.proto 中新增对应的 GraphSchedInfoDef 消息定义,并在图属性组 AscGraphAttrGroupsDef 中挂载该字段,以支持图级调度信息的序列化传输。
主要改动
- 新增图级调度结构
GraphSchedInfo: 在ascendc_ir_def.h中定义,包含axis(当前仅用于接收外部统一调度轴,进入优化流程前可同步到各节点的sched.axis)和loop_axis(默认kIdNone,保留给内部流程使用)两个字段。 - 扩展图属性类
AscGraphAttr: 新增sched字段承载图级调度信息,与节点级调度信息分离存储。 - 新增 protobuf 消息
GraphSchedInfoDef: 在af_ir.proto中定义,包含repeated int64 axis(字段号 1)和int64 loop_axis(字段号 2)。 - 扩展图属性组消息
AscGraphAttrGroupsDef: 新增字段号 5 的sched字段,使图级调度信息可通过 proto 进行序列化/反序列化。


不准确?
atomgit-bot
29 天前 评论:
29 天前 评论:
29 天前 添加了label:cann-cla/yes
此处折叠了55条消息 查看更多
邢智雄
21 天前 评论:
21 天前 评论:
/check-pr


CANN-robot
21 天前 评论:
21 天前 评论:
The following label is not ready.
approved: Please wait for committers to review the code.


yangyongqiang
21 天前 评论:
21 天前 评论:
/approve


21 天前 添加了label:approved
21 天前 合入了pull request
Pull Request
描述
ge适配autofuse后端新增ascgraph图级schedInfo。
变更类型
请选择本次引入的变更类型:
关联的Issue
如何测试
描述测试此变更的步骤和前提条件:
核对清单
其他信息
在此添加任何其他关于本次 PR 的说明。