code: def get_data_parallel_src_rank():
      """Calculate the global rank corresponding to the first local rank in the data parallel group."""
      global_rank = torch.distributed.get_rank()
      data_parallel_size: int = get_data_parallel_world_size()
      num_data_parallel_groups = (
          torch.distributed.get_world_size() // data_parallel_size
      )
      return global_rank % num_data_parallel_groups
  
imports: from typing import Tuple, Optional
  import warnings
  import os
  import torch
  from apex.transformer.log_util import get_transformer_logger
  from apex.transformer._ucc_util import HAS_UCC
label_desc: Use image node_id 44 for calling a global function with example usage: get_data_parallel_src_rank() and returns: unknown