4. GPU概述

GPU(Graphics Processing Unit,图形处理器)最初为图形渲染而生,如今已成为AI训练与HPC(高性能计算)的核心算力引擎。理解GPU的架构演进、核心计算单元以及NVIDIA产品线,是掌握DPU/GPU/NVLink异构互联体系的基础。本章从架构、核心、应用与产品四个维度展开。

一、GPU架构演进:从固定管线到通用并行计算

GPU架构的演进大致可分为四个阶段。第一阶段是固定功能图形管线,硬件电路专用于顶点变换、光栅化与像素着色,不可编程。第二阶段是可编程着色器,顶点与像素着色器支持指令编程,但资源调度仍以图形任务为中心。第三阶段是统一着色器架构(Unified Shader),NVIDIA在2006年推出G80(Tesla架构),将顶点、几何、像素着色器统一为通用流处理器,首次引入CUDA,使GPU具备通用并行计算能力。第四阶段是面向AI的专用加速架构,从Volta开始引入Tensor Core,Ampere、Hopper、Blackwell持续强化低精度矩阵运算、稀疏计算与显存带宽。

架构演进的核心逻辑是:把更多晶体管从图形固定逻辑转移到可编程并行计算单元,并针对矩阵乘加等AI核心算子做专用加速。这一趋势直接决定了GPU在异构计算中的角色——从图形协处理器变为通用算力加速器。

二、CUDA核心与Tensor核心:两种计算单元的分工

CUDA核心(CUDA Core)是GPU的通用标量/向量浮点运算单元,每个核心每时钟周期可执行一次FMA(融合乘加)操作,擅长处理元素级运算、控制逻辑、非规则计算。一个SM(Streaming Multiprocessor)通常包含数十至上百个CUDA核心,配合寄存器堆、共享内存与调度器组成完整计算单元。

Tensor核心(Tensor Core)是专为矩阵乘加设计的脉动阵列单元,可在单条指令内完成4×4或更大矩阵的乘加运算,支持FP16、BF16、TF32、FP8、INT8等低精度格式。以Hopper架构为例,单个Tensor Core的矩阵吞吐量可达同代CUDA核心的数十倍。两者分工明确:CUDA核心负责通用逻辑与数据搬运,Tensor核心负责密集矩阵运算。

GPU内部计算单元分工与数据流 全局显存 HBM 高带宽数据存储 L2 缓存 数据复用与聚合 SM 流多处理器 调度与执行核心 CUDA Core 标量/向量运算 FMA · 控制逻辑 Tensor Core 矩阵乘加阵列 FP16/BF16/FP8 图形渲染任务 光栅化 · 着色 HPC 科学计算 FP64 · 仿真 AI 训练/推理 矩阵乘 · 卷积

三、GPU在AI/HPC中的作用

在AI领域,GPU承担了深度学习训练与推理的绝大部分算力。训练阶段,前向传播与反向传播涉及大量矩阵乘法、卷积和归一化操作,Tensor Core可将这些操作加速数十倍;推理阶段,GPU通过批处理与低精度量化实现高吞吐低延迟。在HPC领域,GPU用于分子动力学、气象模拟、流体仿真、金融建模等场景,CUDA核心提供FP64双精度算力,Tensor Core则用于混合精度迭代求解。

GPU在异构互联中的定位是算力密集型任务的加速器:CPU负责控制流与串行逻辑,GPU负责数据并行与矩阵密集计算,DPU负责网络、存储与安全卸载,三者通过NVLink、PCIe等高速链路协同。理解这一分工,才能理解后续NVLink与DPU章节的设计动机。

四、NVIDIA GPU产品线

NVIDIA GPU产品线按应用场景可分为数据中心、专业可视化、消费级与嵌入式四大类。数据中心产品是AI/HPC的主力,包括Tesla/V100/A100/H100/H200/B200等;专业可视化包括Quadro/RTX A系列;消费级为GeForce RTX系列;嵌入式为Jetson系列。下表列出数据中心主力架构的关键参数对比。

架构 代表型号 CUDA核心数 Tensor Core 显存类型 互联方式
Volta V100 5120 第1代 HBM2 32GB NVLink 2.0
Ampere A100 6912 第3代 HBM2e 80GB NVLink 3.0
Hopper H100 16896 第4代 HBM3 80GB NVLink 4.0
Blackwell B200 — 第5代 HBM3e 192GB NVLink 5.0

五、常见误区与复盘

误区一:CUDA核心越多性能越强。实际性能还取决于显存带宽、Tensor Core数量、SM调度效率与软件优化。A100的CUDA核心少于某些消费卡,但AI性能远超后者。

误区二:Tensor Core可替代CUDA核心。Tensor Core仅加速矩阵运算,非矩阵逻辑、数据搬运、控制流仍依赖CUDA核心,两者互补而非替代。

误区三:GPU可独立完成所有计算。GPU需要CPU进行任务调度与I/O管理,在异构系统中通过PCIe/NVLink与CPU、DPU协同,单卡无法独立构成完整计算系统。

复盘要点:架构演进的主线是“通用并行+专用加速”;CUDA核心管通用,Tensor Core管矩阵;GPU在AI/HPC中承担算力加速角色;产品线按场景分层,数据中心系列是互联课程的重点对象。掌握这些,才能理解后续NVLink为何成为GPU间互联的关键。

# 查看GPU型号与CUDA核心数(Linux)
nvidia-smi --query-gpu=name,compute_cap,memory.total --format=csv

# 典型输出
# name, compute_cap, memory.total [MiB]
# NVIDIA H100 PCIe, 9.0, 81559 MiB
# NVIDIA A100-SXM4-80GB, 8.0, 81920 MiB

综上,GPU已从图形专用处理器演变为AI/HPC的通用算力引擎。CUDA核心与Tensor核心的分工、产品线的场景分层,以及GPU在异构系统中的加速器定位,共同构成了理解DPU/GPU/NVLink互联体系的起点。