code: def _gen_save_on_cpu_context():
      """
      Generate save on cpu context
      """
  
      context = (
          "with torch.autograd.graph.save_on_cpu(pin_memory=True):\n"
      )
      return context
  
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 3 for calling a global function with example usage: _gen_save_on_cpu_context() and returns: context