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

switch get_kernel -> get_program [pr] (#10817)

Browse Source 
* switch get_kernel -> get_program [pr]

* fix tests
This commit is contained in:
George Hotz
2025-06-15 12:26:50 -07:00
committed by GitHub
parent a36b09a715
commit 5dc1bc6070
4 changed files with 12 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docs/abstractions2.py
View File

@@ -59,11 +59,11 @@ st_0 = UOp(Ops.STORE, dtypes.void, (output_buf.view(ShapeTracker.from_shape((1,)
s = UOp(Ops.SINK, dtypes.void, (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()
from tinygrad.engine.realize import get_program, CompiledRunner
program = get_program(Device[DEVICE].renderer, s)
# compile a program (and print the source)
fxn = CompiledRunner(kernel.to_program())
fxn = CompiledRunner(program)
print(fxn.p.src)
# NOTE: fxn.clprg is the CPUProgram