普安| 东乡| 新民| 宾阳| 怀宁| 卓资| 邛崃| 白云| 元氏| 长岛| 娄烦| 化隆| 武冈| 湘东| 驻马店| 大同市| 兴平| 满洲里| 沁阳| 姚安| 镶黄旗| 东辽| 宝清| 田阳| 松滋| 修文| 乌兰浩特| 安丘| 肇源| 潍坊| 六安| 通城| 南海镇| 佛山| 江油| 八公山| 澎湖| 门头沟| 莱西| 台中市| 建湖| 叶城| 谢通门| 环县| 攸县| 福建| 南和| 威县| 济南| 雷山| 恩平| 郧西| 平阳| 丰南| 永春| 霍城| 贺兰| 遵义县| 牟定| 清徐| 彭山| 青州| 红河| 印江| 六枝| 蒙阴| 色达| 三门| 米林| 海淀| 江油| 西山| 林周| 敦煌| 漠河| 高淳| 烟台| 上虞| 鄂伦春自治旗| 巍山| 绿春| 绩溪| 清丰| 大方| 木兰| 麦积| 循化| 灵寿| 瓦房店| 德昌| 鄂州| 洛川| 兴和| 平鲁| 青岛| 紫金| 瑞丽| 西昌| 武城| 镇康| 澳门| 洛南| 乌拉特中旗| 成县| 巴东| 霍州| 万州| 富蕴| 东台| 渠县| 通许| 揭东| 克拉玛依| 阜南| 宁远| 同德| 琼结| 瑞丽| 遵化| 泗阳| 巴中| 德保| 加格达奇| 漳县| 五常| 利川| 佛坪| 铁力| 鸡西| 临沭| 河间| 巴马| 峨眉山| 仙游| 启东| 恩平| 韶山| 曲水| 澧县| 武都| 和林格尔| 盐都| 左云| 于田| 大埔| 赤峰| 林州| 岚皋| 奉节| 巢湖| 寿县| 武陵源| 行唐| 景洪| 日喀则| 泗洪| 渑池| 德惠| 苏家屯| 原平| 吉木萨尔| 阜阳| 安西| 兰坪| 恩施| 耒阳| 岐山| 平潭| 牡丹江| 铁岭县| 东沙岛| 岱岳| 黑山| 长春| 海宁| 大庆| 婺源| 古县| 相城| 常州| 土默特左旗| 赵县| 离石| 黄陵| 永修| 武胜| 资源| 邹平| 商城| 友好| 大安| 阿坝| 康乐| 达孜| 呈贡| 仪陇| 潢川| 怀远| 蓟县| 南和| 依兰| 巴马| 阿勒泰| 巴中| 札达| 宜君| 金湖| 泸定| 隆昌| 抚松| 昭平| 三都| 大洼| 蒙阴| 云阳| 诸城| 贵阳| 乐陵| 岳阳市| 赤峰| 张北| 长葛| 贵阳| 凯里| 全南| 朔州| 蕲春| 库车| 夏河| 遵化| 甘谷| 贡嘎| 通榆| 上甘岭| 利川| 涪陵| 肥乡| 怀来| 岢岚| 六安| 高要| 滦县| 武穴| 阜平| 鲁甸| 沈阳| 镇巴| 丹江口| 确山| 大方| 敖汉旗| 洪湖| 富拉尔基| 泾源| 陇县| 桂阳| 鹰潭| 邕宁| 株洲县| 赞皇| 新兴| 康定| 广南| 绥中| 智能驾驶峰会
您正在使用IE低版浏览器,为了您的雷锋网账号安全和更好的产品体验,强烈建议使用更快更安全的浏览器
雷锋网 读懂智能&未来
人工智能 正文
发私信给程弢
发送

0

谷歌硬件工程师揭秘,TPU为何会比CPU、GPU快30倍?

本文作者:程弢 2018-09-25 14:03
导语:从这次发布的测试结果来看,TPU似乎已经超出了业界的预期,但是藏在这一芯片背后的内部架构究竟有什么秘密呢,我们从Jouppi此前发布的论文当中,可以找到答案。
谷歌硬件工程师揭秘,TPU为何会比CPU、GPU快30倍?

在谷歌发布TPU一年后,这款机器学习定制芯片的神秘面纱终于被揭开了。

昨日,谷歌资深硬件工程师Norman Jouppi刊文表示,谷歌的专用机器学习芯片TPU处理速度要比GPU和CPU快15-30倍(和TPU对比的是英特尔Haswell CPU以及Nvidia Tesla K80 GPU),而在能效上,TPU更是提升了30到80倍。

从这次发布的测试结果来看,TPU似乎已经超出了业界的预期,但是藏在这一芯片背后的内部架构究竟有什么秘密呢,我们从Jouppi此前发布的论文当中,可以找到答案。

据雷锋网(公众号:雷锋网)了解,早在四年前,谷歌内部就开始使用消耗大量计算资源的深度学习模型,这对CPU、GPU组合而言是一个巨大的挑战,谷歌深知如果基于现有硬件,他们将不得不将数据中心数量翻一番来支持这些复杂的计算任务。

所以谷歌开始研发一种新的架构,Jouppi称之为“下一个平台”。Jouppi曾是MIPS处理器的首席架构师之一,他开创了内存系统中的新技术。三年前他加入谷歌的时候,公司上下正在用CPU、GPU混合架构上来进行深度学习的训练。

Jouppi表示,谷歌的硬件工程团队在转向定制ASIC之前,早期还曾用FPGA来解决廉价、高效和高性能推理的问题。但他指出,FPGA的性能和每瓦性能相比ASIC都有很大的差距。他解释说,“TPU可以像CPU或GPU一样可编程,它可以在不同的网络(卷积神经网络,LSTM模型和大规模完全连接的模型)上执行CISC指令,而不是为某个专用的神经网络模型设计的。一言以蔽之,TPU兼具了CPU和ASIC的有点,它不仅是可编程的,而且比CPU、GPU和FPGA拥有更高的效率和更低的能耗。

TPU的内部架构

谷歌硬件工程师揭秘,TPU为何会比CPU、GPU快30倍?

该图显示了TPU上的内部结构,除了外挂的DDR3内存,左侧是主机界面。指令从主机发送到队列中(没有循环)。这些激活控制逻辑可以根据指令多次运行相同的指令。

TPU并非一款复杂的硬件,它看起来像是雷达应用的信号处理引擎,而不是标准的X86衍生架构。Jouppi说,尽管它有众多的矩阵乘法单元,但是它GPU更精于浮点单元的协处理。另外,需要注意的是,TPU没有任何存储的程序,它可以直接从主机发送指令。

TPU上的DRAM作为一个单元并行运行,因为需要获取更多的权重以馈送到矩阵乘法单元(算下来,吞吐量达到了64,000)。Jouppi并没有提到是他们是如何缩放(systolic)数据流的,但他表示,使用主机软件加速器都将成为瓶颈。

谷歌硬件工程师揭秘,TPU为何会比CPU、GPU快30倍?

256×256阵列缩放数据流引擎,经过矩阵乘法积累后实现非线性输出

从第二张图片可以看出,TPU有两个内存单元,以及一个用于模型中参数的外部DDR3 DRAM。参数进来后,可从顶部加载到矩阵乘法单元中。同时,可以从左边加载激活(或从“神经元”输出)。那些以收缩的方式进入矩阵单元以产生矩阵乘法,它可以在每个周期中进行64,000次累加。

毋庸置疑,谷歌可能使用了一些新的技巧和技术来加快TPU的性能和效率。例如,使用高带宽内存或混合3D内存。然而,谷歌的问题在于保持分布式硬件的一致性。

TPU对比Haswell处理器

在和英特尔“Haswell”Xeon E5 v3处理器来的对比中,我们可以看到,TPU各方面的表现都要强于前者。

在Google的测试中,使用64位浮点数学运算器的18核心运行在2.3 GHz的Haswell Xeon E5-2699 v3处理器能够处理每秒1.3 TOPS的运算,并提供51GB/秒的内存带宽;Haswell芯片功耗为145瓦,其系统(拥有256 GB内存)满载时消耗455瓦特。

相比之下,TPU使用8位整数数学运算器,拥有256GB的主机内存以及32GB的内存,能够实现34GB/秒的内存带宽,处理速度高达92 TOPS ,这比Haswell提升了71倍,此外,TPU服务器的热功率只有384瓦。

谷歌硬件工程师揭秘,TPU为何会比CPU、GPU快30倍?

除此之外,谷歌还测试了CPU、GPU和TPU处理不同批量大小的每秒推断的吞吐量。

谷歌硬件工程师揭秘,TPU为何会比CPU、GPU快30倍?

如上图所示,在小批量任务中(16),Haswell CPU的响应时间接近7毫秒,其每秒提供5482次推断(IPS),其可以实现的最大批量任务(64)每秒则可以完成13194次推断,但其响应时间为21.3毫秒。相比之下,TPU可以做到批量大小为200,而响应时间低于7毫秒,并提供225000个IPS运行推理基准,是其峰值性能的80%,当批量大小为250,响应时间为10毫秒。

不过需要注意的是,谷歌所测试的Haswell Xeon处理器似乎也不能完全说明问题,英特尔Broadwell Xeon E5 v4处理器和最新的“Skylake”Xeon E5,每核心时钟(IPC)的指令比这款处理器提升了约5%。在Skylake是28核,而Haswell为18核,所以Xeon的总体吞吐量可能会上升80%。当然,这样的提升与TPU相比仍有差距。

最后雷锋网需要强调的是,TPU是一个推理芯片,它并非是要取代GPU,可以确定的是,TPU与CPU一起使用对训练分析更加有益。但对于CPU制造商而言,如何研发出像ASIC一样兼顾性能和能效的芯片是现在以及未来要做的。

可以确定的是,谷歌已经逐渐在自己的数据中心部署TPU,但是上述测试数据只是理论结果,实际应用表现如何?我们还不得而知。

Via nextplatform,雷锋网编译

雷锋网版权文章,未经授权禁止转载。详情见转载须知

谷歌硬件工程师揭秘,TPU为何会比CPU、GPU快30倍?
分享:
相关文章

文章点评:

表情

略懂技术的小编

关注一切有未来感的产品及技术!
当月热门文章
最新文章
为了您的账户安全,请验证邮箱
您的邮箱还未验证,完成可获20积分哟!
请验证您的邮箱
立即验证
完善账号信息
您的账号已经绑定,现在您可以设置密码以方便用邮箱登录
立即设置 以后再说
请填写申请人资料
姓名
电话
邮箱
微信号
作品链接
个人简介
高排村 中兴大道 罗宗 王五沟 布拉格苏木
候岭乡 上黄竹塘 燕郊中燕小区 冯家窑 雷祖殿
汀田镇 浙江路桥区新桥镇 东山峰农场 开发区第四大街 石狮市蚶江镇拘留所
色达县 箕漕街 前进道四新里 向家村 安阳市北关区
百度