昇腾社区首页
中文
注册
开发者
下载

单算子API执行

本章介绍了以“单算子API执行”方式调用算子的接口,接口列表如表1所示,包括aclnn、acldvpp类接口,它们是CANN算子库(即算子库接口参考)的重要组成部分。

通过介绍接口功能、原型、参数、调用示例等,帮助用户快速掌握算子的使用,支持AI业务的高效应用开发。

表1 算子接口列表

接口分类

说明

接口文档

算子接口(aclnn)> Math类接口

数学计算类算子库,提供Add、Abs等算子。接口前缀为aclnnXxx

参考算子库接口参考中。

算子接口(aclnn)> NN类接口

Neural Network,即神经网络类算子库,提供Matmul等算子。接口前缀为aclnnXxx

目前该类算子API在整个算子库中占最大比重。

算子接口(aclnn)> CV类接口

Computer Vision,即计算机视觉类算子库,提供GridSample等算子。接口前缀为aclnnXxx

算子接口(aclnn)> Transformer类接口

大模型计算类算子库,提供FlashAttention、MC2(通算融合)、MoE(Mixture of Experts)等算子。

接口前缀为aclnnXxx

媒体数据处理接口(acldvpp)

Digital Vision Pre-Processing算子提供高性能视频/图片编解码、图像裁剪缩放等预处理API。接口前缀acldvppXxx。

参考算子库接口参考媒体数据处理接口(acldvpp)