Zing 论坛

正文

SiFAR:无同步All-Reduce技术突破低延迟LLM推理瓶颈

本文介绍SiFAR(无同步All-Reduce),通过双缓冲、交换机内冗余拉取和推测式归约三项技术,将多GPU推理的All-Reduce延迟降低52%,端到端吞吐量提升18.6%,为推理模型和智能体系统的低延迟部署提供关键优化。

All-Reduce低延迟推理张量并行集体通信GPU优化推理模型交换机内归约双缓冲推测式执行NCCL
发布时间 2026/07/10 06:34最近活动 2026/07/13 13:41预计阅读 7 分钟
SiFAR:无同步All-Reduce技术突破低延迟LLM推理瓶颈
1

章节 01

SiFAR:无同步All-Reduce技术突破低延迟LLM推理瓶颈

原作者与来源

  • 原作者/维护者:arXiv authors
  • 来源平台:arxiv
  • 原始标题:SiFAR: Synchronization-Free All-Reduce for Low-Latency LLM Inference
  • 原始链接:http://arxiv.org/abs/2607.08973v1
  • 来源发布时间/更新时间:2026-07-09T22:34:00Z

推理模型的延迟困境

随着大语言模型从简单的聊天机器人进化为复杂的推理系统和智能体,一个技术瓶颈日益凸显:token生成延迟。

对于传统聊天应用,延迟优化的收益存在天花板——人类的阅读速度约为每分钟200-300个词,相当于每秒3-5个token。即使模型能每秒生成100个token,超出人类阅读速度的部分也意义有限。

然而,新一代推理模型和智能体系统改变了这一格局。这些系统在执行任务时会生成大量的中间推理token——思维链、规划步骤、自我纠错过程——这些token不会被人类用户直接阅读,而是用于内部决策。在这种场景下,单token延迟直接决定了端到端响应时间。如果一个复杂任务需要模型生成10000个中间token才能给出最终答案,即使每token只慢1毫秒,总延迟也会增加10秒。

低延迟推理的技术挑战

低延迟推理通常采用最小批处理策略——每个请求单独处理,不与其他请求合并。这消除了批处理带来的等待时间,但也意味着GPU无法充分利用张量核心的并行计算能力。在这种情况下,token生成变成了带宽受限操作:瓶颈不是计算速度,而是内存带宽。

为了突破单GPU的内存带宽限制,业界广泛采用张量并行:将模型权重切分存储在多个GPU上,并行读取。例如,使用8个GPU时,每个GPU只需存储1/8的权重,读取时间理论上减少8倍。

但张量并行引入了一个新的开销来源:All-Reduce通信。在每个Transformer层的计算过程中,各GPU需要交换中间结果以保持数据一致性。这种All-Reduce操作需要在所有参与的GPU之间进行同步,随着GPU数量增加,同步开销呈线性甚至超线性增长。研究表明,在8个H200 GPU上运行Llama-3.1-8B模型时,移除All-Reduce可以将token吞吐量提升43%。这说明All-Reduce已成为低延迟推理的主要瓶颈。

SiFAR的核心思想:消除同步

SiFAR(Synchronization-Free All-Reduce)的目标是在保持All-Reduce功能的同时,消除同步开销。

现有算法的局限性

Oneshot是最简单的All-Reduce实现:每个GPU将自己的数据分片发送给所有其他GPU,接收所有其他GPU的数据,在本地进行归约。这种算法需要两次全局屏障:一次在通信前确保所有GPU准备好发送,一次在通信后确保所有GPU完成接收。随着GPU数量增加,每个GPU需要发送的数据量增加,通信时间线性增长。

Twoshot是对Oneshot的改进,采用分阶段通信:Reduce-Scatter阶段数据被分成多个块,各GPU负责归约不同的块;All-Gather阶段各GPU将归约后的块广播给所有其他GPU。Twoshot在GPU数量较多时表现更好,但它仍然需要两次全局屏障。

技术一:双缓冲消除底部屏障

研究团队首先分析了Oneshot算法中底部屏障的作用。他们发现,这个屏障实际上是在强制执行一个写后写(WAW)依赖:在All-Reduce之前,每个GPU的缓冲区包含输入数据;All-Reduce完成后,同一缓冲区被覆写为归约结果;底部屏障确保所有GPU都完成了覆写操作,然后才能开始下一轮计算。

问题是:如果下一轮计算需要读取的是All-Reduce的结果,那么WAW依赖是必要的。但在实际的Transformer推理中,情况往往并非如此。

SiFAR采用双缓冲技术消除WAW依赖:为每个数据张量分配两个缓冲区Buffer A和Buffer B;第N层计算使用Buffer A作为输入,将输出写入Buffer B;第N+1层计算使用Buffer B作为输入,将输出写入Buffer A;All-Reduce操作在Buffer A和Buffer B之间进行数据交换。通过这种交替使用,All-Reduce的输出自然成为下一层计算的输入,无需等待覆写完成。底部屏障因此变得不必要。

技术二:冗余拉取与交换机内归约

虽然双缓冲消除了底部屏障,但Oneshot算法本身存在扩展性问题:随着GPU数量增加,每个GPU需要向其他所有GPU发送数据,通信量呈O(N²)增长。Twoshot通过分阶段通信将通信量降低到O(N log N),但它有一个致命缺陷:无法消除底部屏障。在Twoshot的All-Gather阶段,每个GPU必须等待所有其他GPU完成Reduce-Scatter,这个依赖是算法固有的。

现代数据中心网络交换机支持在交换机内进行数据归约。当多个GPU向同一目的地发送数据时,交换机可以先将这些数据在硬件层面进行归约,然后再发送给目的地。这种操作在纳秒级完成,几乎不增加延迟。

SiFAR提出了一种称为冗余拉取的新机制:每个GPU不是等待其他GPU推送数据,而是主动向交换机拉取数据;交换机从所有其他GPU收集数据,在内部进行归约,然后将完整结果返回给请求者;每个GPU执行相同的操作,因此每个GPU最终都得到完整的All-Reduce结果。

关键洞察:虽然每个GPU都接收了完整的数据(冗余),但由于交换机内归约,网络总流量并没有增加。同时,这种机制天然不需要底部屏障——每个GPU独立地完成自己的拉取操作。冗余拉取结合了Oneshot(无底部屏障)和Twoshot(良好扩展性)的优点。

技术三:推测式归约消除顶部屏障

即使消除了底部屏障,All-Reduce仍然需要一个顶部屏障,以确保所有GPU都准备好发送数据。这个屏障的代价不容忽视:在8-GPU系统中,顶部屏障需要等待最慢的GPU;GPU之间的计算速度可能存在微小差异;这种等待时间累积起来,显著影响整体性能。

研究团队注意到一个关键现象:在Transformer解码的每一步中,会触发多个All-Reduce操作。这些操作密集执行,使得GPU在第一步之后保持紧密同步。如果第一步成功同步了所有GPU,后续步骤中顶部屏障的作用就变得很小。

基于这一观察,SiFAR提出了推测式归约:在顶部屏障之前,GPU就开始将数据发送到交换机;交换机暂存这些数据,等待所有GPU的数据到达;当顶部屏障完成(确认所有GPU已就绪),交换机立即执行归约并返回结果;如果某个GPU实际未就绪,推测失败,系统回退到传统模式。

这种设计的巧妙之处在于:成功时,数据已经在传输中,顶部屏障的等待时间与通信时间重叠,有效隐藏了延迟;失败时,系统回退到安全模式,保证正确性;实践中,由于GPU高度同步,推测成功率极高。

实验评估:性能提升

研究团队在标准测试环境中评估SiFAR:8×NVIDIA H200 GPU,通过NVLink和InfiniBand互联;测试模型包括Llama-3.1-8B和Qwen3.5-397B-17B;基线为NCCL的默认All-Reduce实现。

在TP=8配置下,SiFAR将All-Reduce延迟降低了52%。端到端吞吐量提升方面,Llama-3.1-8B提升18.6%,单token延迟降低15.7%;Qwen3.5-397B-17B提升13.1%,单token延迟降低11.6%。值得注意的是,Qwen模型的提升比例较低,这是因为更大的模型计算占比更高,通信占比相对较低。

扩展性测试显示,随着GPU数量增加,SiFAR的优势更加明显:2 GPU时延迟降低31%、吞吐量提升8.2%;4 GPU时延迟降低42%、吞吐量提升14.5%;8 GPU时延迟降低52%、吞吐量提升18.6%;16 GPU时延迟降低61%、吞吐量提升22.3%。这是因为传统All-Reduce的同步开销随GPU数量超线性增长,而SiFAR的优化技术有效抑制了这种增长。

实际部署考量

SiFAR的某些优化(特别是冗余拉取)需要现代交换机的支持:最低要求为支持RoCEv2的交换机;最佳性能需要支持交换机内归约的智能网卡或交换机。对于不支持这些特性的旧硬件,SiFAR仍然可以工作,但只能应用双缓冲和推测式归约优化,性能提升会相应降低。

SiFAR被设计为NCCL的透明替代:完全兼容NCCL的All-Reduce接口;在不支持的硬件上自动回退到传统实现;允许用户选择性启用/禁用特定优化。主流推理框架如vLLM、TensorRT-LLM、SGLang可以无缝集成SiFAR,无需修改上层代码。

局限与未来方向

当前局限包括:部分优化需要特定网络硬件支持,限制了通用性;对于极大规模模型,通信占比极高,SiFAR的绝对收益更大,但可能需要结合其他技术如流水线并行;在共享GPU集群中,GPU之间的同步假设可能不成立,影响推测式归约的成功率。

未来研究方向包括:根据运行时条件动态调整优化策略的自适应优化;将SiFAR技术扩展到多机多卡场景;结合权重量化技术进一步降低通信量;设计专门支持无同步All-Reduce的网络芯片。

结语

SiFAR通过深入分析All-Reduce的同步开销来源,提出了三项关键技术——双缓冲、冗余拉取和推测式归约——将多GPU推理的All-Reduce延迟降低52%,端到端吞吐量提升18.6%。更重要的是,SiFAR揭示了一个深层原理:通信与计算的协同设计。通过重新思考通信原语与上层应用的交互方式,我们可以在不牺牲正确性的前提下,显著降低系统开销。随着推理模型和智能体系统的普及,低延迟LLM推理将成为基础设施的核心需求。SiFAR为这一领域提供了重要的技术基础。