1. 为什么要把这两个类彻底吃透
做过大模型推理服务的人应该都有感受,真正决定一个推理引擎在高压并发下是“稳如老狗”还是“频繁崩给你看”的,往往不是模型本身,而是请求调度和执行器之间那层数据契约。vLLM之所以能把吞吐做到高一个量级,核心不在某个花哨的算子,而在于它把“一次请求”抽象成了SequenceGroup,又用SequenceGroupMetadata在每个调度流水节拍里锁定执行快照。这两个类,一个管全局生命周期,一个管单步执行输入,是整个动态批处理、KV缓存管理、抢占恢复、前缀命中这些能力的基石。
这篇文章我打算从设计者的角度拆开它们:字段为什么这样定义、方法为什么这样组织、调度器在执行器之间传递时哪些坑是只有踩过才知道的。适合正在看vLLM源码但被各种数据结构绕晕的读者,也适合要在自己引擎里实现类似请求调度的人做参考。我会尽量还原真实的代码路径、调度语义和踩坑过程,而不是停留在类图层面讲概念。
先说个结论:SequenceGroup和SequenceGroupMetadata不是简单的“一个是封装对象,一个是传输对象”的关系。SequenceGroup代表请求的逻辑聚合,贯穿整个请求从进入到结束的全生命周期;SequenceGroupMetadata代表这个聚合在“某一个Scheduler Step”里的物理执行视角,是给模型执行器用的不可变快照。理解清楚这两种视角的切换,后面看再多的调度代码都顺了。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. SequenceGroup:请求的聚合单位
2.1 整体定位与核心字段拆解
先看SequenceGroup在vLLM里到底承担什么角色。一次用户请求,比如一条Chat Completion请求,进入LLMEngine后会被转换成一个SequenceGroup。如果采样参数里没有启用beam search,那这个组里一般只有一个Sequence;如果启用了beam search(比如n个beam),那一个SequenceGroup会包含多个Sequence,它们共享同一份prompt、同一组采样参数,但各自沿着不同的生成路径前进。
核心字段我按自己的理解重新组织一下,比直接贴源码更好记:
- request_id:引擎全局唯一,用来在Executor、Scheduler、Metrics各层之间唯一定位这组请求。
- seqs:Sequence列表,是整个组里最核心的可变状态集合。每个Sequence维护自己的token序列、生成长度、logprobs、block table。
- sampling_params:采样的参数集合,temperature、top_p、max_tokens、stop词等。注意,虽然组内可能有多条Sequence,但采样参数一定是共享的,否则beam search的路径比较就没有意义。
- arrival_time:请求到达时间,FCFS调度最基本的排序依据。vLLM的waiting队列就是按arrival_time排序的,谁来得早谁先被调度。
- metrics:请求级别的指标对象,包含调度延迟、推理延迟、token吞吐等统计。这个字段很多人忽略,但线上排查问题全靠它。
- state:当前调度状态,枚举值一般是WAITING、RUNNING、SWAPPED。这个状态机是整个调度器工作的基础。
- lora_request / prompt_adapter_request:如果开了LoRA或prompt adapter,在组里保存对应的适配器元数据,执行阶段要带着它去查对应的权重。
- encoder_seq:这是给encoder-decoder模型(比如T5、Bart)用的。如果模型是纯decoder-only,这个字段就是None。有了它,SequenceGroup要同时管理encoder侧的KV cache和decoder侧的KV cache,复杂度直接上一个台阶。
这些字段组合在一起,构成了调度器判断“这个请求现在能不能执行、该以什么优先级执行”的全部输入。所以我一直认为,理解了SequenceGroup,你就能理解vLLM的调度器为什么要写成那样。
2.2 为什么一个请求要包多个Sequence
刚开始接触vLLM的人都会有个自然的疑问:OpenAI的接口一次请求不就返回一个结果吗?为什么SequenceGroup里要放一个List[Sequence]?如果只是单序列解码,序列列表长度为1就可以了,为什么不直接用Sequence?
关键在于,一次“逻辑请求”的边界和一次“前向解码”的边界不是一回事。以beam search为例,一次请求可能同时探索5条beam路径,这5条路径各自维护独立的token历史、独立的KV cache block table,但它们共享同一份prompt、同一个request_id、同一套metrics。如果调度器按Sequence来调度,就无法保证同一次请求的多个beam在一个step里同步前进,甚至可能出现某些beam被抢先执行、某些beam还在排队的撕裂状态。SequenceGroup把多个Sequence包在一起,就是要保证调度的原子性:要么这一组请求一起被选中执行,要么都不选。
另外,采样时在batch维度上,不同Sequence的输入长度往往不一致,执行器需要构造一个padding后的张量。如果只按Sequence管理,batch里可能混入“同一个请求的beam”和“不同请求的seq”,采样结果回写时非常容易错乱。有了SequenceGroup这层聚合,执行器至少能明确“哪些seq属于同一组”,即使后续做tensor并行、流水线并行,也能通过seq_group_id追溯归属。
这种聚合思想其实和数据库里的“事务”很像。事务保证多行操作要么全部成功要么全部回滚,SequenceGroup保证一个请求的多条生成路径要么一起被调度要么一起被换出。只是数据库事务强调原子性的一致性,这里强调的是调度的整体性。
2.3 调度状态机如何体现在SequenceGroup上
vLLM的调度器不像很多人想的那样有复杂的优先级队列。它本质上就是把SequenceGroup在WAITING、RUNNING、SWAPPED三个状态里搬来搬去,每个step都重新审视一遍。
请求刚进来时进入WAITING状态,等待被调度为prefill。一旦Scheduler决定执行prefill,就把它在状态迁移到RUNNING,同时分配KV cache block。这个迁移不是一次性完成的,尤其开了chunked prefill之后,一个很大的prompt会被拆成多个chunk,每个step只处理一部分prefill,此时SequenceGroup虽然已经在RUNNING,但它内部维护的“已处理token数”要不停推进,直到整个prompt处理完才进入正常的decode循环。
当显存里的KV cache block不足时,调度器要从RUNNING组里挑受害者换出到SWAPPED。换出的单位依然是SequenceGroup,意味着组内所有Sequence的block table都要被拷贝到CPU内存,并释放GPU上的物理块。换入时再按block table从CPU拷回GPU。这里如果没有SequenceGroup作为原子单位,你根本没法保证换出后还能完整恢复该请求的生成上下文。
我在自己调试多路请求并发时遇到过一个问题:某个组内部因为beam search有多个seq,换出时只回写了其中一个seq的block table,另外几个还留在GPU上。结果重新换入时,解码出来的token序列错位,后面整条生成路径都崩了。后来才发现是自己在改调度代码时把SequenceGroup的原子性破坏了。这个教训让我彻底理解了为什么调度器里所有涉及状态迁移的操作,都是以SequenceGroup为最小单位的。
3. SequenceGroupMetadata:一次调度步骤的“快照”
3.1 它到底承担什么职责
SequenceGroupMetadata和SequenceGroup的最大区别在于:它不是一个长期存活的对象,而是Scheduler在每次step里为每一个被选中执行的SequenceGroup生成的“一次性快照”,传给ModelRunner后用完即弃。
这个设计背后有一个非常重要的思考:Scheduler在调度时操作的是动态变化的数据结构——Sequence的token列表在增长、block table在不断追加、KV cache的物理块可能被重新分配。而ModelRunner要构造的输入张量(input_tokens、input_positions、attention metadata)需要的是某个固定时刻的一致性视图。如果直接把SequenceGroup引用传给执行器,那么执行器在读取字段的间隙,Scheduler可能已经把状态改了,比如某个block刚被换出,或者某个Sequence刚完成了新的token追加。这种竞态条件在单线程里也许不明显,但在异步Executor或者多线程流水线里就是致命的随机崩溃来源。
所以SequenceGroupMetadata本质上是一个序列化后的只读数据包。它打包了执行器一次forward所需的全部输入信息:这个请求的每个Sequence当前的token数据、块表、采样参数、是否prefill、chunk大小等。执行器只依赖这个数据包组织计算,不回改SequenceGroup自身状态。
3.2 关键字段逐个拆
我见过不少人在阅读源码时,被SequenceGroupMetadata的字段搞得晕头转向。按我的理解,字段可以分成几类。
第一类是请求身份类:
- request_id:链路追踪的锚点,所有日志、指标、采样输出最终都靠它关联回SequenceGroup。
第二类是阶段标识类:
- is_prompt:表示这一step是prefill还是decode。prefill阶段,模型的输入是整段prompt(或chunk);decode阶段,输入是最后一个生成的token(以及可能附带的上一条token)。
- token_chunk_size:配合chunked prefill使用,表示当前step实际处理的token数量。对decode而言,这个值一般是1。这个字段非常重要,因为模型执行器计算attention mask和位置编码时,都取决于这个值。
第三类是执行数据类:
- seq_data:一个字典,key是seq_id,value是SequenceData快照对象。SequenceData里封装了当前Sequence的token id列表、输出logprobs、累计生成长度等。注意这个快照不是整个Sequence对象本身,而是它当前的一瞬间数据。在v1版本里,这部分被重构为TokenPool,但v0里seq_data就是执行器的输入金标准。
- block_tables:一个字典,key是seq_id,value是物理block id的列表。这是执行器做attention时查找KV cache位置的依据。如果这组请求有前缀命中,block table中可能有一部分是公用的,通过computed_block_nums标记出来。
- sampling_params:采样参数。可能有人疑惑,为什么每个step都要带上它?因为采样器在输出阶段需要根据参数决定怎么从logits里挑token。即使在decode阶段,参数没有变化,带上它也是为了执行器无状态化,不必回查SequenceGroup。
- multi_modal_data:多模态输入数据,图片、音频等。如果请求带了图像,这个字段就是非空的。执行器在prefill阶段需要把视觉特征拼接到文本embedding里,所以这个字段的传递非常重要。
- sliding_window:滑动窗口attention的窗口大小。设置了这个字段后,attention计算只保留最近N个token,metadata里必须把这个参数同步过来,否则执行器不知道如何截断。
- computed_block_nums:这个字段是前缀缓存命中的关键。它表示当前seq已经计算过并缓存在KV cache中的block编号。调度器把这个列表传给执行器,执行器就知道这次forward不需要重复计算这些block对应的前缀token,只需计算新增部分。在块复用场景下,这个字段的命令战斗价值极高。
3.3 Metadata是对SequenceGroup的复制而非引用
这是我想重点强调的设计取舍。也许你会觉得,每次step都复制一份seq_data、block_tables,不会影响性能吗?确实会有一点拷贝开销,但换来的是模块之间的解耦。vLLM在工程上把Scheduler和ModelRunner放在不同的内聚模块里,Scheduler负责策略,ModelRunner负责执行。如果两者共享同一个可变对象,一旦未来要做多线程、多流或分布式执行器,这个共享对象将成为最大的竞态温床。
更重要的是,SequenceGroup在Scheduler的视角里是会“变”的:decode之后,Sequence的token列表要追加新token;采样结果要回写logprobs;KV cache manager要更新block引用计数。如果在构造metadata时直接持有引用,执行器看到的可能是“已经追加了token但仍标着is_prompt”的混合状态,逻辑根本没法收敛。所以构造metadata时复制当前所需数据,是空间换确定性的经典做法。
这和我们做前端状态管理时用immutable快照的思路是一样的。你永远不会让UI组件直接改全局store里的对象,而是派发一个不可变的新状态。SequenceGroupMetadata就是调度器和执行器之间的那个不可变state。
4. 调度器与执行器之间的协作流程
4.1 代码路径总览
不看代码路径只记字段,理解始终是虚的。我梳理一条从Scheduler产生请求到ModelRunner消费metadata的典型路径,拿伪代码表示,你对照源码看效果更好。
python复制# scheduler.py 简化示意
def schedule(self) -> SchedulerOutput:
running: List[SequenceGroup] = []
swapped: List[SequenceGroup] = []
preempted: List[SequenceGroup] = []
# 1. 先处理running队列的decode请求
while self.running:
seq_group = self.running[0]
if not self._can_allocate(seq_group):
break
self.running.popleft()
running.append(seq_group)
self._allocate(seq_group)
# 2. 处理waiting队列的prefill请求
while self.waiting:
seq_group = self.waiting[0]
if not self._can_allocate(seq_group):
break
self.waiting.popleft()
running.append(seq_group)
self._allocate(seq_group)
# 3. 为每个被选中的组构造metadata
seq_group_metadata_list: List[SequenceGroupMetadata] = []
for seq_group in running:
md = self._create_metadata(seq_group)
seq_group_metadata_list.append(md)
return SchedulerOutput(seq_group_metadata_list)
_create_metadata这一步就是SequenceGroup到SequenceGroupMetadata的转换点。如果你的代码改到这里,想增加一个自定义输入数据传递给模型,你应该改的就是_create_metadata,而不应该去改SequenceGroup的接口。
4.2 prefill阶段的metadata装配细节
prefill阶段是metadata最复杂的场景。因为此时请求刚被调度,KV cache还没写入,block table可能是空的;如果开了chunked prefill,又只分配了一部分block,那block table只覆盖当前chunk可能用到的块。
prefill的metadata装配大致逻辑如下:
python复制def _create_metadata(self, seq_group) -> SequenceGroupMetadata:
is_prompt = seq_group.is_prefill()
seq_data = {}
block_tables = {}
for seq in seq_group.get_seqs():
seq_data[seq.seq_id] = seq.get_data()
block_tables[seq.seq_id] = self.block_manager.get_block_table(seq)
computed_block_nums = self.block_manager.get_computed_block_nums(seq_group)
return SequenceGroupMetadata(
request_id=seq_group.request_id,
is_prompt=is_prompt,
seq_data=seq_data,
block_tables=block_tables,
sampling_params=seq_group.sampling_params,
token_chunk_size=seq_group.get_token_chunk_size(),
computed_block_nums=computed_block_nums,
)
这里有个细节容易踩坑:当is_prompt为True且token_chunk_size等于完整prompt长度时,seq_data中每个Sequence的prompt token会全部参与前向计算。但当chunked prefill启用时,SequenceGroup内部维护了“当前chunk的起始位置”,metadata里的seq_data看起来是完整的token列表,但实际参与计算的只有最后一个chunk的token,其余token的KV cache是之前chunk计算出来的,直接从block table里取。所以你在调试时要时刻提醒自己:token列表完整不代表这一step要重新计算完整列表,起点要看position编码和computed_block_nums。
4.3 decode阶段的metadata有什么不同
decode阶段是模型每步只生成一个token的自回归过程。此时metadata的计算量比prefill小很多,但有几个特殊性需要处理。
第一,is_prompt恒为False,token_chunk_size恒为1。执行器看到这个标志后,不会再对整段prompt做attention mask的全量构造,而只计算新token对已有KV cache的增量attention。
第二,seq_data里的token列表包含历史所有token,但执行器只会取列表末尾的一个token作为当前输入。这个逻辑在prepare_input_tensors里体现为对input_tokens做长度为1的截取。所以metadata里seq_data传输的虽然是全量,但消费方式是有选择性的。
第三,decode阶段的block_tables已经完整分配。每次追加一个token,调度器会从block manager申请一个slot;如果当前block满了,就申请新block并追加到block table末尾。所以decode阶段的block table是逐步增长的。执行器读取block table时要容忍不同seq的长度不一致,attention计算时要用各自的长度做mask。
我记得自己在第一次阅读decode阶段的构造代码时,总想找到一个“当前生成了几个token”的字段,后来发现根本不用找。因为执行器算位置索引时用的是len(seq_data)减去prefill长度,但这个计算也在executor里做,metadata本身不存这些派生的值。这也是一个设计原则:metadata只存原始事实,不存任何可由原始事实推导出的中间量,避免两个来源不一致的bug。
4.4 chunked prefill引入后的变化
chunked prefill是vLLM吞吐优化的杀手锏,也是让SequenceGroupMetadata设计复杂化的主要推手。在没开chunked prefill时,一个请求的prefill要么整个执行,要么排队等待,不允许被打断。这导致长prompt请求会霸占GPU很久,后续短请求只能干等。开启chunked prefill后,长prompt可以被切成一堆chunk,每个step只处理一个chunk,中间插缝执行decode请求,从而让计算资源被填满。
这个优化反映到metadata上就是token_chunk_size字段。调度器在构造metadata时,会根据当前step该请求还能用的预算,决定这次给它几个chunk。比如一个prompt总长4096个token,chunk大小设为512,那么它需要8个prefill step才能把prompt处理完。前7个step,metadata里的is_prompt=True,token_chunk_size=512,computed_block_nums记录了前N个chunk的块;最后一个step,token_chunk_size可能是剩余不足512的实际长度,处理完成后才切换为decode。
这个机制对KV cache block的分配顺序也带来了改变。在没有chunked prefill时,prefill前一次性把整个prompt的block都分配好;开启后,block要按chunk粒度逐个分配、逐步增长block table。所以metadata里看到的block_tables可能是“当前已计算的块”而不包含尚未计算的块。这种“先算到哪儿,才分配到哪儿”的策略,让显存利用率更高,但也意味着你在自定义执行器时,绝不能假设一个prefill请求的block table在第一个step就是完整的。
4.5 从旧版到新版的设计演进
我最早看的vLLM版本里,SequenceGroup还持有block_manager引用,每个Sequence自己维护自己的block table,metadata构造时直接从Sequence对象上扒数据。随着版本演进,尤其是v1重构之后,SequenceGroup和Executor之间的耦合被进一步削弱,metadata的构造越来越像一种纯函数转换。
演进的核心方向可以总结为三点。第一,把Sequence内部的block table管理职责彻底剥离,交给中央化的BlockManager或BlockPool,避免在Seq对象里散落一大堆显存管理逻辑。第二,把采样参数、LoRA信息等原本散落在metadata里的各种可选参数统一收拢,减少if判断分支。第三,引入TokenPool这类内存池化的结构来替代每次step构造seq_data的临时分配,降低高频调度下的拷贝开销。
我个人的观点是,理解旧版的SequenceGroupMetadata能帮你更好地读懂新版的演进逻辑。因为新版只是把“快照”做得更彻底、更高效,并没有推翻“请求聚合于SequenceGroup、单步执行快照于Metadata”这套基本范式。
5. 实现中的关键难点与盘点
5.1 Metadata中的顺序稳定性问题
这是我自己踩过最隐蔽的坑。SequenceGroupMetadata传入ModelRunner后,执行器要遍历这个列表,按顺序构造batch张量。如果metadata列表的顺序和采样回写的顺序不一致,那么输出logits和seq_id的对映就会错位。特别在并行采样、beam search场景下,一个组内有多个seq,错位后生成的文本会被写入到错误的Sequence里,表面上看代码逻辑一切正常,但结果就是“驴唇不对马嘴”。
vLLM的做法是在构造metadata时用有序列表而不是无序字典,并要求执行器在遍历时严格保持列表顺序。早期有些第三方扩展在自定义ModelRunner时,把seq_data dict转list时用了集合推导,导致顺序漂移。这种问题不会立刻崩,但会让单测偶发失败、线上结果诡异。我的建议是,凡是涉及metadata到张量的转换,必须显式按seq_id排序,不能依赖Python dict的默认插入序。
5.2 抢占与Swap会让metadata失效
有一个新手很容易忽略的事实:SequenceGroupMetadata是某一步的调度快照,一旦发生了抢占或swap,这个快照就彻底作废。比如一个running状态的请求被调度器选中执行decode,但在构造完metadata之后、模型执行之前,显存突然告急,调度器决定换出这个请求。此时metadata里的block table还指向已经释放的物理块,如果执行器没有重新检查有效性,就会访问到非法显存地址。
vLLM内部通过SchedulerOutput把这几个阶段严格串行化:先做调度决策,再构造metadata,然后立刻执行,期间不允许插入抢占逻辑。如果你要在多线程环境里做异步预填充或异步执行,就必须为metadata引入版本号或epoch机制,在真正forward之前校验metadata是否仍然有效。我自己在做多流推理时,就因为这个竞态吃过亏,后来在metadata里加了generation_id字段,每次调度器状态变化就递增,执行器看到版本不匹配就重新拉取。
5.3 Beam Search场景下的块表与COW
beam search是SequenceGroup存在价值最充分的场景,但也是block table管理最复杂的地方。多个beam共享同一个prompt前缀,如果不做任何优化,每个beam都要为整个前缀分配一份物理块,显存浪费极其严重。vLLM的解决方式是Copy-on-Write(COW):当某个beam要修改一个共享块时,先复制一份新块再写入,原块继续被其他beam使用。
这个逻辑落到metadata里,就需要在每个step构造block_table时检查哪些块是共享的、哪些块是独享的。如果共享块被两个beam同时引用,且其中一个beam要追加新token到该块,那么block manager必须触发COW,把新块追加到该beam的block table里,同时更新引用计数。如果这个流程处理不当,两个beam的KV cache就会互相污染,生成结果出现交叉串扰。
在调试时,我建议你把KV cache的block table打印出来,手动检查每个block地址是否被多个beam引用,再对照生成结果是否串扰。这种问题用单测不好查,但通过日志定位到block地址后,基本一眼就能看出问题。
5.4 前缀缓存命中时computed_block_nums的正确使用
前缀缓存是vLLM里另一个复杂功能。请求A和请求B共享同一个系统提示词前缀,调度器可以复用请求A已经计算过的KV cache块,请求B只需要计算新增的差异部分。这个优化依赖metadata里的computed_block_nums字段,但使用这个字段时有一个容易出错的点:它标记的是“已经计算过的block”,而不是“本次要计算的block”。
我见过有扩展者把computed_block_nums当作本step的注意范围,结果模型执行器只对缓存过的块做attention,完全跳过了新增token的计算,导致输出全是乱码。正确理解是:computed_block_nums用于告诉执行器哪些前缀不需要重复计算,但attention的输入、mask和位置编码仍然要覆盖整个序列。执行器内部会做一次“合并”:先把缓存块对应的KV cache从物理块中取出来,再和当前step新增token的KV cache拼接成完整的KV序列。
如果你的自定义模型要支持前缀缓存,就必须在模型forward里显式处理这个拼接逻辑,否则就算调度器给了正确的computed_block_nums,你也用不上。很多模型因为attention实现是朴素的,没写拼接逻辑,导致前缀缓存功能打开后性能反而下降或结果错误,这基本都是metadata消费端的问题。
5.5 多模态输入如何扩展Metadata
自定义模型时最常遇到的一个需求是往推理里塞多模态输入。基础的SequenceGroupMetadata里虽然有multi_modal_data字段,但对具体模态的类型定义是开放的。图像、音频、视频、甚至点云数据,都要通过这个字段传给执行器。
很多人在扩展时犯的错误是只往multi_modal_data里塞原始数据,却没有同步修改input processing阶段的预处理逻辑。执行器拿到的是原始像素或波形,但模型forward要求的是embedding特征。所以正确的做法是在metadata里附带两个东西:一个是多模态原始输入,另一个是预处理后的特征或特征shape信息,然后由执行器决定何时调用预处理器、何时直接拼接。
另外,多模态输入的长度对chunked prefill也有影响。一张图片往往占几十到几百个视觉token,它们在prefill阶段和文本token拼接后共享同一个KV cache空间。如果你的metadata里只计算了文本token的数量,忽略了视觉token的数量,KV cache分配就会不足,attention mask也会出现越界。我自己实现图像输入时,就在SequenceGroup里额外维护了一个visual_token_count字段,每次构造metadata时把它加到token_chunk_size总量里,问题才彻底解决。
6. 常见问题速查表
我在实际使用和给别人解答问题过程中,积累了下面一批频率极高的问题,整理成速查表供大家定位:
| 现象 | 可能原因 | 排查方向 |
|---|---|---|
| 生成结果出现串扰另一个请求的内容 | Sequence的block table共享后未触发COW | 打印KV cache block引用计数,检查beam search共享块 |
| 某个请求在decode阶段突然丢失上下文 | 抢占换出后block table恢复不完整 | 检查swap in时block table是否正确回填 |
| prefill阶段显存使用远超预期 | token_chunk_size计算未包括多模态token | 检查metadata里token_chunk_size与multi_modal_data一致性 |
| 开启前缀缓存后结果错误 | computed_block_nums被错误当作完整注意力范围 | 检查执行器是否合并了缓存KV和新增KV |
| 多线程异步执行时偶发崩溃 | metadata引用被并发修改,未做版本校验 | 增加generation_id或深拷贝metadata |
| 自定义模型时采样结果与请求错位 | metadata列表顺序与采样器回写顺序不一致 | 强制按seq_id排序,不要依赖dict顺序 |
这些问题的根因,大部分都要回到“SequenceGroup负责动态变化、Metadata负责静态快照”这个根本设计原则上去理解。凡是破坏了这个边界,就一定会出问题。
7. 一点个人实操体会
做了这么久的推理引擎适配,我越来越觉得,vLLM里SequenceGroup和SequenceGroupMetadata这两个类的设计,是把传统批处理服务里隐式管理的“请求上下文”显式化、工程化的范本。如果你只是调用API,确实不需要关心它们;但如果你想做二次开发、优化调度、接入自定义算子,这就是你绕不开的骨架。
我的实操建议是,读源码时不要一上来就钻算法细节,先花半天时间把这两个类以及SchedulerOutput的字段彻底背下来,然后跟着一次请求的生命周期走一遍:从add_request创建SequenceGroup,到调度器生成SequenceGroupMetadata,再到ModelRunner消费metadata,最后采样回写更新Sequence状态。这条链路走通了,你对vLLM的整体把握会远超那些只背论文的人。
工具上建议配合vLLM自带的日志开关开启KV cache block table的调试输出,遇到显存异常时,用block级别日志和metadata里的block_tables字段对照,定位问题的速度会快很多。我个人还习惯在SequenceGroupMetadata构造入口加一行结构化日志,输出request_id、is_prompt、token_chunk_size、block_tables长度这四个字段,排查线上问题基本够用。
如果后续要扩展自己的推理引擎,这套“运行态对象+调度快照”的双层结构是值得直接借鉴的。把动态状态和单步执行输入剥离开,能让调度策略、模型执行、显存管理三个模块各自独立演进,这也是现代推理框架里被反复验证过的路线。
