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

Reuse toposort in recursive_property (#13993)

Browse Source
This commit is contained in:
kamilisjon
2026-01-04 08:04:13 +02:00
committed by GitHub
parent 9a9564118c
commit 280790e438
1 changed files with 1 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tinygrad/uop/ops.py
View File

@@ -104,15 +104,7 @@ class recursive_property(property):
self.__doc__ = fxn.__doc__ self.__doc__ = fxn.__doc__
def __get__(self, x:UOp|None, owner=None): def __get__(self, x:UOp|None, owner=None):
if x is None: return self if x is None: return self
# this is very similar to toposort/topovisit for node in x.toposort(gate=lambda node: self.nm not in node.__dict__): node.__dict__[self.nm] = self.fxn(node)
stack: list[tuple[UOp, bool]] = [(x, False)]
while stack:
node, visited = stack.pop()
if self.nm in node.__dict__: continue
if not visited:
stack.append((node, True))
for s in reversed(node.src): stack.append((s, False))
else: node.__dict__[self.nm] = self.fxn(node)
return x.__dict__[self.nm] return x.__dict__[self.nm]
# we import this late so we can use resolve/smax in mixins # we import this late so we can use resolve/smax in mixins