自定义运算符的实现和寄存 - 2.5 简体中文

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 简体中文

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

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

要获取完整的参考黄金代码,请参阅 https://github.com/Xilinx/Vitis-AI/examples/Custom_OP_Demo/op_add

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

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