hotfix optimize_local_size (TODO: add regression test)

2023-11-12 10:29:00 -08:00
parent f5a62a1b42
commit 270f747065
1 changed files with 1 additions and 1 deletions
@@ -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):