本文是《大模型推理系统揭秘:从 vLLM 看 LLM Serving Infra 核心技术》系列的第 3 篇(共十五篇)。上一篇:如何衡量一个 LLM Serving 系统?;下一篇:Scheduler:GPU 这一轮到底给谁用?。
NOTE 本文基于 vLLM v0.27.1(tag
6e448d0, 2026-08-11)源码剖析。文中文件路径、类名和函数名均以该版本为准;vLLM 迭代很快,阅读时请以你手上的版本对照。
前两篇分别定义了问题(LLM Serving 为什么难)和尺子(用什么指标衡量)。在深入调度、KV Cache、GPU 执行等单个战场之前,需要先有一张全景图:一个请求进入 vLLM 之后,会经过哪些模块,以什么形式被传递,最后如何变成 token 返回给用户。
本篇的核心问题是:
一个请求如何穿过 vLLM 的整个推理系统——哪个模块负责决策、哪个模块负责执行,它们之间传递的到底是什么?1
一、总览:静态拓扑、动态生命周期与数据流
1. 三个视角看同一个系统
vLLM V1 的整体架构遵循控制面/数据面分离的经典设计哲学。本篇从三个视角鸟瞰它:先自顶向下看静态系统拓扑——API Server、AsyncLLM、EngineCore、Scheduler / KVCacheManager、Executor / Worker / ModelRunner 各自负责什么;再沿时间轴看一次请求的完整生命周期——从 HTTP 进入到 SSE 返回 [DONE] 的每一步交互;最后跟着数据走一遍,看文本如何变成 token ids、调度结果、GPU 张量、logits,再变回文本。
三个视角合起来回答的是模块之间的分工:EngineCore 驱动循环,Scheduler 决定这一轮谁跑、跑多少 token,Executor 负责分发,ModelRunner 负责真正调用 GPU 算。
2. 本文的章节安排
| 章 | 主题 | 内容 |
|---|---|---|
| 二 | 静态系统拓扑 | 自顶向下:API Server 与 AsyncLLM、EngineCore、Scheduler 与 KVCacheManager、Worker 与 Executor |
| 三 | 一次请求的完整生命周期 | 从 HTTP 请求到 SSE [DONE] 的时序图 |
| 四 | 数据流 | Token 如何穿过整个 Serving 栈 |
| 五 | 本文小结 | 模块分工,以及与第一篇“四问”的对应 |
| 六 | 自测 | 5 道题 |
二、静态系统拓扑(自顶向下)
vLLM V1 的整体架构遵循控制面/数据面分离的经典设计哲学。我们自顶向下,逐层解剖其系统拓扑。
%% 图:vLLM V1 的静态系统拓扑:API 层 → 引擎层 → 调度层 → 执行层,控制面与数据面分离
graph TB
subgraph "API Layer (控制面入口)"
Client[Client / OpenAI SDK]
API[API Server<br/>FastAPI + uvicorn]
end
subgraph "Engine Layer (异步引擎)"
AsyncLLM[AsyncLLM<br/>异步请求管理]
EngineCore[EngineCore<br/>核心调度循环]
end
subgraph "Scheduling Layer (调度层)"
Scheduler[Scheduler<br/>请求调度 + 资源管理]
KVCacheManager[KVCacheManager<br/>显存块管理]
BlockPool[BlockPool<br/>物理块池]
SOManager[StructuredOutputManager<br/>约束输出]
end
subgraph "Execution Layer (执行层)"
Executor[Executor<br/>分布式执行抽象]
Worker0[GPU Worker 0]
Worker1[GPU Worker 1]
WorkerN[GPU Worker N]
end
subgraph "Model Layer (模型层)"
MR0[ModelRunner<br/>模型前向 + 采样]
Attn[Attention Backend<br/>FlashAttn / FlashInfer]
KVCache[GPU KV Cache<br/>物理显存]
end
Client -->|HTTP/gRPC| API
API -->|add_request| AsyncLLM
AsyncLLM -->|EngineCoreRequest| EngineCore
EngineCore --> Scheduler
Scheduler --> KVCacheManager
KVCacheManager --> BlockPool
EngineCore --> SOManager
EngineCore -->|SchedulerOutput| Executor
Executor --> Worker0
Executor --> Worker1
Executor --> WorkerN
Worker0 --> MR0
MR0 --> Attn
Attn --> KVCache
MR0 -->|ModelRunnerOutput| EngineCore
从这张图看,从 Client 到 EngineCore 是请求入队路径,传递的是”生成什么”;EngineCore 到 Worker 再到 ModelRunner 是执行路径,携带的是这一轮调度出来的 batch 和 block 布局;而 ModelRunner 返回的只是采样结果,不是下一次 forward 的完整状态。调度状态留在 EngineCore,模型侧负责的是单次前向计算。
1. API Server 与 AsyncLLM:异步并发处理的桥头堡
入口位于 vllm/entrypoints/openai/api_server.py,基于 FastAPI 构建的 HTTP 服务器,实现了 OpenAI 兼容的 /v1/chat/completions、/v1/completions 等 REST API。
AsyncLLM(vllm/v1/engine/async_llm.py)是面向外部的异步引擎接口。它自己不做重活,而是把工作交给三个同目录下的协作者:
InputProcessor(input_processor.py):把文本 / 多模态输入 tokenize、校验采样参数,产出EngineCoreRequest;EngineCoreClient(core_client.py):与EngineCore通信的客户端。默认部署下EngineCore运行在独立进程中,两者通过 ZMQ + msgspec 收发EngineCoreRequest/EngineCoreOutputs(AsyncMPClient);单进程调试时用InprocClient。这条进程边界是 vLLM V1 让 HTTP 处理、tokenize / detokenize 与 GPU 调度循环互不阻塞的关键;OutputProcessor(output_processor.py):接收EngineCoreOutputs,做 detokenize、stop 条件检查、logprobs 整理,产出RequestOutput交给 API 层流式返回。
因此 AsyncLLM 的职责是:管理请求的异步生命周期、串起上述三者、支持 SSE 流式响应,以及数据并行(DP)场景下的多引擎协调。
2. EngineCore:推理系统的中央调度大脑
EngineCore(vllm/v1/engine/core.py)是 vLLM V1 引擎的核心,它的 step() 方法驱动整个推理循环:
# vllm/v1/engine/core.py (简化)
class EngineCore:
"""Inner loop of vLLM's Engine."""
def __init__(self, vllm_config, executor_class, ...):
# 1. 初始化模型执行器
self.model_executor = executor_class(vllm_config)
# 2. 初始化 KV Cache
kv_cache_config = self._initialize_kv_caches(vllm_config)
# 3. 初始化调度器
Scheduler = vllm_config.scheduler_config.get_scheduler_cls()
self.scheduler = Scheduler(
vllm_config=vllm_config,
kv_cache_config=kv_cache_config,
...
)
3. Scheduler 与 KVCacheManager:资源管理双核
Scheduler(vllm/v1/core/sched/scheduler.py,约 3000 行)是调度的大脑,决定每一轮迭代中哪些请求参与推理、分配多少 token 预算。
KVCacheManager(vllm/v1/core/kv_cache_manager.py)管理 GPU 显存上的 KV Cache 物理块——分配、释放、前缀缓存复用、驱逐。
4. Worker 与 Model Executor:模型执行的设备抽象
%% 图:Executor 与 Worker:一次 execute_model() 分发给多个 GPU Worker,各自的 ModelRunner 持有分片权重与 KV Cache
graph TB
EX["Executor<br/><i>执行抽象:单设备或多设备模型调用</i>"]
EX --> W0 & W1 & WN
subgraph W0["Worker 0 (GPU 0)"]
M0["ModelRunner"] --- S0["Model Weights(分片)<br/>KV Cache"]
end
subgraph W1["Worker 1 (GPU 1)"]
M1["ModelRunner"] --- S1["Model Weights(分片)<br/>KV Cache"]
end
subgraph WN["Worker N (GPU N)"]
MN["ModelRunner"] --- SN["Model Weights(分片)<br/>KV Cache"]
end
S0 <-.->|NCCL| S1
S1 <-.->|NCCL| SN
常见 GPU 部署中,Executor 会把一次 execute_model() 调用分发给一个或多个 Worker;Worker 再调用本地的 ModelRunner 执行模型前向、采样和 KV Cache 读写。单卡、同机多进程、Ray 分布式和 external launcher 的进程/设备映射并不完全相同,因此不应把“一个 GPU 固定绑定一个 Worker 进程”写成绝对规则。
每个 Worker 通常负责:
GPUModelRunner(vllm/v1/worker/gpu_model_runner.py):负责输入准备(_prepare_inputs())、模型前向(execute_model())、采样(sample_tokens())。本系列图中简写为 ModelRunner。v0.27.1 另有一份新实现vllm/v1/worker/gpu/model_runner.py,通过VLLM_USE_V2_MODEL_RUNNER=1启用,文中以默认实现为主线- 模型权重的分片(Tensor Parallel 下每卡持有一部分)
- KV Cache 物理显存
Executor(vllm/v1/executor/abstract.py)是模型执行抽象层,负责在一个设备或多个设备上执行模型,而不是 Scheduler 本身。v0.27.1 中可见的主要实现包括:
UniProcExecutor:单进程单卡MultiprocExecutor:多进程多卡(同机)RayDistributedExecutor:Ray 分布式(直接继承Executor)RayExecutorV2:Ray 分布式的新实现,注意它继承的是MultiprocExecutor而非Executor——即复用同一套 worker 进程管理逻辑,只把进程的拉起方式换成 Ray(RayWorkerProc(WorkerProc))ExecutorWithExternalLauncher:外部 launcher 场景(继承UniProcExecutor)
这条继承链本身就说明了一件事:“用不用 Ray”是部署方式的差异,不是执行模型的差异。 Executor 这层抽象的价值就在于把”进程怎么起、卡怎么分”和”一轮 batch 怎么执行”彻底分开,所以 V2 才能靠换掉进程拉起方式来复用同机多进程的全部逻辑。
5. 进程视角:三类进程与两道 IPC 边界
上面几节按”模块”切分,但真正决定谁会阻塞谁的是进程边界。默认的 vllm serve + MultiprocExecutor 部署下,一个请求要跨越三类进程,中间是两种完全不同的 IPC 机制:
%% 图:三类进程与两道 IPC 边界:API Server 与 EngineCoreProc 之间是 ZMQ,EngineCore 与 Worker 之间是共享内存消息队列
flowchart TB
subgraph PA["进程 A:API Server(uvicorn 事件循环 + AsyncLLM)"]
direction TB
HTTP["FastAPI 路由<br/>/v1/chat/completions"]
IPROC["InputProcessor<br/>tokenize → EngineCoreRequest"]
OPROC["OutputProcessor<br/>detokenize + stop 检查 → RequestOutput"]
MPC["AsyncMPClient<br/>input_socket / output_socket"]
end
subgraph PB["进程 B:EngineCoreProc"]
direction TB
INT["输入线程 process_input_sockets<br/>msgspec 反序列化 → input_queue"]
LOOP["主线程 run_busy_loop<br/>Scheduler.schedule() / update_from_output()"]
OUTT["输出线程 process_output_sockets<br/>output_queue → msgspec 序列化"]
EXEC["MultiprocExecutor<br/>rpc_broadcast_mq / worker_response_mq"]
end
subgraph PC["进程 C…:WorkerProc × (TP × PP)"]
direction TB
WK["Worker → GPUModelRunner<br/>execute_model() / sample_tokens()"]
GPU["GPU:模型权重分片 + KV Cache"]
end
HTTP --> IPROC --> MPC
MPC -- "ZMQ + msgspec<br/>EngineCoreRequest" --> INT --> LOOP
LOOP -- "SchedulerOutput<br/>共享内存 MessageQueue(广播)" --> EXEC --> WK --> GPU
WK -. "ModelRunnerOutput<br/>worker_response_mq(仅 output_rank 回传)" .-> EXEC
EXEC -.-> LOOP
LOOP --> OUTT -- "ZMQ + msgspec<br/>EngineCoreOutputs(只有 token ids)" --> MPC --> OPROC --> HTTP
classDef ipc fill:#fff4e0,stroke:#d9822b;
class MPC,INT,OUTT,EXEC ipc;
三点值得留意:
- A ↔ B 走 ZMQ socket + msgspec(
vllm/v1/engine/core_client.py→AsyncMPClient;vllm/v1/engine/core.py→EngineCoreProc)。跨越这条边界的是EngineCoreRequest(token ids 已经算好)和EngineCoreOutputs(只有新 token ids,没有文本)——所以 tokenize 和 detokenize 都留在进程 A,GPU 循环所在的进程 B 从不碰字符串。 - B ↔ C 走共享内存
MessageQueue(vllm/distributed/device_communicators/shm_broadcast.py)。SchedulerOutput经rpc_broadcast_mq一次广播给所有 Worker(TP 各 rank 需要同一份调度结果),而ModelRunnerOutput默认只由output_rank那一个 Worker 经自己的worker_response_mq回传,避免 N 份重复结果。 - 进程 B 内部又分三个线程:输入线程负责反序列化、输出线程负责序列化,主线程只跑
schedule → execute → update_from_output这个 busy loop。这样序列化开销不会插进调度循环的关键路径。UniProcExecutor时进程 C 退化为进程 B 内的一个对象,B ↔ C 边界消失,但 A ↔ B 依旧存在(InprocClient除外)。
三、一次请求的完整生命周期
%% 图:一次请求的完整生命周期:从 POST /v1/chat/completions 经 AsyncLLM、EngineCore、Scheduler、Executor 到 GPU,再流式返回
sequenceDiagram
participant C as Client
participant API as API Server
participant E as AsyncLLM
participant EC as EngineCore
participant S as Scheduler
participant KV as KVCacheManager
participant EX as Executor
participant W as GPU Worker
participant MR as ModelRunner
participant GPU as GPU
C->>API: POST /v1/chat/completions
API->>API: 参数解析 & 校验
API->>E: add_request(prompt, params)
E->>E: InputProcessor: tokenize → EngineCoreRequest
E->>EC: EngineCoreClient 经 ZMQ 发送(跨进程)
EC->>S: add_request(Request)
Note over S: Request.status = WAITING
rect rgb(230, 245, 255)
Note over EC,GPU: === 调度循环 step() ===
S->>S: schedule() — 选择可运行请求
S->>KV: get_computed_blocks() — 查 Prefix Cache
KV-->>S: 缓存命中块 + 未命中数
S->>KV: allocate_slots() — 分配新块
KV-->>S: KVCacheBlocks
Note over S: Request.status = RUNNING
S-->>EC: SchedulerOutput
end
EC->>EX: execute_model(SchedulerOutput)
EX->>W: forward pass
rect rgb(255, 245, 230)
Note over W,GPU: === Prefill 阶段 ===
W->>MR: execute_model(SchedulerOutput)
MR->>MR: _prepare_inputs(): input_ids / positions / slot_mapping
MR->>GPU: 模型 Forward (所有 prompt tokens)
GPU->>GPU: Attention 计算 + KV Cache 写入
GPU->>GPU: MLP 计算
GPU-->>MR: logits
W->>MR: sample_tokens()
MR->>MR: Sampling → 第 1 个 output token
end
MR-->>EC: ModelRunnerOutput
EC->>S: update_from_output()
EC-->>E: EngineCoreOutputs
E-->>API: 第 1 个 token (TTFT)
API-->>C: SSE: data: {"token": "Hello"}
loop Decode 循环 (每步 1 token)
rect rgb(245, 255, 230)
S->>S: schedule()
S->>KV: allocate_slots(1 new token)
EC->>EX: execute_model()
W->>MR: execute_model() → _prepare_inputs(1 token per request)
MR->>GPU: Forward (读历史 KV Cache + 计算新 token)
GPU-->>MR: logits
MR->>MR: Sampling → next token
MR-->>EC: ModelRunnerOutput
EC->>S: update_from_output()
end
EC-->>E: EngineCoreOutputs
E->>E: OutputProcessor: detokenize + stop 检查
E-->>API: RequestOutput (stream)
API-->>C: SSE: data: {"token": "..."}
end
Note over S: 遇到 stop token / max_tokens
Note over S: Request.status = FINISHED_STOPPED
S->>KV: free(request) — 释放所有块
API-->>C: SSE: data: [DONE]
1. 流式输出:一个 token 从 GPU 到客户端
上面的时序图把 EngineCore 和 AsyncLLM 之间画成了同步的请求-应答,实际上它们跨进程、各自有独立的循环。下面把 decode 循环中一个 token 的回程放大,重点看两件事:detokenize 发生在哪个进程,以及引擎循环为什么不需要等它。
%% 图:一个 token 的回程:ModelRunner 采样后 D2H,EngineCore 主线程立刻回到 schedule(),序列化与 detokenize 分别在输出线程与进程 A 完成
sequenceDiagram
participant MR as ModelRunner<br/>(Worker 进程, GPU)
participant EC as EngineCore 主线程<br/>(进程 B)
participant OT as EngineCore 输出线程<br/>(进程 B)
participant OH as AsyncLLM output_handler<br/>(进程 A)
participant GEN as generate() → SSE<br/>(进程 A)
Note over MR: 第 N 步 forward 结束, logits 已在 GPU
MR->>MR: _sample(): sampled_token_ids (GPU tensor)
MR->>MR: _bookkeeping_sync(): D2H 拷贝 → Python 嵌套 list
MR-->>EC: ModelRunnerOutput (仅 token ids, 经 worker_response_mq)
EC->>EC: Scheduler.update_from_output(): 追加 output_token_ids, 判 max_tokens / stop_token_ids
EC->>OT: output_queue.put(EngineCoreOutputs)
par 进程 B 继续下一步
EC->>EC: schedule() 第 N+1 步, 下发 execute_model
MR->>MR: 第 N+1 步 forward 已在 GPU 上运行
and 进程 A 处理第 N 步的 token
OT->>OH: ZMQ 发送 msgspec 序列化的 EngineCoreOutputs
OH->>OH: OutputProcessor.process_outputs(): IncrementalDetokenizer.update() → 文本增量
OH->>OH: 检查 stop 字符串, 命中则 abort_requests_async 通知进程 B
OH->>GEN: RequestOutputCollector.put(RequestOutput)
GEN->>GEN: await collector.get(), 序列化为 SSE chunk 写回客户端
end
Note over OH,GEN: 若消费端慢于生产端, Collector 会把多个 delta 合并成一个 RequestOutput
这张图解释了第二章第 5 节那两道进程边界的实际收益:进程 B 的主线程把 EngineCoreOutputs 丢进 output_queue 后立刻回到 schedule(),序列化由输出线程做,detokenize 和 stop 字符串检查由进程 A 的 output_handler 协程做,三者互不等待。代价是 stop 字符串(而非 stop token id)的判定要晚一步——进程 A 检测到后反向发 abort,GPU 可能已经为这个请求多算了一步。另外,RequestOutputCollector 在 DELTA 模式下会把积压的输出合并,因此客户端收到的一个 SSE chunk 不一定恰好对应一个 decode 步。
四、数据流:Token 如何穿过整个 Serving 栈
┌──────┐ ┌─────────┐ ┌────────┐ ┌───────────┐
│Client│────▶│API Server│────▶│AsyncLLM│────▶│EngineCore │
└──────┘ └─────────┘ └────────┘ └─────┬─────┘
"Hello, HTTP JSON add_request EngineCoreReq
tell me (text) (token_ids)
a joke" │
Tokenizer
[15496, 11, 2425,
757, 257, 9707]
│
┌──────▼──────┐
│ Scheduler │
│ schedule() │
└──────┬──────┘
SchedulerOutput
(scheduled_new_reqs[NewRequestData:
prompt_token_ids, block_ids, ...],
scheduled_cached_reqs[new_block_ids],
num_scheduled_tokens{req_id: n},
finished_req_ids, ...)
│
┌──────▼──────┐
│ Executor │
│ → Worker │
└──────┬──────┘
│
┌──────▼──────┐
│GPUModelRunner│
│_prepare_inputs│
└──────┬──────┘
input_ids, positions,
block_table, slot_mapping
│
┌──────▼──────┐
│ GPU │
│ Forward │
│ Pass │
└──────┬──────┘
logits [vocab_size]
│
┌──────▼──────┐
│ Sampling │
│ (top-p/top-k│
│ /temp) │
└──────┬──────┘
sampled_token_id
│
┌──────▼──────┐
│Detokenizer │
│ → "Sure" │
└──────┬──────┘
│
┌──────┐ ┌─────────┐ ┌──────▼──────┐
│Client│◀────│SSE Stream│◀───────────────────│ Response │
└──────┘ └─────────┘ └─────────────┘
"Sure, here's a joke..."
上图沿着数据走了一遍,但同一个请求在每一层其实是不同的对象——它们定义在不同文件、活在不同进程、寿命也不同。把这些对象排成一张表,就能看出”谁持有请求的真状态、谁只是一次性的快照”:
| 对象 | 定义位置 | 所在进程 | 生存期 | 它是什么 |
|---|---|---|---|---|
HTTP JSON → ChatCompletionRequest |
vllm/entrypoints/openai/chat_completion/protocol.py |
A(API Server) | 一次 HTTP 连接 | 文本 prompt + 采样参数(用户视角) |
EngineCoreRequest |
vllm/v1/engine/__init__.py |
A → B(msgspec 序列化过 ZMQ) | 只传一次 | 扁平的 IPC 载荷:prompt_token_ids、sampling_params、arrival_time… 已 tokenize,无状态 |
Request |
vllm/v1/request.py(from_engine_core_request) |
B(Scheduler 内部) | 从入队到 free() |
请求状态的唯一权威副本:status 状态机、num_computed_tokens、output_token_ids、block_hashes |
NewRequestData / CachedRequestData |
vllm/v1/core/sched/output.py |
B → C(共享内存广播) | 一个 step | SchedulerOutput 里的两种”订单”:首次调度的请求带全量 prompt_token_ids + block_ids;已在 batch 里的只带增量 new_token_ids + new_block_ids |
SchedulerOutput |
vllm/v1/core/sched/output.py |
B → C | 一个 step | 上面两者 + num_scheduled_tokens{req_id: n} + finished_req_ids 等:这一轮”谁跑、跑多少、块在哪” |
CachedRequestState + InputBatch 行 |
vllm/v1/worker/gpu_input_batch.py |
C(Worker 常驻) | 从首次调度到 finished_req_ids |
Worker 侧对 Request 的镜像,靠增量订单保持同步;InputBatch 是按行排列的持久 batch(token_ids_cpu、block_table) |
ModelRunnerOutput |
vllm/v1/outputs.py |
C → B(worker_response_mq) |
一个 step | req_ids + sampled_token_ids: list[list[int]] + logprobs:只有采样结果,不含任何状态 |
EngineCoreOutput(s) |
vllm/v1/engine/__init__.py |
B → A(ZMQ) | 一个 step | 每个请求的 new_token_ids + finish_reason;仍然只有 token ids |
RequestOutput |
vllm/outputs.py |
A(OutputProcessor 产出) |
一个 SSE chunk | 第一次出现文本:detokenize 后的增量 text + token_ids |
表里有两条规律。第一,状态只在 B 和 C 各有一份:Request 是权威,CachedRequestState 是靠每步增量同步的镜像;所有跨进程的载荷(EngineCoreRequest、SchedulerOutput、ModelRunnerOutput、EngineCoreOutputs)都是无状态的一次性消息,读完即弃,也因此可以随意序列化、走任何 IPC 通道。第二,文本只在进程 A 出现:从 EngineCoreRequest 到 EngineCoreOutputs 全程都是 token ids,进程 B、C 完全不需要 tokenizer。
五、本文小结
这一章主要要关注的是模块之间的分工:
EngineCore 驱动循环,Scheduler 决定这一轮谁跑、跑多少 token,Executor 负责把任务分发下去,ModelRunner 负责真正调用 GPU 算。
把它和第一篇的四问对齐,就得到全系列的骨架:
| 四问 | 承担模块 |
|---|---|
| 一、这一轮谁执行、执行多少 | Scheduler(+ EngineCore 驱动) |
| 二、状态放哪、怎么复用 | KVCacheManager / BlockPool |
| 三、怎么算得更快 | ModelRunner / Attention Backend / Kernel |
| 四、怎么扩出去 | Executor / Worker / 集合通信 |
📂 本章源码导航
入口与引擎循环
| 想看什么 | 从哪开始 |
|---|---|
| HTTP 入口、OpenAI 兼容接口 | vllm/entrypoints/openai/api_server.py(路由)、vllm/entrypoints/openai/chat_completion/(chat 接口实现) |
| 异步请求生命周期、流式响应 | vllm/v1/engine/async_llm.py |
| tokenize 与请求构造 | vllm/v1/engine/input_processor.py → InputProcessor |
| 与 EngineCore 进程通信 | vllm/v1/engine/core_client.py → EngineCoreClient / AsyncMPClient |
| detokenize 与输出整理 | vllm/v1/engine/output_processor.py → OutputProcessor |
| 推理主循环(建议从这里入手) | vllm/v1/engine/core.py → EngineCore.step() |
| 调度器给执行层的”订单” | vllm/v1/core/sched/output.py → SchedulerOutput / NewRequestData / CachedRequestData |
| 执行抽象与各种部署形态 | vllm/v1/executor/abstract.py |
| 一轮 batch 在 GPU 上怎么跑 | vllm/v1/worker/gpu_model_runner.py → GPUModelRunner.execute_model() / sample_tokens() |
六、自测
-
vLLM V1 里
AsyncLLM与EngineCore为什么分成两个进程?中间传什么?答案
把 tokenize / detokenize、HTTP、流式响应这些 CPU 密集且抖动大的工作从引擎循环里摘出去,让
EngineCore.step()稳定地驱动 GPU;中间经 ZMQ 传EngineCoreRequest(token id + 采样参数)与EngineCoreOutputs(新 token id),不传 tensor。 -
SchedulerOutput里有什么、没有什么?为什么 ModelRunner 能只凭它构造这一步的输入?答案
有:每个被调度请求的 id、这一轮的新 token 数、已计算 token 数、block table、采样参数变化、被抢占 / 完成的请求;没有:任何 tensor。ModelRunner 维护持久的
InputBatch,凭这些元数据更新位置与 slot mapping、从 KV 块里读,不需要 Scheduler 传数据。 -
一个请求的状态机有哪几个主要状态?从 WAITING 到 RUNNING 要过哪些检查?
答案
WAITING → RUNNING → FINISHED(或被抢占回 WAITING / PREEMPTED;结构化输出有 WAITING_FOR_STRUCTURED_OUTPUT_GRAMMAR);准入要 KV 块够(
KVCacheManager.allocate_slots)、token budget 够、max_num_seqs未满、(有 LoRA 时)max_loras未满。 -
TP=8 时 Executor 把一步分发给 8 个 Worker,谁做采样?logits 在哪?
答案
8 个 Worker 各算自己那 1/8 的权重,激活经 all-reduce 汇合;lm_head 的输出(logits)在所有 rank 都完整(或 gather 到 rank 0),采样在 rank 0(driver)做,其他 rank 的结果丢弃——所以采样相关的 CPU 工作只在一个进程上。
-
前端 detokenize 为什么放在
AsyncLLM而不是 EngineCore?流式输出时“一个 token 一次返回”有什么坑?答案
detokenize 是纯 CPU 且要维护每请求的字符串状态,放引擎里会拖慢 step;坑:BPE 的 token 边界与 UTF-8 字符边界不对齐,一个多字节字符可能跨两个 token,
IncrementalDetokenizer要缓存到能安全输出的完整字符再发。
下一篇
-
链路是固定的:入口(
api_server的 OpenAI 兼容路由)→AsyncLLM→InputProcessor(tokenize、构造EngineCoreRequest)→ 经EngineCoreClient跨进程送进 EngineCore。EngineCore 的step()是驱动循环:Scheduler 决策——这一轮哪些请求跑、每个推进多少 token,产出SchedulerOutput;Executor 分发到一个或多个 Worker;ModelRunner 执行——准备InputBatch、调 attention backend 与模型前向、采样,产出ModelRunnerOutput;EngineCore 把结果交回 Scheduler 更新请求状态,并把EngineCoreOutputs送回前端 detokenize、流式返回。传递的是什么:Scheduler 与 ModelRunner 之间传的不是 tensor 而是元数据——每个请求这一轮的 token 数、block table、slot mapping、采样参数;Worker 之间传的是激活(TP 的 all-reduce、PP 的 P2P);前后端之间传的是 token id。详见第二至四章。 ↩
系列:大模型推理系统揭秘:从 vLLM 看 LLM Serving Infra 核心技术
- 为什么 LLM Serving 比传统 DL 推理难?
- 如何衡量一个 LLM Serving 系统?
- 鸟瞰 vLLM:一个请求如何穿过整个推理系统?
- Scheduler:GPU 这一轮到底给谁用?
- KV Cache:LLM Serving 的第一号内存问题
- GPU 执行:如何让每个 Token 算得更快?
- 解码的扩展:采样、投机解码与结构化输出
- Multi-GPU:一张卡不够时如何扩展?
- 模型适配:如何跟上变化极快的模型世界?
- 请求形态的扩展:multi-LoRA 与多模态
- 硬件解耦:如何不让芯片差异污染 Serving 核心?
- PD 分离:从资源混部走向计算解耦
- Serving Infra 的下一站:从模型执行器到分布式智能操作系统
- 回到源码:一次请求在 vLLM 内部的真实旅程
- 系列总结与通关自测
本文由 arganzheng 创作,采用 CC BY 4.0 许可协议。在保留原文作者、署名以及完整原文链接(https://arganzheng.life/vllm-request-lifecycle-overview.html)的前提下,欢迎各种形式的转载、翻译或商业引用。
COMMENTS
评论存放在 GitHub Discussions, 用 GitHub 账号登录即可发表,支持 Markdown。 想针对正文某句话说?选中那段文字,点浮出的「评论」即可划线评论;觉得哪里写错了,发表时勾上「同时提交 Issue」。 有人回复你时 GitHub 会按你的通知设置发邮件,不用守在这里。