config: the runner and scheduler read resolved config from the bags (#34095)
This commit is contained in:
@@ -1553,3 +1553,18 @@ def configured_moe_dp_size() -> int:
|
||||
|
||||
def configured_attn_cp_size() -> int:
|
||||
return _configured_parallel("attn_cp_size")
|
||||
|
||||
|
||||
def is_ep_joiner() -> bool:
|
||||
"""True in a process launched as an elastic-EP joiner (scale or recover).
|
||||
|
||||
A predicate over the published ``exec.moe.ep_join_mode`` leaf, so it follows
|
||||
a post-publish override; the same-named ``ServerArgs`` property is the
|
||||
pre-publish equivalent.
|
||||
"""
|
||||
return get_exec().moe.ep_join_mode in ("scale", "recover")
|
||||
|
||||
|
||||
def is_ep_scale_joiner() -> bool:
|
||||
"""True in a process launched as an elastic-EP scale-up joiner."""
|
||||
return get_exec().moe.ep_join_mode == "scale"
|
||||
|
||||
Reference in New Issue
Block a user