Toybrick

标题: 【自定义OP】已经用cuda C实现了算子，怎么快速转到rknn上去 [打印本页]

作者: zhangyichent 时间: 2022-1-20 12:03
标题: 【自定义OP】已经用cuda C实现了算子，怎么快速转到rknn上去
目前：我已经用cuda实现了一个自定义算子，如何才能将这个算子移植到npu上呢。
有没有可能官方提供一个现成的转换工具，否则就要先熟悉了rknn再从头开始写。

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

麻烦指条路

欢迎光临 Toybrick (https://t.rock-chips.com/)