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

remove iter from uopgraph (#6110)

Browse Source 
* remove iter from uopgraph

* linearize returns uops

* fix tests

* linearize in linearize

* tests fix

* touchup

* test failures
This commit is contained in:
George Hotz
2024-08-16 15:58:29 -07:00
committed by GitHub
parent 28c75bf2a6
commit 74ee9febec
15 changed files with 104 additions and 129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/abstractions2.py
View File

@@ -62,7 +62,6 @@ s = UOp(UOps.SINK, None, (st_0,))
# convert the computation to a "linearized" format (print the format)
from tinygrad.engine.realize import get_kernel, CompiledRunner
kernel = get_kernel(Device[DEVICE].renderer, s).linearize()
kernel.uops.print()
# compile a program (and print the source)
fxn = CompiledRunner(kernel.to_program())