diff --git a/tinygrad/codegen/opt/search.py b/tinygrad/codegen/opt/search.py index 20e45c9a46..084352c486 100644 --- a/tinygrad/codegen/opt/search.py +++ b/tinygrad/codegen/opt/search.py @@ -92,9 +92,9 @@ def _ensure_buffer_alloc(bufs:list[Buffer]) -> list[Buffer]: return [buf.ensure_ # *** external API *** # get dictionary of all possible actions -def get_kernel_actions(s:Scheduler, include_0=True, candidates:list[Opt]|None=None) -> dict[int, Scheduler]: +def get_kernel_actions(s:Scheduler, include_0=True) -> dict[int, Scheduler]: acted, max_up, max_lcl = {0:s} if include_0 else {}, getenv("BEAM_UPCAST_MAX", 256), getenv("BEAM_LOCAL_MAX", 1024) - kernel_actions = (actions if candidates is None else candidates).copy() + kernel_actions = actions.copy() for i,a in enumerate(kernel_actions): if a.axis is not None and a.op is not OptOps.TC: