From 270f74706540d0011aed14ed85aa8488b2b31f7a Mon Sep 17 00:00:00 2001 From: George Hotz Date: Sun, 12 Nov 2023 10:29:00 -0800 Subject: [PATCH] hotfix optimize_local_size (TODO: add regression test) --- tinygrad/features/search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/features/search.py b/tinygrad/features/search.py index 9d402cf824..58f1ce1ac6 100644 --- a/tinygrad/features/search.py +++ b/tinygrad/features/search.py @@ -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):