pop 0 is slow [pr] (#10007)

This commit is contained in:
George Hotz
2025-04-23 17:00:59 +01:00
committed by GitHub
parent d3a8d5c128
commit de0504276b
2 changed files with 3 additions and 3 deletions

View File

@@ -3,7 +3,7 @@ from pathlib import Path
import multiprocessing
from tinygrad import Device, GlobalCounters, Tensor, TinyJit, dtypes
from tinygrad.helpers import getenv, BEAM, WINO, round_up, diskcache_clear, FUSE_CONV_BW
from tinygrad.helpers import getenv, BEAM, WINO, round_up, diskcache_clear, FUSE_CONV_BW, Profiling
from tinygrad.nn.state import get_parameters, get_state_dict, safe_load, safe_save
from tinygrad.nn.optim import LAMB, LARS, SGD, OptimizerGroup, Adam
@@ -1211,4 +1211,4 @@ if __name__ == "__main__":
nm = f"train_{m}"
if nm in globals():
print(f"training {m}")
globals()[nm]()
with Profiling(enabled=getenv("PYPROFILE")): globals()[nm]()

View File

@@ -26,7 +26,7 @@ def _apply_map_to_tensors(applied_map:dict[UOp, UOp], name:str|None=None) -> Non
all_uops: set[UOp] = set()
search_uops = list(applied_map)
while len(search_uops):
x = search_uops.pop(0)
x = search_uops.pop()
if x in all_uops: continue
all_uops.add(x)
search_uops.extend([u for c in x.children if (u:=c()) is not None])