已合并
【PR】: [feat] [autofuse] add ut. #4401
邢智雄创建于 7 天前
【PR】: [feat] [autofuse] add ut. #4401
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 邢智雄 的贡献)7 天前 添加了label:enhancement
atomgit-bot
7 天前 评论:
7 天前 评论:
变更摘要
本次 PR 主要为 autofuse 模块补充单元测试,全部改动集中在 tests/autofuse/ut/autofuse/fusion_strategy_solver_unittest.cpp。该变更围绕 LowerSplitHelper 与 ConcatFusionStrategy 新增了 3 个测试用例,并扩展了测试辅助函数 CreatSplitDoubleOutPutsAscGraph 以及相关头文件引入,用于覆盖 split 融合的 lifting 判断、无 split 节点的异常路径以及 concat 反向链接 split 的融合策略。
主要改动
- 新增
LowerSplitHelperNeedLiftingCoversSplitCases测试: 通过内部create_split_backend构造多种 split 场景(首维拆分、非首维拆分、同 shape、constant tail、非 split origin、不同输出数量),验证LowerSplitHelper::NeedLifting的返回值为GRAPH_SUCCESS及need_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.h、can_fuse/strategy/concat_fusion_strategy.h、graph/ascendc_ir/utils/asc_graph_utils.h与graph/utils/graph_utils.h,以支持上述测试逻辑。


atomgit-bot
7 天前 评论:
7 天前 评论:
7 天前 添加了label:cann-cla/yes
此处折叠了47条消息 查看更多
6 天前 添加了label:ci-pipeline-passed
yangyongqiang
6 天前 评论:
6 天前 评论:
/approve


6 天前 添加了label:lgtmapproved
6 天前 合入了pull request
Pull Request
描述
补充autofuse相关ut。
变更类型
请选择本次引入的变更类型:
关联的Issue
如何测试
描述测试此变更的步骤和前提条件:
核对清单
其他信息
在此添加任何其他关于本次 PR 的说明。