【未来虫教育】TPU 的多『芯片』互联层级结构(未来虫教育是正规公司吗)

【未来虫教育】TPU 的多『芯片』互联层级结构(未来虫教育是正规公司吗)

现在升级到更高层级,观察 TPU 在多『芯片』环境中的运作方式。

4.1 托盘层级(即 "板卡";含 4 个『芯片』)

今日霍州(www.jrhz.info)©️

单块 TPU 托盘包含 4 个 TPU 『芯片』或 8 个 TensorCore(简称 "核心")。每块托盘配备独立 CPU 主机(注:推理型 TPU 的每个主机可访问 2 块托盘,因其每『芯片』仅含 1 个核心)。

主机(Host) ⇔ 『芯片』(Chip)的连接采用 PCIe 接口,但『芯片』(Chip)⇔『芯片』(Chip)之间通过 Inter-Core Interconnect(ICI)连接,该接口具备更高带宽。

不过 ICI 连接还可进一步扩展至多块托盘。为此,我们需要继续提升到机架层级(Rack level)。

4.2 机架层级(4x4x4 『芯片』)

TPU 最令人兴奋的特性在于其可扩展性,这一点从机架层级开始显现。

一个 TPU 机架包含 64 个 TPU 『芯片』,通过 4x4x4 三维环面网络互联。如果您看过谷歌的 TPU 宣传资料(如下图),这张图展示的是 8 个 TPU 机架的集群。

今日霍州(www.jrhz.info)©️

8 个 TPU 机架(TPUv4)

但在深入讨论机架之前,我们需要澄清几个容易混淆的术语:机架(Rack)、Pod 和切片(Slice)的区别。

问:TPU 机架、TPU Pod 和 TPU 切片有何不同?

不同谷歌资料对这些术语的使用存在差异,有时甚至混用 "TPU Pod" 和 "TPU Slice"。本文采用谷歌 TPU 论文和 GCP 官方文档的定义([3][7][9]):

1)TPU 机架(Rack)

包含 64 块『芯片』的物理单元,也称为 “立方体(cube)”。

2)TPU Pod

通过 ICI 和光纤连接的 TPU 最大单元。

又称 "Superpod" 或 "Full Pod"。例如 TPUv4 的 TPU Pod 包含 4096 块『芯片』(或 64 个机架)。

3)TPU 切片(Slice)

介于 4 块『芯片』到 Superpod 规模之间的任何 TPU 配置组合。

主要区别在于,TPU 机架和 TPU Pod 是物理计量单位,而 TPU 切片是抽象计量单位。当然,TPU 切片的设置涉及重要的物理拓扑约束,但现阶段我们暂不展开讨论。

现在,我们将聚焦物理计量单位:TPU 机架和 TPU Pod。这是因为,理解 TPU 系统的物理连接方式,能更深入地掌握其设计哲学。

现在回到 TPUv4 机架的具体结构:

单个 TPU 机架通过 ICI 和 OCS(Optical Circuit Switching)技术连接 64 个『芯片』。实质上,我们通过组合多个托盘(trays)来构建一个 64 『芯片』的完整系统。这种 "将小型单元组装成超级计算机" 的设计理念将持续贯穿后续层级。

下图展示了 TPUv4 单个机架的拓扑结构。它采用 4x4x4 三维环面网络,其中每个节点都代表一块『芯片』,蓝色箭头表示 ICI 链路,而各个面上的连接线则代表 OCS(根据文献 [7] 重绘)。

今日霍州(www.jrhz.info)©️

使用 OCS 的 TPU 单机架架构

然而,这张图表引出了两个关键问题:为何 OCS 仅应用于环面结构的表面?换句话说 —— 使用 OCS 的核心优势是什么?共有三大核心优势,我们将在后文再详述另外两点。

OCS 的优势 #1:环绕连接 (Wraparound)

通过环形拓扑优化节点间的通信效率。

OCS 还承担特定 TPU 配置的环绕连接功能。该设计将两节点间的跳数从最坏情况下 N-1 跳降至每轴 (N-1)/2 跳,因为每条轴均形成一个环形(一维环面拓扑)。

随着规模的进一步扩大,这种影响变得更加重要,因为降低『芯片』间的通信延迟对于高度并行化的实现至关重要。

附注:并非所有 TPU 都采用 3D 环面拓扑

注意,早期 TPU(如 TPUv2/v3)及推理专用 TPU(如 TPUv5e/v6e)使用 2D 环面拓扑而非下文所述的 3D 环面。不过 TPUv7"Ironwood" 虽定位为推理『芯片』,但其拓扑疑似 3D 环面(注:仅根据官方宣传材料推测)。

今日霍州(www.jrhz.info)©️

2D 环面拓扑示意图

特别声明:[【未来虫教育】TPU 的多『芯片』互联层级结构(未来虫教育是正规公司吗)] 该文观点仅代表作者本人,今日霍州系信息发布平台,霍州网仅提供信息存储空间服务。

猜你喜欢

腾讯首播!16集悬疑剧!黑色幽默犯罪!沙溢、王彦霖、魏翔主演!(腾讯首播电视剧)

沙溢成功塑造了一个既严肃又充满魅力的警察形象,让人看到他喜剧之外的另一面。 这部剧的案情发展就像过山车般刺激:绑匪本想绑架普通小孩,却误打误撞绑了富商独子;赎金从50万飙到700万;绑匪团队内讧不断,甚至当街…

腾讯首播!16集悬疑剧!黑色幽默犯罪!沙溢、王彦霖、魏翔主演!(腾讯首播电视剧)

货车工具箱🧰的升级之选:外挂铁工具箱🧰与长安跨越王等车型适配分析(小货车工具箱🧰)

为何外挂铁工具箱🧰成为众多车主的青睐之选?本篇文章深入剖析其特点、功能以及与长安跨越王T20、跃进、福田、金杯、江淮等主流商用车型的适配情况,帮助车主找到适合自己的解决方案。 外挂铁工具箱🧰不仅是空间拓展利器,更是提高工作效率的关键一环。

货车工具箱🧰的升级之选:外挂铁工具箱🧰与长安跨越王等车型适配分析(小货车工具箱🧰)

带你读懂永磁体与磁矩的核心特性与实践意义(永磁体的优缺点)

ü 固有性与稳定性:永磁体的磁矩源于内部磁畴的有序排列,在无外磁场干扰时,这种排列相对稳定,磁矩保持恒定,这是其 “永磁” 特性的本质。 磁矩的大小和方向决定了永磁体的磁化强度和方向,从而影响了永磁体的性能…

带你读懂永磁体与磁矩的核心特性与实践意义(永磁体的优缺点)

威尼斯评委一句话:辛芷蕾是近十年来威尼斯最佳女演员最好的表演(威尼斯评委一共有几位)

导演蔡尚君曾提到,辛芷蕾在第82届威尼斯国际电影节上荣获沃尔皮杯最佳女演员奖时,评委给出了如此评价:这是近十年来,威尼斯最佳女演员的最佳表演!让我更为惊讶的是,辛芷蕾不仅在电视剧中表现出色,甚至在电影中的表现…

威尼斯评委一句话:辛芷蕾是近十年来威尼斯最佳女演员最好的表演(威尼斯评委一共有几位)

励志又上头!8部大女主“经商古装剧”,爱情事业两不误!(励志一些)

和蒋长扬的合作不仅使她自己事业有成,也帮助了困境中的其他女性♀️,逐步带领她们走向了幸福的生活。她不仅成功打败了商业对手,还从寡妇逆袭成为陕西的女首富,她的商界传奇至今为人称道。 这些剧集中,女主们不仅仅是依赖爱…

励志又上头!8部大女主“经商古装剧”,爱情事业两不误!(励志一些)