code: def get_num_kv_heads():
      global num_kv_heads
      return num_kv_heads
  
imports: from deepspeed import comm
label_desc: Use image node_id 2 for calling a global function with example usage: get_num_kv_heads() and returns: num_kv_heads