已合并
TopKV2算子代码优化 #6036
cy_hw创建于 9月29日
TopKV2算子代码优化 #6036
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 cy_hw 的贡献)atomgit-bot
9月29日 评论:
9月29日 评论:
变更摘要
本次变更针对 math/top_k_v2/op_host/arch35/top_k_v2_tiling_base.h 中 TopKV2 算子 tiling 相关辅助函数的声明进行代码整理,主要工作是统一并规范函数形参的命名风格,使参数名更简洁且语义更明确。改动仅涉及函数声明的形参名,未改变函数签名中的类型、返回值和调用契约,属于不影响外部行为的可读性优化。
主要改动
- 统一数据类型参数命名: 将
GetTopkPreferredInnerChunk、UseTopkNonLastMergeSort、GetTopkNonLastSortDtype、GetTopkNonLastSortDtypeSize、GetTopkNonLastSortTmpSize等函数的首个数据类型形参由dataType统一更名为dtype。 - 简化索引与长度类参数名:
GetTopkPreferredInnerChunk的index改为candidateIdx,UseTopkNonLastMergeSort的axisLen改为axisLength,ComputeTopkAxisDimProducts的axis改为sortAxisIdx,使参数含义更具体。 - 统一布尔开关类参数命名: 将
useMergeSort统一改为mergeSortOn、isDescend改为descendOn,涉及GetTopkNonLastSortDtype、GetTopkNonLastSortDtypeSize、GetTopkNonLastSortTmpSize及ComputeTopkNonLastLayout等声明。 - 规范缓冲区与计数参数名:
GetTopkNonLastSortDtypeSize的dtypeSize改为elemSize,GetTopkNonLastSortTmpSize的sortCount、tmpUbSize分别改为sortElemCount、tempUbSize。 - 调整搜索规划函数形参名并换行:
SearchTopkNonLastSmallAxisPlan的usableUb、estimateUb、best、selectedInfo分别改为availableUb、estimateUbFn、bestCand、chosenInfo,并对estimateUbFn的类型声明进行了换行排版调整。


不准确?
atomgit-bot
9月29日 评论:
9月29日 评论:
9月29日 添加了label:cann-cla/yes
CANN-robot
9月29日 评论:
9月29日 评论:
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.
You can self-configure the PR merge rules for this repository. For more details, please refer to Here.
For more, you also can visit HICANN.
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| 9.2.0 | ✅ gcw_kUomxQ2l, 宋恺, 王瑞 (3/2) | ✅ gcw_kUomxQ2l (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
caoyan_huawei, thanks for your pull request. All authors of the commits have signed the CLA. 👍


9月29日 将yue-ma,zhou-qilong,wangrui_,rxtfeng,loov1,zhangzijie,TianXL,jia0liang,zl_hw,songkai111,jiangtao_rts,gitcode-chenjiao,chenxingyu18,hanxiaolong,pingchuantang,llimwang,gubaocheng,andong_hw,xuejinghui,zhanw_coding,gcw_kUomxQ2l,Andy-lb,clinglai0517设为评审人
9月29日 将yue-ma,zhou-qilong,wangrui_,rxtfeng,loov1,zhangzijie,TianXL,jia0liang,zl_hw,songkai111,jiangtao_rts,gitcode-chenjiao,chenxingyu18,hanxiaolong,pingchuantang,llimwang,gubaocheng,andong_hw,xuejinghui,zhanw_coding,gcw_kUomxQ2l,Andy-lb,clinglai0517设为评审人
9月29日 将TianXL,jiangtao_rts,gitcode-chenjiao,hanxiaolong,gcw_kUomxQ2l,Andy-lb设为审查人
cy_hw
9月29日 评论:
9月29日 评论:
compile


9月29日 预合并成功(commit_id: e6f6a4ccf859974405b9cdab5bd71e9233c0bcb1)
CANN-robot
9月29日 评论:
9月29日 评论:
| 🚀 CI 流水线已启动 |
|---|
| 📋 执行详情: 点击查看流水线 |


9月29日 添加了label:ci-pipeline-running
此处折叠了9条事件消息 查看更多
9月29日 修改了pull request 的描述
songkai111
9月29日 评论:
9月29日 评论:
/approve


songkai111
9月29日 评论:
9月29日 评论:
/lgtm


9月29日 添加了label:lgtm
liulun0308
9月29日 评论:
9月29日 评论:
/approve


CANN-robot
9月29日 评论:
9月29日 评论:
9月29日 添加了label:approved
9月29日 关闭了关联的issue
9月29日 合入了pull request
描述
本次变更针对 math/top_k_v2/op_host/arch35/top_k_v2_tiling_base.h 中 TopKV2 算子 tiling 相关辅助函数的声明进行代码整理,主要工作是统一并规范函数形参的命名风格,使参数名更简洁且语义更明确。改动仅涉及函数声明的形参名,未改变函数签名中的类型、返回值和调用契约,属于不影响外部行为的可读性优化。
关联的Issue
https://gitcode.com/cann/ops-math/issues/3352
测试
二级冒烟,st用例测试通过
文档更新
类型标签