跳转到内容

序列与上下文并行

当序列长度达到数十万 token,单卡放不下一个序列的激活。上下文并行(Ring Attention、DeepSpeed-Ulysses)在序列维度切分并在注意力计算中交换 KV。本章比较两者的通信模式与适用场景。

  • Ring Attention 通信量随序列长度线性增长但可与计算重叠
  • Ulysses 用 all-to-all 切换分片维度,受限于头数
  • 因果掩码带来的负载不均衡需要特殊调度