MegEngine / MegCC

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

MegCC中添加 MegEngine 中的loader opr #10

Closed chenqy4933 closed 1 year ago

chenqy4933 commented 2 years ago

目前很多模型都会选择NPU来进行加速,但是很多模型需要的前后处理NPU不支持,需要在CPU上运行。为了解决这个问题,MegEngine中使用https://github.com/MegEngine/MegEngine/blob/master/src/serialization/include/megbrain/serialization/extern_c_opr.h extern c opr的形式来描述这个NPU模型,然后在MegEngine中视为一个Operator。为了让这类模型可以正常编译,需要在MegEngine中支持这个Op的编译。