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: simplify row line logic (#14227)

Browse Source
This commit is contained in:
qazal
2026-01-19 10:00:28 -05:00
committed by GitHub
parent 01653db4fd
commit ab426cb671
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -325,7 +325,7 @@ async function renderProfiler(path, unit, opts) {
const eventType = u8(), eventsLen = u32();
const [pcolor, scolor] = path.includes("pkts") ? ["#00c72f", "#858b9d"] : ["#9ea2ad", null];
// last row doesn't get a border
const rowBorderColor = i<layoutsLen ? "#22232a" : null;
const rowBorderColor = i<layoutsLen-1 ? "#22232a" : null;
if (rowBorderColor != null) div.style("border-bottom", `1px solid ${rowBorderColor}`);
if (eventType === EventTypes.EXEC) {
const levelHeight = (baseHeight-padding)*(opts.heightScale ?? 1);
@@ -377,9 +377,7 @@ async function renderProfiler(path, unit, opts) {
shapes.push({x:e.st, y:levelHeight*depth, width:e.dur, height:levelHeight, arg, label:opts.hideLabels ? null : label, fillColor });
if (j === 0) data.first = data.first == null ? e.st : Math.min(data.first, e.st);
}
const height = levelHeight*levels.length;
div.style("height", height+padding+"px").style("pointerEvents", "none");
data.tracks.get(k).height = Math.max(rect(div.node()).height-padding, levelHeight*levels.length);
div.style("height", levelHeight*levels.length+padding+"px").style("pointerEvents", "none");
} else {
const peak = u64();
let x = 0, y = 0;
@@ -485,7 +483,7 @@ async function renderProfiler(path, unit, opts) {
ctx.textBaseline = "middle";
// draw shapes
const paths = [];
for (const [_, { shapes, eventType, visible, offsetY, valueMap, pcolor, scolor, height, rowBorderColor }] of data.tracks) {
for (const [k, { shapes, eventType, visible, offsetY, valueMap, pcolor, scolor, rowBorderColor }] of data.tracks) {
visible.length = 0;
const addBorder = scolor != null ? (p,w) => { if (w > 10) { ctx.strokeStyle = scolor; ctx.stroke(p); } } : null;
for (const e of shapes) {
@@ -519,7 +517,7 @@ async function renderProfiler(path, unit, opts) {
}
// draw row line
if (rowBorderColor != null) {
const y = offsetY+height+padding/2 - 0.5;
const y = offsetY+rect(document.getElementById(k)).height-padding/2 - 0.5;
drawLine(ctx, [0, canvasWidth], [y, y], { color:rowBorderColor });
}
}