mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-08 22:48:25 -05:00
pop 0 is slow [pr] (#10007)
This commit is contained in:
@@ -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]()
|
||||
|
||||
@@ -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])
|
||||
|
||||
Reference in New Issue
Block a user