【自定义OP】已经用cuda C实现了算子，怎么快速转到rknn上去

[复制链接] · 发表于 2022-1-20 12:03:34

目前：我已经用cuda实现了一个自定义算子，如何才能将这个算子移植到npu上呢。
有没有可能官方提供一个现成的转换工具，否则就要先熟悉了rknn再从头开始写。

我这边也有对应的libtorch 的算子，从libtorch转到rknn会不会容易一点呢。

麻烦指条路