已合并
提升topk int64索引的部分shape的性能 #2505
JimmyLam2创建于 4月27日
提升topk int64索引的部分shape的性能 #2505
已合并
JimmyLam2创建于 4月27日
JimmyLam2成员
4月27日

描述

在k大于2000,且尾轴小于10000000时走int32的索引,最后cast回int64

关联的Issue

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

测试

二级冒烟与aclnn用例已更过

文档更新

类型标签

  • [x ] 性能优化
likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 JimmyLam2 的贡献)
JJimmyLam2成员
4月27日 创建了 pull request,commit f59b91d7
CANN-robot
CANN-robot成员
4月27日 评论:

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 宋凯, zhangzijie (2/2) 宋凯, zhangzijie (2/1)

💡 Tip:

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

CLA Signature Pass

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

likedislike
CANN-robotCANN-robot成员
4月27日 添加了label:cann-cla/yes
CANN-robotCANN-robot成员
4月27日 将zhou-qilong,wangrui_,rxtfeng,loov1,zhangzijie,jia0liang,songkai111,llimwang,gubaocheng设为评审人
CANN-robotCANN-robot成员
4月27日 将zhou-qilong,wangrui_,rxtfeng,loov1,zhangzijie,jia0liang,songkai111,llimwang,gubaocheng设为审查人
JJimmyLam2成员
4月27日 update merge request[project id: 7649531, iid: 2505, commit_id: 8b810fafd453b01a8615ae16a83dc5673b67e9c3] virtual merging success
JJimmyLam2成员
4月27日 强制推送  1 个提交:454bb768-修复int64索引的性能问题
JJimmyLam2成员
4月27日 update merge request[project id: 7649531, iid: 2505, commit_id: cbca8aeed9c13ffa8c34cc3d5e45f45b4fd53d92] virtual merging success
JimmyLam2成员
4月27日 评论:

/compile

likedislike
JimmyLam2成员
4月27日 评论:

compile

likedislike
CANN-robotCANN-robot成员
4月27日 添加了label:ci-pipeline-running
CANN-robot
CANN-robot成员
4月27日 评论:

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

任务名称状态日志下载链接
Compile_Ascend_X86 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_mobile_station ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_experimental ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_experimental ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_single ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_A5 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_A5 ✅ SUCCESS >>>>> >>>>>
pre_comment ✅ SUCCESS >>>>>
Compile_Ascend_X86_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_X86_mobile_station_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_X86_experimental_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_X86_A5_ubuntu24 ✅ SUCCESS >>>>>
harmony-infer ✅ SUCCESS
Compile_Ascend_ARM_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_ARM_experimental_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_single_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_ARM_A5_ubuntu24 ✅ SUCCESS >>>>>
Compile_X86_monitor_910b ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910c ✅ SUCCESS >>>>> >>>>>
UT_Test ✅ SUCCESS
UT_Test_experimental ✅ SUCCESS
UT_Test_kernel ✅ SUCCESS
PreSmoke_A900 ✅ SUCCESS >>>>>
API_Check ✅ SUCCESS >>>>>
ATK_Test_A2 ✅ SUCCESS

[2026-04-27 19:55:57]    CI执行结束

likedislike
CANN-robotCANN-robot成员
4月27日 添加了label:api-check-pass
CANN-robotCANN-robot成员
4月27日 删除了label:api-check-pass
CANN-robot
CANN-robot成员
4月27日 评论:

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

任务名称状态日志下载链接
codecheck ✅ SUCCESS >>>>>
SCA ✅ SUCCESS >>>>>
anti_virus ✅ SUCCESS >>>>>
Check_Pr ✅ SUCCESS >>>>>
StaticCheck_codespell_check ✅ SUCCESS
StaticCheck_link_validity_check ✅ SUCCESS
StaticCheck_resource_existence_check ✅ SUCCESS
StaticCheck_tag_closed_check ✅ SUCCESS
StaticCheck_markdownlint ✅ SUCCESS
cocheck_codestyle ✅ SUCCESS
precommit ✅ SUCCESS >>>>> >>>>>

[2026-04-27 20:10:28]    CI执行结束

likedislike
CANN-robotCANN-robot成员
4月27日 删除了label:ci-pipeline-running
此处折叠了5条事件消息 查看更多
JJimmyLam2成员
4月28日 修改了pull request 的描述
songkai111成员
4月28日 评论:

/lgtm
/approve

likedislike
CANN-robotCANN-robot成员
4月28日 添加了label:approved
zhangzijie成员
4月28日 评论:

/lgtm
/approve

likedislike
CANN-robotCANN-robot成员
4月28日 添加了label:lgtm
CANN-robot
CANN-robot成员
4月28日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: 宋凯, zhangzijie.
Reviewers who wrote a comment of /lgtm are: 宋凯, zhangzijie.

likedislike
CANN-robotCANN-robot成员
4月28日 关闭了关联的issue
CANN-robotCANN-robot成员
4月28日 合入了pull request