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