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: fix n² tiny device linking in profiler (#15028)

Browse Source
This commit is contained in:
qazal
2026-02-27 01:25:39 +08:00
committed by GitHub
parent 88d650d606
commit c61fe57cfd
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -375,8 +375,10 @@ async function renderProfiler(path, unit, opts) {
if (shapeRef != null) { ref = {ctx:e.ref, step:0}; shapeRef = ref; }
else if (ref != null) {
const start = ref.step>0 ? ref.step+1 : 0;
const stepIdx = ctxs[ref.ctx+1].steps.findIndex((s, i) => i >= start && s.name == e.name);
if (stepIdx !== -1) { ref.step = stepIdx; shapeRef = ref; }
const steps = ctxs[ref.ctx+1].steps;
for (let si=start; si<steps.length; si++) {
if (steps[si].name == e.name) { ref.step = si; shapeRef = ref; break; }
}
} else {
const steps = ctxs[state.currentCtx].steps;
for (let i=state.currentStep+1; i<steps.length; i++) {