github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-28 08:17:58 -05:00
Code Issues Packages Projects Releases Wiki Activity

remove ordered parents, seems like dead code [pr] (#8092)

Browse Source 
* remove ordered parents, seems like dead code

* no need to dedup
This commit is contained in:
mesozoic-egg
2024-12-07 05:19:37 +08:00
committed by GitHub
parent e180a31c5e
commit aaf2379f97
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/codegen/kernel.py
View File

@@ -63,9 +63,7 @@ class Kernel:
print(self.ast)
raise e
@functools.lru_cache(None)
def ordered_parents(op:UOp) -> List[UOp]: return dedup([item for x in op.src for item in ordered_parents(x)] + [op])
self.reduceops = dedup([x for x in ordered_parents(self.ast) if x.op is Ops.REDUCE_AXIS])
self.reduceops = [x for x in self.ast.toposort if x.op is Ops.REDUCE_AXIS]
self.vars: List[Variable] = self.ast.variables()
# NOTE: this requires a specific order with the [::-1], this is likely a bug