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

move graph/search to engine (#4596)

Browse Source
This commit is contained in:
George Hotz
2024-05-14 23:12:59 -07:00
committed by GitHub
parent afa9753d39
commit ff64bcab69
28 changed files with 33 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/handcode_resnet50_opt.py
View File

@@ -4,12 +4,12 @@ from tinygrad import Tensor, nn
from tinygrad.ops import LoadOps, get_lazyop_info
from tinygrad.device import Device, Compiled
from tinygrad.codegen.linearizer import Linearizer
from tinygrad.features.search import time_linearizer, beam_search, bufs_from_lin
from tinygrad.engine.search import time_linearizer, beam_search, bufs_from_lin
from tinygrad.helpers import ansilen, DEBUG, getenv
from tinygrad.shape.symbolic import sym_infer
from tinygrad.dtype import dtypes
from tinygrad.engine.schedule import create_schedule
from tinygrad.features.graph import print_tree
from tinygrad.engine.graph import print_tree
if __name__ == "__main__":
if getenv("HALF"):