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

reorder a few things (#1915)

Browse Source 
* reorder a few things

* huh, that has to be there

* move apply shapetracker

* BufferOps

* only for type checking
This commit is contained in:
George Hotz
2023-09-25 10:17:21 +08:00
committed by GitHub
parent 25a767cd5d
commit c907efbf4a
10 changed files with 96 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/abstractions.py
View File

@@ -266,9 +266,9 @@ from tinygrad.tensor import Tensor
result = Tensor(2).realize() + Tensor(3).realize()
# use the real Linearizer to linearize 2+3
from tinygrad.lazy import _replace_loadops
from tinygrad.lazy import _replace_bufferops
from tinygrad.codegen.linearizer import Linearizer
op, _ = _replace_loadops(result.lazydata.op)
op, _ = _replace_bufferops(result.lazydata.op)
linearizer = Linearizer(op)
linearizer.linearize()