Closed
sort类算子arch35 MicroAPI旧接口别名整改 #2665
tujun6Add on  Aug 17Closed at  Aug 20
tujun6Collaborators
Aug 17 Create

一、问题描述

CANN 9.1.0 的 arch35(ascend950)MicroAPI 中,大量旧接口名与新接口名是同一 Impl 函数的别名。sort 类算子(sort、sort_with_index、top_k_v2、kth_value)的 op_kernel/arch35 目录下 .h 文件中仍使用旧接口名,需统一替换为新接口名,保证代码规范并维持二进制完全等价。

涉及的旧接口别名映射(共 28 类):

  • __local_mem__ → __ubuf__
  • DataCopy(RegTensor 首参)→ LoadAlign
  • DataCopy(addr 首参)→ StoreAlign
  • CompareScalar → Compares
  • MaskAnd/Or/Xor/Mov/Sel/Not/Pack → And/Or/Xor/Move/Select/Not/Pack
  • ReduceSum → Reduce<ReduceType::SUM>(仅 MicroAPI 层级 3 参数,排除传统高阶 4 参数 + LocalTensor API)
  • DataCopyGather → Gather

二、涉及算子及文件

算子 目录 修改文件数 替换处数
sort math/sort 5 158
sort_with_index math/sort_with_index 1 2
top_k_v2 math/top_k_v2 10 293
kth_value math/kth_value 2 62
合计 18 515

三、环境信息

  • CANN 版本:9.1.0
  • SoC:ascend950(arch35)

四、预期结果

  1. 所有可替换的旧 MicroAPI 接口零残留(不可替换项如 GatherMask、PrefixSum、传统高阶 API、参数结构体等保持不变)。
  2. 编译生成的 .o 文件与替换前 md5 完全一致,证明二进制等价。

五、关联 PR

https://gitcode.com/cann/ops-math/pull/4790

likedislike
陈思
陈思Collaborators
Aug 17 Comments:

/assign @tujun6

likedislike
CANN-robotCANN-robotCollaborators
Aug 17 Assign tujun6 as responsible person
CANN-robotCANN-robotCollaborators
Aug 20 Close issue
CANN-robotCANN-robotCollaborators
Aug 20 Added label: resolved