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

doc: variable names in abstractions2.py (#5591)

Browse Source
This commit is contained in:
chenyu
2024-07-19 17:06:15 -04:00
committed by GitHub
parent 00c01f6f4d
commit 3acd8559f4
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docs/abstractions2.py
View File

@@ -53,15 +53,15 @@ ld_1 = LazyOp(BufferOps.LOAD, (), MemBuffer(1, dtypes.int32, ShapeTracker.from_s
ld_2 = LazyOp(BufferOps.LOAD, (), MemBuffer(2, dtypes.int32, ShapeTracker.from_shape((1,))))
alu = LazyOp(BinaryOps.ADD, (ld_1, ld_2))
st_0 = LazyOp(BufferOps.STORE, (alu,), MemBuffer(0, dtypes.int32, ShapeTracker.from_shape((1,))))
sink = LazyOp(MetaOps.KERNEL, (st_0,))
k = 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()
lin.uops.print()
kernel = get_kernel(Device[DEVICE].renderer, k).linearize()
kernel.uops.print()
# compile a program (and print the source)
fxn = CompiledRunner(lin.to_program())
fxn = CompiledRunner(kernel.to_program())
print(fxn.p.src)
# NOTE: fxn.clprg is the ClangProgram