diff --git a/tinygrad/viz/js/index.js b/tinygrad/viz/js/index.js index 08c363d39d..c6d694c781 100644 --- a/tinygrad/viz/js/index.js +++ b/tinygrad/viz/js/index.js @@ -221,7 +221,13 @@ async function renderProfiler() { ctx.closePath(); ctx.fill(); // NOTE: y coordinates are in reverse order - for (let i = 0; i < x.length - 1; i++) rectLst.push({ x0:x[i], x1:x[i+1], y0:e.y1[i], y1:e.y0[i], arg:e.arg }); + for (let i = 0; i < x.length - 1; i++) { + let tooltipText = e.arg.tooltipText; + if (yscale != null && ((yaxisVal=yscale.invert(e.y1[i]))>0)) { + tooltipText += `\nTotal: ${formatUnit(yaxisVal, data.axes.y.fmt)}`; + } + rectLst.push({ x0:x[i], x1:x[i+1], y0:e.y1[i], y1:e.y0[i], arg:{...e.arg, tooltipText} }); + } continue; } // contiguous rect