[Fix] Read the DSA prefill CP flag from the parallel config bag in bootstrap (#35110)

2026-08-17 00:21:41 -07:00
parent 721e359ca7
commit 0d8c850a35
1 changed files with 1 additions and 1 deletions
@@ -239,7 +239,7 @@ def _init_parallel_groups(
        duplicate_attn_cp_group=(
            is_hip()
            and server_args.enable_two_batch_overlap
-            and server_args.enable_dsa_prefill_context_parallel
+            and get_parallel().enable_dsa_prefill_context_parallel
        ),
        enable_symm_mem=server_args.enable_symm_mem,
        recovered_rank=is_ep_joiner,