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