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

Add insert_before to Linearizer Functions (#4320)

Browse Source 
* adding insert_before to linearizer functions

* uop insert_before test case

* formatting

* more formatting

* more formatting

* syntax

* removing self.cast

* addressing err

* removing noqa s
This commit is contained in:
Timmy
2024-04-28 11:38:36 -04:00
committed by GitHub
parent 3372bea322
commit 664b563c91
3 changed files with 83 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
test/test_uop_graph.py
View File

@@ -46,5 +46,14 @@ class TestUOpGraph(unittest.TestCase):
self.assertEqual(out.uop, UOps.CONST)
self.assertEqual(out.arg, 0)
def test_insert_before(self):
g = UOpGraph()
g.add(UOps.CONST, dtypes.int, arg=0)
three = g.add(UOps.CONST, dtypes.int, arg=3)
g.add(UOps.CONST, dtypes.int, arg=1, insert_before=three)
g.add(UOps.CONST, dtypes.int, arg=2, insert_before=three)
g.add(UOps.CONST, dtypes.int, arg=4)
for i,uop in enumerate(g.uops): self.assertEqual(i, uop.arg)
if __name__ == '__main__':
unittest.main(verbosity=2)