From ccb5dcf3b8f3ddf7508a1699d81dfd0e6e6a7d7d Mon Sep 17 00:00:00 2001 From: George Hotz Date: Wed, 4 Mar 2026 13:01:35 +0800 Subject: [PATCH] fix --- tinygrad/engine/schedule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/engine/schedule.py b/tinygrad/engine/schedule.py index 32d2bf329d..ed01648d92 100644 --- a/tinygrad/engine/schedule.py +++ b/tinygrad/engine/schedule.py @@ -136,7 +136,7 @@ def soft_allreduce(c:UOp, a:UOp): to = c.src[1].param_like(0) src = c.src[2].param_like(1) red = UOp(Ops.ALLREDUCE, dtype=a.arg, src=(src, a.src[1]), arg=a.arg) - return handle_allreduce(src, red).assign(to).sink().call(*c.src[1:]) + return to.assign(handle_allreduce(src, red)).sink().call(*c.src[1:]) pm_schedule = PatternMatcher([ (UPat(Ops.SINK, name="function"), lower_sink_to_linear),