github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

move colorize_float to helpers.py (#5490)

Browse Source 
* add colorize_float to helpers.py

* update references
This commit is contained in:
Edward Wang
2024-07-15 14:29:03 -04:00
committed by GitHub
parent a347d91e0e
commit 9a7d5a148e
4 changed files with 4 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
test/external/speed_compare_cuda_nv.py vendored
View File

@@ -1,5 +1,5 @@
from tinygrad import Device, dtypes
from tinygrad.helpers import getenv, colored
from tinygrad.helpers import getenv, colorize_float
from extra.optimization.helpers import load_worlds, ast_str_to_lin
from test.external.fuzz_linearizer import get_fuzz_rawbufs
from tinygrad.engine.search import bufs_from_lin
@@ -7,13 +7,6 @@ from tinygrad.engine.realize import CompiledRunner
from tinygrad.tensor import _to_np_dtype
import numpy as np
# move to helpers?
def colorize_float(x):
ret = f"{x:7.2f}x"
if x < 0.75: return colored(ret, 'green')
elif x > 1.15: return colored(ret, 'red')
else: return colored(ret, 'yellow')
if __name__ == "__main__":
ast_strs = load_worlds(filter_reduce=False, filter_novariable=True)
cudev = Device["CUDA"]