config: a parallel leaf with no live counterpart is read bare (#36620)
Co-authored-by: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 5
parent
7c3b5a6732
commit
ca1d7ed8e6
@@ -911,11 +911,11 @@ class TestForwardFlags(_IsolatedServerArgs):
|
||||
@torch.compile(fullgraph=True, backend="eager", dynamic=False)
|
||||
def probe(x):
|
||||
par = get_parallel()
|
||||
if par.config.enable_prefill_context_parallel:
|
||||
if par.enable_prefill_context_parallel:
|
||||
x = x + 1
|
||||
if par.config.moe_dense_tp_size == 1:
|
||||
if par.moe_dense_tp_size == 1:
|
||||
x = x + 2
|
||||
if par.config.dwdp_size > 1:
|
||||
if par.dwdp_size > 1:
|
||||
x = x + 4
|
||||
return x
|
||||
|
||||
|
||||
Reference in New Issue
Block a user