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

hotfix: process_replay/diff_schedule.py to LBScheduleItem

Browse Source
This commit is contained in:
George Hotz
2024-08-06 16:59:37 -07:00
parent aee737bd9e
commit 3a0515ea22
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/external/process_replay/diff_schedule.py vendored
View File

@@ -12,7 +12,7 @@ def diff_schedule(s):
si_for_buf: DefaultDict[LazyBuffer, List[ScheduleItem]] = defaultdict(list)
for _,prescheduled in s:
for ps in prescheduled.values():
for buf in ps[0]: si_for_buf[buf].append(ScheduleItem(ps[1], tuple(x.buffer for x in ps[0]+ps[2] if x.size != 0), ps[4]))
for buf in ps.outputs: si_for_buf[buf].append(ScheduleItem(ps.ast, tuple(x.buffer for x in ps.outputs+ps.inputs if x.size != 0), ps.metadata))
changed = 0
seen_diff: Set[Tuple[LazyOp, LazyOp]] = set()
for buf, si in si_for_buf.items():