github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

remove iter from uopgraph (#6110)

Browse Source 
* remove iter from uopgraph

* linearize returns uops

* fix tests

* linearize in linearize

* tests fix

* touchup

* test failures
This commit is contained in:
George Hotz
2024-08-16 15:58:29 -07:00
committed by GitHub
parent 28c75bf2a6
commit 74ee9febec
15 changed files with 104 additions and 129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extra/optimization/get_action_space.py
View File

@@ -28,7 +28,7 @@ if __name__ == "__main__":
# confirm linearize can be called twice
uops1 = lin.linearize().uops
uops2 = lin.linearize().uops
for x,y in zip(uops1.uops, uops2.uops):
for x,y in zip(uops1, uops2):
# for some reason DEFINE_ACC is changing the arg
if x.op != y.op or x.dtype != y.dtype: # or x.arg != y.arg:
uops1.print()