TVM及深度学习编译技术入门分享

点击上方“小白学视觉”,选择加"星标"或“置顶

重磅干货,第一时间送达

给实验室同学们做了一次TVM分享,介绍了深度学习编译技术的背景、主要组件、主要过程等。资料来源TVM相关论文和slides,每一页给出来源,感兴趣的可以深挖下。

背景

eefff6449b98a31690cff8bd18ec4b50.png 602918244ddd8c47e280d8e613f812ec.png 7b1e0b60bd4c08f30be3979553b03d04.png 760663425d553891c0a54a96e68f8606.png f7c2a0db0cc672ea817e8becc8416c88.png e8bf495b5c5c9ad77ab5256f70e42a54.png

上图来源:M. Li et al., “The Deep Learning Compiler: A Comprehensive Survey,” IEEE Trans. Parallel Distrib. Syst., vol. 32, no. 3, pp. 708–727, Mar. 2021, doi: 10.1109/TPDS.2020.3030548.

99c8de3f9a712de5b3ea64f0bb11095a.png b0455e3b4eb503ca26cad403c7d6703a.png abdd3d65a46912ad9d41f84eedcd2bb9.png deeb2d73bc61fe663eaf8ddbb3f4c590.png

TVM架构

d79e3a0da801ef1f70342a8c6c2d6b10.png a46c19ed7077f564702f08327298a519.png 818252bad03275e7135ccf7ae6328c8c.png

J. Roesch et al., “Relay: A High-Level Compiler for Deep Learning.” arXiv, Aug. 24, 2019. Accessed: Jun. 27, 2022. [Online]. Available: http://arxiv.org/abs/1904.08368

6ea45ad6bc4b045882008230d78303ab.png 193e290efbae5143f512f96681e645cc.png 9b9456336fa8cc98cc4b32f10e5ec024.png fb11f85fa15a35022aa561d829a855ff.png

TVM扩展:支持新的硬件

064ade681948d6d586fbbca82ba99e6f.png

Z. Chen et al., “Bring Your Own Codegen to Deep Learning Compiler.” arXiv, May 03, 2021. Accessed: Jun. 27, 2022. [Online]. Available: Bring Your Own Codegen to Deep Learning Compiler

0881069c18a03d3d9e4c1fae4f1e56b6.png f2a39708e8cbad752a8d5e45e359d423.png a048f9110e5954d2aa54e975e718d69c.png

TVM对神经网络量化的支持

6e79a5dbc7b4078d5a8d5a5218673c70.png 4e9b551e7d66b83720e9db5ed8b7dc4f.png f19ba0d1c4bdf84d77c1774cf3a13d62.png

JAIN A, BHATTACHARYA S, MASUDA M, et al. Efficient Execution of Quantized Deep Learning Models: A Compiler Approach[J]. arXiv, 2020.

TVM应用案例

02a0e0ab524c3747fc5870b8f17273ee.png b945c7ad81996ef7b8167551f1c5da84.png

相关资源:

https://mlc.ai/mlc-llm/

https://github.com/mlc-ai/mlc-llm

https://mlc.ai/mlc-llm/docs/index.html

4569d99efe895f481d04e17cfd669052.png

相关资源:

希姆计算

ba19c5ddc29aa747c05a70ce07f08eb0.png

相关资源:

MindSpore/akg
https://gitee.com/mindspore/akg

79191dff8f8af870a65b0c66c9ed46dd.png
下载1:OpenCV-Contrib扩展模块中文版教程

在「小白学视觉」公众号后台回复:扩展模块中文教程,即可下载全网第一份OpenCV扩展模块教程中文版,涵盖扩展模块安装、SFM算法、立体视觉、目标跟踪、生物视觉、超分辨率处理等二十多章内容。


下载2:Python视觉实战项目52讲
在「小白学视觉」公众号后台回复:Python视觉实战项目,即可下载包括图像分割、口罩检测、车道线检测、车辆计数、添加眼线、车牌识别、字符识别、情绪检测、文本内容提取、面部识别等31个视觉实战项目,助力快速学校计算机视觉。


下载3:OpenCV实战项目20讲
在「小白学视觉」公众号后台回复:OpenCV实战项目20讲,即可下载含有20个基于OpenCV实现20个实战项目,实现OpenCV学习进阶。


交流群

欢迎加入公众号读者群一起和同行交流,目前有SLAM、三维视觉、传感器、自动驾驶、计算摄影、检测、分割、识别、医学影像、GAN、算法竞赛等微信群(以后会逐渐细分),请扫描下面微信号加群,备注:”昵称+学校/公司+研究方向“,例如:”张三 + 上海交大 + 视觉SLAM“。请按照格式备注,否则不予通过。添加成功后会根据研究方向邀请进入相关微信群。请勿在群内发送广告,否则会请出群,谢谢理解~
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值