code: def get_world_size() -> int:
      if _WORLD_SIZE is None:
          raise RuntimeError("model parallelism was not initialized")
      return _WORLD_SIZE
  
file_path: xformers/examples/llama_inference/mp_utils.py
imports: import functools
  import os
  from typing import Optional
  import torch
  from torch.distributed import ProcessGroup
label_desc: Use image node_id 2 for calling a global function with example usage: get_world_size() and returns: _WORLD_SIZE