代码拉取完成,页面将自动刷新
本样例介绍Add算子的核函数直调方法。
├── 3_add_kernellaunch // 使用核函数直调的方式调用Add自定义算子
│ ├── AddKernelInvocationNeo // Kernel Launch方式调用核函数样例
│ ├── AddKernelInvocationTilingNeo // Kernel Launch方式调用核函数样例,带有Tiling
│ └── CppExtensions // pybind方式调用核函数样例,带有Tiling
Add算子实现了两个数据相加,返回相加结果的功能。对应的数学表达式为:
z = x + y
算子类型(OpType) | Add | |||
算子输入 | name | shape | data type | format |
x | 8 * 2048 | float | ND | |
y | 8 * 2048 | float | ND | |
算子输出 | z | 8 * 2048 | float | ND |
核函数名 | add_custom |
本样例支持如下产品型号:
针对自定义算子工程,编译运行包含如下步骤:
详细操作如下所示。
编译运行此样例前,请参考准备:获取样例代码完成源码包获取。
时间 | 更新事项 | 注意事项 |
---|---|---|
2023/10/09 | 新增AddCustomSample/KernelLaunch样例 | |
2024/01/04 | 新增AddKernelInvocationNeo样例 | 需要基于社区CANN包7.0.0.alpha003及之后版本运行 |
2024/01/04 | 新增AddKernelInvocationTilingNeo样例 | 需要基于社区CANN包7.0.0.alpha003及之后版本运行 |
2024/02/02 | 新增AddCustomSample/KernelLaunch/CppExtensions样例 | 需要基于社区CANN包7.0.0.alpha003及之后版本运行 |
2024/05/22 | 更新readme结构 | 需要基于社区CANN包7.0.0.alpha003及之后版本运行 |
2024/06/06 | AddKernelInvocation样例转维护,不再更新,不推荐使用 | |
2024/08/11 | 删除AddKernelInvocation样例 | |
2024/11/11 | 样例目录调整 |
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。