github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

viz: convert to function_name in server [pr] (#10951)

Browse Source 
* viz: convert to function_name in server [pr]

* it exists
This commit is contained in:
qazal
2025-06-24 13:59:37 +03:00
committed by GitHub
parent c0d9cf09e0
commit ab8c5d04ab
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -252,7 +252,7 @@ async function renderProfiler() {
}
}
const kernelMap = new Map();
for (const [i, c] of ctxs.entries()) kernelMap.set(c.name.replace(/\x1b\[\d+m(.*?)\x1b\[0m/g, "$1"), { name:c.name, i });
for (const [i, c] of ctxs.entries()) kernelMap.set(c.function_name, { name:c.name, i });
// place devices on the y axis and set vertical positions
const [tickSize, padding] = [10, 8];
const deviceList = document.getElementById("device-list");

2
tinygrad/viz/js/worker.js
View File

@@ -12,7 +12,7 @@ onmessage = (e) => {
for (let [k, {label, src, ref, ...rest }] of Object.entries(graph)) {
const idx = ref ? ctxs.findIndex(k => k.ref === ref) : -1;
// replace colors in label
if (idx != -1) label += `\ncodegen@${ctxs[idx].name.replace(/\x1b\[\d+m(.*?)\x1b\[0m/g, "$1")}`;
if (idx != -1) label += `\ncodegen@${ctxs[idx].function_name}`;
// adjust node dims by label size + add padding
let [width, height] = [0, 0];
for (line of label.split("\n")) {

2
tinygrad/viz/serve.py
View File

@@ -25,7 +25,7 @@ def get_metadata(keys:list[Any], contexts:list[list[TrackedGraphRewrite]]) -> li
ret = []
for k,v in zip(keys, contexts):
steps = [{"name":s.name, "loc":s.loc, "depth":s.depth, "match_count":len(s.matches), "code_line":lines(s.loc[0])[s.loc[1]-1].strip()} for s in v]
if isinstance(k, ProgramSpec): ret.append({"name":k.name, "kernel_code":k.src, "ref":id(k.ast), "steps":steps})
if isinstance(k, ProgramSpec): ret.append({"name":k.name, "kernel_code":k.src, "ref":id(k.ast), "function_name":k.function_name, "steps":steps})
else: ret.append({"name":str(k), "steps":steps})
return ret