nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 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
parent 05cd2279d0
commit 2df8a3474e
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/viz/js/index.js
+4 -2
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));