1.0 技术全景:从 GPU 到 Agent
理解大模型系统,需要同时回答两个问题:一次计算怎样执行,一项任务怎样完成。前者涉及模型、算子和硬件;后者还需要数据、服务、工具与治理。图 1-0 将这些技术放在同一张地图中,后文再逐层展开。
1.0.1 先看层次,再看两条工作路径

图 1-0:从应用需求向下追踪依赖。纵向箭头表示“依赖”,不是一次请求的执行顺序;横向箭头表示训练制品交给推理系统加载。基础设施框内表示组织关系,不是软件调用链。绘图源文件
计算与框架:CUDA/Triton · 融合/CUDA Graph · NCCL · 模型计算 · 分布式训练框架 · vLLM · SGLang · TensorRT-LLM。
服务与应用:服务治理 · RAG/工作流 · Agent 工具调用。优化技术与讲解深度见下表。
自下而上看能力。 GPU 执行计算,显存保存权重与中间状态;服务器组合计算、主机内存和输入输出设备;集群把多台服务器组织成可调度的资源;数据中心提供机房、供电和散热。集群是逻辑资源组织,数据中心是物理设施,二者并非一一对应。软件在这些资源之上实现算子、模型执行、在线服务与应用。
沿两条路径看工作。 训练读取样本,通过前向、损失、反向和参数更新产生模型。推理加载模型,先由 Prefill 处理整段输入并预测首个新词元,再由 Decode 逐轮处理已生成的词元、预测下一个。两侧共享算子与硬件,但管理的状态不同:训练还需梯度、优化器与恢复状态,推理重点管理请求队列和 KV 缓存。RLHF 等在线后训练会反复调用生成服务,图中的两条路径并不隔绝。
区分模型与框架。 Transformer 定义计算结构,PyTorch 等框架组织计算,vLLM 等引擎组织高效生成,Agent 宿主组织工具与任务。它们不是相互替代的产品。图按主要职责定位,实际软件可跨层:例如 vLLM 同时提供 API 服务入口、调度与模型执行,不能理解成只包含一个方框。其实现分工可对照 vLLM 架构说明。
1.0.2 图中的技术,到哪里学
下表按职责指向正文。概念简介解释用途与边界,机制详解展开计算与设计,实践配方说明输入、执行、输出与排障;三者不等同于完整的框架使用手册。
预训练与分布式执行
数据流水线、损失与优化器决定学什么;DDP、FSDP/ZeRO、张量/流水线并行决定如何分摊计算与状态
机制详解:5.5 数据、第六章 优化、第七章 分布式;实践配方:DDP/FSDP2、Megatron-Core/DeepSpeed
推理执行
GQA/MQA 减少 KV 头数,KV Cache 避免重复计算;FlashAttention 减少注意力访存;FP8 等量化降低表示开销;投机解码尝试减少串行等待
推理引擎
PagedAttention 管理分页 KV 的访问,Continuous Batching 动态组织请求;vLLM、SGLang、TensorRT-LLM 集成执行机制
PyTorch、FSDP、DeepSpeed、Megatron 和 TRL 的职责及组合方式集中在 A.7,不要求每个项目都使用整套工具。先选需要的机制,再选能支持该机制、模型与硬件的实现。
1.0.3 用一个问题串起整张地图
假设用户询问“公司的差旅住宿上限是多少”。模型已在服务启动时加载,本次请求按以下顺序执行:
应用整理任务。 RAG 检索当前用户有权读取的制度片段,将来源与问题放入上下文;固定的检索问答流程不一定需要 Agent。
服务接收请求。 网关校验身份和配额,路由到可用模型实例。这里处理的是请求,不是矩阵。
Prefill:处理整段输入。 引擎安排批次与 KV 空间,将问题和制度片段送入模型;模型计算各层表示、建立 KV 缓存,利用最后位置的输出预测首个新词元。
Decode:继续生成。 每轮输入上一轮生成的词元,读取并更新 KV 缓存,完成模型前向并预测下一个词元;结果可立即流式返回,不必等待完整答案。重复直到满足停止条件,随后回收请求状态;可复用的前缀缓存可按引擎策略保留。
应用检查并交付。 应用展示答案与引用;若需要执行报销等动作,则另行校验参数、权限与授权,不能把模型生成的调用当作执行许可。
步骤 3、4 每次都包含同一条计算链:模型张量运算 → GPU 内核与必要的跨卡通信 → 计算结果。GPU 从显存读取权重与状态,网络在跨节点计算时传输数据;这些工作发生在 Prefill/Decode 内部,不是生成结束后的另两个阶段。输入与输出矩阵的维度、具体计算和 KV 缓存更新见 3.8 节。
模型权重在上线前由训练路径产生;这次问答通常不更新权重。检索到新制度、复用 KV 缓存和把知识训练进模型,是三个不同过程。
排障也沿这张地图定位:答案依据错误,先检查检索与生成;首词元慢,拆开排队与 Prefill;逐词元输出慢,检查 Decode、访存和通信;扩卡不增吞吐,检查切分与互连。不要用同一种优化解决所有层的问题,具体判据见 11.13 节。
最后更新于
