code: def _current_device(module: torch.nn.Module):
      try:
          return next(module.parameters()).device
      except:
          return _default_device()
  
file_path: colossalai/colossalai/_analyzer/fx/tracer/symbolic_trace.py
imports: from typing import Any, Callable, Dict, Optional, Union
  import torch
  from torch.fx import Tracer
  from torch.utils._pytree import tree_map
  from colossalai._analyzer._subclasses import MetaTensor
  from ..graph_module import ColoGraphModule
  from .tracer import ColoTracer
label_desc: Use image node_id 2 for calling a global function with example usage: _current_device(module) and returns: next, _default_device