config: spell the parallel config tier at the call site (#36250)

2026-08-26 03:00:28 -07:00
parent 689ade69d1
commit 8005df61d3
136 changed files with 1033 additions and 795 deletions
@@ -542,7 +542,7 @@ def _maybe_prepare_mlp_sync_batch(batch: ScheduleBatch, model_runner):
        prepare_mlp_sync_batch_raw(
            batch,
            model_runner=model_runner,
-            dp_size=get_parallel().dp_size,
+            dp_size=get_parallel().config.dp_size,
            attn_tp_size=get_parallel().attn_tp_size,
            attn_cp_size=model_runner.ps.attn_cp_size,
            tp_group=model_runner.tp_group,