sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 8 Activity

clean up pm_remove_invalid [PR] (#16895)

Browse Source
This commit is contained in:
chenyu
2026-07-06 20:10:52 -04:00
committed by GitHub
parent 4faed79216
commit eb3cb16cc1
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/uop/symbolic.py
+2 -2
View File
@@ -98,9 +98,9 @@ pm_data_invalid = PatternMatcher([
propagate_invalid = pm_index_invalid + pm_data_invalid
# TODO: this does nothing now
# NOTE: this happens in padded WMMA, so rewrite to 0
pm_remove_invalid = PatternMatcher([
(UPat(Ops.CONST, arg=Invalid, name="i"), lambda i: i.const_like(0) if i.dtype.scalar() is not dtypes.weakint else None),
(invalid_pat, lambda i: i.const_like(0) if i.dtype.scalar() is not dtypes.weakint else None),
])
symbolic_simple = propagate_invalid + PatternMatcher([