commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

hotfix optimize_local_size (TODO: add regression test)

This commit is contained in:
George Hotz 2023-11-12 10:29:00 -08:00
parent f5a62a1b42
commit 270f747065
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/features/search.py
View File

@ -45,7 +45,7 @@ def time_linearizer(lin:Linearizer, rawbufs:List[RawBuffer], allow_test_size=Tru
# TODO: this is copied from prg.__call__
global_size, local_size = prg.launch_dims(var_vals)
if global_size is not None and local_size is None:
local_size = prg.optimize_local_size(global_size, rawbufs)
local_size = optimize_local_size(prg, global_size, rawbufs)
global_size = [g//l if g%l == 0 else g/l for g,l in zip(global_size, local_size)]
tms = []
for _ in range(cnt):