megatron.core.parallel_state.get_model_parallel_group#

megatron.core.parallel_state.get_model_parallel_group()#

Get the model parallel group the caller rank belongs to.