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

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
test/test_uops_stats.py
View File

@@ -116,7 +116,7 @@ class TestUOpsStats(unittest.TestCase):
u4 = UOp(UOps.ALU, dtypes.int, (u1,u2,u3), TernaryOps.MULACC)
uops_fma = UOpGraph([u4])
self.assertEqual(flops_mem(uops.uops), flops_mem(uops_fma.uops))
self.assertEqual(flops_mem(uops.linearize()), flops_mem(uops_fma.linearize()))
N = 100
@unittest.skipIf(getenv("PTX"), "wrong in PTX") # maybe?