code: def _get_cuda_home():
      if TORCH_VERSION == "parrots":
          from parrots.utils.build_extension import CUDA_HOME
      else:
          if is_rocm_pytorch():
              from torch.utils.cpp_extension import ROCM_HOME
  
              CUDA_HOME = ROCM_HOME
          else:
              from torch.utils.cpp_extension import CUDA_HOME
      return CUDA_HOME
  
file_path: lavis/lavis/common/annotator/uniformer/mmcv/utils/parrots_wrapper.py
imports: from functools import partial
  import torch
label_desc: Use image node_id 2 for calling a global function with example usage: _get_cuda_home() and returns: CUDA_HOME