From 617aa2b248d14446c7c7fb431ff68532efc1e857 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Sun, 7 Sep 2025 02:12:32 -0700 Subject: [PATCH] [Auto Sync] Update parallel_state.py (20250907) (#10126) Co-authored-by: github-actions[bot] Co-authored-by: jzhou-xai --- python/sglang/srt/distributed/parallel_state.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/python/sglang/srt/distributed/parallel_state.py b/python/sglang/srt/distributed/parallel_state.py index bba83a95f..984582351 100644 --- a/python/sglang/srt/distributed/parallel_state.py +++ b/python/sglang/srt/distributed/parallel_state.py @@ -1596,6 +1596,16 @@ def get_tensor_model_parallel_rank(): return get_tp_group().rank_in_group +def get_pipeline_model_parallel_world_size(): + """Return world size for the pipeline model parallel group.""" + return get_pp_group().world_size + + +def get_pipeline_model_parallel_rank(): + """Return my rank for the pipeline model parallel group.""" + return get_pp_group().rank_in_group + + def get_moe_expert_parallel_world_size(): """Return world size for the moe expert parallel group.""" return get_moe_ep_group().world_size