已合并
【PR】: [feat] [autofuse] add ut. #4401
【PR】: [feat] [autofuse] add ut. #4401
已合并
邢智雄创建于 7 天前
邢智雄
邢智雄成员
7 天前

Pull Request

描述

补充autofuse相关ut。

变更类型

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

关联的Issue

如何测试

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

  1. 蓝区流水线通过

核对清单

其他信息

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

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 邢智雄 的贡献)
邢智雄邢智雄成员
7 天前 添加了label:enhancement
邢智雄邢智雄成员
7 天前 创建了 pull request,commit 5e59dd96
atomgit-bot
atomgit-bot
7 天前 评论:

变更摘要

本次 PR 主要为 autofuse 模块补充单元测试,全部改动集中在 tests/autofuse/ut/autofuse/fusion_strategy_solver_unittest.cpp。该变更围绕 LowerSplitHelperConcatFusionStrategy 新增了 3 个测试用例,并扩展了测试辅助函数 CreatSplitDoubleOutPutsAscGraph 以及相关头文件引入,用于覆盖 split 融合的 lifting 判断、无 split 节点的异常路径以及 concat 反向链接 split 的融合策略。

主要改动

  • 新增 LowerSplitHelperNeedLiftingCoversSplitCases 测试: 通过内部 create_split_backend 构造多种 split 场景(首维拆分、非首维拆分、同 shape、constant tail、非 split origin、不同输出数量),验证 LowerSplitHelper::NeedLifting 的返回值为 GRAPH_SUCCESSneed_lifting 是否符合预期。
  • 新增 LowerSplitHelperRejectsGraphWithoutSplit 测试: 使用 CreatAddAscGraph 构造不含 split 节点的图,断言 LowerSplitHelper::NeedLifting 返回非 GRAPH_SUCCESS,覆盖异常/失败路径。
  • 新增 ConcatStrategyCoversBackwardSplitLinkHelpers 测试: 同时构建 split 与 concat 的 AscGraph 及后端节点、边和 fuse 属性,验证 ConcatFusionStrategy::CanFuse(split_backend, fused_backend) 返回 true
  • 扩展 CreatSplitDoubleOutPutsAscGraph 辅助函数: 新增 bool non_store_peer = false 参数,开启时在 split.y[0] 之后插入一个双输入均为 split.y[0]Add 节点,用于构造非 Store 直连 peer 的场景。
  • 补充头文件引入: 新增 lowering/op_helper/lower_split_helper.hcan_fuse/strategy/concat_fusion_strategy.hgraph/ascendc_ir/utils/asc_graph_utils.hgraph/utils/graph_utils.h,以支持上述测试逻辑。
likedislike
atomgit-bot
atomgit-bot
7 天前 评论:

代码审查

✅ 未发现问题

likedislike
CANN-robotCANN-robot成员
7 天前 添加了label:cann-cla/yes
此处折叠了47条消息 查看更多
CANN-robotCANN-robot成员
6 天前 添加了label:ci-pipeline-passed
ClarkXie成员
6 天前 评论:

/lgtm

likedislike
yangyongqiang
yangyongqiang成员
6 天前 评论:

/approve

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