[BCG][3/N] Enable bcg on dsa & deepep a2a backend (#31987)

Co-authored-by: Claude Fable 5 <noreply@anthropic.com>
2026-07-31 16:45:18 -07:00
parent 3c5f115741
commit 3e0f7c3f30
13 changed files with 319 additions and 116 deletions
@@ -536,6 +536,7 @@ def _maybe_prepare_mlp_sync_batch(batch: ScheduleBatch, model_runner):
    if require_mlp_sync(model_runner.server_args):
        prepare_mlp_sync_batch_raw(
            batch,
+            model_runner=model_runner,
            dp_size=model_runner.server_args.dp_size,
            attn_tp_size=get_parallel().attn_tp_size,
            attn_cp_size=model_runner.ps.attn_cp_size,