move time_linearizer to extra.optimization.helpers [pr] (#9048)

no longer used in tinygrad
This commit is contained in:
chenyu
2025-02-12 15:49:58 -05:00
committed by GitHub
parent c15486cf39
commit f4f56d7c15
12 changed files with 41 additions and 36 deletions

View File

@@ -5,8 +5,9 @@ from tinygrad import Tensor, Device, dtypes, nn
from tinygrad.codegen.kernel import Kernel
from tinygrad.ops import Ops, sym_infer
from tinygrad.device import Compiled
from tinygrad.engine.search import time_linearizer, beam_search, bufs_from_lin
from tinygrad.engine.search import beam_search, bufs_from_lin
from tinygrad.helpers import DEBUG, ansilen, getenv, colored, TRACEMETA
from extra.optimization.helpers import time_linearizer
def get_sched_resnet():
mdl = ResNet50()