diff --git a/tinygrad/viz/js/index.js b/tinygrad/viz/js/index.js index 9a4fd1c602..26cfa18c2f 100644 --- a/tinygrad/viz/js/index.js +++ b/tinygrad/viz/js/index.js @@ -197,12 +197,15 @@ async function renderProfiler() { // rescale to match current zoom const xscale = d3.scaleLinear().domain([0, et-st]).range([0, canvas.clientWidth]); xscale.domain(xscale.range().map(zoomLevel.invertX, zoomLevel).map(xscale.invert, xscale)); + const zoomDomain = transform != null ? xscale.domain() : null; let yscale = null; if (data.axes.y != null) { yscale = d3.scaleLinear().domain(data.axes.y.domain).range(data.axes.y.range); } // draw shapes for (const e of data.shapes) { + const [start, end] = Array.isArray(e.x) ? [e.x[0], e.x[e.x.length-1]] : [e.x, e.x+e.dur]; + if (zoomDomain != null && (start>zoomDomain[1]|| end