Function fmt

examples/tools/bandwidth_test.py:26–28

(x)

      bw[i][j] = SZ / t / 1e9  # GB/s

  def fmt(x):
    c = "green" if x > 50 else "yellow" if x > 20 else "red"
    return colored(f"{x:6.1f}", c)

  # header
  print(" " * 8 + " ".join(f"{'d'+str(j):>6}" for j in range(GPUS)))