code: def clear_cached_mem_on_dev(
      device: Union[ivy.Device, torch.device], /
  ) -> None:
      torch_dev = as_native_dev(device)
      if torch_dev.type == "cuda":
          torch.cuda.empty_cache()
      elif torch_dev.type == "mps":
          from torch import mps
  
          mps.empty_cache()
  
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 5 for calling a global function with example usage: clear_cached_mem_on_dev() without return types