mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

opt weights

Browse source
This commit is contained in:
George Hotz 2025-03-31 11:02:03 +08:00
commit bcfcd60f55
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tinygrad/engine/jit.py
View file

@ -314,6 +314,7 @@ class TinyJit(Generic[ReturnType]):
# set this for next run
self.captured = CapturedJit(ret, jit_cache, input_replace, extra_view_inputs, names, st_vars_dtype_device)
self.captured.optimize_weights()
elif self.cnt >= 2:
# jit exec
assert self.captured is not None