code: def __enter__(self):
      global CURRENT_DEVICE
      self.old_device = CURRENT_DEVICE
      CURRENT_DEVICE = self.device
      return super().__enter__()
  
file_path: pytorch/torch/utils/_device.py
imports: import torch
  from torch.overrides import TorchFunctionMode
  from torch.utils._contextlib import context_decorator
  import functools
label_desc: Use image node_id 2 for calling the DeviceContext obj's underlying member method code with example usage: obj.__enter__() and returns: super