111_cOd9G5 · 2023年07月14日 · 浙江

Zhouyi NPU如何自定义算子?

为Parse和Optimizer模块添加了plugin,使用aipuparse和aipuopt转包含自定义算子的模型时,能够正常输出权重文件和IR。
为GBuiler添加了plugin,使用aipugb时会提示:
[E] [subgraphop.cpp: 443] Unsupport node type:ZhouyiMax
当我移除此plugin的时候,会提示:
[E] [IRChecker] IRChecker: Invalid IR. Please double check the IR
是不是意味着为GBuiler添加plugin是OK的,只是plugin实现有问题?
有没有一个完整自定义算子的案例?

你的回答