host侧算子实现概述

host侧算子实现开发包括Tiling实现、Shape推导等函数实现、算子原型注册。具体介绍如下:

上述内容均在host侧的算子类中实现,完成上述实现后,需要通过OP_ADD接口,将该算子类注册到算子原型库中。举例如下:

OP_ADD(AddCustom); // 此处AddCustom即为host侧的算子类

下文的介绍仅呈现代码片段,您可以点击LINK,在op_host目录下获取完整样例代码,并在op_kernel中获取对应的kernel侧实现。您可以点击LINK获取Add算子的单算子调用程序样例,来完成算子的调用和验证。