tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 19:36:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

use get_buf in exec_kernel (#16356)

Browse Source
This commit is contained in:
nimlgen
2026-05-25 15:13:40 +03:00
committed by GitHub
parent 5d5e02871f
commit 64095cf2e2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/engine/realize.py
+1 -1
View File
@@ -175,7 +175,7 @@ def exec_kernel(ctx:ExecContext, call:UOp, ast:UOp) -> float|None:
rt = get_runtime(device:=bufs[0].device, ast, cache=ctx.cache)
global_size, local_size = ast.arg.launch_dims(var_vals)
with track_stats(ctx, call, device, prg_bufs, var_vals) as tm:
et = tm[0] = rt(*[b._buf for b in prg_bufs], global_size=global_size, local_size=local_size, vals=ast.arg.vals(var_vals),
et = tm[0] = rt(*[b.get_buf(device) for b in prg_bufs], global_size=global_size, local_size=local_size, vals=ast.arg.vals(var_vals),
wait=ctx.wait, timeout=ctx.timeout)
return et