megatron.core.parallel_state.set_pipeline_model_parallel_rank#

megatron.core.parallel_state.set_pipeline_model_parallel_rank(rank)#

Set pipeline model parallel rank.