CANN 9.1.0 的 arch35(ascend950)MicroAPI 中,大量旧接口名与新接口名是同一 Impl 函数的别名。sort 类算子(sort、sort_with_index、top_k_v2、kth_value)的 op_kernel/arch35 目录下 .h 文件中仍使用旧接口名,需统一替换为新接口名,保证代码规范并维持二进制完全等价。
Impl
sort
sort_with_index
top_k_v2
kth_value
op_kernel/arch35
.h
涉及的旧接口别名映射(共 28 类):
__local_mem__
__ubuf__
DataCopy
LoadAlign
StoreAlign
CompareScalar
Compares
MaskAnd/Or/Xor/Mov/Sel/Not/Pack
And/Or/Xor/Move/Select/Not/Pack
ReduceSum
Reduce<ReduceType::SUM>
DataCopyGather
Gather
GatherMask
PrefixSum
.o
https://gitcode.com/cann/ops-math/pull/4790
/assign @tujun6
一、问题描述
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 首参)→LoadAlignDataCopy(addr 首参)→StoreAlignCompareScalar→ComparesMaskAnd/Or/Xor/Mov/Sel/Not/Pack→And/Or/Xor/Move/Select/Not/PackReduceSum→Reduce<ReduceType::SUM>(仅 MicroAPI 层级 3 参数,排除传统高阶 4 参数 + LocalTensor API)DataCopyGather→Gather二、涉及算子及文件
三、环境信息
四、预期结果
GatherMask、PrefixSum、传统高阶 API、参数结构体等保持不变)。.o文件与替换前 md5 完全一致,证明二进制等价。五、关联 PR
https://gitcode.com/cann/ops-math/pull/4790