华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
昇腾社区首页
昇腾论坛
0
/100
支持
消息
我的论坛主页
论坛首页
/
Ascend C算子开发后怎么在torch_npu调用
点赞
0
收藏
0
回复
0
分享
举报
只看楼主
Ascend C算子开发后怎么在torch_npu调用
已解决
新人帖
发表于2025-05-29 10:18:50
0
查看
目前已经按照算子开发教程完成了算子的编译和部署,并且也完成了pytorch框架的适配,在python中能够通过torch_npu.add_custom()调用自定义的算子。但是目前的疑惑是:
当前主机中有自带的自定义的算子:torch.ops.fbgemm.jagged_to_padded_dense,这个算子是如何完成pytorch的注册和框架适配的,为什么可以通过
torch.ops.fbgemm.****的方式调用。
匿名回复
发表
我要发帖子
我们使用cookie来确保您的高速浏览体验。继续浏览本站,即表示您同意我们使用cookie。
查看详情
当前主机中有自带的自定义的算子:torch.ops.fbgemm.jagged_to_padded_dense,这个算子是如何完成pytorch的注册和框架适配的,为什么可以通过
torch.ops.fbgemm.****的方式调用。