在此示例中,实现了一个 XIR 运算符 add。它添加了 2 个输入张量,并假定这 2 个张量具有相同形状。
要寄存新的 XIR 运算符,请参阅 Vitis AI 用户指南(UG1414)。此示例假定 add 运算符已寄存在 Xmodel 计算图内,也就是说 Vitis AI xcompiler 已成功编译含 add 运算符的模型。
要获取完整的参考黄金代码,请参阅自定义 XIR 运算符示例。
要实现 XIR 运算符,请执行以下步骤:
- 编写 C++ 类。
- 编写构造函数。
- 编写 calculate 函数。
- 寄存含宏的实现。
- 构建共享库。
- 部署该库。