code: def get_shard_size_list(total_size, mp_size):
      shard_sizes = []
      for i in range(mp_size):
          shard_sizes.append(get_shard_size(total_size, mp_size, i))
      return shard_sizes
  
imports: from deepspeed import comm
label_desc: Use image node_id 4 for calling a global function with example usage: get_shard_size_list(total_size, mp_size) and returns: shard_sizes