github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 23:48:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

viz: support list of sidebar info (#13612)

Browse Source
This commit is contained in:
qazal
2025-12-08 03:09:43 +08:00
committed by GitHub
parent b981b6f89e
commit 73093314bd
2 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tinygrad/viz/js/index.js
View File

@@ -800,10 +800,13 @@ async function main() {
if (ret.cols != null) {
renderTable(root, ret);
} else root.append(() => codeBlock(ret.src, ret.lang || "txt"));
if (ret.metadata != null) metadata.appendChild(tabulate(ret.metadata.map(({ label, value, idx }) => {
const div = d3.create("div").style("background", cycleColors(colorScheme.CATEGORICAL, idx)).style("width", "100%").style("height", "100%");
return [label.trim(), div.text(typeof value === "string" ? value : formatUnit(value)).node()];
})).node());
ret.metadata?.forEach(m => {
if (Array.isArray(m)) return metadata.appendChild(tabulate(m.map(({ label, value, idx }) => {
const div = d3.create("div").style("background", cycleColors(colorScheme.CATEGORICAL, idx)).style("width", "100%").style("height", "100%");
return [label.trim(), div.text(typeof value === "string" ? value : formatUnit(value)).node()];
})).node());
metadata.appendChild(codeBlock(m.src, "txt")).classList.add("full-height")
});
return document.querySelector("#custom").replaceChildren(root.node());
}
// ** UOp view (default)

8
tinygrad/viz/serve.py
View File

@@ -346,10 +346,10 @@ def get_llvm_mca(asm:str, mtriple:str, mcpu:str) -> dict:
instr_usage.setdefault(i:=d["InstructionIndex"], {}).setdefault(r:=d["ResourceIndex"], 0)
instr_usage[i][r] += d["ResourceUsage"]
# last row is the usage summary
metadata = [{"idx":k, "label":resource_labels[k], "value":v} for k,v in instr_usage.pop(len(rows), {}).items()]
summary = [{"idx":k, "label":resource_labels[k], "value":v} for k,v in instr_usage.pop(len(rows), {}).items()]
max_usage = max([sum(v.values()) for i,v in instr_usage.items() if i<len(rows)], default=0)
for i,usage in instr_usage.items(): rows[i].append([[k, v, (v/max_usage)*100] for k,v in usage.items()])
return {"rows":rows, "cols":["Instruction", "Latency", {"title":"HW Resources", "labels":resource_labels}], "metadata":metadata}
return {"rows":rows, "cols":["Instruction", "Latency", {"title":"HW Resources", "labels":resource_labels}], "metadata":[summary]}
def get_stdout(f: Callable) -> str:
buf = io.StringIO()
@@ -394,9 +394,9 @@ def get_render(i:int, j:int, fmt:str) -> dict:
inst["stall"] += e.stall
inst["hits"]["rows"].append((inst["hit_count"]-1, e.time, max(0, e.time-prev_instr), e.dur, e.stall))
prev_instr = max(prev_instr, e.time + e.dur)
metadata = [{"label":"Total Cycles", "value":w.end_time-w.begin_time}, {"label":"SE", "value":w.se}, {"label":"CU", "value":w.cu},
summary = [{"label":"Total Cycles", "value":w.end_time-w.begin_time}, {"label":"SE", "value":w.se}, {"label":"CU", "value":w.cu},
{"label":"SIMD", "value":w.simd}, {"label":"Wave ID", "value":w.wave_id}, {"label":"Run number", "value":data["run_number"]}]
return {"rows":[tuple(v.values()) for v in rows.values()], "cols":columns, "metadata":metadata}
return {"rows":[tuple(v.values()) for v in rows.values()], "cols":columns, "metadata":[summary]}
return data
# ** HTTP server