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

move graph_dedup out of class [run_process_replay] (#5197)

Browse Source
This commit is contained in:
George Hotz
2024-06-27 12:04:00 -07:00
committed by GitHub
parent d094a6828f
commit 345bcc2099
2 changed files with 31 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/test_uop_graph.py
View File

@@ -3,7 +3,7 @@ from test.helpers import TestUOps
from tinygrad import dtypes, Variable
from tinygrad.dtype import PtrDType
from tinygrad.ops import BinaryOps, TernaryOps, UnaryOps
from tinygrad.codegen.uops import UOpGraph, UOps, UOp, PatternMatcher, graph_rewrite
from tinygrad.codegen.uops import UOpGraph, UOps, UOp, PatternMatcher, graph_rewrite, graph_dedup
#from tinygrad.engine.graph import print_tree
simple_pm = PatternMatcher([
@@ -14,6 +14,12 @@ simple_pm = PatternMatcher([
])
class TestGraphRewrite(unittest.TestCase):
def test_dedup(self):
v1 = UOp(UOps.DEFINE_VAR, dtypes.float)
v2 = UOp(UOps.DEFINE_VAR, dtypes.float)
nout = graph_dedup(v1+v2)
self.assertIs(nout.src[0], nout.src[1])
def test_simple(self):
c1 = UOp.const(dtypes.float, 1.0)
c2 = UOp.const(dtypes.float, 2.0)