Drop ModelRunner's duplicated parallel-degree fields and read them via self.ps (#31165)

2026-07-14 16:02:40 +08:00
parent cfd17301a8
commit 6999007a13
20 changed files with 107 additions and 138 deletions
@@ -530,7 +530,7 @@ def _maybe_prepare_mlp_sync_batch(batch: ScheduleBatch, model_runner):
            batch,
            dp_size=model_runner.server_args.dp_size,
            attn_tp_size=get_parallel().attn_tp_size,
-            attn_cp_size=model_runner.attn_cp_size,
+            attn_cp_size=model_runner.ps.attn_cp_size,
            tp_group=model_runner.tp_group,
            get_idle_batch=None,
            disable_cuda_graph=model_runner.server_args.disable_cuda_graph,