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/) Powered by Discuz! X3.3