已合并
fix(topk): A2/A3 rank-0输入增加L0防御校验（#3968） #6808
baijinxiang_hw创建于 9 天前
fix(topk): A2/A3 rank-0输入增加L0防御校验（#3968） #6808
已合并
Pull Request已成功合入, 合并人@CANN-robot
（感谢 baijinxiang_hw 的贡献）9 天前 创建了 pull request，commit fb5284ff
Bbaijinxiang_hw
9 天前 关联了issue：[Bug-Report|缺陷反馈]: aclnnTopkV2 rank-0 self 未被第一段拦截，内核触发设备侧 vector core exception（phase1=0 / phase2=561103 / sync=507035）
9 天前 关联了issue：[Bug-Report|缺陷反馈]: aclnnTopkV2 rank-0 self 未被第一段拦截，内核触发设备侧 vector core exception（phase1=0 / phase2=561103 / sync=507035）
atomgit-bot
9 天前 评论：
9 天前 评论：
变更摘要
本次变更针对 TopkCommon 缺乏 rank 防护的问题：在 math/topk/op_api/topk.cpp 的 TopkCommon 入口新增 rank-0 防御性校验。当 self 的维度数为 0（超出文档声明的 1–8 维支持范围）时，记录参数错误日志并直接返回空结果，从而在后续 GetDim(dimNum - 1) 负索引越界读、SetDim(dim, k) 越界写以及 tiling 失败后内核仍被下发导致设备侧异常之前完成拦截。rank≥1 的各条路径行为保持不变。
主要改动
- 新增 rank 入口校验：在
TopkCommon起始处通过self->GetViewShape().GetDimNum() == 0判断，将 rank-0 标量输入前置拦截，避免负索引越界与矛盾元数据下发。 - 返回参数错误并输出日志：以
OP_LOGE(ACLNN_ERR_PARAM_INVALID, ...)记录可诊断的参数错误信息，明确TopK输入张量 rank 需在 [1, 8] 范围内。 - 失败时返回空结果：校验不通过时返回
std::tuple<aclTensor*, aclTensor*>(nullptr, nullptr)，使调用方在构造输出形状之前即终止，不再进入后续outShape.SetDim(dim, k)等处理流程。
🧩 本次 Review 已使用自定义 Skill：
不准确？
atomgit-bot
9 天前 评论：
9 天前 评论：
不准确？
9 天前 添加了label：cann-cla/yes
CANN-robot
9 天前 评论：
9 天前 评论：
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 |
|---|---|---|
| repo-cann/ops-math | ✅ 陈兴宇, yue-ma (2/2) | ✅ yue-ma, 陈兴宇 (2/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
doufloat, thanks for your pull request. All authors of the commits have signed the CLA. 👍
CANN-robot9 天前 将yue-ma,zhou-qilong,wangrui_,rxtfeng,loov1,zhangzijie,jia0liang,zl_hw,songkai111,chenxingyu18,pingchuantang,llimwang,gubaocheng,andong_hw,xuejinghui,onanfield,zhanw_coding,clinglai0517设为评审人
9 天前 将yue-ma,zhou-qilong,wangrui_,rxtfeng,loov1,zhangzijie,jia0liang,zl_hw,songkai111,chenxingyu18,pingchuantang,llimwang,gubaocheng,andong_hw,xuejinghui,onanfield,zhanw_coding,clinglai0517设为评审人
CANN-robot9 天前 将yue-ma,zhou-qilong,wangrui_,rxtfeng,loov1,zhangzijie,jia0liang,zl_hw,songkai111,chenxingyu18,pingchuantang,llimwang,gubaocheng,andong_hw,xuejinghui,onanfield,zhanw_coding,clinglai0517设为审查人
9 天前 将yue-ma,zhou-qilong,wangrui_,rxtfeng,loov1,zhangzijie,jia0liang,zl_hw,songkai111,chenxingyu18,pingchuantang,llimwang,gubaocheng,andong_hw,xuejinghui,onanfield,zhanw_coding,clinglai0517设为审查人
baijinxiang_hw
9 天前 评论：
9 天前 评论：
/compile
9 天前 预合并成功（commit_id: 6d40838e9fdc8fc9ce7e270e5ee63ca026cad100）
CANN-robot
9 天前 评论：
9 天前 评论：
| 🚀 CI 流水线已启动 |
|---|
| 📋 执行详情: 点击查看流水线 |
9 天前 添加了label：ci-pipeline-running
9 天前 删除了label：ci-pipeline-running
9 天前 添加了label：ci-pipeline-passed
chenyifeng
8 天前 评论：
8 天前 评论：
您好，PR已满足合入指标，请尽快联系Committer进行合入。如暂时不需要合入，请在PR标题前加[WIP]
8 天前 预合并成功（commit_id: 689c612ac4dcd9992d08c76bf41791a520f3a6a7）
8 天前 强制推送 1 个提交:3d1ead96-fix(topk): A2/A3 rank-0输入增加L0防御校验（#3968）
8 天前 预合并成功（commit_id: b5138e2bc379d35f6257d16e4f6c0343ffdef39a）
8 天前 删除了label：cann-cla/yes
8 天前 删除了label：ci-pipeline-passed
CANN-robot
8 天前 评论：
8 天前 评论：
8 天前 添加了label：cann-cla/yes
Bbaijinxiang_hw
8 天前 修改标题为 “fix(topk): A2/A3 rank-0输入增加L0防御校验（#3968）”，原标题为“fix(topk): l0op TopkCommon新增rank>=1防御校验，避免rank-0设备异常（#3968）”
8 天前 修改标题为 “fix(topk): A2/A3 rank-0输入增加L0防御校验（#3968）”，原标题为“fix(topk): l0op TopkCommon新增rank>=1防御校验，避免rank-0设备异常（#3968）”
8 天前 修改了pull request 的描述
baijinxiang_hw
8 天前 评论：
8 天前 评论：
/compile
8 天前 预合并成功（commit_id: a2eef8eed3b4d689124315485df87701963c3d6c）
CANN-robot
8 天前 评论：
8 天前 评论：
| 🚀 CI 流水线已启动 |
|---|
| 📋 执行详情: 点击查看流水线 |
8 天前 添加了label：ci-pipeline-running
8 天前 删除了label：ci-pipeline-running
8 天前 添加了label：ci-pipeline-passed
chenxingyu18
7 天前 评论：
7 天前 评论：
/lgtm
/approve
7 天前 添加了label：approved
7 天前 添加了label：lgtm
7 天前 关闭了关联的issue
7 天前 合入了pull request
描述
问题
TopkCommon在构造输出形状和访问最后轴之前没有 rank-0 防御。A2/A3 使用相同的DAV_2201非 RegBase 路径，self为 rank-0 时GetDim(dimNum - 1)会形成无效的最后轴访问，后续可能触发设备侧异常。兼容性取舍
DAV_2201路径，实测 rank-0、k=1无法正常完成，增加前置防御。RegBase/DAV_3510路径与 A2/A3 不同，实测 rank-0、k=0/1可正常返回；为保持 A5 现有兼容行为，本 PR 不拦截 A5。TopkCommon的 L0 防御，只对NpuArch::DAV_2201生效，rank>=1 路径不变。边界说明
aclnnTopkV2GetWorkspaceSize的公共 ACLNN wrapper 不在本仓；第一段统一返回161002的检查需要在对应 ACLNN 代码中完成。本 PR 只负责阻止 A2/A3 的 L0 负索引访问，不能替代 wrapper 的第一段校验。关联的Issue
https://gitcode.com/cann/ops-math/issues/3968
测试
fp32/fp16/bf16 rank-0 k=1实测失败；int32 rank-0 k=1也失败，错误表现不同。fp32/fp16/bf16/int32 rank-0 k=0/1实测均为gws=0, exec=0, sync=0；k=1回读value=1/index=0，k=2返回161002。DAV_2201条件，因此保持现有 rank-0 行为；合法 rank>=1 路径不改变。