diff --git a/tinygrad/viz/js/index.js b/tinygrad/viz/js/index.js index 82eb89541c..9bc7cbb92b 100644 --- a/tinygrad/viz/js/index.js +++ b/tinygrad/viz/js/index.js @@ -353,7 +353,11 @@ async function renderProfiler(path, unit, opts) { colorMap.set(colorKey, d3.rgb(color)); } const fillColor = colorMap.get(colorKey).brighter(0.3*depth).toString(); - const label = parseColors(e.name).map(({ color, st }) => ({ color, st, width:ctx.measureText(st).width })); + const label = parseColors(e.name).flatMap(({ color, st }) => { + const parts = []; + for (let i=0; i