code: def run_matmul(X, weight):
      torch.cuda.synchronize()
      start = time.time()
      for i in range(N):
          x = X[i % 10]
          U = x.matmul(weight)
          loss = U.mean()
          loss.backward()
      torch.cuda.synchronize()
      print("matmul() takes: {}".format(time.time() - start))
  
file_path: sru/misc/test_mm.py
imports: import time
  import torch
  import torch.nn
label_desc: Use image node_id 2 for calling a global function with example usage: run_matmul(X, weight) without return types