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

fix abstracions2 printout (#5557)

Browse Source
This commit is contained in:
kormann
2024-07-18 20:21:45 +02:00
committed by GitHub
parent a7fec05acc
commit c951bc99af
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/abstractions2.py
View File

@@ -58,7 +58,7 @@ sink = LazyOp(MetaOps.KERNEL, (st_0,))
# convert the computation to a "linearized" format (print the format)
from tinygrad.engine.realize import get_kernel, CompiledRunner
lin = get_kernel(Device[DEVICE].renderer, sink).linearize()
for u in lin.uops: print(u)
lin.uops.print()
# compile a program (and print the source)
fxn = CompiledRunner(lin.to_program())