今天看啥  ›  专栏  ›  机器之心

陈天奇等人提出TVM:深度学习自动优化代码生成器

机器之心  · 公众号  · AI  · 2018-02-17 11:26
选自arXiv作者:陈天奇等机器之心编译参与:李泽南、路雪TVM 是由华盛顿大学在读博士陈天奇等人提出的深度学习自动代码生成方法,去年 8 月机器之心曾对其进行过简要介绍。该技术能自动为大多数计算硬件生成可部署优化代码,其性能可与当前最优的供应商提供的优化计算库相比,且可以适应新型专用加速器后端。近日,这项研究的论文《TVM: End-to-End Optimization Stack for Deep Learning》终于完成,内容包含新方法的介绍与讨论,以及 TVM 在英伟达、AMD 的 GPU、树莓派及一些 FPGA 上的性能评估。项目链接:https://github.com/dmlc/tvm深度学习模型可以识别图像、处理自然语言,以及在部分具有挑战性的策略游戏中击败人类。在其技术发展的过程中,现代硬件稳步推进的计算 ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照