【2023 · CANN训练营第一季】TIK C++算子开发流程

原创于 2023-05-28 20:22:14 发布 · 212 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #人工智能

文章介绍了TIKC++算子的开发流程，包括快速和标准两种模式。快速模式适用于单算子调用和快速验证，而标准模式涉及更多代码文件，适用于单算子网络或整网部署。开发流程包括算子分析、核函数定义和算子类的实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

TIK C+算子开发流程

快速TIK C++算子开发流程：

完成算子核函数的开发
基于内核调用符方式进行算子运行验证

标准TIK C++算子开发流程：

完成算子核函数的开发
完成单算子网络应用程序的开发
基于ACL单算子调用方式进行算子运行验证

两种开发模式对比

	快速开发模式	标准开发模式
代码文件	少	多
开发时间	短	长
使用场景	单算子调用，快速验证算法逻辑	单算子网络/整网部署使用
推荐开发顺序	先	后

算子开发流程

a. 算子分析：分析算子的数学表达式、输入、输出以及计算逻辑的实现，明确需要调用的TIK C++接口。

b. 核函数定义：定义TIK C++算子入口函数。根据矢量编程范式实现

c. 算子类：完成核函数的内部实现。

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-8TRa1IRb-1685276419208)(../../../../AppData/Roaming/Typora/typora-user-images/image-20230528200312161.png)]
.(img-8TRa1IRb-1685276419208)]