megatron.core.parallel_state.get_pipeline_model_parallel_group

megatron.core.parallel_state.get_pipeline_model_parallel_group()

Get the pipeline model parallel group the caller rank belongs to.