code: def _test_perf(param, optimizer_func):
      optimizer = optimizer_func(param)
      avg = 0
      for i in range(NUM_ITERS):
          for i, p in enumerate(param):
              p.grad = torch.ones_like(p) * 2
          start = time.time()
          optimizer.step()
          stop = time.time()
          avg += stop - start
  
      return avg / NUM_ITERS
  
imports: import torch
  from deepspeed.ops.adam import DeepSpeedCPUAdam
  import time
label_desc: Use image node_id 1 for calling a global function with example usage: _test_perf(param, optimizer_func) and returns: unknown