code: def _tracer_override(self):
      # override the tracer to support custom modules and checkpointing
      if self.trace_act_ckpt:
          orig_ckpt_func_apply = (
              torch.utils.checkpoint.CheckpointFunction.apply
          )
          orig_ckpt_func_without_reentrant = (
              torch.utils.checkpoint._checkpoint_without_reentrant
          )
  
          def checkpoint(run_function, preserve_rng_state=False, *args):
              self.ckpt_regions.append(self.ckpt_idx)
              out = run_function(*args)
              self.ckpt_idx = self.ckpt_regions.pop(-1) + 1
              return out
  
          # override the checkpoint function
          torch.utils.checkpoint.CheckpointFunction.apply = checkpoint
          torch.utils.checkpoint._checkpoint_without_reentrant = (
              checkpoint
          )
  
      # override the custom functions
      ColoProxy._func_dispatch.update(
          {k: v for k, v in self._custom_impl.items()}
      )
  
      # override the bias addition functions
      if self.bias_addition_split:
          ColoProxy._func_dispatch.update(
              {k: v for k, v in self._bias_addition_impl.items()}
          )
  
      yield
  
      if self.trace_act_ckpt:
          # recover the checkpoint function upon exit
          torch.utils.checkpoint.CheckpointFunction.apply = (
              orig_ckpt_func_apply
          )
          torch.utils.checkpoint._checkpoint_reentrant = (
              orig_ckpt_func_without_reentrant
          )
  
      ColoProxy._func_dispatch = {}
  
imports: import functools
  import inspect
  from contextlib import contextmanager
  from typing import Any, Callable, Dict, Iterable, Optional, Set, Tuple, Type, Union
  import torch
  import torch.nn
  from torch.fx import Graph, Node, Proxy, Tracer
  from torch.utils._pytree import tree_map
  from colossalai._analyzer._subclasses import _TensorPropertyMethod, _TorchFactoryMethod
  from ..node_util import MetaInfo
  from .proxy import ColoProxy
label_desc: Use image node_id 8 for calling the ColoTracer obj's underlying member method code with example usage: obj._tracer_override() and returns: out