gpu cuda计算加速如何改造成npu加速
收藏回复举报
gpu cuda计算加速如何改造成npu加速
t('forum.solved') 已解决
发表于2025-09-05 10:50:22
0 查看

向各位前辈请教一个问题

目前有一个项目里使用了英伟达cuda进行计算加速,非深度学习模型

只是调用cuda算子与组合基础算子进行计算加速,具体实现在.cu文件内,然后使用python调用算子(具体实现可能不是这样,只是我的理解)

如果想把这个加速过程改到npu上,我该怎么做呢?

目前了解了一些CANN自定义算子开发的内容,但感觉这种开发算子工程、再开发算子调用工程,对我来说上手有些慢

另外因为原来的加速并不是基于torch模型这种,也不好直接转换成om模型

有没有其他方法可以快速将原来在gpu上的计算加速移植到npu上呢?

我要发帖子