code: def bench(f, iters=100, warmup=10):
      for _ in range(warmup):
          f()
      begin = time.time()
      for _ in range(iters):
          f()
      print(time.time() - begin)
  
file_path: pytorch/functorch/examples/compilation/linear_train.py
imports: import time
  import torch
  import torch.nn
  from functorch import make_functional
  from functorch.compile import nnc_jit
label_desc: Use image node_id 1 for calling a global function with example usage: bench(f, iters, warmup) without return types