MegEngine / MegCC

MegCC是一个运行时超轻量,高效,移植简单的深度学习模型编译器
Apache License 2.0
474 stars 56 forks source link

如何支持新的算子? #26

Closed tpoisonooo closed 1 year ago

tpoisonooo commented 1 year ago

我看算子列表的 opr 个数并不多,如果碰到个不支持的 (例如 topk)应该怎么办?

chenqy4933 commented 1 year ago

kernel 数量确实不多,但是topk还真有了,另外已经有了一个粗糙的加 kernel 的文档,https://github.com/MegEngine/MegCC/blob/main/doc/how-to-add-new-kernel.md , 可能不太清晰,后续补充一下