NPU有什么新架构?

wen IT资讯 4

本文目录导读:

NPU有什么新架构?

  1. 宏观架构层面的核心变革:从“计算中心”到“数据流中心”
  2. 微观计算单元的革新:从“MAC”到“多样算力”
  3. 具体芯片和架构实例(截至2025年5月)
  4. 未来的关键方向

这是一个很有深度的问题,NPU(神经网络处理单元)的架构正处于快速演进期,核心目标是在更低的功耗和延迟下,高效处理Transformer等大模型,同时兼顾传统CNN(卷积神经网络)任务。

主流的NPU新架构趋势可以从宏观架构设计和微观计算单元两个层面来看:

宏观架构层面的核心变革:从“计算中心”到“数据流中心”

传统的NPU(如早期手机SoC中的)主要针对CNN设计,采用脉动阵列(Systolic Array),但Transformer模型带来了巨大的挑战:

  1. 大模型与“内存墙”: 模型参数动辄数十亿,对片外带宽(DDR/HBM)和片上缓存(SRAM)提出了极高要求。
  2. 复杂的注意力机制: 需要灵活的非线性计算和数据搬运,脉动阵列的固定数据流模式效率不高。

新架构的宏观设计发生了几大变化:

  • 存内计算(In-Memory Computing, IMC):

    • 思想: 直接在存储单元(如SRAM、RRAM、MRAM)内完成矩阵乘法运算,而不是把数据搬来搬去。
    • 优势: 极大减少“冯·诺依曼瓶颈”导致的数据搬运功耗和时间。
    • 代表: 许多AI芯片初创公司(如国内的知存科技、国外的Mythic)都在采用此技术,三星的HBM-PIM也属于一种3D封装下的存内计算,但目前工艺成熟度和精度控制仍是挑战。
  • 近存计算(Near-Memory Computing):

    • 思想: 将计算单元(MAC阵列)和存储单元(SRAM)通过先进的3D堆叠技术(如TSV、混合键合)紧密耦合。
    • 优势: 比传统分离式设计带宽提升数十倍,功耗大幅下降,是目前更务实的解决方案。
    • 代表: AMD的XDNA(源于赛灵思) 架构、NVIDIA的Grace Hopper超芯片英特尔Meteor Lake的NPU都采用了类似思路。
  • 数据流架构(Dataflow Architecture):

    • 思想: 不再依赖于中央控制,而是由数据本身触发计算,计算单元按流水线方式排列,数据在单元间“流过”。
    • 优势: 硬件利用率高,非常适合Transformer这种层数深、结构相对固定的模型。
    • 代表: Google的TPU v1/v4、Groq的LPU(语言处理单元)、Cerebras的晶圆级芯片(WSE-2/3)、以及国内寒武纪的某些系列。

微观计算单元的革新:从“MAC”到“多样算力”

在具体的计算单元(PE,Processing Element)设计上,也开始变得多样化:

  1. Transformer专属引擎(Transformer Engine):

    • 核心: 对Attention机制中的矩阵乘法(QK.V)进行硬件加速,并引入FP8(8-bit浮点) 作为主要的训练和推理格式,通过硬件自动缩放精度,在保证精度的同时大幅提升计算速度。
    • 代表: NVIDIA的Hopper (H100) 和 Blackwell (B200) GPU上的Transformer Engine
  2. 可重构计算单元:

    • 核心: 单元内集成了乘法器、加法树、以及专用的非线性函数单元(如Softmax、LayerNorm、GELU的近似计算),这些单元可以根据不同的算子需求(卷积、全连接、注意力)动态配置。
    • 代表: 华为的达芬奇架构(昇腾系列)苹果的Neural Engine
  3. 稀疏计算支持:

    • 核心: 对权重或激活值的稀疏性(大量零值)进行加速,通过硬件检测并跳过零值计算,或者使用结构化稀疏。
    • 代表: NVIDIA的Ampere及之后的GPU高通Snapdragon的Adreno/NPU英特尔Gaudi系列

具体芯片和架构实例(截至2025年5月)

以下是当前市场上或学术界最具代表性的新架构:

架构/芯片 核心创新 适用场景 关键特点
NVIDIA Blackwell GPU Transformer Engine + 第二代稀疏 云端训练/推理 FP8/FP4支持;第五代NVLink (900GB/s);专为万亿参数模型设计。
AMD Ryzen AI (XDNA) 近存计算 + 自适应数据流 PC/笔记本端推理 基于赛灵思AI引擎,2000 TOPS(目前仅宣传,实际算力在46亿参数模型)。
英特尔 Meteor Lake/Lunar Lake NPU 可扩展的数据流架构 PC端推理 支持动态的INT8/FP16计算,低延迟,芯片级内存一致。
Google TPU v5e/v5p 大规模2D网格数据流 + 矩阵单元 云端推理/训练 针对Transformer优化,高效的大模型部署。
高通Snapdragon X Elite / 8 Gen (Hexagon NPU) 混合精度 + 张量加速器 + 向量加速器 移动端/笔记本端推理 独立的标量、向量、张量单元,支持INT8/FP16/BF16。
华为昇腾910B/910C (达芬奇架构V2) Cube + Vector + Scalar三维指令集 云端推理/训练 可重构计算单元,强调通用性和高性能。
Groq LPU 确定性数据流架构 云端推理 无缓存一致性开销,极低延迟,但DDR带宽是瓶颈。
Cerebras WSE-3 晶圆级芯片 + 片内全部SRAM 云端训练 不存在显存墙和通信墙,支持全精度。

未来的关键方向

  • Tiling & 3D封装: 成为主流,将计算、存储、I/O通过3D方式堆叠,打破带宽瓶颈。
  • 混合精度计算: 支持FP8、甚至FP4/INT2,用更低的精度换取更高的吞吐和效率。
  • 可编程性增强: 不再仅仅是固定的矩阵乘法器,而是支持Transformer、MoE(混合专家)、扩散模型、Mamba等新型架构的通用加速器。
  • 片上网络(NoC): 支持更复杂的片内数据通信,例如针对MoE的门控路由、专家间的灵活通信。

一句话总结: 新一代NPU不再是单纯的“矩阵乘法器”,而是“数据流导向 + 近存/存内计算 + 混合精度 + 可重构算子”的智能数据处理器。 苹果、高通、AMD、英特尔、华为等厂商都在朝这个方向全速前进。

抱歉,评论功能暂时关闭!