github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

change create_schedule_with_vars api to big_sink [pr] (#8677)

Browse Source
This commit is contained in:
George Hotz
2025-01-19 13:30:26 -08:00
committed by GitHub
parent beba490ba8
commit 168c16646a
4 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/abstractions2.py
View File

@@ -91,7 +91,7 @@ b = b.buf_uop_view()
out = a.alu(Ops.ADD, b)
# schedule the computation as a list of kernels
sched, _, becomes_map = create_schedule_with_vars([out])
sched, _, becomes_map = create_schedule_with_vars(out.sink())
for si in sched: print(si.ast.op) # NOTE: the first two convert it to CLANG
# NOTE: UOps are no longer mutable, the scheduler gives you a map to lookup which BUFFER the result was written to
out = becomes_map[out]