code: def graph_it(bottleneck, x):
      print("Graphing")
      with torch.no_grad():
          x = x.clone()
          x.grad = None
          x.requires_grad = True
      return torch.cuda.make_graphed_callables(bottleneck, (x,))
  
imports: import unittest
  import torch
  from torch.testing._internal import common_utils
  from apex.transformer.testing.distributed_test_base import NcclDistributedTestBase
label_desc: Use image node_id 6 for calling a global function with example usage: graph_it(bottleneck, x) and returns: torch