mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

viz: bring back flops and mem in sidebar (#13467)

Browse source
This commit is contained in:
qazal 2025-11-27 17:27:44 +08:00 • committed by GitHub
commit 2df8a3474e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tinygrad/viz/js/index.js
View file

@ -202,8 +202,10 @@ function getMetadata(shape) {
const e = track.shapes[idx];
const html = d3.create("div").classed("info", true);
if (track.eventType === EventTypes.EXEC) {
html.append(() => tabulate([["Name", d3.create("p").html(e.arg.tooltipText.split("\n")[0]).node()],
["Duration", formatTime(e.width)], ["Start Time", formatTime(e.x)]]).node());
const [n, _, ...rest] = e.arg.tooltipText.split("\n");
html.append(() => tabulate([["Name", d3.create("p").html(n).node()], ["Duration", formatTime(e.width)], ["Start Time", formatTime(e.x)]]).node());
const group = html.append("div").classed("args", true);
for (const r of rest) group.append("p").text(r);
if (e.arg.ctx != null) {
const i = e.arg.ctx; s = e.arg.step;
html.append("a").text(ctxs[i+1].steps[s].name).on("click", () => switchCtx(i, s));