Compare commits

...

1 commit

Author SHA1 Message Date
George Hotz
b349e55c66 fix fuse unique 2025-08-04 19:07:49 -07:00

View file

@ -344,7 +344,7 @@ pm_fuse = PatternMatcher([
def do_fusion(x:UOp): def do_fusion(x:UOp):
found_contiguous = {} found_contiguous = {}
def gate_contiguous(x): def gate_contiguous(x):
if is_contiguous:=(x.op is Ops.CONTIGUOUS): found_contiguous[x] = x.replace(src=(UOp(Ops.VIEW, arg=x.st),)) if is_contiguous:=(x.op is Ops.CONTIGUOUS): found_contiguous[x] = x.replace(src=(UOp(Ops.VIEW, arg=x.st), UOp.unique()))
return not is_contiguous return not is_contiguous
x.toposort(gate=gate_contiguous) x.toposort(gate=gate_contiguous)
del gate_contiguous del gate_contiguous