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

fix typing in simplify_valid [pr] (#7216)

Browse Source
This commit is contained in:
chenyu
2024-10-22 10:01:33 -04:00
committed by GitHub
parent cef7078c14
commit 7ce12a4b06
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/codegen/uopgraph.py
View File

@@ -133,7 +133,7 @@ def simplify_valid(valid:UOp) -> Optional[UOp]:
ret:List[UOp] = []
something_changed = False
for stmt in split_uop(valid, BinaryOps.AND):
ret.append(stmt if not ret else uop_given_valid(functools.reduce(operator.and_, ret), stmt))
ret.append(newstmt if ret and (newstmt:=uop_given_valid(functools.reduce(operator.and_, ret), stmt)) is not None else stmt)
if ret[-1] is not stmt: something_changed = True
return functools.reduce(operator.and_, ret) if something_changed else None