【2023 · CANN训练营第一季】核函数简介
【摘要】 核函数简介核函数返回类型为voidTIK C++算子设备侧实现的入口函数是核函数TIK C++矢量算子开发时，需要在核函数内完成算法逻辑实现TIK C++算子开发是在核函数中进行算子类对象的创建和其成员函数的调用，实现算子的所有功能核函数（Kernel Function）是TIK C++算子设备侧的入口。TIK C++允许用户使用核函数这种C/C++函数的语法扩展来管理设备侧的运行代码，用户...
核函数简介
核函数返回类型为void
TIK C++算子设备侧实现的入口函数是核函数
TIK C++矢量算子开发时，需要在核函数内完成算法逻辑实现
TIK C++算子开发是在核函数中进行算子类对象的创建和其成员函数的调用，实现算子的所有功能
核函数（Kernel Function）是TIK C++算子设备侧的入口。TIK C++允许用户使用核函数这种C/C++函数的语法扩展来管理设备侧的运行代码，用户在核函数中实现算子逻辑的编写，例如自定义算子类及其成员函数以实现该算子的所有功能。核函数是主机侧和设备侧连接的桥梁
核函数是直接在设备侧执行的代码。在核函数中，需要为在一个核上执行的代码规定要进行的数据访问和计算操作，当核函数被调用时，多个核将并行执行同一个计算任务
编写核函数
使用函数类型限定符
除了需要按照C/C++函数声明的方式定义核函数之外，还要为核函数加上额外的函数类型限定符，包含__global__和__aicore__
使用__global__函数类型限定符来标识它是一个核函数，可以被<<<...>>>调用；使用__aicore__函数类型限定符来标识该核函数在设备侧AI Core上执行：
__global__ __aicore__ void kernel_name(argument list);
【声明】本内容来自华为云开发者社区博主，不代表华为云及华为云开发者社区的观点和立场。转载时必须标注文章的来源（华为云社区）、文章链接、文章作者等基本信息，否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容，欢迎发送邮件进行举报，并提供相关证据，一经查实，本社区将立刻删除涉嫌侵权内容，举报邮箱：
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论（0）