code: def _gen_ckpt_usage(
      label, input_vars, output_vars, use_reentrant=True
  ):
      """
      Generate the checkpoint function call code text
      """
      outputs = ", ".join(output_vars)
      inputs = ", ".join(input_vars)
      return f"{outputs} = torch.utils.checkpoint.checkpoint(self.checkpoint_{label}, {inputs}, use_reentrant={use_reentrant})"
  
file_path: ColossalAI/colossalai/_analyzer/fx/codegen.py
imports: from typing import Any, Dict, List, Tuple
  import torch
  from torch.fx.graph import PythonCode, _custom_builtins, _format_target, _is_from_torch, _Namespace, _origin_type_map, _register_custom_builtin, inplace_methods, magic_methods
  from torch.fx.node import Argument, Node, _get_qualified_name, _type_repr, map_arg
  import colossalai
  from colossalai.fx._compatibility import compatibility
label_desc: Use image node_id 3 for calling a global function with example usage: _gen_ckpt_usage(label, input_vars, output_vars, use_reentrant) and returns: outputs, str, label, str, inputs, str, use_reentrant, str