已合并
【feat】: 补充split_pass UT #4399
WangYanMale创建于 11 天前
【feat】: 补充split_pass UT #4399
已合并
WangYanMale创建于 11 天前
WangYanMale成员
11 天前

Pull Request

描述

请清晰准确地描述本次 Pull Request 的意图和变更内容。

变更类型

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

关联的Issue

如何测试

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

核对清单

其他信息

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

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

变更摘要

本 PR 主要补充 FlattenSplitPass(split 扁平化 Pass)的单元测试。变更集中在测试文件 tests/autofuse/ut/autofuse/autofuse_pattern_fusion_unittest.cpp,新增 FlattenSplitPassUT 测试夹具及多个测试用例,覆盖 FlattenSplitPass::RunFlattenSplitPass::CanFlatten 在禁用开关、空图、无 Split 节点、单/多 SplitD、不同拆分维度、负维度、多消费者、未知维度数、与 PatternFusion 集成等场景;同时在 tests/depends/aihacb_autofusion/src/autofuse_new_stub.cc 中为 GetAutofuseBackendSpec() 启用了 split 扁平化配置。

主要改动

  • 新增 FlattenSplitPassUT 测试夹具: 在 autofuse_pattern_fusion_unittest.cpp 中新增 FlattenSplitPassUT,通过 SetUp/TearDown 控制 experimental_lowering_split 开关的开启与关闭,为后续用例提供统一环境。
  • 补充 FlattenSplitPass::Run 分支覆盖: 新增 RunSplitDisabledRunNullGraphRunNoSplitNodesRunSingleSplitDNoFusionRunSplitDDifferentDimNoFusionRunSplitDNonSplitPeerNoFusionRunSplitNegativeDimRunSplitDMultiConsumerNoFusionRunSplitDUnknownDimNumRunMultipleSplitNodesInGraph 等用例,覆盖不同图结构和异常/边界输入下 Run 的返回值。
  • 补充 FlattenSplitPass::CanFlatten 分支覆盖: 新增 CanFlattenSingleConsumerCanFlattenMultiConsumerFail,分别验证单消费者可扁平化返回 GRAPH_SUCCESS、多消费者不可扁平化返回 GRAPH_FAILED
  • 新增 PatternFusion 集成用例: 新增 RunPatternFusionIntegration,构造包含多层 SplitD 的图并调用 PatternFusion::RunAllPatternFusion,验证 split 扁平化与 pattern fusion 的集成流程。
  • 启用 split 扁平化 stub 配置: 在 autofuse_new_stub.ccGetAutofuseBackendSpec() 中新增 instance->slice_split_spec.enable_split_flatten = true;,为测试环境的 backend spec 启用 enable_split_flatten
likedislike
atomgit-bot
atomgit-bot
11 天前 评论:

代码审查

✅ 未发现问题

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

/lgtm

likedislike
yangyongqiang
yangyongqiang成员
11 天前 评论:

/approve

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