diff --git a/tinygrad/viz/js/index.js b/tinygrad/viz/js/index.js index 87d8c128c5..037626f915 100644 --- a/tinygrad/viz/js/index.js +++ b/tinygrad/viz/js/index.js @@ -375,7 +375,7 @@ async function renderProfiler(path, unit, opts) { // tiny device events go straight to the rewrite rule const key = k.startsWith("TINY") ? null : `${k}-${j}`; const labelHTML = label.map(l=>`${l.st}`).join(""); - const arg = { tooltipText:labelHTML+"\n"+formatTime(e.dur)+(e.info != null ? "\n"+e.info : ""), bufs:[], key, + const arg = { tooltipText:labelHTML+" N:"+shapes.length+"\n"+formatTime(e.dur)+(e.info != null ? "\n"+e.info : ""), bufs:[], key, ctx:shapeRef?.ctx, step:shapeRef?.step }; if (e.key != null) shapeMap.set(e.key, key); // offset y by depth