Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

能否添加自定义算子? #189

Open
wohaiaini opened this issue Feb 25, 2025 · 3 comments
Open

能否添加自定义算子? #189

wohaiaini opened this issue Feb 25, 2025 · 3 comments

Comments

@wohaiaini
Copy link

wohaiaini commented Feb 25, 2025

手头的文档中没有找到,有没有相关的示例参考下,或者参考链接,谢谢

@waydong
Copy link
Collaborator

waydong commented Feb 25, 2025

你好,能详细描述下需要自定义什么算子吗?

@wohaiaini
Copy link
Author

你好,能详细描述下需要自定义什么算子吗?

目前是有这个想法,具体算子还没确定,想先了解下基本的方法和操作说明~

@wohaiaini
Copy link
Author

你好,能详细描述下需要自定义什么算子吗?

你好,在NPU调用matmul的示例已经弄好,非常感谢。

现在是想添加一个在rk3576板端mali-g52 GPU上运行基本matmul计算的算子,请问能在这个基础上进行修改吗?
https://github.com/airockchip/rknn-toolkit2/tree/master/rknpu2/examples/rknn_custom_op_demo/rknn_custom_gpu_op_demo

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants