code: def profile_it(f, inp):
      for _ in range(5):
          f(inp)
  
      itr = 5
      with profile(
          activities=[ProfilerActivity.CUDA], record_shapes=True
      ) as prof:
          for _ in range(itr):
              f(inp)
  
      timing = prof.key_averages()
      cuda_time_total = 0
      for e in timing:
          cuda_time_total = cuda_time_total + e.cuda_time_total
      return cuda_time_total / itr
  
file_path: pytorch/functorch/benchmarks/cse.py
imports: import torch
  import torch.fx
  from functorch import make_fx
  from torch._functorch.compile_utils import fx_graph_cse
  from torch.profiler import profile, ProfilerActivity
label_desc: Use image node_id 1 for calling a global function with example usage: profile_it(f, inp) and returns: unknown