code: def num_gpus() -> int:
      if (
          hasattr(torch.backends, "mps")
          and torch.backends.mps.is_available()
      ):
          return 1
      return torch.cuda.device_count()
  
imports: import inspect
  import os
  import importlib
  import torch
  from typing import Optional, Union
  from torch.profiler import ProfilerActivity
  from torch.profiler import profile
  import ivy
  from ivy.functional.ivy.device import _shift_native_arrays_on_default_device, Profiler
label_desc: Use image node_id 6 for calling a global function with example usage: num_gpus() and returns: torch, int