AI 模型编译器 MegCC 开源,解决推理引擎体积问题 - V2EX
V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
MegEngineBot
V2EX    推广

AI 模型编译器 MegCC 开源,解决推理引擎体积问题

  •  1
     
  •   MegEngineBot 2022-11-22 15:09:25 +08:00 1292 次点击
    这是一个创于 1055 天前的主题,其中的信息可能已经有所发展或是发生改变。

    目前社区已经有多个移动端深度学习推理框架,如:NCNN 、MNN... 这些推理引擎都给社区的用户带来了在移动端上部署深度学习非常多的便利,但是他们也都有一个共性问题:随着不断地迭代以及性能优化,运行时库会逐渐的增大,特别是在不同算子 fuse 的时候,会导致非常多的长尾算子,这就会增大我们 App 或者 SDK 的体积。

    为了解决这个问题,由MegEngine团队开源的 MegCC 创新使用模型预编译的方案,生成模型推理必要的代码,去除掉了和模型推理无关的代码,因此极大程度上减少了推理引擎的体积。主要方法是:

    将传统框架运行时的必要步骤:计算图优化、Kernel 选择、内存分配都移到编译时,从而最大程度上减少了 Runtime 时的二进制体积大小,并根据模型信息做进一步的性能优化。

    该方案有以下优点:

    • 随着框架的迭代将不会使得推理引擎的体积增大
    • 很多的算子融合可以在编译时根据模型信息生成对应的 code
    • 模型编译时可以获得整个计算图的信息,这样可以进一步进行极致的性能优化
    • 可以吸收社区在代码生成方面的经验用于为 MegCC 生成 code

    不同于传统推理框架,MegCC 是一个真真实实的深度学习模型编译器,具备极其轻量的 Runtime 二进制体积,高性能,方便移植,极低内存使用以及快启动等核心特点。用户可在 MLIR 上进行计算图优化,内存规划,最后通过预先写好的 code 模版进行代码生成。目前,MegCC 已支持 Arm64 ,Armv7 ,x86 ,risc-v 以及单片机平台。

    GitHub 开源地址:https://github.com/MegEngine/MegCCgithub.com/MegEngine/MegCC

    使用方法及效果

    使用 MegCC 完成模型部署只需要完成以下 3 步:

    • 模型编译:编译 MegEngine 模型,生成运行这个模型对应的 Kernel 以及优化之后的模型。
    • Runtime 编译:这个阶段会将 Runtime 和上一步中生成的 Kernel 一起编译成一个静态库。
    • 集成到应用中:调用上一步编译的静态库的接口进行推理。

    以 YOLOX 模型为例,运行效果如下图:

    从图中可见,MegCC 生成的推理程序在保证推理性能良好(模型测速结果为 670ms )的情况下,其大小可以达到 95KB 。

    详细操作文档:MegCC/how-to-use-chinese.md at main MegEngine/MegCC

    手把手操作教程:挑战 100KB 可执行程序高性能推理 YOLOX 模型

    未来计划

    目前 MegCC 仅支持 MegEngine 模型作为输入,其他模型格式可以考虑转换到 ONNX ,然后通过mgeconvert进行模型格式转换。

    预计在未来 2 个月内,MegCC 将支持更多的模型格式编译。同时实现以下进阶功能:

    • 支持 ONNX 模型作为输入
    • 做更多的 Kernel fusion
    • 支持更多的后端设备

    大家在使用 MegCC 过程中有任何问题,欢迎随时提 issue 告诉我们,也欢迎提 PR 帮助 MegCC 变得更好。

    GitHub:MegEngine 旷视天元(欢迎 star~

    Gitee:MegEngine/MegEngine

    MegEngine 官网:MegEngine-深度学习,简单开发

    欢迎加入 MegEngine 技术交流 QQ 群:1029741705

    目前尚无回复
    关于     帮助文档     自助推广系统     博客     API     FAQ     Solana     2710 人在线   最高记录 6679       Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 30ms UTC 12:06 PVG 20:06 LAX 05:06 JFK 08:06
    Do have faith in what you're doing.
    ubao snddm index pchome yahoo rakuten mypaper meadowduck bidyahoo youbao zxmzxm asda bnvcg cvbfg dfscv mmhjk xxddc yybgb zznbn ccubao uaitu acv GXCV ET GDG YH FG BCVB FJFH CBRE CBC GDG ET54 WRWR RWER WREW WRWER RWER SDG EW SF DSFSF fbbs ubao fhd dfg ewr dg df ewwr ewwr et ruyut utut dfg fgd gdfgt etg dfgt dfgd ert4 gd fgg wr 235 wer3 we vsdf sdf gdf ert xcv sdf rwer hfd dfg cvb rwf afb dfh jgh bmn lgh rty gfds cxv xcv xcs vdas fdf fgd cv sdf tert sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf shasha9178 shasha9178 shasha9178 shasha9178 shasha9178 liflif2 liflif2 liflif2 liflif2 liflif2 liblib3 liblib3 liblib3 liblib3 liblib3 zhazha444 zhazha444 zhazha444 zhazha444 zhazha444 dende5 dende denden denden2 denden21 fenfen9 fenf619 fen619 fenfe9 fe619 sdf sdf sdf sdf sdf zhazh90 zhazh0 zhaa50 zha90 zh590 zho zhoz zhozh zhozho zhozho2 lislis lls95 lili95 lils5 liss9 sdf0ty987 sdft876 sdft9876 sdf09876 sd0t9876 sdf0ty98 sdf0976 sdf0ty986 sdf0ty96 sdf0t76 sdf0876 df0ty98 sf0t876 sd0ty76 sdy76 sdf76 sdf0t76 sdf0ty9 sdf0ty98 sdf0ty987 sdf0ty98 sdf6676 sdf876 sd876 sd876 sdf6 sdf6 sdf9876 sdf0t sdf06 sdf0ty9776 sdf0ty9776 sdf0ty76 sdf8876 sdf0t sd6 sdf06 s688876 sd688 sdf86