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

首先来了解这款AI算力芯片的研发背景。国内致力于AI专用算力芯片的企业并不多，而中昊芯英正是其中最早押注TPU架构的团队之一。近期，他们正式发布了新一代全自研高性能TPU芯片“须臾”，同步推出的还有搭载该芯片的软硬件一体化智算底座——泰则2.0 AI高性能智算平台。作为初代“刹那”芯片和泰则智算服务器的全面升级版本，新品在底层架构、峰值算力、片上存储、集群互联以及计算能效等关键维度上，均实现了跨越式提升。

具体性能参数方面：单颗“须臾”芯片深度优化了大模型专属的张量计算逻辑，不仅扩大了寄存器容量，还增大了片上缓存。单芯片混合精度浮点算力达到896TFLOPS，这一数值是上一代芯片“刹那”的3倍；8位整数推理算力高达1792TOPS，尤其适合海量词元高并发推理场景。功耗方面，单芯片额定功耗为600W，但值得注意的是，其算力已达到传统芯片的同等水平，而功耗却降低了50%。核心奥秘在于多维张量计算单元与数据复用优化设计，这使得它在执行相同AI任务时，综合计算效能可达传统GPU架构的数倍。

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

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

（受访单位供图）