Ascend C算子开发后怎么在torch_npu调用
收藏回复举报
Ascend C算子开发后怎么在torch_npu调用
t('forum.solved') 已解决
新人帖
发表于2025-05-29 10:18:50
0 查看
目前已经按照算子开发教程完成了算子的编译和部署,并且也完成了pytorch框架的适配,在python中能够通过torch_npu.add_custom()调用自定义的算子。但是目前的疑惑是:
当前主机中有自带的自定义的算子:torch.ops.fbgemm.jagged_to_padded_dense,这个算子是如何完成pytorch的注册和框架适配的,为什么可以通过

torch.ops.fbgemm.****的方式调用。

我要发帖子