diff --git a/tinygrad/schedule/rangeify.py b/tinygrad/schedule/rangeify.py index 4feefa7f3b..89570079c6 100644 --- a/tinygrad/schedule/rangeify.py +++ b/tinygrad/schedule/rangeify.py @@ -135,7 +135,6 @@ earliest_rewrites = mop_cleanup+PatternMatcher([ (UPat(Ops.CALL, name="c"), resolve_call), # resolve allreduce (must be bottom up) - (UPat(Ops.ASSIGN, src=(UPat.var("output"), UPat(Ops.ALLREDUCE, src=(UPat.var("buf"), UPat()), name="red"))), create_allreduce_function), (UPat(Ops.ALLREDUCE, src=(UPat.var("buf"), UPat()), name="red"), create_allreduce_function), # split_reduceop