From 517eea598511517ddd62cddfee63ff880e7f0cff Mon Sep 17 00:00:00 2001 From: chenyu Date: Mon, 1 Jun 2026 17:12:34 -0400 Subject: [PATCH] no CONST(DEVICE) in create_allreduce_function (#16460) --- tinygrad/schedule/allreduce.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/schedule/allreduce.py b/tinygrad/schedule/allreduce.py index b2d6166082..8a974032be 100644 --- a/tinygrad/schedule/allreduce.py +++ b/tinygrad/schedule/allreduce.py @@ -55,7 +55,7 @@ def handle_allreduce(buf:UOp, red:UOp) -> UOp|None: return UOp.usum(*[c.pad(((s,numel-e),)) for (s,e),c in zip(chunks, copied_chunks)]).reshape(shape) def create_allreduce_function(buf:UOp, red:UOp, output:UOp|None=None) -> UOp|None: - if output is None: output = UOp.const(red.dtype, Invalid, red.device, red.shape).clone() + if output is None: output = UOp.const(red.dtype, Invalid, shape=red.shape).clone(device=red.device) to = red.param_like(0) src = buf.param_like(1) red = src.allreduce(red.arg, red.src[1])