Linux 网络内核协议栈深度剖析(八):网络虚拟化与容器网络实现
容器网络是现代云原生基础设施的底座。Docker、Kubernetes 依赖 Linux 内核提供的四大网络虚拟化原语——Network Namespace、veth pair、Linux Bridge、VXLAN——把跑在同一台物理机或跨越多台主机的容器连接成一张逻辑网络。本文基于 Linux 6.4-rc1 源码,逐层拆解这四个模块的内核实现,并在最后串联出一条完整的容器间数据包路径。
容器网络是现代云原生基础设施的底座。Docker、Kubernetes 依赖 Linux 内核提供的四大网络虚拟化原语——Network Namespace、veth pair、Linux Bridge、VXLAN——把跑在同一台物理机或跨越多台主机的容器连接成一张逻辑网络。本文基于 Linux 6.4-rc1 源码,逐层拆解这四个模块的内核实现,并在最后串联出一条完整的容器间数据包路径。
在前六篇中,我们依次拆解了 sk_buff 生命周期、网卡驱动收发路径、TCP/IP 协议栈、路由子系统、套接字层以及 NAPI/GRO 机制。本篇聚焦数据包过滤与处理的核心框架:Netfilter hook 体系、nftables 规则执行引擎、conntrack 连接追踪,以及近年来高速演进的 eBPF 网络程序(XDP、tc BPF、Socket Filter)。所有代码片段均源自 Linux 6.4-rc1。
Socket 是用户态程序与内核网络协议栈之间的唯一接口。它不仅是一个文件描述符,更是一套精心设计的多层抽象体系——从 VFS 层的 struct socket,到协议无关的 struct sock,再到 IPv4 专用的 struct inet_sock 和 TCP 专用的 struct tcp_sock,每一层都有其清晰的职责边界。本文基于 Linux 6.4-rc1 源码,深入剖析 socket 系统调用的完整实现路径、epoll 的内核机制,以及缓冲区管理与内存压力控制。
本系列前五篇分别覆盖了网卡驱动收发、NAPI、协议栈 IP/TCP 层、路由子系统和 netfilter,这一篇聚焦于最接近用户态的 socket 接口层,梳理从系统调用入口到数据进出内核缓冲区的全链路,并以 epoll 的内部实现为重点,阐释高并发服务器底层事件通知的工作机制。
TCP 拥塞控制是互联网稳定运行的核心机制之一。自 1988 年 Van Jacobson 在 SIGCOMM 发表奠基论文以来,拥塞控制算法经历了从 Tahoe、Reno、BIC 到 CUBIC,再到基于测量的 BBR 的演进历程。本文以 Linux 6.4-rc1 源码为基础,深入剖析内核拥塞控制框架的实现细节,覆盖从插件注册到状态机转换、从 CUBIC 三次函数到 BBR 带宽估计的完整链路。
拥塞控制算法面对的核心挑战是:如何在不知道网络容量的前提下,以尽可能高的速率发送数据,同时不引发严重拥塞?这个问题的本质是分布式资源竞争——互联网中每条流都是自私的,拥塞控制协议需要在个体利益(最大化自身吞吐)和集体利益(网络整体公平稳定)之间取得平衡。Jacobson 的原始方案用丢包作为拥塞的唯一信号,这在当时(网络带宽以 Kbps 计)是合理的。但随着网络带宽扩展到 Gbps、Tbps 量级,以及 bufferbloat(路由器缓冲区过大导致的延迟膨胀)问题的出现,基于丢包的方案暴露出越来越多的局限性,催生了 CUBIC 和 BBR 等新一代算法的诞生。
TCP 是互联网的基石协议,其可靠性、有序性和流量控制能力建立在一套精密的状态机和连接管理机制之上。本文基于 Linux 6.4-rc1 内核源码,深入剖析 TCP 协议栈从数据结构设计、三次握手、数据收发,到四次挥手和 TIME_WAIT 的完整生命周期,并给出实用的内核级诊断工具链。
IP 层(网络层)是 Linux 协议栈的核心,负责报文的寻址、路由、转发与分片重组。本文基于 Linux 6.4-rc1 源码,从 ip_rcv 入口函数出发,逐层拆解接收路径、FIB 路由查找、发送路径、分片重组和 netfilter 钩子的完整实现,并给出实用的内核诊断技巧。文中所有代码均来自 net/ipv4/ip_input.c、net/ipv4/ip_output.c、net/ipv4/fib_trie.c、net/ipv4/ip_fragment.c、net/ipv4/ip_forward.c 和 net/netfilter/core.c 等实际文件。
本文是 Linux 网络内核协议栈系列的第二篇,聚焦于网卡驱动框架与 NAPI(New API)收发包机制。我们将深入 Linux 6.4-rc1 源码,逐层拆解从网卡硬件中断到 sk_buff 进入协议栈的完整链路,并对称地分析发送路径。所有代码片段均直接取自内核源文件,确保与实际内核行为一致。
本系列基于 Linux 6.4-rc1 源码(commit
ac9a78681b92),所有代码片段均从真实源文件读取,路径为include/linux/skbuff.h与net/core/skbuff.c。
网络性能调优、eBPF 程序开发、内核模块编写,乃至排查一个”神秘的丢包”——这些工作的底层都有一个共同的核心:sk_buff(socket buffer)。它是 Linux 内核网络子系统中最关键的数据结构,每一个在内核中流动的网络报文都以 sk_buff 的形式存在。
本系列文章将从零到深,系统拆解 Linux 网络协议栈。第一篇先建立全局视图,再深入 sk_buff 的内存模型与操作原语。
从网卡收到一帧,到用户态 read() 返回数据,内核经历的完整调用链如下:
1 | 硬件中断 |
关键节点说明:
NAPI(New API) 是现代驱动的标准轮询机制。硬中断只负责触发 napi_schedule(),将 poll 工作推给软中断 NET_RX_SOFTIRQ,从而避免高速网卡的中断风暴。驱动在 poll 函数中用 napi_build_skb() 或 __napi_alloc_skb() 创建 skb,并调用 napi_gro_receive() 进入 GRO(Generic Receive Offload)聚合层。
netif_receive_skb() 是协议层的分发入口。它通过 ptype_base 哈希表查找与 skb->protocol 对应的协议处理函数(如 ip_rcv、ipv6_rcv)。
ip_rcv() 完成 IP 首部校验、路由查找(ip_route_input_noref()),然后决定报文去向:本地投递还是转发。本地投递进入 ip_local_deliver(),这里完成 IP 分片重组,再调用上层协议(TCP/UDP/ICMP)的接收函数。
tcp_v4_rcv() 根据四元组找到 socket,将报文挂入 sk->sk_receive_queue,并通过 sk->sk_data_ready 回调唤醒阻塞在 read() 的用户进程。
1 | 用户态 write() / sendmsg() |
发送路径的核心在于 TCP 在 sk_write_queue 中维护待发送的 skb。TCP 实现了可靠传输,skb 只有在收到 ACK 后才会通过 tcp_clean_rtx_queue() 释放。发送时,TCP 先调用 skb_clone() 产生一个 fclone(fast-clone),将克隆体交给 IP 层,自己保留原始 skb 用于重传。
sk_buff 是 Linux 内核网络子系统中报文的载体。它本身不直接存储报文数据,而是一个元数据描述符,通过指针指向实际的数据缓冲区。其设计目标是:在协议栈各层之间传递报文时,尽量避免数据拷贝,只修改指针和元数据。
skbuff.h 中的官方注释给出了最清晰的内存模型(源码第 706-728 行):
1 | /** |
四个核心指针在结构体末尾(源码第 1053-1057 行):
1 | /* These elements must be at the end, see alloc_skb() for details. */ |
各指针含义:
| 指针 | 含义 |
|---|---|
head |
分配的缓冲区起始地址,固定不变 |
data |
当前有效数据的起始位置(随 skb_push/pull 移动) |
tail |
当前有效数据的结束位置(随 skb_put 移动) |
end |
缓冲区末尾(紧接 skb_shared_info),固定不变 |
在 64 位系统(NET_SKBUFF_DATA_USES_OFFSET 宏定义时),tail 和 end 存储的是相对于 head 的偏移量,而非指针,节省内存并方便原子操作。
Headroom(头部预留空间) 是 data - head 的字节数,发送时用于各层协议依次 skb_push() 写入自己的首部,无需移动数据。Tailroom 是 end - tail 的字节数(不含 skb_shared_info),接收时用于 skb_put() 追加数据。
这三个字段是初学者最容易混淆的:
1 | unsigned int len, // 源码第 896 行 |
skb->len:报文的逻辑总长度,等于线性区域长度加上所有分片的长度。skb_headlen(skb) + skb->data_len。skb->data_len:仅计算非线性部分(page frags + frag_list)的字节数。线性 skb 的 data_len 为 0。truesize**:内核为这个 skb 实际占用的内存大小,包含 sk_buff 结构体本身、数据缓冲区以及 skb_shared_info。用于 socket 发送/接收缓冲区的内存计费(sk_wmem_alloc / sk_rmem_alloc)。1 | /* return minimum truesize of one skb containing X bytes of data */ |
truesize 的精确计算在 __finalize_skb_around() 中(net/core/skbuff.c 第 352-375 行):
1 | static inline void __finalize_skb_around(struct sk_buff *skb, void *data, |
1 | char cb[48] __aligned(8); // 源码第 880 行 |
这 48 字节是各协议层的私有暂存区,对上下层完全透明。内核使用宏来访问它:
IP 层(include/net/ip.h):
1 | struct inet_skb_parm { |
TCP 层(include/net/tcp.h):
1 | struct tcp_skb_cb { |
TCP 在发包时用 TCP_SKB_CB(skb)->seq 记录序列号,在收包时用 TCP_SKB_CB(skb)->header.h4 读取 IP 层留下的信息。由于 sizeof(struct tcp_skb_cb) <= 48,两者都合法地复用同一块内存。
1 | union { |
skb->sk 指向拥有这个报文的 socket。在发送路径中,skb 一直持有对 socket 的引用(用于内存计费);在接收路径进入 TCP/UDP 处理后也会被赋值。ip_defrag_offset 是 IP 分片重组阶段对 sk 字段的临时复用,无需额外内存。
1 | union { |
skb->dev 指向 skb 当前关联的网络设备(struct net_device)。在接收路径,它指向收包的网卡;在发送路径,它指向出口网卡。某些协议(如 UDP 接收路径)会将 dev 置 NULL 并用 dev_scratch 存储其他临时信息。
1 | union { |
_skb_refdst 存储指向 struct dst_entry 的指针(路由缓存条目)。最低位被用作标志位 SKB_DST_NOREF,表示是否持有引用计数:
1 | #define SKB_DST_NOREF 1UL |
在快速转发路径(如 ip_forward())中,内核使用 skb_dst_set_noref() 以 RCU 方式持有路由缓存,避免昂贵的引用计数操作。
GSO(Generic Segmentation Offload)和 GRO(Generic Receive Offload)是 Linux 将分段/聚合工作推迟到驱动层或硬件的机制,核心信息存储在 skb_shared_info(见下节)中:
1 | struct skb_shared_info { |
发送路径中,TCP 构造一个大 skb(可包含多个 MSS 的数据),设置 gso_size = tp->mss_cache、gso_segs = 数量,并设置 ip_summed = CHECKSUM_PARTIAL,由驱动或网卡完成实际切割和 checksum。
接收路径中,GRO 将多个小包聚合成一个大 skb,减少协议栈处理次数。skb->slow_gro 标志指示该 skb 在 GRO 合并时携带了额外状态(连接跟踪、路由等),需要在释放时特殊处理。
1 | __u8 ip_summed:2; // CHECKSUM_NONE/UNNECESSARY/COMPLETE/PARTIAL |
四种状态的含义(源码 include/linux/skbuff.h 第 99-250 行有完整文档):
| 值 | 场景 | 含义 |
|---|---|---|
CHECKSUM_NONE |
RX | 硬件未校验,软件需要验证 |
CHECKSUM_UNNECESSARY |
RX | 硬件已验证,无需软件重算 |
CHECKSUM_COMPLETE |
RX | 硬件提供了整包 checksum(存于 skb->csum) |
CHECKSUM_PARTIAL |
TX | 软件填写了伪首部,请求硬件完成最终 checksum |
TX 路径中,csum_start 是从 skb->head 开始计算 checksum 的起始偏移,csum_offset 是相对于 csum_start 的 checksum 写入位置。
skb_shared_info 紧跟在数据缓冲区的末尾(end 指针处):
1 | #define skb_shinfo(SKB) ((struct skb_shared_info *)(skb_end_pointer(SKB))) |
完整结构(源码第 574-599 行):
1 | struct skb_shared_info { |
frags[] 是 Scatter-Gather IO 的核心。每个 skb_frag_t(实为 struct bio_vec)描述一个页片段:
1 | typedef struct bio_vec skb_frag_t; |
网卡驱动通过 DMA 将数据直接写入这些页,frags[] 只存储(页, 偏移, 长度)三元组,完全避免数据拷贝。
frag_list 是一个 skb 链表,用于两种场景:
frag_list 链接在一起,skb->len 是全部分片的总长度。SKB_GSO_FRAGLIST 模式下,将多个小 skb 链成链表交给驱动。dataref 是 skb_shared_info 中最重要的计数器,分为两个 16 位字段:低 16 位是总引用数,高 16 位是 payload-only 引用数(nohdr clone)。skb_clone() 共享数据区时递增此计数,skb_release_data() 递减,归零后才真正释放数据缓冲区。
内核为 sk_buff 头部维护了专用的 slab 缓存(net/core/skbuff.c 第 89-90 行):
1 | struct kmem_cache *skbuff_cache __ro_after_init; |
核心分配函数 __alloc_skb()(源码第 625-687 行):
1 | struct sk_buff *__alloc_skb(unsigned int size, gfp_t gfp_mask, |
分配分两步:
sk_buff 结构体(仅元数据,约 232 字节)kmalloc_reserve() 分配数据缓冲区kmalloc_reserve() 内部会优先尝试 skb_small_head_cache(专为 TCP 头部优化的小块缓存,大小为 SKB_HEAD_ALIGN(MAX_TCP_HEADER)),对于不超过 1024 字节的小报文走快速路径。
NAPI 路径还有额外优化:napi_skb_cache_get() 维护一个 per-CPU 的 64 个 skb 的批量缓存,使用 kmem_cache_alloc_bulk() 批量分配,大幅减少 slab 调用开销。
1 | static inline void skb_reserve(struct sk_buff *skb, int len) |
在 skb 刚分配、尚未写入数据时调用,同时移动 data 和 tail,使 head 到 data 之间形成 headroom。后续各层通过 skb_push() 向前扩展,写入自己的首部,无需移动已有数据。
驱动分配接收 skb 时通常调用:
1 | skb_reserve(skb, NET_SKB_PAD); // 驱动预留(NET_SKB_PAD = 32 字节) |
发送路径中,sock_alloc_send_skb() 预留的空间要包含所有协议层首部的总长度:
1 | NET_SKB_PAD + MAX_HEADER(ETH + IP + TCP 首部之和) |
这三个函数是协议栈各层操作数据的核心原语(源码第 2383-2428 行):
skb_put() — 向 tail 方向扩展,用于接收路径追加数据:
1 | void *skb_put(struct sk_buff *skb, unsigned int len) |
skb_push() — 向 head 方向扩展,用于发送路径在 headroom 中写入首部:
1 | void *skb_push(struct sk_buff *skb, unsigned int len) |
skb_pull() — 从 data 头部移除数据,用于接收路径跳过已处理的首部:
1 | void *skb_pull(struct sk_buff *skb, unsigned int len) |
三者关系图示(发送路径,从上层到下层):
1 | 初始状态(alloc + reserve 后): |
**skb_clone()**(源码第 1853-1881 行):
1 | struct sk_buff *skb_clone(struct sk_buff *skb, gfp_t gfp_mask) |
skb_clone() 只复制 sk_buff 元数据结构,共享数据缓冲区(递增 shinfo->dataref)。克隆体不能修改数据内容,但可以修改自己的首部指针(data 等)。适用于:
fclone 优化:TCP 通过 SKB_ALLOC_FCLONE 标志分配 sk_buff_fclones(两个 skb 紧挨着分配),第一次 clone 时直接使用伴生的 skb2,无需额外 slab 分配。
**skb_copy()**(源码第 1933-1953 行):
1 | struct sk_buff *skb_copy(const struct sk_buff *skb, gfp_t gfp_mask) |
skb_copy() 完全复制数据,包括 headroom、线性数据以及所有 page frags,返回一个完全独立的 skb。代价是需要实际的数据拷贝。适用于需要修改数据内容的场景。
| 操作 | 复制元数据 | 复制数据 | 共享 shinfo | 可修改数据 |
|---|---|---|---|---|
skb_clone() |
是 | 否 | 是 | 否 |
skb_copy() |
是 | 是 | 否 | 是 |
pskb_copy() |
是 | 仅线性部分 | 是(frags) | 仅线性部分 |
1 | static inline int skb_linearize(struct sk_buff *skb) |
当代码需要顺序访问报文所有字节,但 skb 是非线性的(data_len > 0,存在 page frags 或 frag_list)时,调用 skb_linearize() 将所有 page frags 的数据拷贝到线性区域,整合为线性 skb。
这是一个较重的操作,会触发内存分配和拷贝。协议栈中有大量 pskb_may_pull() 调用——它只确保 skb 的首部位于线性区域(仅 pull 必要的字节),比完整的 skb_linearize() 代价小得多。
两者都会递减 skb->users 引用计数,归零后调用 __kfree_skb() 释放内存,但语义不同,这对 tracing 工具(如 drop_monitor、skb:kfree_skb tracepoint)至关重要:
1 | // kfree_skb:报文被丢弃(drop),通常意味着异常 |
释放链路(__kfree_skb() 路径):
1 | __kfree_skb(skb) |
skb_frag_t 等价于 struct bio_vec,每个 frag 持有一个 struct page * 引用:
1 | typedef struct bio_vec skb_frag_t; |
驱动在接收时通常这样操作:
page_pool 管理),通过 DMA 映射给网卡skb_add_rx_frag() 将 page 引用填入 frags[]1 | void skb_add_rx_frag(struct sk_buff *skb, int i, struct page *page, |
page_pool 是 6.x 内核引入的高性能页缓存机制,通过 skb->pp_recycle 标志和 skb_pp_recycle() 函数,在 skb 释放时直接将 page 归还给 page_pool 而非系统内存池,避免了 page allocator 的开销,实现”原地回收”。
sendfile() 和 splice() 系统调用走的是 SKBFL_SHARED_FRAG 路径:文件页缓存中的 page 直接被加入 skb 的 frags[],内核标记该 frag 为可能被外部修改:
1 | /* This indicates at least one fragment might be overwritten |
此时 skb 的数据直接来自 page cache,全程无需拷贝到内核网络缓冲区。但如果需要计算 TX checksum(ip_summed == CHECKSUM_PARTIAL 但硬件不支持),内核必须通过 skb_copy_ubufs() 先复制 frags 内容,再计算 checksum,这是 zero-copy sendfile 的一个隐含成本。
用户态零拷贝(MSG_ZEROCOPY,SKBFL_ZEROCOPY_ENABLE)则通过 ubuf_info 回调机制,在 DMA 传输完成后通知用户态释放缓冲区,实现用户态缓冲区到网卡的真正零拷贝。
| 结构体 | 大小 |
|---|---|
struct sk_buff |
~232 字节 |
struct skb_shared_info |
~320 字节(含 17 个 frags) |
skb->cb[48] |
48 字节 |
struct tcp_skb_cb |
44 字节(含 IPv6 时) |
struct inet_skb_parm |
~20 字节 |
skb_frag_t (bio_vec) |
12 字节 |
本系列共计划 10 篇,系统覆盖 Linux 网络内核的核心子系统:
| 篇次 | 标题 | 核心内容 |
|---|---|---|
| Part 1 | 总览架构与 sk_buff 生命周期(本文) | 协议栈全路径、sk_buff 内存模型、操作函数、零拷贝 |
| Part 2 | 网卡驱动与 NAPI 机制 | ndo_open/ndo_start_xmit、中断合并、NAPI poll、page_pool、XDP |
| Part 3 | IP 层:路由、分片与转发 | fib_lookup、路由缓存、ip_fragment、conntrack hook 点 |
| Part 4 | TCP 实现(一):连接管理与状态机 | 三次握手、TIME_WAIT、tcp_rcv_state_process、SYN cookies |
| Part 5 | TCP 实现(二):拥塞控制与发送缓冲区 | CUBIC/BBR、tcp_write_xmit、TSQ、Nagle 算法 |
| Part 6 | TCP 实现(三):接收与流量控制 | tcp_rcv_established、OOO 队列、接收窗口、tcp_recvmsg |
| Part 7 | UDP 与多播 | udp_sendmsg、udp_rcv、GRO for UDP、多播路由 |
| Part 8 | Socket 层与 VFS 接口 | sock_alloc、inet_stream_ops、poll/epoll 实现、零拷贝 recv |
| Part 9 | Netfilter 与 iptables/nftables | hook 框架、连接跟踪(nf_conntrack)、NAT 实现原理 |
| Part 10 | 性能调优与 eBPF 网络编程 | RSS/RPS/RFS、SO_REUSEPORT、XDP、TC eBPF、sockmap |
include/linux/skbuff.h、net/core/skbuff.cDocumentation/networking/skbuff.rst(内核源码树内)vLLM 官方 recipes 仓库 提供了超过 57 个生产级部署配方,覆盖 28+ 主流模型家族。本文深入剖析这些配方背后的部署模式、硬件选择、并行策略与调优技巧,帮助你快速从零到一地在生产环境中部署大型语言模型。
Scheduler 是 vLLM 的核心调度器。在每个微秒级的时间窗口内,它都要做出关键决策:处理哪些 request、计算多少个 token、何时 preempt request,以及如何最大化 GPU 利用率。本文将深入探讨 Scheduler 的算法与实现。
在任意时刻,Scheduler 必须处理:
与传统批处理不同,LLM 推理服务面临独特挑战:
vLLM 的核心创新:continuous batching(又称迭代级批处理)。
1 | # Static batching - wait for batch to fill |
问题:
1 | # Continuous batching - add/remove every iteration |
优点:
文件位置:vllm/v1/core/sched/scheduler.py
Scheduler 维护 requests、waiting、running 和 skipped_waiting,并持有 KVCacheManager / EncoderCacheManager 等资源管理器。其中 max_num_scheduled_tokens 的来源是:
1 | self.max_num_scheduled_tokens = ( |
也就是说,max_num_batched_tokens 现在是 fallback,而不是唯一来源。
当前 V1 的 request 状态比“WAITING → RUNNING → FINISHED”更细:
WAITINGWAITING_FOR_STRUCTURED_OUTPUT_GRAMMARWAITING_FOR_REMOTE_KVSWAITING_FOR_STREAMING_REQRUNNINGPREEMPTEDFINISHED_STOPPED、FINISHED_LENGTH_CAPPED、FINISHED_ABORTED、FINISHED_IGNORED、FINISHED_ERROR、FINISHED_REPETITION因此,更准确的流转应理解为:
WAITING 进入 RUNNINGPREEMPTED 并重新入队PREEMPTED → RUNNINGFINISHED当前 RequestQueue 的核心 API 是:
1 | queue.add_request(request) |
如果启用 priority 策略,优先级大小规律也与很多直觉相反:数值越小,优先级越高。PriorityRequestQueue 直接依赖 Request.__lt__,比较顺序是:
priority(更小者优先)arrival_time(更早者优先)request_idschedule() 的核心思路仍然是“先处理 running,再在条件允许时接纳 waiting”。SchedulerOutput 会分别携带:
scheduled_new_reqsscheduled_cached_reqsnum_scheduled_tokens: dict[str, int]total_num_scheduled_tokensscheduled_spec_decode_tokensscheduled_encoder_inputsfinished_req_idspreempted_req_ids可以把主循环简化理解为:
1 | def schedule(self) -> SchedulerOutput: |
token budget 的概念没有变化:它仍然是当前迭代最重要的“算力 / latency / activation memory”上限之一。当前实现用 self.max_num_scheduled_tokens 作为每轮 budget,并在 chunked prefill、running request 续跑、spec decode 等路径上共享这份预算。
当前 scheduler 不再显式维护“prefill phase / decode phase”两套完全独立的 API。更准确的做法是根据:
request.num_computed_tokensrequest.num_tokens_with_specrequest.num_output_placeholders之间的差值来决定本轮还需要为该 request 安排多少 token。
对 running request 来说,scheduler 主要会:
num_new_tokenslong_prefill_token_threshold 等配置裁剪超长 prefillallocate_slots() 尝试拿到新 slot因此,比起“给 request 写回 request.num_scheduled_tokens = ...”,当前实现更准确的说法是:每轮调度数量被记录在 SchedulerOutput.num_scheduled_tokens[request_id] 中。
一个新 request 要进入本轮 batch,通常会经历这些检查:
waiting 或 skipped_waiting 中选择可调度的队列get_computed_blocks() 查找 prefix cache 命中can_fit_full_sequence() 判断完整序列是否可以被系统接纳allocate_slots() 为本轮真正需要的 token 分配 slot这里有两个关键点:
waiting + skipped_waiting 两条队列,而不是单一 waiting queuepreemption 通过 _preempt_request() 完成,流程更接近:
1 | def _preempt_request(self, request, timestamp): |
需要注意:
PREEMPTED 状态重新入队,而不是简单地回到 WAITINGpriority 模式下,会先牺牲较差优先级的 running request;FCFS 模式下则更接近按当前运行队列尾部顺序处理chunked prefill 仍然存在,但当前实现不是 self.chunk_size 这种字段驱动。更贴近源码的配置包括:
enable_chunked_prefilllong_prefill_token_thresholdmax_num_partial_prefillsmax_long_partial_prefills因此,当前行为更像是:
num_new_tokens当前 vLLM 只支持两种调度策略:
fcfsprioritypriority 模式下数字越小越先执行。可以这样理解:
1 | # Lower numeric value = higher scheduling priority |
当前 scheduler 对 speculative decoding 的感知体现在:
spec_token_idsSchedulerOutput.scheduled_spec_decode_tokensnum_lookahead_tokens因此,比起 main_model.verify(draft_tokens) 这种直观伪代码,更准确的描述是:scheduler 会为 lookahead / draft token 预留 slot,并通过 scheduled_spec_decode_tokens 把本轮 spec decode 信息传给后续执行与输出更新路径。
当前本地源码里并没有一个按“预估 prefix cache 命中率”对 waiting requests 进行重排序的 reorder_for_cache_hits() 实现。waiting 请求的顺序主要由 FCFS / priority queue 决定,再叠加 skipped_waiting 的恢复逻辑。
更贴近源码的准入流程是:
1 | computed_blocks, num_cached_tokens = kv_cache_manager.get_computed_blocks(request) |
当存在 sliding window、external KV transfer、spec decode 或 encoder-decoder cross-attention 时,allocate_slots() 还会综合这些上下文一起计算需要分配的 block 数量。
当前 scheduler 的内存压力处理更接近“尝试为 running request 分配 slot,失败时根据当前策略 preempt”。prefix cache 的显式淘汰接口是 kv_cache_manager.evict_blocks(block_ids),它按 block id 清理缓存索引。
多模态 / encoder-decoder 场景下,当前 request 并不是暴露 encoder_input_ids 这个简化字段。scheduler 会通过 request.mm_features、_try_schedule_encoder_inputs(),以及 EncoderCacheManager.can_allocate() / allocate() / check_and_update_cache() 来判断 encoder side 是否有容量。
SchedulerStats 的重点字段包括:
num_running_reqsnum_waiting_reqskv_cache_usageencoder_cache_usageprefix_cache_statsconnector_prefix_cache_statskv_cache_eviction_eventsspec_decoding_statsperf_stats而“本轮 preempt 了多少 request”这类信息,当前是放在 IterationStats.num_preempted_reqs 里,而不是 SchedulerStats.num_preempted。
因此,当前 scheduler 更像是在平衡这些目标:
下面给出一个更接近当前实现语义的简化时间线。注意这里使用的是 scheduler_output.num_scheduled_tokens[req_id] 这个“每轮输出”,而不是给 request 对象写回一个持久字段。
1 | waiting = [A(prompt=100), B(prompt=50)] |
A 和 B 都还是新请求1 | scheduler_output.num_scheduled_tokens = { |
A、B 进入 running 集合C,scheduler 会先为 running request 预留预算,再根据 long_prefill_token_threshold 与剩余 budget 决定 C 这一轮最多能推进多少 tokenschedule() 里直接调用某个 finish_request() helper,而是要等 model output 返回后,由 scheduler.update_from_output(...) 更新状态并判断 stop / length cap / error 等 finished 原因B 被 preempt,scheduler 会把它以 PREEMPTED 状态重新入队;等资源恢复后,它会再次进入 RUNNING当前 vLLM 暴露的调度策略只有两种:
FCFSRequestQueueadd_request() / pop_request() / peek_request()PriorityRequestQueuepriority 数值越小越先执行;若相同则按 arrival_time,再按 request_id更贴近当前指标的观察方式是关注:
IterationStats.num_preempted_reqsSchedulerStats.kv_cache_usageSchedulerStats.encoder_cache_usage如果 preemption 频繁发生,通常优先检查:
max_num_seqs 是否过高long_prefill_token_threshold 是否合理如果 SchedulerStats.num_waiting_reqs 长时间偏高,通常意味着:
如果 TTFT 偏高,通常优先检查:
max_num_scheduled_tokens 是否过大Continuous batching 通过每次迭代动态增减 request,消除了队头阻塞
Token budget 控制 batch 大小,平衡延迟与吞吐量的权衡
运行中的 request 拥有优先权,以最小化 decode 延迟
Preemption 是内存紧张时的最后手段:它会中断当前运行进度,但后续恢复时仍可能重新命中 prefix cache
Chunked prefill 在处理长 prompt 与交互式 request 之间取得平衡
当前的 SchedulerOutput 是结构化输出,会同时携带 per-request token 分配、spec decode、encoder 输入与 finished/preempted request 信息
在第四部分中,我们将探索 Request Processing——request 如何从 tokenization 到流式输出在系统中流转,以及跨迭代的状态管理机制。
Scheduler 是 vLLM 的大脑,每秒钟要做出成千上万个瞬间决策,在满足延迟 SLA 的同时最大化 GPU 利用率。接下来,我们将看到 request 如何在系统中端到端地流转。