code: def recv_meta_helper(prev_rank, tensor_kwargs):
      recv_ndims = torch.empty((), **tensor_kwargs)
      dist.recv(recv_ndims, prev_rank)
      recv_shape = torch.empty(recv_ndims, **tensor_kwargs)
      dist.recv(recv_shape, prev_rank)
      return recv_shape
  
file_path: colossalai/colossalai/legacy/communication/utils.py
imports: from typing import List, Tuple, Union
  import torch
  import torch.distributed
  from colossalai.legacy.context.parallel_mode import ParallelMode
  from colossalai.legacy.core import global_context
  from colossalai.utils import get_current_device
label_desc: Use image node_id 3 for calling a global function with example usage: recv_meta_helper(prev_rank, tensor_kwargs) and returns: recv_shape