github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: sched comment

Browse Source
This commit is contained in:
George Hotz
2025-01-31 12:10:04 +08:00
parent 1fce864a6d
commit e63d160376
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tinygrad/engine/schedule.py
View File

@@ -494,6 +494,11 @@ def create_schedule_with_vars(big_sink:UOp, skip_check:bool=not __debug__) -> tu
# group realizes into kernels
store_groups = group_realizes(ctx)
graph_rewrite(sink, break_sched, ctx)
# TODO: this should be the break between the "grouper" and the "linearizer"
# here, there should just be one sink UOp with BUFFER/KERNEL/COPY/ASSIGN (assign is the parent if you want the buffer post assign)
# call into `def linearize_schedule(sched_sink:UOp) -> list[ScheduleItem]`
# create schedule items + map buffers to realized tensors
prescheduled: list[ScheduleItem] = []
for store_uops in store_groups: