Toybrick

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

zhangyichent

注册会员

积分
133
楼主
发表于 2022-1-20 12:03:34    查看: 2988|回复: 0 | [复制链接]    打印 | 只看该作者
目前:我已经用cuda实现了一个自定义算子,如何才能将这个算子移植到npu上呢。
有没有可能官方提供一个现成的转换工具,否则就要先熟悉了rknn再从头开始写。

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

麻烦指条路
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表