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

jit: TODO, use abstractions

Browse Source
This commit is contained in:
geohot
2023-05-05 22:51:30 -07:00
parent 0cd3feb452
commit 7fbf96b992
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
openpilot/compile.py
+1 -1
View File
@@ -64,7 +64,7 @@ def compile(dat, output_fn):
# pull out inputs and put them in the jit cache
input_rawbuffers = {k:inputs[k].lazydata.realized for k in inputs.keys()}
for (j,i),idx in model_exec.input_replace.items(): model_exec.jit_cache[j][1][i] = input_rawbuffers[idx]
for (j,i),(idx,_,_) in model_exec.input_replace.items(): model_exec.jit_cache[j][1][i] = input_rawbuffers[idx]
# transform to CL.CACHE
used_ops = 0