下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

算子开发和部署

自定义算子入图前,需要先完成算子开发和部署。以Ascend C算子开发和部署过程为例,具体步骤如下:

  1. 自定义算子开发。

    创建标准自定义算子工程,完成算子原型定义、Kernel侧算子实现、Host侧Tiling实现等,详细介绍参见《CANN Ascend C 算子开发指南》中的“工程化算子开发”章节

  2. 自定义算子入GE图。

    完成shape推导与data type推导等函数实现,适配算子入GE图,详细介绍参见《CANN Ascend C 算子开发指南》中“算子入图(GE图)开发”章节

  3. 自定义算子工程编译部署。

    通过工程编译脚本完成算子编译,并将自定义算子包部署到CANN算子加速库中,详细介绍参见《CANN Ascend C 算子开发指南》中的“ 编译部署”章节

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词