我国企业自主研发TPU芯片“须臾”发布

日前，中昊芯英正式发布新一代全自研高性能TPU AI专用算力芯片“须臾”，同步推出搭载该芯片的软硬件一体化智算底座——泰则2.0 AI高性能智算平台。作为初代“刹那”芯片与泰则智算服务器的全面升级产品，新品在底层架构、算力峰值、片上存储、集群互联、计算能效等方面均实现跨越式提升。

据介绍，单颗“须臾”芯片深度优化大模型专属张量计算逻辑，扩容寄存器与大容量片上缓存。单芯片混合精度浮点算力达896TFLOPS，性能是上一代芯片“刹那”的3倍；8-bit推理算力达1792TOPS，适配海量词元高并发推理场景。单芯片额定功耗600W，相较于算力性能持平的传统算力芯片功耗降低50%。依托多维张量计算单元与数据复用优化设计，其执行同等AI任务时综合计算效能可达传统GPU架构数倍。

“须臾”芯片延续中昊芯英全自研TPU技术路线，实现芯片IP核、专属指令集、底层算子加速库、整机系统软件完整自主研发。目前，中昊芯英核心技术已完整覆盖芯片设计、电路开发、编译工具、模型适配全链条。其同步推出的泰则2.0 AI高性能智算平台，最小计算单元由两路高性能CPU与8片“须臾”TPU协同构成，整机混合精度算力达7.168P，同等任务下整机能耗仅为传统GPU服务器的80%。

作为国内最早投身TPU架构AI专用算力芯片研发的企业之一，中昊芯英2023年成功流片国内首枚高性能TPU芯片“刹那”并实现量产。目前，公司产品已在深圳联通、天津移动、太极股份等建设的多个超大规模智算中心部署，辐射金融、传媒、教育、医疗等行业领域。

（受访单位供图）