谷歌声称TensorFlow的OpenCL可以将推理性能提高一倍

发布时间:2020-08-18 15:53:51 【来源:

谷歌今天为其Android上的TensorFlow框架启动了基于OpenCL的移动GPU推理引擎。它现在可以在TensorFlow Lite库的最新版本中使用,并且该公司声称它使用“合理大小”的AI模型在现有OpenGL后端上提供了两倍的加速。

OpenGL已经使用了将近30年,它是一种平台无关的API,用于渲染2D和3D矢量图形。OpenGL ES 3.1中添加了计算着色器,但TensorFlow团队表示,向后兼容的设计决策限制了它们无法发挥设备GPU的全部潜能。另一方面,OpenCL从一开始就是为使用各种加速器进行计算而设计的,因此与移动GPU推理领域更加相关。这激发了TensorFlow团队对基于OpenCL的移动推理引擎的调查,并最终采用了它。

新的TensorFlow推理引擎具有优化器,该优化器选择合适的工作组大小以提高性能,从而使高通Adreno GPU等硬件的平均速度提高了50%。它本身支持FP16,并要求加速器指定数据类型的可用性,通过加速算法计算来减少内存和带宽的使用以及培训时间。(Google指出,借助FP16支持,某些较旧的GPU(例如,大约2012年的Adreno 305现在可以以其全部功能运行)。)OpenCL通过与物理常量内存(芯片等硬件功能)保持协同作用,能够大大超越OpenGL的性能。 Adreno GPU保留用于存储常量数组和变量的RAM。

在一个基准测试中,TensorFlow团队将MNASNet 1.3(一种所谓的神经体系结构搜索系统)的延迟从基于OpenGL后端的Vivo Z3上的100毫秒减少到了OpenCL替代的25毫秒。在使用对象检测算法SSD MobileNet v3进行的另一项测试中,该团队将等待时间从Huawei Mate 20上的近100毫秒减少到不到25毫秒。

Google指出,OpenCL不属于标准Android发行版,因此某些用户无法使用。作为权宜之计,TensorFlow Lite现在在运行时检查OpenCL的可用性,以便在不可用或无法加载OpenCL的情况下,库退回到旧的OpenGL后端。

TensorFlow软件工程师Juhyun Lee和Raman Sarokin在博客中写道:“虽然TensorFlow Lite GPU团队不断改进现有的基于OpenGL的移动GPU推理引擎,但我们也一直在研究其他技术。”“ OpenCL带来了很多功能,这些功能使我们能够优化移动GPU推理引擎。”

现代和起亚今年前10个月出口逾9万辆电动汽车 同比增长71.3%
中芯国际称产能接近满载 14nm量产良率已达业界量产水准
OPPO和vivo申请绿厂蓝厂商标 引发网友关注
华为申请“HICAR SMART COCKPIT”等商标 国际分类为“科学仪器”
NASA为登月任务组装火箭 预计在2024年将第一位女性送上月球
特斯拉申请召回美国9136辆Model X及401辆Model Y
OPPO Reno5 Pro入网工信部:双曲面屏设计 矩阵式相机模组
特斯拉在欧洲涨价,下一批Model S在德国贵了4万

[ 最新资讯 ]

始祖鸟之家双面派对,设计师系列惊艳亮相

  11月24日,被誉为户外爱马仕的始祖鸟(ARC’TERYX)全新概念店——始祖鸟之家在全国顶级商圈北京国贸开业了。现场门店实拍图片   ...

曝小米显示器34寸本月最后一单出货后将暂停供应 因三星无屏供应原因

  本月初,视讯堂消息称小米的 34 寸带鱼屏显示器型号,之后可能无法再保持 1999 元的价格进行销售。今天,视讯堂发布最新消息,称小 ...

微星Stealth 15M超轻游戏本纯净白即将上架 屏幕为15.6英寸

  根据微星笔记本官方的消息,微星 Stealth 15M (纯净白)即将上架预约,这款笔记本是首款采用 11 代低压酷睿 + RTX 2060 配置的 ...

EG7今天宣布收购《机甲战士5》开发商Piranha Games 增加其游戏公司种类

  Enad Global 7(EG7)今天(11 27)宣布收购《机甲战士5》开发商Piranha Games,增加了其游戏公司的种类。  EG7已同意以2410万美元的 ...

华米将推出GTS 2 mini手表:将于12月1日发布 外观更轻薄更精致

  根据华米创始人黄汪的消息,华米将推出 GTS 2 mini 手表,外观更轻薄更精致。    根据华米微博的消息, GTS 2 mini 手表将 ...

冬季必看的小黑裤穿搭指南|九牧王男裤专家

  伴随着秋冬而来的除了购物欲还有穿搭热情,但如果你正苦于衣服少、缺乏穿搭灵感,那么,一条百搭、舒适的九牧王小黑裤,或许可以帮到你,只 ...