已合并
feat: add matrix_set_diag_v2 UT #4608
zhanw_coding创建于 23 天前
feat: add matrix_set_diag_v2 UT #4608
已合并
zhanw_coding创建于 23 天前
zhanw_coding
zhanw_coding成员
23 天前

描述

  • feat: add matrix_set_diag_v2 UT
  • fix: matrix_set_diag codecheck
  • feat: add matrix_set_diag_v2 classify rule

新增 14 个用例(全部通过)

  • infershape(7 个):
    非 const k 三分支(diag 维过小/过大/合法)、diag 未知前导维 continue、x 未知前导维用 diag 填充、x/diag 未知 rank({-2})→ 输出 {-2}。

  • tiling(7 个):
    V1 切尾路径 3 个用例(主循环/提前返回/CalUbFactor if 分支)、SIMT ubSize 过小失败、GATHER ubFactor=0 回退 CutTail、NoCutTail 优化循环 break、SIMT 直达。

关联的Issue

https://gitcode.com/cann/ops-math/issues/2606

测试

UT

文档更新

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 zhanw_coding 的贡献)
zhanw_codingzhanw_coding成员
23 天前 创建了 pull request,commit 001745c4
atomgit-bot
atomgit-bot
23 天前 评论:

变更摘要

该 PR 主要围绕 matrix_set_diag_v2 算子补充单元测试,并修复 matrix_set_diag 的 codecheck 问题:在 classify_rule.yaml 中新增 matrix_set_diag_v2 的分类规则,在 matrix_set_diag_tiling_arch35.cpp 中调整对角线长度校验逻辑,同时为 matrix_set_diag_v2 的 tiling 与 infershape 实现补充覆盖多种分支的 UT 用例。

主要改动

  • 新增 matrix_set_diag_v2 分类规则:在 classify_rule.yaml 中为 matrix_set_diag_v2 增加 op_hostop_kernelexamplestests 以及 arch35matrix_set_diag_v2_def.cpp 等路径的归类配置。

  • 修复 matrix_set_diag tiling 对角长度校验:在 matrix_set_diag_tiling_arch35.cpp 中将 maxDiagLen 先以 uint64_t 读取再转为 uint32_t 存入 inputInfo_.maxDiagLen,并新增 maxDiagLen != inputInfo_.maxDiagLen 的截断溢出校验,避免大值截断后通过检查。

  • 补充 matrix_set_diag_v2 tiling UT:在 test_matrix_set_diag_tiling_arch35.cpp 中新增 7 个用例,覆盖 Tiling4CutTail 的 V1 路径、GetOptimizeTiling 提前返回、CalUbFactor 分支、SIMT 缓存不足失败、GATHER 回退 CutTailSCATTER NoCutTail 优化以及 SIMT 直通 NoCutTail 等场景。

  • 补充 matrix_set_diag_v2 infershape UT:在 test_matrix_set_diag_infershape.cpp 中新增 8 个用例,覆盖非 const k 的维度校验成功与失败、diag/x 前导维未知时的填充与跳过、以及 x/diag 未知 rank 时输出 {-2} 等分支。

likedislike
atomgit-bot
atomgit-bot
23 天前 评论:

代码审查

✅ 未发现问题

likedislike
CANN-robotCANN-robot成员
23 天前 添加了label:stat/needs-squash
CANN-robotCANN-robot成员
23 天前 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
23 天前 评论:

Thanks for your pull-request.
The full list of commands accepted by me can be found at here
You can get sig-info at here


PR Approval Progress

Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
repo-cann/ops-math wangrui_, andong_hw (2/2) wangrui_ (1/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

zhanw_coding, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
CANN-robotCANN-robot成员
23 天前 将yue-ma,zhou-qilong,wangrui_,rxtfeng,loov1,zhangzijie,jia0liang,zl_hw,songkai111,chenxingyu18,pingchuantang,llimwang,gubaocheng,andong_hw,xuejinghui设为评审人
CANN-robotCANN-robot成员
23 天前 将yue-ma,zhou-qilong,wangrui_,rxtfeng,loov1,zhangzijie,jia0liang,zl_hw,songkai111,chenxingyu18,pingchuantang,llimwang,gubaocheng,andong_hw,xuejinghui设为审查人
zhanw_codingzhanw_coding成员
23 天前 修改了pull request 的描述
zhanw_codingzhanw_coding成员
23 天前 修改了pull request 的描述
zhanw_coding
zhanw_coding成员
23 天前 评论:

compile

likedislike
zhanw_codingzhanw_coding成员
23 天前 update merge request[project id: 7649531, iid: 4608, commit_id: ccf123fef09e9f72d9df579cc34e70601b1a1e20] virtual merging success
CANN-robotCANN-robot成员
23 天前 添加了label:ci-pipeline-running
CANN-robot
CANN-robot成员
23 天前 评论:

流水线任务触发成功
任务链接 [f1104575bc7745d481c728232b251f94][流水线指导]

任务名称状态日志下载链接
pre_comment ✅ SUCCESS >>>>>
Compile_Ascend_X86_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_mobile_station_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_experimental_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_harmony-infer ✅ SUCCESS >>>>>
Compile_Ascend_ARM_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_experimental_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_single_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910b ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910c ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_950 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_A5_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_A5_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_mobile_station_9030_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_single ✅ SUCCESS >>>>> >>>>>
UT_Test ✅ SUCCESS
UT_Test_experimental ✅ SUCCESS
UT_Test_kernel ✅ SUCCESS
PreSmoke_A900 ✅ SUCCESS >>>>>
API_Check ✅ SUCCESS >>>>>
PreSmoke_ATK_Test_A2 ✅ SUCCESS >>>>>

[2026-08-13 11:45:58]    CI执行结束

likedislike
zhanw_codingzhanw_coding成员
23 天前 update merge request[project id: 7649531, iid: 4608, commit_id: b4ea7a115beae250fcfb60fa3dd8d0bd06c545c8] virtual merging success
zhanw_codingzhanw_coding成员
23 天前 强制推送  1 个提交:b4f73645-feat: add matrix_set_diag_v2 UT
zhanw_codingzhanw_coding成员
23 天前 update merge request[project id: 7649531, iid: 4608, commit_id: f5eb8e9787c6e67ef42e1c062536ad35b348f2c9] virtual merging success
CANN-robotCANN-robot成员
23 天前 删除了label:cann-cla/yes
CANN-robotCANN-robot成员
23 天前 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
23 天前 评论:

流水线任务触发成功
任务链接 [68b9467d97cb4beabf99ecaebf2df4c6][流水线指导]

任务名称状态日志下载链接
SCA ✅ SUCCESS >>>>>
antipoison ✅ SUCCESS >>>>>
codecheck_Pr ✅ SUCCESS
StaticCheck_codespell ✅ SUCCESS
StaticCheck_link_validity ✅ SUCCESS
StaticCheck_resource_existence ✅ SUCCESS
StaticCheck_tag_closed ✅ SUCCESS
StaticCheck_markdownlint ✅ SUCCESS
codecheck_precommit ✅ SUCCESS >>>>>

[2026-08-13 11:35:57]    CI执行结束

likedislike
CANN-robotCANN-robot成员
23 天前 添加了label:api-check-pass
此处折叠了6条事件消息 查看更多
CANN-robotCANN-robot成员
22 天前 删除了label:ci-pipeline-passed
CANN-robot
CANN-robot成员
22 天前 评论:

Notification

This pull request has been changed(code update) or closed, so removes the following label(s): ci-pipeline-passed.

likedislike
CANN-robotCANN-robot成员
22 天前 添加了label:cann-cla/yes
zhanw_coding
zhanw_coding成员
22 天前 评论:

compile

likedislike
zhanw_codingzhanw_coding成员
22 天前 update merge request[project id: 7649531, iid: 4608, commit_id: 111b5e30e8166bf84015e646d57796e924f5bb6a] virtual merging success
CANN-robotCANN-robot成员
22 天前 删除了label:api-check-pass
CANN-robotCANN-robot成员
22 天前 添加了label:ci-pipeline-running
CANN-robot
CANN-robot成员
22 天前 评论:

流水线任务触发成功
任务链接 [9cdb6209edbb4a8a85423fd343148b81][流水线指导]

任务名称状态日志下载链接
pre_comment ✅ SUCCESS >>>>>
Compile_Ascend_X86_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_mobile_station_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_experimental_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_harmony-infer ✅ SUCCESS >>>>>
Compile_Ascend_ARM_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_experimental_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_single_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910b ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910c ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_950 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_A5_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_A5_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_mobile_station_9030_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_single ✅ SUCCESS >>>>> >>>>>
UT_Test ✅ SUCCESS
UT_Test_experimental ✅ SUCCESS
UT_Test_kernel ✅ SUCCESS
PreSmoke_A900 ✅ SUCCESS >>>>>
API_Check ✅ SUCCESS >>>>>
PreSmoke_ATK_Test_A2 ✅ SUCCESS >>>>>

[2026-08-13 22:26:31]    CI执行结束

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

流水线任务触发成功
任务链接 [2cbbef9da6d84469a70ce4bd22415027][流水线指导]

任务名称状态日志下载链接
SCA ✅ SUCCESS >>>>>
antipoison ✅ SUCCESS >>>>>
codecheck_Pr ✅ SUCCESS
StaticCheck_codespell ✅ SUCCESS
StaticCheck_link_validity ✅ SUCCESS
StaticCheck_resource_existence ✅ SUCCESS
StaticCheck_tag_closed ✅ SUCCESS
StaticCheck_markdownlint ✅ SUCCESS
codecheck_precommit ✅ SUCCESS >>>>>

[2026-08-13 22:15:35]    CI执行结束

likedislike
CANN-robotCANN-robot成员
22 天前 添加了label:api-check-pass
CANN-robotCANN-robot成员
22 天前 删除了label:ci-pipeline-running
CANN-robotCANN-robot成员
22 天前 添加了label:ci-pipeline-passed
RuiWang_成员
22 天前 评论:

/lgtm
/approve

likedislike
CANN-robotCANN-robot成员
22 天前 添加了label:approved
andong_hw成员
22 天前 评论:

/lgtm

likedislike
CANN-robotCANN-robot成员
22 天前 添加了label:lgtm
CANN-robotCANN-robot成员
22 天前 关闭了关联的issue
CANN-robotCANN-robot成员
22 天前 合入了pull request