日前Cerebras公司发布了全球最大的芯片WSE(Wafer Scale Engine),专注于AI运算,总计1.2万亿个晶体管,核心面积超过46225mm2,集成了40万个核心以及18GB SRAM缓存,带宽超过100Pb/s。
单独说这些参数可能没什么感觉,这个芯片到底有多大呢?要知道NVIDIA用于AI加速的GV100大核心集成了211亿晶体管,核心面积815mm2,所以WSE芯片晶体管数量是最强GPU芯片的60倍,面积则是它的56倍多。
如果与当前的CPU芯片相比,WSE可以说更震撼了,AMD的64核EPYC二代处理器才320亿晶体管,封装总面积也不过4410mm2,WSE光是核心面积就是它的10倍多。
WSE芯片采用了台积电的16nm工艺生产,集成了40万个AI运算核心,如此大规模的并行运算需要解决很多问题,包括核心核心之间的通讯及缓存,所以WSE配备了18GB SRAM缓存,同时带宽达到了100Pb/s,要知道通常的计算芯片SRAM缓存也不过几百兆,带宽也就Tb/ss级别,比如AMD的EPYC二代处理器L1/L2/L3缓存加起来不超过300MB,PCIe 4.0总线带宽不过128GB/s,NVIDIA的NVLink 2.0最大带宽也不过300GB/.s,算下来也就是2.4Tb/s,WSE的内部带宽是现有水平的3.3万倍之多。
总之,WSE芯片在工程技术上绝对是一次奇迹,不过官方并没有提及WSE芯片的具体性能,也没有提到量产及上市时间,这让它的前途也面临一丝怀疑,毕竟这样的芯片实在是太特殊了。