ZJIT 线性扫描寄存器分配器：从本地到全局的架构跃迁

Ruby 生态正在经历一场底层编译基础设施的重构。作为 Shopify 主导的下一代 JIT 项目，ZJIT 近期合入了一项关键更新 —— 全新的寄存器分配器。这一改动不仅替换了从 YJIT 继承的本地分配策略，更通过引入基于 SSA 的线性扫描算法，为方法内联等高级优化铺平了道路。

为什么需要新的分配器

寄存器分配是编译器后端的核心环节。当编译器将中间表示（IR）转换为目标机器码时，必须决定每个变量驻留在物理寄存器还是内存中。由于物理寄存器数量有限，分配器需要在变量生命周期重叠时合理分配寄存器，或在寄存器不足时将变量 "溢出" 到栈上。

ZJIT 早期沿用了 YJIT 的本地寄存器分配器。这种设计每次只处理单个基本块，在每个块边界处将所有活跃变量强制存入固定位置，以便下一个基本块能够正确读取。这种策略对于 YJIT 的 "惰性逐块编译" 模式是合理的，因为 YJIT 本身就不持有完整函数的控制流图。

但 ZJIT 采用 "全方法编译" 架构，编译时已经拥有完整的控制流信息。本地分配器的块边界同步机制反而成为负担：频繁的栈存取操作增加了指令开销，更重要的是，它阻碍了跨基本块的寄存器复用和方法内联的实现。

SSA 与线性扫描的核心机制

ZJIT 后端采用静态单赋值（SSA）形式作为中间表示。在 SSA 中，每个变量只能被赋值一次，这天然契合了寄存器分配对 "值" 而非 "变量" 的追踪需求。

新分配器基于 Christian Wimmer 的论文《Linear Scan Register Allocation on SSA Form》实现。其核心流程分为三个阶段：

活跃区间计算：通过逆向数据流分析遍历控制流图，追踪每个 SSA 值从定义到最后使用的范围。在 ZJIT 的调试输出中，活跃区间以可视化图表呈现，使用 v 标记变量诞生点，实心块表示存活区间，^ 标记最后一次使用。

干扰图构建：当两个值的活跃区间重叠时，它们互相 "干扰"，不能共享同一寄存器。干扰图以节点表示值，边表示干扰关系。虽然图着色算法在干扰图上能生成最优分配，但构建和操作图结构的时间 / 空间开销对 JIT 场景过于昂贵。

线性扫描分配：按起始位置排序所有活跃区间，依次分配空闲寄存器。当寄存器池耗尽时，选择溢出代价最小的区间将其暂存到内存。这种贪心策略的时间复杂度接近线性，非常适合 JIT 的快速编译需求。

全局分配带来的能力跃迁

从本地到全局的切换是此次更新的关键架构升级。全局分配器以整个函数为作用域，允许单个值的活跃区间跨越多个基本块，并在整个生命周期内保持寄存器分配的一致性。

这一改变带来了三方面收益：

消除块边界开销：跨块的活跃值无需在边界处强制同步到内存，减少了冗余的存储 / 加载指令。在循环密集型代码中，这种优化尤为明显。

简化优化 pass：以往在基本块层面插入、删除或重排块时，需要维护复杂的变量位置映射。全局分配器将寄存器分配与块结构解耦，使得控制流优化更加灵活。

解锁方法内联：方法内联将被调用函数的代码体合并到调用方的控制流图中。全局分配器确保内联后的跨函数值能够参与统一的寄存器分配，这是本地分配器无法实现的。

目前，ZJIT 的方法内联功能正在积极开发中，它直接依赖于新分配器提供的全局视野。

工程实践：调试与观察

ZJIT 提供了调试选项用于观察分配器行为：

ruby --zjit-call-threshold=2 --zjit-dump-lir=live_intervals test.rb

该命令输出每个基本块的活跃区间图，展示各 SSA 变量在指令序列中的生命周期分布。对于编译器开发者，这是诊断寄存器压力热点和溢出决策的有效工具。

在实际调优中，关注以下指标：

• 活跃区间密度：单个指令位置同时存活的值数量峰值，直接对应寄存器压力
• 溢出频率：被迫存入内存的变量比例，过高可能提示需要调整内联阈值或拆分热点函数
• 跨块寄存器复用率：全局分配器优化效果的直接体现

局限与后续工作

当前实现将每个值的活跃区间视为单一连续段，尚未处理 "活跃区间空洞"（lifetime holes）。考虑如下代码：

def example(cond, a, b)
  x = a + b
  if cond
    y = a * 2
    use(y)
  else
    use(x)
  end
end

变量 x 在 if 的真分支中未被使用，但当前分配器仍将其视为活跃，导致寄存器在真分支中被无效占用。精确建模这类空洞将允许在间隙中复用寄存器，进一步减少溢出。

此外，线性扫描的贪心本质意味着它可能错过全局最优分配。对于寄存器压力极高的函数，未来可考虑在热路径上回退到图着色算法，或引入迭代优化策略。

总结

ZJIT 的新寄存器分配器代表了 Ruby JIT 编译器从 "快速够用" 向 "全局优化" 的转型。通过 SSA 形式的线性扫描算法，配合全局作用域的分配策略，它不仅降低了块边界开销，更为方法内联等高级优化奠定了基础。对于关注 Ruby 运行时性能的开发者，理解这一变化有助于更好地评估 ZJIT 的适用场景，并为未来的性能调优做好准备。

参考来源

• Aaron Patterson, "A new Register Allocator for ZJIT", Rails at Scale, 2026
• Takashi Kokubun, "ZJIT: Building a New JIT Compiler for Ruby", REBASE 2025
• Christian Wimmer, "Linear Scan Register Allocation on SSA Form"

compilers