github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

pop 0 is slow [pr] (#10007)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/mlperf/model_train.py
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]()