code: def _gen_ckpt_usage(
      label,
      activation_offload,
      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} = colossalai.utils.activation_checkpoint.checkpoint(self.checkpoint_{label}, {activation_offload}, {inputs}, use_reentrant={use_reentrant})"
  
file_path: ColossalAI/colossalai/fx/codegen/activation_checkpoint_codegen.py
imports: from typing import Any, Dict, Iterable, List, Tuple
  import torch
  import colossalai
label_desc: Use image node_id 9 for calling a global function with example usage: _gen_ckpt_usage(label, activation_offload, input_vars, output_vars, use_reentrant) and returns: outputs, str, label, str, activation_offload, str, inputs, str, use_reentrant, str