记者从中国科学院计算技术研究所获悉,日前,我国科研团队在人工智能芯片领域取得重大突破——成功研发出新一代自主架构AI芯片,其峰值算力达到每秒520万亿次浮点运算(520 TFLOPS),全面超越当前国际主流同级别产品。这项成果标志着我国在AI芯片核心架构上实现了从“跟跑”到“领跑”的关键跨越。
算力跃升背后的架构革命
这款被命名为“天枢”系列的最新芯片,最引人瞩目的并非单纯算力数字的攀升,而是其内在架构的根本性变革。据介绍,研发团队摒弃了传统冯·诺依曼架构中“计算与存储分离”的设计思路,开创性地采用了“存算一体二维脉动阵列”架构。
传统芯片在执行AI计算时,大量时间浪费在数据在内存与处理器之间的搬运上,即所谓的“存储墙”瓶颈。而天枢芯片将计算单元与存储单元深度融合,让数据“就近计算”,大幅降低了数据迁移的延迟与能耗。同时,二维脉动阵列结构使得数据可以在芯片内部的多个处理单元间像波浪一样高效流动,实现高度并行计算。这种架构设计使芯片在520 TFLOPS的峰值算力下,能效比相比上一代提升了4倍,达到国际领先水平。
突破性技术细节
除了架构创新,天枢芯片在工艺与系统优化上同样亮点频频。该芯片采用7纳米先进制程,集成了超过500亿个晶体管。研发团队还为其量身定制了全新的指令集——基于“可重构数据流”的ISA,使得芯片能够根据AI模型的计算特点动态重组计算路径,大幅提升对不同类型神经网络(如卷积神经网络、Transformer等)的适配性。
在存储方面,芯片内部搭载了高带宽的HBM3e显存,总带宽超过3 TB/s,确保巨量参数在520 TFLOPS算力下不会被“饿死”。同时,针对大模型训练中常见的稀疏化计算场景,芯片设计了专用的稀疏计算引擎,可以将无效计算自动跳过,实际有效算力提升约30%。
从“能用”到“好用”的关键一步
中国工程院院士、计算机专家李教授评价:“天枢芯片的架构突破,是我国在高端AI芯片领域从‘追赶’到‘并跑’,再到部分‘领跑’的标志性事件。过去我们总在追赶英伟达的架构路线,如今中国团队提出了原创性方案,且实测性能在多个主流AI基准测试中排名前三。”
据悉,天枢芯片已在中科院计算所的“智能超算”平台上完成部署测试。在GPT-3规模的大语言模型训练中,其训练效率相比上一代国产芯片提升了6倍,功耗则下降了40%。与此同时,在自动驾驶领域的视觉感知模型推理任务中,其延迟控制在1毫秒以内,完全满足实时性要求。
产业应用前景广阔
随着AI大模型、自动驾驶、智慧医疗等场景对算力的需求呈指数级增长,天枢芯片的问世将为我国数字经济核心底座提供有力支撑。目前,研发团队已与国内多家头部云计算厂商、服务器厂商展开合作,预计明年将实现量产并推向市场。
“架构上的‘弯道超车’只是第一步,接下来我们还要在软件生态、工具链上持续投入,让更多开发者能够轻松地使用国产芯片做AI创新。”项目负责人表示,团队已同步发布了适配天枢芯片的编译器与深度学习框架补丁,兼容PyTorch、TensorFlow等主流框架。
从“卡脖子”到“杀手锏”,天枢芯片的架构突破不仅展现了我国科研人员的创新智慧,更预示着全球AI芯片竞争格局正在发生深刻变化。这一枚小小的芯片,正在为中国的智能未来注入澎湃算力。