新的內(nèi)存方案出現(xiàn)，可將帶寬再提升一個臺階。 ? 在 2024 年，如果需要將數(shù)十個、數(shù)百個、數(shù)千個甚至數(shù)萬個加速器拼接在一起，那么互連就是個大課題了。

英偉達（Nvidia）?擁有 NVLink 和 InfiniBand。Google 的 TPU 吊艙使用光電路開關 （OCS） 相互通信。AMD 擁有 Infinity Fabric，用于芯片到芯片、芯片到芯片以及即將推出的節(jié)點到節(jié)點流量。當然，還有好的老式以太網(wǎng)。

這里的訣竅不是構建足夠大的網(wǎng)格，而是抵御與離包相關的大量性能損失和帶寬瓶頸。它也沒有做任何事情來解決這樣一個事實，即所有這些 AI 處理所依賴的 HBM 內(nèi)存都以固定的比例與計算相關聯(lián)。

“這個行業(yè)正在使用Nvidia GPU作為世界上最昂貴的內(nèi)存控制器，”Dave Lazovsky說，他的公司Celestial AI剛剛在USIT和許多其他風險投資巨頭支持的C輪融資中獲得了1.75億美元，以將其光子織物商業(yè)化。

去年夏天，我們研究了Celestial的光子結構，其中包括一系列硅光子學互連器、中介層和小芯片，旨在將AI計算從內(nèi)存中分解出來。不到一年后，他們正在與幾家超大規(guī)模客戶和一家大型處理器制造商合作，將其技術集成到他們的產(chǎn)品中。Lazovsky沒有指名道姓。

但事實上，Celestial將AMD Ventures視為其支持者之一，其高級副總裁兼產(chǎn)品技術架構師Sam Naffziger在公告發(fā)布的同一天討論了共同封裝硅光子小芯片的可能性，這無疑引起了一些人的注意。話雖如此，AMD為光子學初創(chuàng)公司提供資金并不意味著我們將永遠在Epyc CPU或Instinct GPU加速器中看到Celestial的小芯片。

雖然 Lazovsky 無法透露 Celestial 與誰合作，但他確實提供了一些關于該技術如何集成的線索，以及即將推出的 HBM 內(nèi)存設備的先睹為快。

正如我們在最初涉足Celestial的產(chǎn)品戰(zhàn)略時所討論的那樣，該公司的零件分為三大類：小芯片、中介層和英特爾EMIB或臺積電CoWoS的光學旋轉，稱為OMIB。

不出所料，Celestial的大部分吸引力都集中在小芯片上?！拔覀儧]有做的是試圖強迫我們的客戶采用任何一種特定的產(chǎn)品實施。目前，為光子結構提供接口的風險最低、最快、最不復雜的方法是通過小芯片，“Lazovsky告訴 The Next Platform。

從廣義上講，這些小芯片可以以兩種方式使用：要么增加額外的 HBM 內(nèi)存容量，要么作為芯片到芯片的互連，分類或類似于光學 NVLink 或 Infinity Fabric。

這些小芯片比 HBM 堆棧小一點，提供光電互連，片外總帶寬為 14.4 Tb/s或 1.8 GB/s。

話雖如此，我們被告知可以制造一個小芯片來支持更高的帶寬。第一代技術可以支持每平方毫米約 1.8 Tb/s的速度。與此同時，Celestial的第二代Photonic結構將從56 Gb/s提高到112 Gb/s的PAM4 SerDes，并將通道數(shù)量從4個增加到8個，從而有效地將帶寬翻兩番。

因此，14.4 Tb/s不是上限，而是現(xiàn)有芯片架構能夠處理的結果。這是有道理的，否則任何額外的容量都會被浪費。

這種連接性意味著 Celestial 可以實現(xiàn)類似于 NVLink 的互連速度，只是沿途的步驟更少。

雖然芯片到芯片的連接相對不言自明——在每個封裝上放一個光子織物小芯片并對齊光纖連接——但內(nèi)存擴展完全是另一種動物。雖然 14.4 Tb/s的速度并不慢，但對于多個 HBM3 或 HBM3e 堆棧來說，它仍然是一個瓶頸。這意味著添加更多的 HBM 只會讓您的容量超過某個點。盡管如此，用兩個 HBM3e 堆棧代替一個堆棧并不算什么。

Celestial 有一個有趣的解決方法，即它的內(nèi)存擴展模塊。由于帶寬的上限為 1.8 GB/s，因此該模塊將僅包含兩個總計 72 GB 的 HBM 堆棧。此外，還將配備一組 4 個 DDR5 DIMM，支持高達 2 TB 的額外容量。

Lazovsky不愿將所有豆子都灑在產(chǎn)品上，但確實告訴我們，它將使用Celestial的硅光子學中介層技術作為HBM，互連和控制器邏輯之間的接口。

說到模塊的控制器，我們被告知 5nm開關 ASIC 有效地將 HBM 變成 DDR5 的直寫緩存?！八鼮槟峁┝?DDR 的容量和成本以及帶寬和 HBM 互連的 32 個偽通道的所有優(yōu)勢，從而隱藏了延遲，”Lazovsky 解釋道。

他補充說，這與英特爾對至強Max所做的或英偉達對其GH200超級芯片所做的事情相去不遠?！八旧鲜且粋€增壓的 Grace-Hopper，沒有所有的成本開銷，而且效率更高?！?/p>

效率提高多少？“我們的內(nèi)存事務能量開銷約為每比特 6.2 皮焦耳，而通過 NVLink、NVSwitch 進行遠程內(nèi)存事務的開銷約為 62.5 皮焦耳，”Lazovsky稱，并補充說延遲也不高。

“這些遠程內(nèi)存事務的總往返延遲，包括通過光子結構的兩次旅行和內(nèi)存讀取時間，為120納秒，”他補充道：“因此，它將比大約80納秒的本地內(nèi)存多一點，但它比去Grace并讀取參數(shù)并將其拉到Hopper要快?！?/p>

據(jù)我們了解，這些內(nèi)存模塊中的 16 個可以嚙合并為一個內(nèi)存交換機，并且可以使用光纖隨機播放連接多個這些設備。

這意味著，除了計算、存儲和管理網(wǎng)絡之外，使用Celestial互連構建的芯片不僅能夠相互連接，而且能夠共享內(nèi)存池。

“這允許你以一種非常非常有效的方式進行機器學習操作，例如廣播和減少，而無需切換，” Lazovsky說。

Celestial面臨的挑戰(zhàn)是時機。Lazovsky告訴我們，他預計將在 2025 年下半年的某個時候開始向客戶提供光子織物小芯片的樣品。然后，他預計至少還需要一年時間，我們才能看到使用該設計的產(chǎn)品投放市場，并在2027年實現(xiàn)銷量增長。

然而，Celestial并不是唯一一家追求硅光子學的初創(chuàng)公司。另一家獲得英特爾投資支持的光子學初創(chuàng)公司Ayar Labs已經(jīng)將其光子學互連集成到原型加速器中。

然后是 Lightmatter，它在去年 12 月獲得了 1.55 億美元的 C 輪融資，并試圖通過其 Passage 中介層做一些與 Celestial 非常相似的事情。當時，Lightmatter 首席執(zhí)行官尼克·哈里斯 （Nick Harris） 聲稱，它有客戶使用 Passage 來“擴展到 300,000 臺節(jié)點的超級計算機”。當然，和拉佐夫斯基一樣，哈里斯也不會告訴我們它的客戶是誰。

還有 Eliyan，它正試圖通過其 NuLink PHY 完全擺脫中介層——或者如果你必須擁有它們，可以提高中介層的性能和規(guī)模。

無論誰在這場競賽中脫穎而出，向共封裝光學器件和硅光子中介層的轉變似乎只是時間問題。

審核編輯：黃飛