diff --git a/tinygrad/engine/schedule.py b/tinygrad/engine/schedule.py index c007271092..cbca43f4ff 100644 --- a/tinygrad/engine/schedule.py +++ b/tinygrad/engine/schedule.py @@ -429,7 +429,8 @@ def create_schedule_with_vars(outs:List[LazyBuffer]) -> Tuple[List[ScheduleItem] for out in lsi.outputs: del out.srcs # can only schedule once schedule.append(si:=ScheduleItem(lsi.ast, tuple(x.buffer for x in lsi.bufs if x.size != 0), lsi.metadata)) if (m:=BUF_LIMIT.get(device:=si.outputs[0].device)) and len(si.bufs) >= m: - raise RuntimeError(f"{si} exceeded the buffer count limit for {device}: {len(si.bufs)} >= {m}") + if DEBUG >= 3: print(si) + raise RuntimeError(f"Kernel for {si.metadata} exceeded the {m} buffer count limit for {device} with {len(si.bufs)} buffers.") for x in graph[lsi]: in_degree[x] -= 1 if in_degree[x] == 0: queue.append(x)