实现并寄存定制运算符 - 3.5 简体中文

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2023-06-29
Version
3.5 简体中文

在此示例中,实现了一个 XIR 运算符 add。它添加了 2 个输入张量,并假定这 2 个张量具有相同形状。

要寄存新的 XIR 运算符,请参阅 Vitis AI 用户指南(UG1414)。此示例假定 add 运算符已寄存在 Xmodel 计算图内,也就是说 Vitis AI xcompiler 已成功编译含 add 运算符的模型。

要获取完整的参考黄金代码,请参阅自定义 XIR 运算符示例

要实现 XIR 运算符,请执行以下步骤:

  1. 编写 C++ 类。
  2. 编写构造函数。
  3. 编写 calculate 函数。
  4. 寄存含宏的实现。
  5. 构建共享库。
  6. 部署该库。