已合并
TopKV2算子代码优化 #6036
TopKV2算子代码优化 #6036
已合并
cy_hw创建于 9月29日
cy_hw
cy_hw成员
9月29日

描述

本次变更针对 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用例测试通过

文档更新

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 cy_hw 的贡献)
cy_hwcy_hw成员
9月29日 创建了 pull request,commit b00da9c0
atomgit-bot
atomgit-bot
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 的类型声明进行了换行排版调整。
likedislike
不准确?
atomgit-bot
atomgit-bot
9月29日 评论:

代码审查

✅ 未发现问题

likedislike
不准确?
CANN-robotCANN-robot成员
9月29日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
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 /approve or /lgtm
  • Commenting /approve implies 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. 👍

likedislike
CANN-robotCANN-robot成员
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设为评审人
CANN-robotCANN-robot成员
9月29日 将TianXL,jiangtao_rts,gitcode-chenjiao,hanxiaolong,gcw_kUomxQ2l,Andy-lb设为审查人
cy_hw
cy_hw成员
9月29日 评论:

compile

likedislike
cy_hwcy_hw成员
9月29日 预合并成功(commit_id: e6f6a4ccf859974405b9cdab5bd71e9233c0bcb1)
CANN-robot
CANN-robot成员
9月29日 评论:
🚀 CI 流水线已启动
📋 执行详情: 点击查看流水线
likedislike
CANN-robotCANN-robot成员
9月29日 添加了label:ci-pipeline-running
此处折叠了9条事件消息 查看更多
cy_hwcy_hw成员
9月29日 修改了pull request 的描述
songkai111成员
9月29日 评论:

/approve

likedislike
songkai111成员
9月29日 评论:

/lgtm

likedislike
RuiWang_成员
9月29日 评论:

/lgtm
/approve

likedislike
CANN-robotCANN-robot成员
9月29日 添加了label:lgtm
liulun0308成员
9月29日 评论:

/approve

likedislike
CANN-robot
CANN-robot成员
9月29日 评论:

The following users do not have permission to comment /lgtm or /approve on any module in this PR:
/approve: 宋恺, 王瑞

likedislike
CANN-robotCANN-robot成员
9月29日 添加了label:approved
CANN-robotCANN-robot成员
9月29日 关闭了关联的issue
CANN-robotCANN-robot成员
9月29日 合入了pull request