code: def track_graph_compiling(aot_config, graph_name):
      global graph_being_compiled
      # TODO: Don't shove the aot_id in here; set it in the context
      graph_being_compiled = [f"{aot_config.aot_id}_{graph_name}"]
      try:
          yield
      finally:
          global nth_graph
          nth_graph += 1
          graph_being_compiled = []
  
file_path: pytorch/torch/_functorch/_aot_autograd/logging_utils.py
imports: import collections
  from contextlib import contextmanager
  from typing import List, Tuple
  import torch
  import torch.fx.traceback
label_desc: Use image node_id 4 for calling a global function with example usage: track_graph_compiling(aot_config, graph_name) without return types