當前 AI 技術盛行,不少廠商也紛紛推出針對 AI 加速所設計的晶片,而一家名為 Cerebra 宣布推出第二代 AI 晶片 WSE-2 ,其名稱 WSE 指的是 Wafer Scale Engine ,也意味這款 WSE-2 是一款晶圓級的 AI 處理器,具備高達 2.6 兆個電晶體,遠超越 280 億個電晶體的 NVIDIA Ampere ,整個晶片也是龐然大物,猶如餐桌上的金色餐墊。
▲ Ampere 在 WSE-2 面前僅是一顆小小的晶片
WSE-2 是 Cerebras 第二代產品,其前身 WSE 在 2019 年宣布,當時採用 16nm 製程達到 1.2 兆個電晶體密度,而 WSE-2 則採用台積電當前炙手可熱的 7nm 製程,使得電晶體密度倍增,其架構內包括 85 萬個核心 ,彼此透過 2D 網狀網路連接,晶片溝通的頻寬達 220 Pbps ,相較當前最高效能的 GPU 頻寬高出 4.5 萬倍,溝通的延遲低於 1ms ,同時配有達 20PBps 頻寬的 40GB 的晶片級 SRAM,使得這款龐然大物能進行即時 AI 處理。
▲專用系統 CS-2 採用 15U 大小,裡面僅有一顆 WSE-2
當然畢竟這是一顆超巨大晶片,也不可能用一般常見的 1U 到 4U 等級的系統機架安裝, Cerebras 為 WSE-2 開發一套 15U 的超大機架系統 CS-2 ,當中就是一顆 WSE-2 ,使用高達 23kW 的電源,以 9+3 向電源設計為晶片進行供電,機架系統採用水冷設計,具備 12 個 100GbE 乙太網路。
雖然 WSE-2 是一顆特殊的 AI 級單晶片,不過為了能提供主流業界所需的環境,它們也針對當前業界主流提供相容性,透過編譯器可相容在 CS-I 或 GPU 執行的 AI 應用, Cerebra 預計在今年第三季開始出貨搭載 WSE-2 的 CS-2 系統。
17 則回應