code: def test_reg_count(nthread=1, ngrp=1):
      base = reg_count(nthread, ngrp, 1)
      return [
          (nreg, (reg_count(nthread, ngrp, nreg) - base) / nreg)
          for nreg in trange(4, 513, 4)
      ]
  
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 5 for calling a global function with example usage: test_reg_count(nthread, ngrp) and returns: unknown