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

move graph.py and jit.py into features (#3376)

Browse Source 
* move graph.py into features

* move jit into features

* fix quickstart
This commit is contained in:
George Hotz
2024-02-12 17:34:34 +01:00
committed by GitHub
parent 0f6cde243d
commit 41efaa848c
41 changed files with 77 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_schedule.py
View File

@@ -9,7 +9,7 @@ from tinygrad.ops import LoadOps
from tinygrad.device import Device, Compiled
from tinygrad.helpers import DEBUG, GRAPH
from tinygrad.codegen.linearizer import Linearizer
from tinygrad.graph import print_tree, realized_lazybuffer
from tinygrad.features.graph import print_tree, realized_lazybuffer
from tinygrad import nn, dtypes
def check_schedule(t:Tensor, allowed:int, to_prerealize:Optional[List[Tensor]]=None, filter_loadops=True):