已合并
【PR】: [feat] [autofuse] 增加图级schedInfo. #4530
【PR】: [feat] [autofuse] 增加图级schedInfo. #4530
已合并
邢智雄创建于 29 天前
邢智雄
邢智雄成员
29 天前

Pull Request

描述

ge适配autofuse后端新增ascgraph图级schedInfo。

变更类型

请选择本次引入的变更类型:

关联的Issue

如何测试

描述测试此变更的步骤和前提条件:

  1. 前端lowering新增图级schedInfo到ascgraph上,然后和后端进行联调,目标不影响功能/精度/性能
  2. 验证修改前后新老版本CANN包的兼容性

核对清单

其他信息

在此添加任何其他关于本次 PR 的说明。

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 邢智雄 的贡献)
邢智雄邢智雄成员
29 天前 添加了label:enhancement
邢智雄邢智雄成员
29 天前 创建了 pull request,commit 1714a6ea
atomgit-bot
atomgit-bot
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 进行序列化/反序列化。
likedislike
不准确?
atomgit-bot
atomgit-bot
29 天前 评论:

代码审查

✅ 未发现问题

likedislike
不准确?
CANN-robotCANN-robot成员
29 天前 添加了label:cann-cla/yes
此处折叠了55条消息 查看更多
邢智雄
邢智雄成员
21 天前 评论:

/check-pr

likedislike
CANN-robot
CANN-robot成员
21 天前 评论:

The following label is not ready.

approved: Please wait for committers to review the code.

likedislike
yangyongqiang
yangyongqiang成员
21 天前 评论:

/approve

likedislike
CANN-robotCANN-robot成员
21 天前 添加了label:approved
CANN-robotCANN-robot成员
21 天前 合入了pull request