From 7703f8b8059680534e2fffec1af791620132e88d Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Wed, 6 Aug 2025 03:12:52 +0800 Subject: [PATCH] viz: skip flops info if estimates is symbolic (#11513) --- tinygrad/viz/serve.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tinygrad/viz/serve.py b/tinygrad/viz/serve.py index e9a0b11a18..e41ae59d1f 100755 --- a/tinygrad/viz/serve.py +++ b/tinygrad/viz/serve.py @@ -132,7 +132,8 @@ def timeline_layout(events:list[tuple[int, int, float, DevEvent]]) -> dict: name, cat, info = e.name, None, None if (ref:=ref_map.get(name)) is not None: name = ctxs[ref]["name"] - if isinstance(p:=contexts[0][ref].ret, ProgramSpec): + # TODO: support symbolic by capturing var_vals in profile events + if isinstance(p:=contexts[0][ref].ret, ProgramSpec) and all(isinstance(es,int) for es in [p.estimates.ops, p.estimates.mem, p.estimates.lds]): info = f"{p.estimates.ops/(t:=dur*1e3):.2f} GFLOPS {p.estimates.mem/t:4.1f}|{p.estimates.lds/t:.1f} GB/s" elif isinstance(e.name, TracingKey): name, cat = e.name.display_name, e.name.cat