code: def register_test(fxn):
      tests[fxn.__name__] = fxn
  
file_path: tinygrad/extra/archprobe.py
imports: import numpy
  import pickle
  from tinygrad.runtime.ops_gpu import CLProgram, CLBuffer
  from tinygrad import dtypes
  from tqdm import trange, tqdm
  from matplotlib import pyplot
label_desc: Use image node_id 1 for calling a global function with example usage: register_test(fxn) without return types