From 4042bc2399cf4e9fb058265798a2c858ead95c84 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Mon, 22 Jul 2024 20:11:15 -0700 Subject: [PATCH] hotfix: put that space back in DEBUG=2 --- tinygrad/engine/realize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/engine/realize.py b/tinygrad/engine/realize.py index 334606edb3..4cf1c18424 100644 --- a/tinygrad/engine/realize.py +++ b/tinygrad/engine/realize.py @@ -183,7 +183,7 @@ class ExecItem: lds_est = sym_infer(self.prg.lds_estimate, var_vals) ptm = (colored(f"{et*1e3:9.2f}ms", "yellow") if et > 0.01 else f"{et*1e6:9.2f}us") if et is not None else "" print(f"{colored(f'*** {self.prg.dname[:7]:7s} {GlobalCounters.kernel_count:4d}', 'magenta' if jit else ('green' if self.prg.first_run else None))} {self.prg.display_name+' '*(38-ansilen(self.prg.display_name))} mem {GlobalCounters.mem_used/1e9:5.2f} GB " + # noqa: E501 - (str() if et is None else f"tm {ptm}/{GlobalCounters.time_sum_s*1e3:9.2f}ms ({op_est/((et or 1e-20)*1e9):8.2f} GFLOPS {mem_est/((et or 1e-20)*1e9):6.1f}|{lds_est/((et or 1e-20)*1e9):<7.1f}GB/s)" + # noqa: E501 + (str() if et is None else f"tm {ptm}/{GlobalCounters.time_sum_s*1e3:9.2f}ms ({op_est/((et or 1e-20)*1e9):8.2f} GFLOPS {mem_est/((et or 1e-20)*1e9):6.1f}|{lds_est/((et or 1e-20)*1e9):<7.1f} GB/s)" + # noqa: E501 f" {[repr(m) if TRACEMETA >= 2 else str(m) for m in self.metadata] if self.metadata else ''}")) self.prg.first_run = False return et