code: def base_nk(h, i):
      return [(3 * h, h), (h, h), (i, h), (h, i)]
  
file_path: lightseq/lightseq/training/ops/pytorch/gemm_test.py
imports: import os
  import functools
  from dataclasses import dataclass
  from copy import deepcopy
  from tqdm import tqdm
  from lightseq.training.ops.pytorch.builder import KernelBuilder
label_desc: Use image node_id 1 for calling a global function with example usage: base_nk(h, i) and returns: list