code: def get_balance_profile():
      global balance_dict
      return balance_dict
  
file_path: thu-pacman-faster-moe/fmoe/megatron/balance.py
imports: import torch
  from fmoe.balance import reset_balance_profile
  from fmoe.balance import update_balance_profile
  from fmoe.utils import get_torch_default_comm
  from .distributed import get_moe_group
label_desc: Use image node_id 2 for calling a global function with example usage: get_balance_profile() and returns: balance_dict