执行摘要

• 一句话：移除 EPLB 同步路径中无用的 cuda sync
• 推荐动作：值得合并，代码清理明确且验证充分。

功能与动机

对应 Issue #32028 中的痛点 6：同步 EPLB 路径中存在一个原始作者也不清楚为何需要的神秘同步调用。删除它可提高代码可维护性并消除误导。

实现拆解

1. 在 vllm/distributed/eplb/rebalance_execute.py 的 rearrange_expert_weights_inplace 函数中，删除第 587-589 行的 torch.accelerator.synchronize() 调用及其注释。
2. 验证 SYNC 路径所有操作（torch.empty_like、b.copy_(w, non_blocking=True)、torch.distributed.send/recv）均在默认 CUDA 流上执行，没有跨流风险。
3. ASYNC 路径不受影响，因为它使用独立的 cuda_stream.synchronize() 和 CpuGpuEvent 机制。

关键文件：

• vllm/distributed/eplb/rebalance_execute.py（模块 分布式；类别 source；类型 core-logic；符号 rearrange_expert_weights_inplace）: 唯一变更文件，移除无用的 GPU 同步调用

关键符号：rearrange_expert_weights_inplace

关键源码片段

vllm/distributed/eplb/rebalance_execute.py

唯一变更文件，移除无用的 GPU 同步调用

# 删除前：
# # NOTE(bowen): We need this synchronize to run, but I don't know why.
# # If you figure out the reason, please let me know -- thank you!
# torch.accelerator.synchronize()
#
# 删除后：该段完全移除，紧接着进行索引转换和循环
    old_global_expert_indices_cpu = old_global_expert_indices.cpu().numpy()
    new_global_expert_indices_cpu = new_global_expert_indices.cpu().numpy()
​
    for layer_idx in range(num_moe_layers):
        transfer_metadata = move_to_buffer(...)
        move_from_buffer(...)

评论区精华

gemini-code-assist[bot] 建议完全删除注释掉的代码和过时的 NOTE 注释，而不是仅注释掉，以提高代码可维护性。ilmarkov 表示同意。

• 暂无高价值评论线程

风险与影响

• 风险：低风险。仅删除一行注释为“不知道为什么需要”的同步调用，且 PR 提供了 100k 次迭代的稳定性测试（2x A100）和字节码审计证明该函数不再调用任何同步原语。ASYC 路径完全不受影响。
• 影响：仅影响同步 EPLB 路径的专家权重重排流程，移除一个无用的全局同步点，理论上可带来微小的性能提升（约 1-2 us 延迟降低）。对用户无功能性影响。
• 风险标记：低风险

关联脉络

• 暂无明显关联 PR