已合并
refactor: arch35 算子 MicroAPI/Reg 旧接口替换为新接口 #8450
yuanbin_22创建于 15 天前
refactor: arch35 算子 MicroAPI/Reg 旧接口替换为新接口 #8450
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 yuanbin_22 的贡献)15 天前 创建了 pull request,commit 06ccccc2
atomgit-bot
15 天前 评论:
15 天前 评论:
变更摘要
本次变更对多个激活算子(activation)和量化算子(quant)中使用的 MicroAPI 旧版接口进行了系统性替换,将其升级为对应的新版接口。改动涉及比较操作、归约操作、地址空间标识符、寄存器类型以及融合运算等多个 API 层面,属于纯重构性质的接口迁移,不改变算子的实际计算逻辑。
主要改动
CompareScalar→Compares:在elu_dag.h、mish_dag.h、mish_grad_dag.h中将MicroAPI::CompareScalar替换为MicroAPI::Compares,接口名由单数形式改为复数形式,模板参数和调用方式保持一致。ReduceMax→Reduce<ReduceType::MAX>:在gelu_dynamic_quant_regbase.h和gelu_dynamic_quant_workspace_regbase.h中将专用的MicroAPI::ReduceMax替换为模板化的MicroAPI::Reduce<ReduceType::MAX>,统一了归约操作的接口风格。__local_mem__→__ubuf__:在sigmoid、sigmoid_grad、swish及ascend_quant相关算子内核文件中,将局部内存地址空间标识符__local_mem__统一替换为__ubuf__,以匹配新版 MicroAPI 的统一缓冲区(Unified Buffer)地址空间语义。UnalignReg→UnalignRegForLoad:在gelu_dynamic_quant_workspace_regbase.h中将MicroAPI::UnalignReg替换为MicroAPI::UnalignRegForLoad,明确了该非对齐寄存器仅用于加载场景的语义。FusedMulDstAdd→MulDstAdd:在swish_grad_dag.h中将MicroAPI::FusedMulDstAdd替换为MicroAPI::MulDstAdd,去掉了接口名中的Fused前缀,语义更简洁清晰。


atomgit-bot
15 天前 评论:
15 天前 评论:
15 天前 添加了label:cann-cla/yes
CANN-robot
15 天前 评论:
15 天前 评论:
此处折叠了388条消息 查看更多
范其瑞
6 天前 评论:
6 天前 评论:
/lgtm
/approve


su-yueming
6 天前 评论:
6 天前 评论:
/lgtm
/approve


6 天前 添加了label:lgtmapproved
6 天前 关闭了关联的issue
6 天前 合入了pull request
描述
将 ops-nn 仓库中 34 个算子的 arch35 kernel 实现从 MicroAPI/Reg 旧接口替换为新接口,同时将地址空间限定符
__local_mem__统一替换为__ubuf__,并对 3 个 UT 测试文件的ubFormer参数进行调整,使代码与 AscendC SDK 最新命名规范保持一致。改动原因
AscendC SDK 新版本对 MicroAPI 和 Reg 层 API 进行了重命名,旧 API(如 DataCopy、CompareScalar、MaskAnd、ReduceMax、ReduceSum、UnalignReg 等)将逐步废弃。本 PR 同步更新仓库中 arch35 算子 kernel 的接口调用,避免使用将废弃的 API。
改动方法
对 108 个文件进行批量接口替换,其中 105 个为 arch35 kernel 头文件、3 个为 UT 测试文件,涉及 3 个算子域共 34 个算子:
接口替换映射如下:
此外,3 个 UT 测试文件(fast_gelu、fast_gelu_grad、log_sigmoid)的
ubFormer参数从 256 调整为 1024,并补充文件末尾换行符。关联的Issue
https://gitcode.com/cann/ops-nn/issues/4689
测试
接口替换为纯命名变更,不涉及计算逻辑修改,通过编译验证和已有 ST/UT 用例回归验证。UT 测试文件
ubFormer参数调整(256→1024)用于适配新接口下的 UB 对齐要求。文档更新
无
类型标签
AI/Agent生成声明