GPU、TPU、RISC-V的区别与联系

GPU(Graphics Processing Unit,图形处理器),通俗来说就是电脑里负责图像显示和大规模并行计算的"超级大脑"。

TPU (Tensor Processing Unit),张量处理单元,专用AI加速芯片,是谷歌为支撑其庞大AI业务(搜索、翻译等)而研发的"核武器",经过近十年发展,已成为能与NVIDIA GPU抗衡的成熟技术。

RISC-V(读作"risk-five")是一个开源、免费且高度灵活的指令集架构(ISA)。你可以把它理解成一本公开的、设计处理器的"万能说明书"或"乐高图纸"。它定义了处理器最底层的"语言"和"规则",但本身不是一颗具体的芯片。

RISC-V与传统架构(如x86、ARM)相比,其核心优势在于:

  • 开源与免费:这是RISC-V最颠覆性的特点。任何人都可以自由地将其用于任何目的,设计、制造并销售RISC-V芯片和软件,无需支付高额的授权费和专利费。这极大地降低了芯片设计的门槛。

  • 简洁与高效:RISC-V属于精简指令集(RISC) 架构,其基础指令集(如RV32I)仅包含40多条核心指令。这种极简设计让指令解码和执行更高效,从而提升了处理器的性能与能效。

  • 模块化与可扩展:这是RISC-V设计的精髓。它像搭积木一样,由一个必须实现的基础整数指令集(如RV32I)和一系列可选的标准扩展组成。你可以根据需求,自由组合扩展指令集,比如:

    • M:整数乘除法扩展

    • A:原子操作扩展

    • F/D:单/双精度浮点扩展

    • C:压缩指令扩展

    • V:向量运算扩展

    这种灵活性让RISC-V能适用于从微小的嵌入式传感器到高性能服务器等各种场景

这三者最根本的区别在于:GPU和TPU是实实在在的"硬件芯片",而RISC-V是一本"设计芯片的说明书"

可以用一个简单的图表来概括:

一、GPU与RISC-V的结合

传统GPU巨头如英伟达、AMD长期使用闭源架构,而RISC-V的开源特性正为GPU领域带来新的可能性。

  • 用RISC-V核心取代GPU的通用计算部分:NVIDIA GPU中的CUDA CoreWarp Scheduler等通用计算部件,原理与CPU相近,理论上可以用RISC-V处理器来替代。同时,RISC-V的向量扩展(RVV) 与GPU的SIMT(单指令多线程) 执行模型在底层高度相似,很适合用于实现GPU的核心并行计算。

  • 作为GPU内部的控制与管理核心:早在2018年,英伟达就在其Turing架构的GPU中,用基于RISC-V的GSP(GPU系统处理器) 取代了自研的Falcon MCU,负责驱动程序功能卸载和GPU资源管理。如今,在更复杂的芯粒(Chiplet)架构GPU中,也会集成RISC-V核心作为控制单元,负责任务调度和动态负载均衡。

  • 催生新一代开源GPGPU项目:学术界和工业界涌现出多个基于RISC-V的开源GPGPU项目:

    • Vortex:来自佐治亚理工学院,是一个基于RISC-V的、支持OpenCL和OpenGL的开源GPGPU研究平台。

    • Ventus(承影):由清华大学开发,巧妙地利用RISC-V向量扩展(RVV)来"模拟"GPU的SIMT执行模型,实现GPGPU功能。

    • Think Silicon NEOX:业界首个基于RISC-V指令集的多核、多线程GPU架构IP。

    • 初创公司Bolt Graphics:正研发采用RISC-V与自研架构的GPU,宣称其路径追踪性能可达英伟达RTX 5090的2.5倍。

二、TPU与RISC-V的关系

RISC-V与TPU的关系,本质上是一个开放、通用的处理器架构,与一个专用、高效的AI加速芯片之间的互补与融合。

核心关系:RISC-V是TPU的"大脑"与"管家"

这是两者最直接、最重要的关系。在谷歌的TPU中,RISC-V架构的处理器核心扮演着至关重要的控制与管理角色。

  • 作为加速器控制单元(ACU):谷歌在其数据中心TPU中,选择了基于RISC-V架构的SiFive Intelligence X280处理器IP作为"加速器控制单元"。你可以把它理解为TPU的"大脑"或"管家"。

  • 具体职责:这个RISC-V核心不负责高强度的矩阵乘法(那是TPU中矩阵乘法单元MXU的工作),而是负责:

    • 命令下发与控制:接收并解析来自系统主机的指令,然后精确地控制MXU执行任务。

    • 任务卸载:处理那些不适合MXU加速的复杂管理代码和任务。

    • 数据搬运:高效地管理和调度数据,将其喂给MXU进行处理。

  • 选择RISC-V的原因:谷歌选择RISC-V核心,是因为它灵活、高性能、紧凑,并且能通过其VCIX(矢量协处理器接口扩展)总线,以仅10个时钟周期的极低延迟与TPU的MXU无缝协同工作。这比传统的PCIe方案延迟低得多。

谷歌的这一选择,也推动了RISC-V核心提供商SiFive的发展。TPU v5的微架构变化之一,就是引入了RISC-V核心

GPU、TPU 与 RISC-V 全方位对比

对比维度 GPU(图形处理器) TPU(张量处理器) RISC-V(指令集架构)
本质属性 物理硬件(处理器芯片) 物理硬件(处理器芯片) 抽象规范(ISA,即指令集架构)
通俗类比 万能工具箱,里面有扳手、锤子,啥都能修 专用豆浆机,只干磨豆浆(矩阵乘法),但效率极高 一套"制造工具的国家标准",厂家可按标准生产任意工具箱
设计目标 高吞吐量并行计算,擅长图形渲染和通用计算 极致高效的张量(矩阵)运算,专为神经网络设计 提供开源、免费、可定制的芯片设计基础
核心架构 SIMT(单指令多线程),拥有数千个小核心 脉动阵列(Systolic Array),数据像流水线一样流动计算 精简指令集(RISC),基础指令少,支持模块化扩展
代表产品 NVIDIA H100、AMD MI300X Google TPU v6(Trillium) SiFive 处理器、阿里玄铁系列
主要战场 AI训练、科学计算、3D游戏 Google云端的AI训练与推理(如搜索、翻译) 设计各类处理器(CPU、AI芯片、物联网MCU)
是否开源 闭源(NVIDIA/AMD专有) 闭源(Google内部自用) 开源、开放(指令集标准公开)

GPU vs TPU:谁干活更快?

两者都是"干活"的芯片,但分工不同:

  • GPU是"通才":它不仅要算AI,还得会画游戏画面、做物理模拟。它的核心很多,但每个核心功能比较通用。就像一个大食堂,什么菜都能做。

  • TPU是"专才":它几乎只会做一件事------矩阵乘法(AI训练的核心动作),但做这件事时,它的效率远超GPU。就像专门做拉面的师傅,一天能拉一万碗,但不会做炒菜。

结果:在Google内部,TPU训练特定模型的能效比通常比GPU高,但它不够灵活,换一套新算法可能就跑不满了。

GPU/TPU 与 RISC-V:谁是制造者,谁是制造标准?

GPU和TPU目前都不使用RISC-V指令集,但这三者可以发生关系:

  • 现状:NVIDIA的GPU用的是自研的CUDA架构(指令集叫PTX),Google的TPU也是自研的内部架构。它们都不基于RISC-V。

  • 未来关系(可能性):RISC-V是一本"菜谱"(指令集),芯片公司可以拿着这本菜谱去设计GPU或AI芯片。比如,如果Google愿意,它完全可以用RISC-V的向量扩展(RVV)指令集,去设计一个"基于RISC-V的TPU"。目前已经有开源项目(如Vortex)在做基于RISC-V的GPGPU。

  • GPU :一个通用并行计算猛将,什么AI活都能接,但功耗大。

  • TPU :一个极致偏科的矩阵计算怪兽,只干AI训练/推理这一件事,但干得特别快还省电。

  • RISC-V :一本公开免费、全球通用的处理器设计蓝皮书,它自己不干活,但谁拿着它都能造出GPU或AI芯片,且不会被卡脖子。